Ollama MCP 数据库助手

一个交互式聊天界面，它将 Ollama 的 LLM 功能与通过模型上下文协议 (MCP) 访问 PostgreSQL 数据库的能力相结合。用自然语言提问关于您数据的问题，并获得由真实 SQL 查询支持的 AI 驱动的响应。

特性

• 连接到 PostgreSQL 数据库的自然语言界面
• 自动 SQL 查询生成
• 模式感知响应
• 交互式聊天界面
• 安全的只读数据库访问

前提条件

• Node.js 16 或更高版本
• 正在运行的 PostgreSQL 数据库
• 已安装并在本地运行的 Ollama
• 在 Ollama 中拉取的 qwen2.5-coder:7b-instruct 模型

设置

1. 克隆存储库：

git clone [your-repo-url]
cd [your-repo-name]

2. 安装依赖项：

npm install

3. 拉取所需的 Ollama 模型：

ollama pull qwen2.5-coder:7b-instruct

4. 在项目根目录中创建一个 .env 文件：

DATABASE_URL=postgresql://user:password@localhost:5432/dbname
OLLAMA_MODEL=qwen2.5-coder:7b-instruct  # 可选 - 这是默认值

用法

1. 启动聊天界面：

npm start

2. 用自然语言提问关于您数据的问题：

已连接到数据库。您现在可以提问关于您数据的问题。
输入 "exit" 退出。

您想了解关于您数据的什么信息？ 哪些产品上个月产生的收入最多？
正在分析...

[AI 将生成并执行 SQL 查询，然后解释结果]

3. 输入 'exit' 退出应用程序。

工作原理

1. 应用程序通过 PostgreSQL MCP 服务器 连接到您的 PostgreSQL 数据库
2. 它加载并缓存您的数据库模式
3. 当您提出问题时：
   • 模式和问题被发送到 Ollama
   • Ollama 生成适当的 SQL 查询
   • 查询通过 MCP 执行
   • 结果被发送回 Ollama 进行解释
   • 您收到自然语言响应

环境变量

变量	描述	默认值
DATABASE_URL	PostgreSQL 连接字符串	必需
OLLAMA_MODEL	要使用的 Ollama 模型	qwen2.5-coder:7b-instruct

安全性

• 所有数据库访问都是只读的
• SQL 查询仅限于 SELECT 语句
• 数据库凭据安全地保存在您的 .env 文件中

开发

使用以下技术构建：

• TypeScript
• 模型上下文协议 (MCP)
• Ollama
• PostgreSQL

故障排除

常见问题

1. "Failed to connect to database"（无法连接到数据库）

   • 检查 .env 中的 DATABASE_URL
   • 验证 PostgreSQL 是否正在运行
   • 检查网络连接

2. "Failed to connect to Ollama"（无法连接到 Ollama）

   • 确保 Ollama 正在运行 (ollama serve)
   • 验证模型是否已安装 (ollama list)

3. "Error executing query"（执行查询时出错）

   • 检查数据库权限
   • 验证模式中的表/列名称

许可证

MIT

贡献

1. Fork 存储库
2. 创建您的功能分支
3. 提交您的更改
4. 推送到分支
5. 打开一个 Pull Request