Update rag_llamaindex.py

rag_llamaindex.py

@@ -121,14 +121,13 @@ class LlamaIndexRAG(BaseRAG):
         service_context = self.get_service_context(config)
         
         query_engine = index.as_query_engine(
-            text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"]),
+            response_mode = "compact",
             service_context = service_context,
-            #max_top_k = config["k"], #3631 token
-            #similarity_top_k = config["k"], #5315 token
-            #top_k = config["k"] #3631 token
+            similarity_top_k = config["k"],
+            text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"])
         )
 
-        completion = query_engine.query(prompt, response_mode = "compact")
+        completion = query_engine.query(prompt)
         callback = self.get_callback(
             service_context.callback_manager.handlers[0])