Sfoglia il codice sorgente

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

Engel Nyst 1 anno fa
parent
commit
3328669b89
1 ha cambiato i file con 3 aggiunte e 0 eliminazioni
  1. 3 0
      agenthub/codeact_agent/codeact_agent.py

+ 3 - 0
agenthub/codeact_agent/codeact_agent.py

@@ -121,6 +121,8 @@ class CodeActAgent(Agent):
             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 @@ class CodeActAgent(Agent):
             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',