From 8d325d72f77e4a692d4fcb58b0b281435599214d Mon Sep 17 00:00:00 2001 From: hailin Date: Thu, 8 May 2025 16:46:38 +0800 Subject: [PATCH] . --- scripts/rag_build_query.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py index 17a0947..f59f6b8 100644 --- a/scripts/rag_build_query.py +++ b/scripts/rag_build_query.py @@ -1,17 +1,16 @@ import os import faiss from typing import List -from llama_index import ( - SimpleDirectoryReader, - VectorStoreIndex, - ServiceContext, - PromptTemplate, -) +from llama_index.readers.file import SimpleDirectoryReader +from llama_index.indices.vector_store import VectorStoreIndex +from llama_index.service_context import ServiceContext +from llama_index.prompts import PromptTemplate from llama_index.embeddings.huggingface import HuggingFaceEmbedding from llama_index.vector_stores.faiss import FaissVectorStore from app.core.config import settings from scripts.permissions import get_user_allowed_indexes + USER_INDEX_PATH = "index_data" USER_DOC_PATH = "docs"