feat(core): Upgrade pydantic to 2.x (#1428)

dbgpt/agent/expand/dashboard_assistant_agent.py

@@ -34,8 +34,6 @@ class DashboardAssistantAgent(ConversableAgent):
         "professional reports"
     )
 
-    max_retry_count: int = 3
-
     def __init__(self, **kwargs):
         """Create a new instance of DashboardAssistantAgent."""
         super().__init__(**kwargs)

dbgpt/agent/expand/data_scientist_agent.py

@@ -17,7 +17,7 @@ logger = logging.getLogger(__name__)
 class DataScientistAgent(ConversableAgent):
     """Data Scientist Agent."""
 
-    name = "Edgar"
+    name: str = "Edgar"
     profile: str = "DataScientist"
     goal: str = (
         "Use correct {dialect} SQL to analyze and solve tasks based on the data"

dbgpt/agent/expand/plugin_assistant_agent.py

@@ -16,7 +16,7 @@ class PluginAssistantAgent(ConversableAgent):
 
     plugin_generator: Optional[PluginPromptGenerator] = None
 
-    name = "LuBan"
+    name: str = "LuBan"
     profile: str = "ToolExpert"
     goal: str = (
         "Read and understand the tool information given in the resources below to "

dbgpt/agent/expand/retrieve_summary_assistant_agent.py

@@ -244,7 +244,7 @@ class RetrieveSummaryAssistantAgent(ConversableAgent):
                 **act_extent_param,
             )
             if act_out:
-                reply_message.action_report = act_out.dict()
+                reply_message.action_report = act_out.to_dict()
             # 4.Reply information verification
             check_pass, reason = await self.verify(reply_message, sender, reviewer)
             is_success = check_pass

dbgpt/agent/expand/summary_assistant_agent.py

@@ -11,7 +11,7 @@ logger = logging.getLogger(__name__)
 class SummaryAssistantAgent(ConversableAgent):
     """Summary Assistant Agent."""
 
-    name = "Aristotle"
+    name: str = "Aristotle"
     profile: str = "Summarizer"
     goal: str = (
         "Summarize answer summaries based on user questions from provided "