Update rag_llamaindex.py

rag_llamaindex.py

@@ -123,8 +123,9 @@ class LlamaIndexRAG(BaseRAG):
         query_engine = index.as_query_engine(
             text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"]),
             service_context = service_context,
-            #similarity_top_k = config["k"]
-            max_top_k = config["k"]
+            max_top_k = config["k"],
+            similarity_top_k = config["k"],
+            top_k = config["k"]
         )
 
         completion = query_engine.query(prompt)