fix: delegateAgent select micro agents name bug. (#1601)

* fix: delegateAgent select micro agents name bug.

* refactor: last_observation

---------

Co-authored-by: aaren.xzh <aaren.xzh@antfin.com>
This commit is contained in:
Xia Zhenhua
2024-05-06 20:27:03 +08:00
committed by GitHub
parent 16400e40ca
commit c9970e6817

View File

@@ -42,40 +42,40 @@ class DelegatorAgent(Agent):
agent='StudyRepoForTaskAgent', inputs={'task': state.plan.main_goal}
)
lastObservation = state.history[-1][1]
if not isinstance(lastObservation, AgentDelegateObservation):
last_observation = state.history[-1][1]
if not isinstance(last_observation, AgentDelegateObservation):
raise Exception('Last observation is not an AgentDelegateObservation')
if self.current_delegate == 'study':
self.current_delegate = 'coder'
return AgentDelegateAction(
agent='Coder',
agent='CoderAgent',
inputs={
'task': state.plan.main_goal,
'summary': lastObservation.outputs['summary'],
'summary': last_observation.outputs['summary'],
},
)
elif self.current_delegate == 'coder':
self.current_delegate = 'verifier'
return AgentDelegateAction(
agent='Verifier',
agent='VerifierAgent',
inputs={
'task': state.plan.main_goal,
},
)
elif self.current_delegate == 'verifier':
if (
'completed' in lastObservation.outputs
and lastObservation.outputs['completed']
'completed' in last_observation.outputs
and last_observation.outputs['completed']
):
return AgentFinishAction()
else:
self.current_delegate = 'coder'
return AgentDelegateAction(
agent='Coder',
agent='CoderAgent',
inputs={
'task': state.plan.main_goal,
'summary': lastObservation.outputs['summary'],
'summary': last_observation.outputs['summary'],
},
)
else: