From 430b586b52c7cb6645b8e6a182ff4193466b9fbc Mon Sep 17 00:00:00 2001 From: hailin Date: Sun, 11 May 2025 12:10:01 +0800 Subject: [PATCH] . --- app/api/search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/api/search.py b/app/api/search.py index 3dc6612..5fe0bca 100644 --- a/app/api/search.py +++ b/app/api/search.py @@ -48,7 +48,7 @@ def search_docs(request: QueryRequest, user_id: str = Query(..., description=" storage_context = StorageContext.from_defaults(persist_dir=index_path, vector_store=vector_store) logger.info("Storage context created successfully.") - # 创建 ServiceContext 实例 + # 创建 ServiceContext 实例，确保使用本地嵌入模型，且明确禁用 LLM service_context = ServiceContext.from_defaults(embed_model=embedder, llm=None) logger.info("Service context created successfully.") @@ -61,7 +61,7 @@ def search_docs(request: QueryRequest, user_id: str = Query(..., description=" logger.info(f"Generated query embedding: {query_vector}") # 使用 VectorStoreIndex 创建查询引擎 - query_engine = index.as_query_engine() + query_engine = index.as_query_engine(service_context=service_context) logger.info(f"Query engine created successfully.") # 执行查询