github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-31 16:39:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(chat_history): chat_history_db mysql message query bug (#989)

Browse Source
This commit is contained in:
明天
2023-12-28 14:26:22 +08:00
committed by GitHub
parent fd3a5d2bfa
commit 96474e9148
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
dbgpt/storage/chat_history/chat_history_db.py
View File

@@ -106,4 +106,11 @@ class ChatHistoryDao(BaseDao):
chat_history.delete()
def get_by_uid(self, conv_uid: str) -> ChatHistoryEntity:
return ChatHistoryEntity.query.filter_by(conv_uid=conv_uid).first()
# return ChatHistoryEntity.query.filter_by(conv_uid=conv_uid).first()
session = self.get_raw_session()
chat_history = session.query(ChatHistoryEntity)
chat_history = chat_history.filter(ChatHistoryEntity.conv_uid == conv_uid)
result = chat_history.first()
session.close()
return result