fix(chat): update data keys (#923)

QuivrHQ · Aug 10, 2023 · 21db719 · 21db719
1 parent 0cc6dcf
commit 21db719
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/backend/core/llm/qa_base.py b/backend/core/llm/qa_base.py
@@ -140,7 +140,7 @@ def generate_answer(
 
         model_response = qa(
             {
-                "question": question,
+                "question": question.question,
                 "chat_history": transformed_history,
                 "custom_personality": self.get_prompt(),
             }
@@ -227,7 +227,7 @@ async def wrap_done(fn: Awaitable, event: asyncio.Event):
             wrap_done(
                 qa.acall(
                     {
-                        "question": question,
+                        "question": question.question,
                         "chat_history": transformed_history,
                         "custom_personality": self.get_prompt(),
                     }

diff --git a/backend/core/repository/chat/get_chat_history.py b/backend/core/repository/chat/get_chat_history.py
@@ -20,7 +20,7 @@ class GetChatHistoryOutput(BaseModel):
 
     def dict(self, *args, **kwargs):
         chat_history = super().dict(*args, **kwargs)
-        chat_history["chat_id"] = str(chat_history.get("prompt_id"))
+        chat_history["chat_id"] = str(chat_history.get("chat_id"))
         chat_history["message_id"] = str(chat_history.get("message_id"))
 
         return chat_history