Spaces:
Build error
Build error
Add max_steps parameter to ToolCallingAgent in graph.py for improved agent control
Browse files
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
|