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-11 01:07:53 +08:00
parent f992ca30c6
commit 8bb35d12ea
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/api/search.py
View File

@ -3,7 +3,7 @@ from pydantic import BaseModel
from app.core.embedding import embedder from app.core.embedding import embedder
from app.core.config import settings from app.core.config import settings
from llama_index.vector_stores.faiss import FaissVectorStore 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 os
import logging import logging
import faiss # 引入faiss 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) service_context = ServiceContext.from_defaults(embed_model=embedder, llm=None)
logger.info("Service context created successfully.") logger.info("Service context created successfully.")
# 创建 VectorStoreIndex 并加载文本与向量 # 使用 load_index_from_storage 加载索引
index = VectorStoreIndex.from_storage_context(storage_context) index = load_index_from_storage(storage_context)
logger.info("VectorStoreIndex created successfully.") logger.info("VectorStoreIndex loaded successfully.")
# 检索结果（包含文本） # 检索结果（包含文本）
retriever = index.as_retriever(similarity_top_k=settings.TOP_K) retriever = index.as_retriever(similarity_top_k=settings.TOP_K)