This commit is contained in:
parent
b89f81229e
commit
430b586b52
|
|
@ -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.")
|
||||
|
||||
# 执行查询
|
||||
|
|
|
|||
Loading…
Reference in New Issue