diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py index 6ed457b..2e1424c 100644 --- a/scripts/rag_build_query.py +++ b/scripts/rag_build_query.py @@ -1,8 +1,4 @@ -from llama_index.core.readers.file.base 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.core import SimpleDirectoryReader, VectorStoreIndex, ServiceContext, PromptTemplate from llama_index.embeddings.huggingface import HuggingFaceEmbedding from llama_index.vector_stores.faiss import FaissVectorStore from app.core.config import settings @@ -10,6 +6,7 @@ from scripts.permissions import get_user_allowed_indexes + USER_INDEX_PATH = "index_data" USER_DOC_PATH = "docs"