Add max steps

agent.py

@@ -23,7 +23,7 @@ class WikipediaAgentTool(Tool):
     inputs = {"query": {"type": "string", "description": "Search query for Wikipedia or Web search."}}
     output_type = "string"
     def forward(self, query: str) -> str:
-        return agent_wikipedia.run(query)
+        return agent_wikipedia.run(query,max_steps=5)
 
 class MediaAgentTool(Tool):
     name = "MediaAgentTool"
@@ -34,7 +34,7 @@ class MediaAgentTool(Tool):
 
     def forward(self, query: str) -> str:
         # Utilise l'agent vidéo pour rechercher sur YouTube
-        return agent_multimedia.run(query)
+        return agent_multimedia.run(query,max_steps=5)
 
 
 class CodeAgentTool(Tool):
@@ -45,7 +45,7 @@ class CodeAgentTool(Tool):
 
     def forward(self, query: str) -> str:
         # Utilise l'agent code pour exécuter du code Python
-        return agent_code.run(query)
+        return agent_code.run(query,max_steps=5)
 
 
 
@@ -69,7 +69,7 @@ manager_agent = ToolCallingAgent(
 # Pour interagir avec l'agent manager
 async def run_multi_agent_system(prompt: str):
     print(f"\nRequête au Manager : {prompt}")
-    return manager_agent.run(prompt)
+    return manager_agent.run(prompt,max_steps=5)
 
 
 class BasicAgent: