chromadb-fastapi-mcp
by osushinekotan
This project provides a FastAPI-based server for ChromaDB, enabling easy management and querying of vector embeddings. It also integrates with the MCP (Machine Communication Protocol) for enhanced tool and resource discovery.
Last updated: N/A
Installation
- Clone the repository
- Install dependencies:
uv sync
- Copy
.env.example
to.env
and configure settings:cp .env.example .env
- Edit
.env
with your preferred settings
Configuration
Configure the following environment variables in your .env
file:
CHROMA_CLIENT_TYPE
: Eitherephemeral
orpersistent
CHROMA_DATA_DIR
: Directory for storage when using persistent clientOPENAI_API_KEY
: Your OpenAI API key for embeddings
Usage
Run the server:
cd /path/to/chromadb-fastapi-mcp
uv run python -m app.main
Or with uvicorn directly:
uv run uvicorn app.main:app --reload
The server will start at http://localhost:8000, and API documentation is available at http://localhost:8000/docs.
API Endpoints
Collections
GET /api/collections/
- List all collectionsPOST /api/collections/
- Create a new collectionGET /api/collections/{collection_name}/peek
- Peek at documents in a collectionGET /api/collections/{collection_name}/info
- Get collection informationGET /api/collections/{collection_name}/count
- Get document count in a collectionPUT /api/collections/{collection_name}
- Modify a collectionDELETE /api/collections/{collection_name}
- Delete a collection
Documents
POST /api/documents/add
- Add documents to a collectionPOST /api/documents/query
- Query documents from a collectionPOST /api/documents/get
- Get documents from a collectionPUT /api/documents/update
- Update documents in a collectionDELETE /api/documents/delete
- Delete documents from a collection
MCP server
https://github.com/tadata-org/fastapi_mcp
Connecting to the MCP Server using SSE
Once your FastAPI app with MCP integration is running, you can connect to it with any MCP client supporting SSE, such as Cursor:
-
Run your application.
-
In Cursor -> Settings -> MCP, use the URL of your MCP server endpoint (e.g.,
http://localhost:8000/mcp
) as sse. -
Cursor will discover all available tools and resources automatically.
Connecting to the MCP Server using mcp-proxy stdio
If your MCP client does not support SSE, for example Claude Desktop:
-
Run your application.
-
Install mcp-proxy, for example:
uv tool install mcp-proxy
. -
Add in Claude Desktop MCP config file (
claude_desktop_config.json
):
On Windows:
{
"mcpServers": {
"my-api-mcp-proxy": {
"command": "mcp-proxy",
"args": ["http://127.0.0.1:8000/mcp"]
}
}
}
On MacOS:
Find the path to mcp-proxy by running in Terminal: which mcp-proxy
.
{
"mcpServers": {
"my-api-mcp-proxy": {
"command": "/Full/Path/To/Your/Executable/mcp-proxy",
"args": ["http://127.0.0.1:8000/mcp"]
}
}
}
Find the path to mcp-proxy by running in Terminal: which uvx
.
{
"mcpServers": {
"my-api-mcp-proxy": {
"command": "/Full/Path/To/Your/uvx",
"args": ["mcp-proxy", "http://127.0.0.1:8000/mcp"]
}
}
}
- Claude Desktop will discover all available tools and resources automatically