Spaces:
Sleeping
Sleeping
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
|