fix Finish action to sent its 'thoughts' in the prompt (OpenHands#3149)

enyst · web-flow · commit 3328669b8986 · 2024-07-27T17:37:44.000Z
diff --git a/agenthub/codeact_agent/codeact_agent.py b/agenthub/codeact_agent/codeact_agent.py
@@ -121,6 +121,8 @@ def action_to_str(self, action: Action) -> str:
             return f'{action.thought}\n<execute_browse>\n{action.inputs["task"]}\n</execute_browse>'
         elif isinstance(action, MessageAction):
             return action.content
+        elif isinstance(action, AgentFinishAction) and action.source == 'agent':
+            return action.thought
         return ''
 
     def get_action_message(self, action: Action) -> dict[str, str] | None:
@@ -129,6 +131,7 @@ def get_action_message(self, action: Action) -> dict[str, str] | None:
             or isinstance(action, CmdRunAction)
             or isinstance(action, IPythonRunCellAction)
             or isinstance(action, MessageAction)
+            or (isinstance(action, AgentFinishAction) and action.source == 'agent')
         ):
             return {
                 'role': 'user' if action.source == 'user' else 'assistant',
