Update rag_llamaindex.py

rag_llamaindex.py

@@ -123,12 +123,13 @@ 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)
+        completion = query_engine.query(
+            prompt, similarity_top_k = config["k"])
         callback = self.get_callback(
             service_context.callback_manager.handlers[0])