diff --git a/examples/workarena/agent.py b/examples/workarena/agent.py index 7d2ce92..d99e32c 100644 --- a/examples/workarena/agent.py +++ b/examples/workarena/agent.py @@ -107,7 +107,7 @@ def prepare_tape(self, tape: WorkArenaTape, max_chars: int = 100): class WorkArenaAgent(Agent): @classmethod - def create(cls, llm: LLM): + def create(cls, llm: LLM, max_iterations: int = 4): return super().create( llm, nodes=[ @@ -115,5 +115,5 @@ def create(cls, llm: LLM): WorkArenaNode(name="reflect", guidance=PromptRegistry.reflect), WorkArenaNode(name="act", guidance=PromptRegistry.act, next_node="reflect"), ], - max_iterations=2, + max_iterations=max_iterations, )