MCP Pymilvus Code Generator

🚀 mcp-pymilvus-code-generate-helper

A model context protocol server for retrieving relevant code snippets or documentation to assist in generating pymilvus code.

🚀 Quick Start

✨ Prerequisites

Before using this MCP server, ensure that you meet the following requirements:

• Python 3.10 or higher
• A running Milvus instance (local or remote)
• uv installed (recommended for running the server)

💻 Usage Examples

The recommended way to use this MCP server is to run it directly via uv without installation. Claude Desktop and Cursor are examples configured in this way. The current server supports two modes: SSE and STDIO.

SSE

Run the Server

uv run src/mcp_pymilvus_code_generate_helper/sse_server.py
# If milvus_uri is not provided, it will connect to the local Milvus service (http://localhost:19530) by default.
# To connect to a remote Milvus service, specify milvus_uri, for example:
# uv run src/mcp_pymilvus_code_generate_helper/sse_server.py --milvus_uri http://<your-server-ip>:<your-server-port>

Use with Cursor

1. Go to Cursor > Preferences > Cursor Settings > MCP.
2. Click the + Add New MCP Server button.
3. Fill out the form:
   • Name: pymilvus-code-generate-helper (or any name you like).
   • Type: Select SSE.
   • Server URL: http://localhost:23333/sse (replace with your server IP address and port).
4. Click Save.

You can also directly edit mcp.json as follows:

{
  "mcpServers": {
    "pymilvus-code-generate-helper": {
      "url": "http://localhost:23333/sse"
    }
  }
}

Use with Claude Desktop

⚠️ Important Note

Claude Desktop currently only supports partial functionality.

STDIO

Run the Server

uv run src/mcp_pymilvus_code_generate_helper/stdio_server.py --milvus_uri http://localhost:19530

Use with Claude Desktop

1. Go to Claude > Preferences > MCP Configuration.
2. Add a new MCP server:
   • Name: pymilvus-code-generate-helper.
   • Type: Select STDIO.
   • Executable File Path: src/mcp_pymilvus_code_generate_helper/stdio_server.py.
3. Configure environment variables:

   export OPENAI_API_KEY="your-api-key"
   

4. Restart Claude Desktop.

📚 Documentation

The server provides the following tools:

• milvus-pypmilvus-code-generate-helper: Searches for relevant pymilvus code/documentation based on the user's natural language query to assist in generating code.

  • Parameters:
    • query: The user's query for generating code.

• milvus-translate-orm-to-milvus-client-code-helper: Searches for relevant ORM and pymilvus client code/documentation based on the user's natural language query to assist in converting ORM code to Milvus client code.

  • Parameters:
    • query: The user's query for translating ORM code.

🔧 Contribution

Contributions are welcome! If you have any ideas for improving the retrieval results, please submit a pull request or open an issue.

📄 License

This project is licensed under the MIT License.