Chromadb Fastapi MCP

🚀 chromadb-fastapi-mcp項目

本項目提供了一個基於FastAPI的服務，藉助Chroma數據庫進行文檔管理，並可與MCP服務器集成，為用戶提供便捷的文檔操作和交互體驗。

🚀 快速開始

安裝步驟

1. 克隆倉庫。
2. 安裝依賴：

   uv sync
   

3. 複製 .env.example 到 .env 並配置設置：

   cp .env.example .env
   

4. 使用你喜歡的設置編輯 .env

配置環境變量

在你的 .env 文件中配置以下環境變量：

• CHROMA_CLIENT_TYPE：選擇 ephemeral 或 persistent。
• CHROMA_DATA_DIR：使用持久化客戶端時的數據目錄。
• OPENAI_API_KEY：你的 OpenAI API 密鑰用於嵌入。

啟動服務器

運行服務器：

cd /path/to/chromadb-fastapi-mcp
uv run python -m app.main

或直接使用 uvicorn：

uv run uvicorn app.main:app --reload

服務器將在 http://localhost:8000 啟動，API 文檔可在 http://localhost:8000/docs 查看。

📦 安裝指南

依賴安裝

uv sync

環境配置

複製 .env.example 到 .env 並進行編輯：

cp .env.example .env

💻 使用示例

基礎用法

啟動服務器：

cd /path/to/chromadb-fastapi-mcp
uv run python -m app.main

高級用法

使用 uvicorn 啟動並開啟熱重載：

uv run uvicorn app.main:app --reload

📚 詳細文檔

API 端點

收藏集

• GET /api/collections/：列出所有收藏集。
• POST /api/collections/：創建新的收藏集。
• GET /api/collections/{collection_name}/peek：查看收藏集中內容的概覽。
• GET /api/collections/{collection_name}/info：獲取收藏集信息。
• GET /api/collections/{collection_name}/count：獲取收藏集中的文檔數量。
• PUT /api/collections/{collection_name}：修改收藏集。
• DELETE /api/collections/{collection_name}：刪除收藏集。

文檔

• POST /api/documents/add：向收藏集中添加文檔。
• POST /api/documents/query：從收藏集中查詢文檔。
• POST /api/documents/get：獲取收藏集中的文檔。
• PUT /api/documents/update：更新收藏集中的文檔。
• DELETE /api/documents/delete：刪除收藏集中的文檔。

MCP 服務器

使用 SSE 連接到 MCP 服務器

運行你的 FastAPI 應用後：

1. 在 Cursor -> 設置 -> MCP 中，使用你的 MCP 服務器端點 URL (例如 http://localhost:8000/mcp) 作為 sse。
2. Cursor 將自動發現所有可用工具和資源。

使用 mcp-proxy stdio 連接到 MCP 服務器

在 Claude Desktop 上配置：

1. 安裝 mcp-proxy：

   pip install mcp-proxy
   

2. 在終端中運行以下命令以創建配置文件：

   python -m mcp_proxy.start_server
   

3. 修改 ~/.config/mcp_proxy/config.json 文件，添加你的模型端點信息。
4. 使用 Claude 應用連接到 MCP 服務器。