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

Co-authored-by: 刘家星 <liujiaxing@fulan.com.cn>

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

@@ -983,6 +983,7 @@ class GptsAppDao(BaseDao):
                         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 @@ class GptsConversationsDao(BaseDao):
         )
         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 @@ class GptsMessagesDao(BaseDao):
         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