diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py index 86be09a..583d01f 100644 --- a/scripts/rag_build_query.py +++ b/scripts/rag_build_query.py @@ -174,30 +174,35 @@ def build_user_index(user_id: str): - # # 持久化存储之后,加载已保存的存储上下文信息 - # logger.info(f"开始加载持久化存储的数据...") - # # 加载 Faiss 索引 - # faiss_index_file = os.path.join(persist_dir, "index.faiss") - # faiss_index = faiss.read_index(faiss_index_file) - # # 创建 FaissVectorStore 实例 - # vector_store = FaissVectorStore(faiss_index=faiss_index) - # # 创建一个新的 StorageContext 实例,使用相同的目录 - # storage_context = StorageContext.from_defaults( - # persist_dir=persist_dir, # 使用与之前相同的目录 - # vector_store=vector_store # 使用 Faiss 向量存储 - # ) - # # 确认存储是否加载成功,检查索引数据 - # logger.info("已成功加载存储上下文。") + # 持久化存储之后,加载已保存的存储上下文信息 + logger.info(f"开始加载持久化存储的数据...") - # # 使用 load_index_from_storage 加载索引 - # loaded_index = load_index_from_storage(storage_context) + # 加载 Faiss 索引 + faiss_index_file = os.path.join(persist_dir, "index.faiss") + faiss_index = faiss.read_index(faiss_index_file) - # # 检查加载后的索引 - # logger.info(f"加载的索引数量: {len(loaded_index.get_documents())}") + # 创建 FaissVectorStore 实例 + vector_store = FaissVectorStore(faiss_index=faiss_index) + + # 创建一个新的 StorageContext 实例,使用相同的目录 + storage_context = StorageContext.from_defaults( + persist_dir=persist_dir, # 使用与之前相同的目录 + vector_store=vector_store, # 使用 Faiss 向量存储 + llm=None # 明确指定不使用 LLM + ) + + # 确认存储是否加载成功,检查索引数据 + logger.info("已成功加载存储上下文。") + + # 使用 load_index_from_storage 加载索引 + loaded_index = load_index_from_storage(storage_context) + + # 检查加载后的索引 + logger.info(f"加载的索引数量: {len(loaded_index.get_documents())}") except Exception as e: