hailin
/
faiss_rag_enterprise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-05-11 12:20:40 +08:00
parent 430b586b52
commit 0130a76c6b
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/search.py
View File

@ -56,6 +56,10 @@ def search_docs(request: QueryRequest, user_id: str = Query(..., description="
index = load_index_from_storage(storage_context) index = load_index_from_storage(storage_context)
logger.info("VectorStoreIndex loaded successfully.") logger.info("VectorStoreIndex loaded successfully.")
# 设置索引的 ServiceContext
index.set_service_context(service_context)
logger.info("Service context set to index.")
# 将用户查询通过本地模型生成嵌入向量 # 将用户查询通过本地模型生成嵌入向量
query_vector = embedder.encode([request.query]) # 使用本地模型生成查询的嵌入向量 query_vector = embedder.encode([request.query]) # 使用本地模型生成查询的嵌入向量
logger.info(f"Generated query embedding: {query_vector}") logger.info(f"Generated query embedding: {query_vector}")