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

This commit is contained in:
Engel Nyst
2024-07-27 19:37:44 +02:00
committed by GitHub
parent 9ed95abf83
commit 3328669b89

View File

@@ -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',