This commit is contained in:
hailin 2025-05-11 01:07:53 +08:00
parent f992ca30c6
commit 8bb35d12ea
1 changed files with 4 additions and 4 deletions

View File

@ -3,7 +3,7 @@ from pydantic import BaseModel
from app.core.embedding import embedder
from app.core.config import settings
from llama_index.vector_stores.faiss import FaissVectorStore
from llama_index import VectorStoreIndex, ServiceContext, StorageContext
from llama_index import VectorStoreIndex, ServiceContext, StorageContext, load_index_from_storage
import os
import logging
import faiss # 引入faiss
@ -52,9 +52,9 @@ def search_docs(request: QueryRequest, user_id: str = Query(..., description="
service_context = ServiceContext.from_defaults(embed_model=embedder, llm=None)
logger.info("Service context created successfully.")
# 创建 VectorStoreIndex 并加载文本与向量
index = VectorStoreIndex.from_storage_context(storage_context)
logger.info("VectorStoreIndex created successfully.")
# 使用 load_index_from_storage 加载索引
index = load_index_from_storage(storage_context)
logger.info("VectorStoreIndex loaded successfully.")
# 检索结果(包含文本)
retriever = index.as_retriever(similarity_top_k=settings.TOP_K)