Codebase RAG MCP Server

此MCP服务器通过使用嵌入模型和PostgreSQL数据库提供语义代码库搜索和文件检索功能。它提供了搜索代码、列出文件以及检索文件内容的工具。

功能

• 语义代码搜索：使用Voyage嵌入模型将查询转换为向量，并根据相似性搜索代码片段。
• 文件列表：列出代码库中所有被跟踪的文件。
• 文件内容检索：通过文件名获取文件的全部内容。

工具

• search_codebase

  • 返回与查询相关的代码片段。
  • 输入：
    • query (字符串)：搜索词

• get_files

  • 列出代码库中的所有文件。

• get_file_content

  • 返回指定文件的内容。
  • 输入：
    • filename (字符串)：文件名

配置

获取Voyage API密钥

1. 注册一个Voyage API账户
2. 在仪表板中生成您的API密钥

环境变量

• DATABASE_URL：PostgreSQL连接字符串。
• VOYAGE_API_KEY：嵌入模型的API密钥。
• EMBEDDING_TABLE：代码嵌入表名。
• TRACKING_TABLE：文件跟踪表名。

Docker

{
  "mcpServers": {
    "codebase-rag": {
      "command": "docker",
      "args": [
        "run",
        "-i",
        "--rm",
        "-e",
        "DATABASE_URL",
        "-e",
        "VOYAGE_API_KEY",
        "-e",
        "EMBEDDING_TABLE",
        "-e",
        "TRACKING_TABLE",
        "-p",
        "8080:8080",
        "allentcm/mcp-codebase-rag:1.0.0"
      ],
      "env": {
        "DATABASE_URL": "postgres://cocoindex:cocoindex@localhost/cocoindex",
        "VOYAGE_API_KEY": "voyage_api_key",
        "EMBEDDING_TABLE": "codebase__code_embeddings",
        "TRACKING_TABLE": "codebase__cocoindex_tracking"
      }
    }
  }
}

许可证

此MCP服务器采用MIT许可证。您可以在遵守MIT许可证条款和条件的前提下自由使用、修改和分发该软件。更多详情，请参见项目仓库中的LICENSE文件。