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

packages/dbgpt-core/src/dbgpt/core/interface/message.py

@@ -1150,6 +1150,12 @@ def delete(self) -> None:
 
     def clear(self) -> None:
         """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
         message_list = self._get_message_items()
         message_ids = [message.identifier for message in message_list]

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

@@ -983,6 +983,7 @@ def edit(self, gpts_app: GptsApp):
                         params=json.dumps(recommend_question.params),
                         valid=recommend_question.valid,
                         chat_mode=chat_scene,
+                        user_code=gpts_app.user_code,
                     )
                 )
             session.add_all(recommend_questions)

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

@@ -111,3 +111,13 @@ def update(self, conv_id: str, state: str):
         )
         session.commit()
         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

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

@@ -177,3 +177,11 @@ def get_last_message(self, conv_id: str) -> Optional[GptsMessagesEntity]:
         result = gpts_messages.first()
         session.close()
         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