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 13:40:45 +08:00
parent d46ff27835
commit 636bd451f5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/rag_build_query.py
View File

@ -3,7 +3,7 @@ from typing import List # 导入 List 用于类型注解
import os import os
import logging import logging
import faiss import faiss
from llama_index import SimpleDirectoryReader, VectorStoreIndex, ServiceContext, StorageContext, load_index_from_storage from llama_index import SimpleDirectoryReader, VectorStoreIndex, ServiceContext, StorageContext, load_index_from_storage, Settings
from llama_index.embeddings.huggingface import HuggingFaceEmbedding from llama_index.embeddings.huggingface import HuggingFaceEmbedding
from llama_index.vector_stores.faiss import FaissVectorStore from llama_index.vector_stores.faiss import FaissVectorStore
from app.core.config import settings # 导入应用配置 from app.core.config import settings # 导入应用配置
@ -135,6 +135,9 @@ def build_user_index(user_id: str):
) )
try: try:
# 禁用 LLM
Settings.llm = None
# 构建索引，并使用 `storage_context.persist()` 方法保存索引 # 构建索引，并使用 `storage_context.persist()` 方法保存索引
index = VectorStoreIndex.from_documents( index = VectorStoreIndex.from_documents(
documents, documents,