From 78c036846303b818e30bec2e028b8838a35df030 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=98=8E=E5=A4=A9?= <460342015@qq.com> Date: Sat, 21 Dec 2024 12:47:33 +0800 Subject: [PATCH] feat(agent):fix agent overrid bug (#2235) Co-authored-by: Fangyin Cheng --- dbgpt/serve/agent/agents/expand/app_start_assisant_agent.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/dbgpt/serve/agent/agents/expand/app_start_assisant_agent.py b/dbgpt/serve/agent/agents/expand/app_start_assisant_agent.py index fc0cf2755..3c5182b36 100644 --- a/dbgpt/serve/agent/agents/expand/app_start_assisant_agent.py +++ b/dbgpt/serve/agent/agents/expand/app_start_assisant_agent.py @@ -55,6 +55,7 @@ def prepare_act_param( received_message: Optional[AgentMessage], sender: Agent, rely_messages: Optional[List[AgentMessage]] = None, + **kwargs, ) -> Dict[str, Any]: return { "user_input": received_message.content, @@ -67,9 +68,10 @@ async def _load_thinking_messages( received_message: AgentMessage, sender: Agent, rely_messages: Optional[List[AgentMessage]] = None, + historical_dialogues: Optional[List[AgentMessage]] = None, context: Optional[Dict[str, Any]] = None, - is_retry_chat: Optional[bool] = False, - ) -> tuple[List[AgentMessage], Optional[Dict]]: + is_retry_chat: bool = False, + ) -> Tuple[List[AgentMessage], Optional[Dict]]: if rely_messages and len(rely_messages) > 0: return rely_messages[-1:], None else: