google / google/adk-python-community

Feature Request: Add Milvus Vector Store Integration for RAG

未关闭
#70 2 条评论 1 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Python
星标
184
派生
76
PR 合并指标
30 天内没有已合并 PR

描述

### Description

Add [Milvus](https://milvus.io/) as a vector database option for knowledge base RAG workflows, following the existing Spanner integration pattern.

### Motivation

ADK currently supports Vertex AI RAG, LlamaIndex, and Spanner for vector-based retrieval. Milvus is a widely adopted open-source vector database purpose-built for AI applications. Adding Milvus support would give developers a lightweight, self-hosted (or cloud-managed via [Zilliz](https://zilliz.com/)) alternative for building RAG agents.

### Proposed Solution

- `MilvusVectorStore`: Utility class for collection setup, data ingestion, and similarity search.
- `MilvusToolset`: `BaseToolset` implementation exposing a `similarity_search` tool to LLM agents.
- `MilvusToolSettings` / `MilvusVectorStoreSettings`: Pydantic configuration classes.
- Users provide their own embedding function (e.g., Google GenAI `gemini-embedding-001`).
- `pymilvus>=2.5.0` as an optional dependency (`google-adk[milvus]`).

### Related PR

google/adk-python#4417

贡献指南

打开贡献指南

调研方向

先阅读现有的 Spanner 集成模式和 BaseToolset 入口点。将 MilvusVectorStore、MilvusToolset、MilvusToolSettings 和 MilvusVectorStoreSettings 作为范围,并通过 google-adk[milvus] 提供 pymilvus>=2.5.0。完成的标准是涵盖 collection 设置、摄取、similarity_search 暴露以及用户提供的 embeddings。

由索引模型根据 Issue 内容生成。

评估

技术栈
python
领域
ai, databases
Issue 类型
功能
难度
5/5
预计耗时
一周以上
活跃度
停滞
描述清晰度
基本清楚
新手友好度
25/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。