StanGirard/quivr
1
1
Fork 0
mirror of https://github.com/StanGirard/quivr.git synced 2024-11-09 17:15:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Mamadou DICKO 2023-08-10 18:35:30 +02:00 committed by GitHub
parent 0cc6dcfa47
commit 21db719796
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/core/llm/qa_base.py
View File

@ -140,7 +140,7 @@ class QABaseBrainPicking(BaseBrainPicking):
model_response = qa(
{
"question": question,
"question": question.question,
"chat_history": transformed_history,
"custom_personality": self.get_prompt(),
}
@ -227,7 +227,7 @@ class QABaseBrainPicking(BaseBrainPicking):
wrap_done(
qa.acall(
{
"question": question,
"question": question.question,
"chat_history": transformed_history,
"custom_personality": self.get_prompt(),
}

2
backend/core/repository/chat/get_chat_history.py
View File

@ -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