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

feat(dbgpt-serve): Agent fixes and parameter additions (#2457)

Browse Source 
Co-authored-by: 刘家星 <liujiaxing@fulan.com.cn>
This commit is contained in:
北极 2025-03-13 14:43:34 +08:00 committed by GitHub
parent a115c57faa
commit ccfce5d279
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
packages/dbgpt-core/src/dbgpt/core/interface/message.py
View File

@ -1150,6 +1150,12 @@ class StorageConversation(OnceConversation, StorageItem):
def clear(self) -> None: def clear(self) -> None:
"""Clear all the messages and conversation.""" """Clear all the messages and conversation."""
from dbgpt_serve.agent.db import GptsConversationsDao, GptsMessagesDao
# 清除Agent聊天记录
GptsMessagesDao().delete_chat_message(conv_id=self.conv_uid)
GptsConversationsDao().delete_chat_message(conv_id=self.conv_uid)
# Clear messages first # Clear messages first
message_list = self._get_message_items() message_list = self._get_message_items()
message_ids = [message.identifier for message in message_list] message_ids = [message.identifier for message in message_list]

1
packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_app.py
View File

@ -983,6 +983,7 @@ class GptsAppDao(BaseDao):
params=json.dumps(recommend_question.params), params=json.dumps(recommend_question.params),
valid=recommend_question.valid, valid=recommend_question.valid,
chat_mode=chat_scene, chat_mode=chat_scene,
user_code=gpts_app.user_code,
) )
) )
session.add_all(recommend_questions) session.add_all(recommend_questions)

10
packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_conversations_db.py
View File

@ -111,3 +111,13 @@ class GptsConversationsDao(BaseDao):
) )
session.commit() session.commit()
session.close() session.close()
def delete_chat_message(self, conv_id: str) -> bool:
session = self.get_raw_session()
gpts_messages = session.query(GptsConversationsEntity)
gpts_messages.filter(
GptsConversationsEntity.conv_id.like(f"%{conv_id}%")
).delete()
session.commit()
session.close()
return True

8
packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_messages_db.py
View File

@ -177,3 +177,11 @@ class GptsMessagesDao(BaseDao):
result = gpts_messages.first() result = gpts_messages.first()
session.close() session.close()
return result return result
def delete_chat_message(self, conv_id: str) -> bool:
session = self.get_raw_session()
gpts_messages = session.query(GptsMessagesEntity)
gpts_messages.filter(GptsMessagesEntity.conv_id.like(f"%{conv_id}%")).delete()
session.commit()
session.close()
return True