From 727305fbd40ba2c6f281bd13ac79f964998e812a Mon Sep 17 00:00:00 2001 From: hailin Date: Fri, 9 May 2025 22:27:40 +0800 Subject: [PATCH] . --- scripts/rag_build_query.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py index 2b8035c..d1e9777 100644 --- a/scripts/rag_build_query.py +++ b/scripts/rag_build_query.py @@ -57,7 +57,7 @@ def build_user_index(user_id: str): documents = SimpleDirectoryReader(doc_dir).load_data() 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( 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: 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 = []