From 8bb35d12eacb4157b50f18ea9204dbf51cda3df3 Mon Sep 17 00:00:00 2001 From: hailin Date: Sun, 11 May 2025 01:07:53 +0800 Subject: [PATCH] . --- app/api/search.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/app/api/search.py b/app/api/search.py index 5f91c6e..10d93e8 100644 --- a/app/api/search.py +++ b/app/api/search.py @@ -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)