From b8b9648de3d5f5dfbcd1ce29b5164add2fc8a562 Mon Sep 17 00:00:00 2001 From: hailin Date: Sat, 10 May 2025 10:20:52 +0800 Subject: [PATCH] . --- scripts/rag_build_query.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py index 4340685..d72284d 100644 --- a/scripts/rag_build_query.py +++ b/scripts/rag_build_query.py @@ -101,9 +101,11 @@ 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() 保存其他索引数据 - storage_context.persist(persist_dir=persist_dir) - logger.info(f"索引数据已保存到 {persist_dir}") + # 使用 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}") except Exception as e: logger.error(f"索引构建失败: {e}")