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:48:59 +08:00
parent 8d325d72f7
commit 8d4107569e
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/rag_build_query.py
View File

@ -1,7 +1,4 @@
import os from llama_index.readers.file.base import SimpleDirectoryReader
import faiss
from typing import List
from llama_index.readers.file import SimpleDirectoryReader
from llama_index.indices.vector_store import VectorStoreIndex from llama_index.indices.vector_store import VectorStoreIndex
from llama_index.service_context import ServiceContext from llama_index.service_context import ServiceContext
from llama_index.prompts import PromptTemplate from llama_index.prompts import PromptTemplate
@ -11,6 +8,7 @@ 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"