Spaces:

bstraehle
/

rag

Paused

bstraehle commited on Jan 7, 2024

Commit

7a59153

1 Parent(s): f2edd1d

Update rag_llamaindex.py

Files changed (1) hide show

rag_llamaindex.py CHANGED Viewed

@@ -128,7 +128,7 @@ class LlamaIndexRAG(BaseRAG):
             #top_k = config["k"] #3631 token
         )
-        completion = query_engine.query(prompt, similarity_top_k = config["k"])
         callback = self.get_callback(
             service_context.callback_manager.handlers[0])

             #top_k = config["k"] #3631 token
         )
+        completion = query_engine.query(prompt, top_k = config["k"])
         callback = self.get_callback(
             service_context.callback_manager.handlers[0])