| name | rag-service |
| description | 高性能 RAG 多路检索服务。集成 Milvus 向量数据库进行语义检索,并结合 Rerank 模型进行精准重排序,支持海量文档的高效存储与历史内容召回。 |
功能
RAG 多路检索服务,提供:
- 向量语义检索 - 基于 Milvus 的向量相似度搜索
- Rerank 重排序 - 对检索结果进行精排
- 文档存储 - 保存文档到向量数据库
调用方式
from services.rag_service.client import RAGServiceClient
client = RAGServiceClient()
status = client.health()
result = client.retrieve(
query="Python 异步编程最佳实践",
top_k=5,
min_score=0.85,
rerank=True
)
print(result["results"])
texts = client.retrieve_texts(query="Python 异步编程", top_k=5)
client.save(documents=[
{"text": "文档内容...", "metadata": {"title": "标题", "url": "..."}}
])
返回格式
retrieve
{
"query": "Python 异步编程",
"results": [
{
"id": "abc123",
"text": "Python异步编程基于asyncio库...",
"score": 0.92,
"metadata": {"title": "Python官方文档", "url": "..."}
}
],
"total": 3,
"elapsed_ms": 45.2,
"from_cache": false
}
save
{
"saved_count": 5,
"collection_name": "websearch_results"
}