openai-llm-rag

Build error

bstraehle commited on Jan 6, 2024

Commit

e55f593

1 Parent(s): d798c18

Update rag_langchain.py

Files changed (1) hide show

rag_langchain.py CHANGED Viewed

@@ -130,6 +130,6 @@ class LangChainRAG(BaseRAG):
         )
         with get_openai_callback() as callback:
-            completion = rag_chain({"question": prompt})
         return completion, rag_chain, callback

         )
         with get_openai_callback() as callback:
+            completion = rag_chain({"query": prompt})
         return completion, rag_chain, callback