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 10:29:20 +08:00
parent b8b9648de3
commit 48f8f81de2
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/rag_build_query.py
View File

@ -101,11 +101,9 @@ def build_user_index(user_id: str):
faiss.write_index(faiss_index, faiss_index_file) # 使用 Faiss 的 write_index 方法保存索引
logger.info(f"Faiss 索引已保存到 {faiss_index_file}")
# 使用 storage_context.persist() 保存元数据
# 存储非 Faiss 向量数据，如文档的 ID、嵌入向量等
vector_store_file = os.path.join(persist_dir, "vector_store.json")
storage_context.persist(persist_dir=vector_store_file)
logger.info(f"向量存储数据已保存到 {vector_store_file}")
# 使用 storage_context.persist() 保存其他索引数据
storage_context.persist(persist_dir=persist_dir)
logger.info(f"索引数据已保存到 {persist_dir}")
except Exception as e:
logger.error(f"索引构建失败: {e}")