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-08 16:46:38 +08:00
parent 2d0a6c11ee
commit 8d325d72f7
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/rag_build_query.py
View File

@ -1,17 +1,16 @@
import os import os
import faiss import faiss
from typing import List from typing import List
from llama_index import ( from llama_index.readers.file import SimpleDirectoryReader
SimpleDirectoryReader, from llama_index.indices.vector_store import VectorStoreIndex
VectorStoreIndex, from llama_index.service_context import ServiceContext
ServiceContext, from llama_index.prompts import PromptTemplate
PromptTemplate,
)
from llama_index.embeddings.huggingface import HuggingFaceEmbedding from llama_index.embeddings.huggingface import HuggingFaceEmbedding
from llama_index.vector_stores.faiss import FaissVectorStore from llama_index.vector_stores.faiss import FaissVectorStore
from app.core.config import settings from app.core.config import settings
from scripts.permissions import get_user_allowed_indexes from scripts.permissions import get_user_allowed_indexes
USER_INDEX_PATH = "index_data" USER_INDEX_PATH = "index_data"
USER_DOC_PATH = "docs" USER_DOC_PATH = "docs"