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-10 12:34:09 +08:00
parent 256646f426
commit 979e14430e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rag_build_query.py
View File

@ -78,6 +78,10 @@ def build_user_index(user_id: str):
# 创建服务上下文 # 创建服务上下文
service_context = ServiceContext.from_defaults(embed_model=embed_model, llm=None) service_context = ServiceContext.from_defaults(embed_model=embed_model, llm=None)
# 直接检查模型嵌入方法是否被调用
logger.info(f"Embedding method being used: {embed_model._get_query_embedding('test query')}")
# 使用 Faiss 向量存储 # 使用 Faiss 向量存储
faiss_index = faiss.IndexFlatL2(1024) faiss_index = faiss.IndexFlatL2(1024)
vector_store = FaissVectorStore(faiss_index=faiss_index) vector_store = FaissVectorStore(faiss_index=faiss_index)