MCP Milvus

🚀 MCP Milvus

這是一個為 Milvus 向量數據庫打造的 模型上下文協議（Model Context Protocol，MCP） 服務器，可提供全面的向量數據庫操作功能。

🚀 快速開始

MCP Milvus 為 Milvus 向量數據庫提供了全面的操作能力。你可以通過它對數據庫、集合和索引進行全生命週期管理，還能進行高性能的向量搜索。以下是使用該項目的快速步驟：

1. 確保滿足安裝的前提條件，即 Go 1.24 或更高版本以及一個正在運行的 Milvus 實例。
2. 選擇從源代碼安裝或使用 Docker 進行部署。
3. 啟動服務器並連接到 Milvus 實例，之後就可以使用各種工具進行數據庫操作了。

✨ 主要特性

• 完整的 Milvus 操作：對數據庫、集合和索引進行全生命週期管理。
• 高性能向量搜索：支持相似性搜索、混合搜索等多種檢索方法。
• 智能會話管理：基於 Ristretto 緩存實現高效的連接池。
• 工程架構：模塊化設計，便於擴展和維護。
• 中間件支持：內置日誌記錄、身份驗證等中間件。
• Docker 支持：提供完整的容器化部署解決方案。
• 類型安全：Go 的強類型系統確保 API 的安全性。

📋 支持的工具

數據庫管理

• milvus_create_database - 創建數據庫
• milvus_list_databases - 列出所有數據庫
• milvus_use_database - 切換數據庫

集合管理

• milvus_create_collection - 創建集合
• milvus_drop_collection - 刪除集合
• milvus_list_collections - 列出集合
• milvus_get_collection_info - 獲取集合信息
• milvus_rename_collection - 重命名集合
• milvus_load_collection - 將集合加載到內存中
• milvus_release_collection - 從內存中釋放集合

索引管理

• milvus_create_index - 創建索引
• milvus_drop_index - 刪除索引

數據操作

• milvus_insert_data - 插入數據
• milvus_upsert - 插入或更新數據
• milvus_delete_entities - 刪除實體
• milvus_query - 條件查詢
• milvus_vector_search - 向量相似性搜索

連接管理

• milvus_connector - 建立 Milvus 連接

📦 安裝指南

前提條件

• Go 1.24 或更高版本
• 一個正在運行的 Milvus 實例

從源代碼安裝

git clone https://github.com/tailabs/mcp-milvus.git
cd mcp-milvus
make deps
make build

或者直接使用 Go 進行安裝：

git clone https://github.com/tailabs/mcp-milvus.git
cd mcp-milvus
go mod download
go build -o mcp-milvus ./cmd/mcp-milvus

Docker 部署

# 構建鏡像
docker build -t mcp-milvus .

# 運行容器
docker run -p 8080:8080 mcp-milvus

使用方法

1. 啟動服務器

# 使用 Makefile
make run

# 或者直接運行
./build/mcp-milvus

2. 連接到 Milvus 使用 milvus_connector 工具建立連接：

{
  "address": "localhost:19530",
  "token": "username:password",
  "db_name": "default"
}

3. 執行操作 建立連接後，你可以使用其他工具進行數據庫操作。

📚 詳細文檔

項目結構

mcp-milvus/
├── cmd/mcp-milvus/          # 主應用程序入口
├── internal/
│   ├── middleware/          # 中間件（日誌記錄、身份驗證等）
│   ├── registry/            # 工具註冊表
│   ├── schema/              # 模式構建器
│   ├── session/             # 會話管理
│   └── tools/               # Milvus 工具實現
├── Dockerfile               # Docker 構建文件
├── go.mod                   # Go 模塊定義
└── README.md               # 項目文檔

配置

環境變量

• LOG_LEVEL：日誌級別（debug/info/warn/error），默認值：info
• PORT：服務端口，默認值：8080

連接配置

支持以下連接參數：

• address：Milvus 服務地址
• token：身份驗證令牌（格式：username:password）
• db_name：數據庫名稱

🤝 貢獻代碼

我們歡迎各種形式的貢獻！詳情請參閱 CONTRIBUTING.md。

開發工作流程

1. Fork 此倉庫
2. 創建你的功能分支 (git checkout -b feature/AmazingFeature)
3. 設置開發環境 (make deps && make tools)
4. 進行更改並測試 (make test && make lint)
5. 提交更改 (git commit -m 'Add some AmazingFeature')
6. 推送到該分支 (git push origin feature/AmazingFeature)
7. 打開一個 Pull Request

可用的 Make 命令

運行 make help 查看所有可用命令：

make help          # 顯示所有可用命令
make build         # 構建二進制文件
make test          # 運行測試
make lint          # 運行代碼檢查
make fmt           # 格式化代碼
make run           # 構建並運行
make dev           # 即時重新加載運行
make docker        # 構建 Docker 鏡像
make build-all     # 為所有平臺構建
make release       # 準備發佈
make clean         # 清理構建產物

📄 許可證

本項目採用 MIT 許可證 - 詳情請參閱 LICENSE 文件。

🔗 相關鏈接

• Milvus 官方網站
• 模型上下文協議
• MCP Go SDK

📞 支持

如果你遇到任何問題或有疑問：

1. 查看 Issues 以確認是否存在類似問題。
2. 創建一個新的 Issue 描述你的問題。
3. 加入討論區參與討論。

🙏 致謝

• Milvus - 優秀的向量數據庫
• MCP Go - MCP 的 Go 實現
• 所有貢獻者和用戶

⭐ 如果這個項目對你有幫助，請給它點個 Star！