From 3d82a1b52fc273ecfa24c0efca80562ff050a95d Mon Sep 17 00:00:00 2001 From: yhjun1026 <460342015@qq.com> Date: Wed, 7 May 2025 10:27:49 +0800 Subject: [PATCH] feat(agent):change orm default value --- .../src/dbgpt_serve/agent/db/gpts_messages_db.py | 4 +++- .../dbgpt-serve/src/dbgpt_serve/agent/db/gpts_plans_db.py | 6 +++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_messages_db.py b/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_messages_db.py index 82e0df4c2..080a57eab 100644 --- a/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_messages_db.py +++ b/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_messages_db.py @@ -27,7 +27,7 @@ class GptsMessagesEntity(Model): String(255), nullable=False, comment="The unique id of the conversation record" ) message_id = Column( - String(255), nullable=False, comment="The unique id of the messages" + String(255), nullable=False, comment="The unique id of the messages", default="" ) sender = Column( String(255), @@ -37,6 +37,7 @@ class GptsMessagesEntity(Model): sender_name = Column( String(255), nullable=False, + default = "", comment="Who(name) speaking in the current conversation turn", ) receiver = Column( @@ -47,6 +48,7 @@ class GptsMessagesEntity(Model): receiver_name = Column( String(255), nullable=False, + default="", comment="Who(name) receive message in the current conversation turn", ) model_name = Column(String(255), nullable=True, comment="message generate model") diff --git a/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_plans_db.py b/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_plans_db.py index 2cfc065a5..5060c09e7 100644 --- a/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_plans_db.py +++ b/packages/dbgpt-serve/src/dbgpt_serve/agent/db/gpts_plans_db.py @@ -14,12 +14,12 @@ class GptsPlansEntity(Model): conv_id = Column( String(255), nullable=False, comment="The unique id of the conversation record" ) - task_uid = Column(String(255), nullable=False, comment="The uid of the plan task") + task_uid = Column(String(255), nullable=False, default='', comment="The uid of the plan task") sub_task_num = Column(Integer, nullable=False, comment="Subtask id") - conv_round = Column(Integer, nullable=False, comment="The dialogue turns") + conv_round = Column(Integer, nullable=False, default=0, comment="The dialogue turns") conv_round_id = Column(String(255), nullable=True, comment="The dialogue turns uid") - sub_task_id = Column(String(255), nullable=False, comment="Subtask id") + sub_task_id = Column(String(255), nullable=False, default='', comment="Subtask id") task_parent = Column(String(255), nullable=True, comment="Subtask parent task id") sub_task_title = Column(String(255), nullable=False, comment="subtask title") sub_task_content = Column(Text, nullable=False, comment="subtask content")