mcp-pymilvus-code-generator
@brcarry
A pimilvus code generating helper with SSE-based MCP support
Overview
What is mcp-pymilvus-code-generator?
An MCP server that retrieves relevant code snippets or documents to help generate pymilvus code. It integrates with a running Milvus instance and is intended for developers using AI coding assistants like Cursor or Claude Desktop.
How to use mcp-pymilvus-code-generator?
Run the server with uv run src/mcp_pymilvus_code_generate_helper/sse_server.py (for SSE) or uv run src/mcp_pymilvus_code_generate_helper/stdio_server.py (for STDIO). Configure it in Cursor or Claude Desktop by adding an MCP server entry. The server connects to a local Milvus instance by default; use --milvus_uri to specify a remote one.
Key features of mcp-pymilvus-code-generator
- Retrieves pymilvus code snippets based on natural language queries.
- Converts Milvus ORM code to pymilvus client code and vice versa.
- Translates Milvus code between programming languages.
- Supports both SSE and STDIO transport protocols.
- Can be run locally or via Docker.
Use cases of mcp-pymilvus-code-generator
- Generate pymilvus code for hybrid search from a user description.
- Convert existing Milvus ORM code to pymilvus client code.
- Translate a Milvus code snippet from Python to Java, Node.js, or other languages.
FAQ from mcp-pymilvus-code-generator
What are the prerequisites?
Python 3.10 or higher, a running Milvus instance (local or remote), and uv installed.
Does the server require an API key?
Yes, the OPENAI_API_KEY environment variable must be set when using the STDIO configuration (e.g., in Claude Desktop or Cursor).
What transports does the server support?
It supports both SSE (Server-Sent Events) and STDIO (standard input/output).
Can I run the server in a container?
Yes, a Dockerfile is provided. Build the image with docker build -t milvus-code-helper . and run with environment variables for OPENAI_API_KEY, MILVUS_URI, and MILVUS_TOKEN.