Update rag_llamaindex.py

rag_llamaindex.py

@@ -124,11 +124,11 @@ class LlamaIndexRAG(BaseRAG):
             text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"]),
             service_context = service_context,
             #max_top_k = config["k"], #3631 token
-            similarity_top_k = config["k"],
+            #similarity_top_k = config["k"], #5315 token
             #top_k = config["k"] #3631 token
         )
 
-        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])