This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user