openai-llm-rag

Build error

bstraehle commited on Jan 7, 2024

Commit

be5fb2d

1 Parent(s): 7a59153

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, 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, response_mode = "compact")
         callback = self.get_callback(
             service_context.callback_manager.handlers[0])