Update rag_llamaindex.py

rag_llamaindex.py

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