chroma-mcp

import os import uvicorn import tempfile import shutil # This library is a server application designed to be run, not typically imported # for client-side functionality. The quickstart demonstrates how to run the server locally. # Create a temporary directory for ChromaDB to store its data temp_dir = tempfile.mkdtemp() print(f"ChromaDB data will be stored in: {temp_dir}") # Configure the Chroma MCP server via environment variables. # These are crucial for the server to operate correctly. os.environ["CHROMA_DB_PATH"] = temp_dir # Use a local persistent ChromaDB path os.environ["CHROMA_SERVER_HOST"] = os.environ.get('CHROMA_SERVER_HOST', '127.0.0.1') os.environ["CHROMA_SERVER_PORT"] = os.environ.get('CHROMA_SERVER_PORT', '8000') os.environ["CHROMA_SERVER_ROOT_PATH"] = os.environ.get('CHROMA_SERVER_ROOT_PATH', '') os.environ["LOG_LEVEL"] = os.environ.get('LOG_LEVEL', 'info') # Optional: set log level print(f"Starting Chroma MCP Server on http://{os.environ['CHROMA_SERVER_HOST']}:{os.environ['CHROMA_SERVER_PORT']}") print("Press Ctrl+C to stop the server.") try: # Run the Chroma MCP FastAPI application using uvicorn. # The application instance is found at 'chroma_mcp.app:app'. uvicorn.run( "chroma_mcp.app:app", host=os.environ["CHROMA_SERVER_HOST"], port=int(os.environ["CHROMA_SERVER_PORT"]), log_level=os.environ["LOG_LEVEL"], reload=False # Set to True for development, False for production ) except KeyboardInterrupt: print("\nServer stopped.") finally: # Clean up the temporary directory used by ChromaDB print(f"Cleaning up temporary ChromaDB data directory: {temp_dir}") shutil.rmtree(temp_dir)