feat(agent):fix agent overrid bug (#2235)

Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
eosphoros-ai · Dec 21, 2024 · 78c0368 · 78c0368
1 parent 1bdf695
commit 78c0368
Showing 1 changed file with 4 additions and 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
@@ -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: