@@ -411,7 +411,7 @@ class WriteEmail(BaseNode[State]):
411
411
prompt,
412
412
message_history = ctx.state.write_agent_messages,
413
413
)
414
- ctx.state.write_agent_messages += result.all_messages ()
414
+ ctx.state.write_agent_messages += result.new_messages ()
415
415
return Feedback(result.output)
416
416
417
417
@@ -681,7 +681,7 @@ Instead of running the entire graph in a single process invocation, we run the g
681
681
'Ask a simple question with a single correct answer.',
682
682
message_history=ctx.state.ask_agent_messages,
683
683
)
684
- ctx.state.ask_agent_messages += result.all_messages ()
684
+ ctx.state.ask_agent_messages += result.new_messages ()
685
685
ctx.state.question = result.output
686
686
return Answer(result.output)
687
687
@@ -722,7 +722,7 @@ Instead of running the entire graph in a single process invocation, we run the g
722
722
format_as_xml({'question': ctx.state.question, 'answer': self.answer}),
723
723
message_history=ctx.state.evaluate_agent_messages,
724
724
)
725
- ctx.state.evaluate_agent_messages += result.all_messages ()
725
+ ctx.state.evaluate_agent_messages += result.new_messages ()
726
726
if result.output.correct:
727
727
return End(result.output.comment)
728
728
else:
0 commit comments