mjschock commited on
Commit
dfc37bb
·
unverified ·
1 Parent(s): bc3bc22

Add max_steps parameter to ToolCallingAgent in graph.py for improved agent control

Browse files
Files changed (1) hide show
  1. graph.py +1 -0
graph.py CHANGED
@@ -56,6 +56,7 @@ class AgentNode:
56
  )
57
 
58
  self.agent = ToolCallingAgent(
 
59
  model=self.model,
60
  prompt_templates=prompt_templates,
61
  tools=tools
 
56
  )
57
 
58
  self.agent = ToolCallingAgent(
59
+ max_steps=1,
60
  model=self.model,
61
  prompt_templates=prompt_templates,
62
  tools=tools