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

eosphoros-ai · Dec 28, 2023 · 96474e9 · 96474e9
1 parent fd3a5d2
commit 96474e9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/dbgpt/storage/chat_history/chat_history_db.py b/dbgpt/storage/chat_history/chat_history_db.py
@@ -106,4 +106,11 @@ def raw_delete(self, conv_uid: int):
             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