Update model.py

ThisIs-Developer · Jan 11, 2024 · 0138ee1 · 0138ee1
1 parent 116b9f9
commit 0138ee1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -83,7 +83,7 @@ async def main(message):
         stream_final_answer=True, answer_prefix_tokens=["FINAL", "ANSWER"]
     )
     cb.answer_reached = True
-    res = await chain.acall(message, callbacks=[cb])
+    res = await chain.acall(message.content, callbacks=[cb])
     answer = res["result"]
     sources = res["source_documents"]