修改readme
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m28s

This commit is contained in:
2026-05-05 13:30:31 +08:00
parent 128aad0c22
commit 8c021c264e
5 changed files with 730 additions and 1037 deletions

View File

@@ -20,12 +20,11 @@ def is_initialized() -> bool:
return _initialized
async def init_rag_tool(local_llm_creator, force: bool = False):
async def init_rag_tool(force: bool = False):
"""
初始化 RAG 工具(注册到模块级变量)
初始化 RAG 工具(注册到模块级变量,内部获取所需服务
Args:
local_llm_creator: 返回 LLM 实例的函数
force: 是否强制重新初始化
Returns:
@@ -39,20 +38,22 @@ async def init_rag_tool(local_llm_creator, force: bool = False):
return _rag_tool
try:
from app.model_services.chat_services import get_chat_service
info("🔄 正在初始化 RAG 检索系统...")
embeddings = get_embedding_service()
retriever = create_parent_hybrid_retriever(
collection_name="rag_documents",
search_k=5,
embeddings=embeddings
embeddings=embeddings,
)
rewrite_llm = local_llm_creator()
rewrite_llm = get_chat_service()
rag_tool = create_rag_tool(
retriever=retriever,
llm=rewrite_llm,
num_queries=3,
rerank_top_n=5
rerank_top_n=5,
)
_rag_tool = rag_tool