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-09 22:27:40 +08:00
parent afdbdd0fe2
commit 727305fbd4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/rag_build_query.py
View File

@ -57,7 +57,7 @@ def build_user_index(user_id: str):
documents = SimpleDirectoryReader(doc_dir).load_data() documents = SimpleDirectoryReader(doc_dir).load_data()
embed_model = HuggingFaceEmbedding(model_name=settings.MODEL_NAME) embed_model = HuggingFaceEmbedding(model_name=settings.MODEL_NAME)
service_context = ServiceContext.from_defaults(embed_model=embed_model) service_context = ServiceContext.from_defaults(embed_model=embed_model, llm=None)
index = VectorStoreIndex.from_documents( index = VectorStoreIndex.from_documents(
documents, documents,
@ -71,7 +71,7 @@ def build_user_index(user_id: str):
def query_user_rag(user_id: str, question: str, top_k: int = 4) -> str: def query_user_rag(user_id: str, question: str, top_k: int = 4) -> str:
embed_model = HuggingFaceEmbedding(model_name=settings.MODEL_NAME) embed_model = HuggingFaceEmbedding(model_name=settings.MODEL_NAME)
service_context = ServiceContext.from_defaults(embed_model=embed_model) service_context = ServiceContext.from_defaults(embed_model=embed_model, llm=None)
all_nodes = [] all_nodes = []