AI导哈

数据库代理系统

一个基于LLM的PostgreSQL数据库代理系统，支持通过自然语言进行数据库操作，具有工具驱动执行和模块化工作流特性。

  • by 00VALAK00
  • Python
  • 本地部署
  • 1 星标
  • 更新于
在 GitHub 查看 00VALAK00/MCP-LLamaIndex-CRUD-Agent

使用说明

MCP-AgentDB

概述

该项目提供了一个代理的、MCP工具驱动的系统，用于使用LLM与PostgreSQL数据库进行交互。它利用LlamaIndex、Ollama和自定义工作流来解释用户请求，选择适当的数据库工具，并透明地执行操作。该系统的设计具有可扩展性和易用性。

特性

  • 自然语言数据库操作： 使用简明英语查询、插入、更新和管理数据库。
  • 工具驱动执行： 所有操作都是通过明确的工具调用执行的，以确保透明度和可审计性。
  • PostgreSQL后端： 使用PostgreSQL作为默认数据库（可配置）。
  • 代理工作流： 具有内存、事件驱动步骤和LLM驱动推理的模块化工作流。
  • Docker支持： 使用Docker Compose轻松设置应用程序和数据库。

建筑

flowchart LR
    subgraph UserAgent["FunctionAgent (LlamaIndex)"]
        A[FunctionAgent]
    end
    subgraph MCP["MCP Server"]
        B[MCP Server]
    end
    subgraph Ollama["Ollama Server"]
        C[LLM]
    end
    subgraph DB["PostgreSQL DB"]
        D[PostgreSQL DB]
    end

    A -- "MCP Protocol Requests/Responses" --> B
    C -- "Integration" --> A
    B -- "APIs" --> D
  

    style A fill:#cef,stroke:#333,stroke-width:2px,stroke-length:4px
    style B fill:#eef,stroke:#333,stroke-width:2px,stroke-length:8px
    style C fill:#ffe,stroke:#333,stroke-width:2px
    style D fill:#fcf,stroke:#333,stroke-width:2px
  • main.py： 入口点；运行代理工作流循环。
  • scripts/workflow.py： 定义 DatabaseWorkflow，协调LLM、工具选择和执行。
  • mcp/mcpserver.py： 实现MCP服务器公开数据库工具（CRUD、模式等）。
  • config/settings.py： 从环境变量加载Ollama（LLM）和数据库的配置。
  • config/prompts.py： 系统提示指导代理的行为。
  • Dockerfile和docker-compose.yml： 应用程序和PostgreSQL的容器化设置。

设置

1.Docker Compose（推荐）

确保你已经安装了Docker和Docker Compose。

git clone <this-repo-url>
docker-compose up --build
  • 该应用程序将在 Agent_MCP_Server 集装箱。
  • PostgreSQL运行在 postgres_db 容器（默认用户： postgres，密码： postgres，db： testdb).
  • 您可以在中自定义环境变量 .env.docker.

2.手动设置

  • 安装Python 3.12+
  • 安装依赖项：
    pip install -r requirements.txt  # or use pyproject.toml with pip/uv
  • 确保PostgreSQL正在运行且可访问（请参阅 config/settings.py 默认值）。
  • 根据需要设置环境变量（请参见 .env.docker 例如）。
  • 启动MCP服务器和主工作流：
    # With Docker
docker-compose exec -it app bash
python main.py

# Without Docker
python main.py # This will spin up the server and initiate the workflow

用法

  • 启动时，代理将提示： What would you like to do?
  • 输入自然语言请求，例如：
    • Show me all the tables in the database.
    • Add a new customer: 'Alice', 'alice@email.com', 25 to the customers table.
  • 代理人将：
    1. 解释您的请求
    2. 选择并调用适当的数据库工具
    3. 返回结果

交互示例

What would you like to do? Show me all the tables in the database.
Tool: list_tables()
Output: The available tables are: customers, products, orders

目录结构

config/           # Prompts and settings
mcp/              # MCP server and database tool definitions
scripts/          # Workflow and event logic
main.py           # Entry point
Dockerfile        # App container
docker-compose    # Initiates the app, database and MCP server (Make sure ollama is serving)

依赖项

  • Python 3.12+
  • Calma-Index-LLMS-Ollama
  • mcp[cli]
  • mlflow
  • 奥拉马
  • psycopg2二进制
  • python dotenv
  • （参见 pyproject.toml 完整列表）

配置

  • Ollama LLM： 通过环境变量设置（请参见 config/settings.py).
  • 数据库： 通过环境变量设置（请参见 config/settings.py).

享受实验的乐趣

相关 MCP