Update agent.py

agent.py

@@ -2,6 +2,7 @@ import os
 import json
 from smolagents import MultiStepAgent, PythonInterpreterTool
 from smolagents.agents import ActionOutput
+from smolagents.utils import AgentToolExecutionError
 from models import ModelManager
 from tools import search_web, scrape_website, read_file
 
@@ -86,27 +87,26 @@ If you have the final answer, respond with an empty plan and the answer in the '
             memory_step.tool_calls = tool_calls
             yield tool_call
 
-            # On exécute l'outil
-            tool_output_value = self.execute_tool_call(tool_call.name, tool_call.arguments)
+            tool_name = tool_call.function.name
+            tool_arguments = tool_call.function.arguments
             
-            # On vérifie si c'est la réponse finale
-            if tool_call.name == "final_answer":
+            tool_output_value = self.execute_tool_call(tool_name, tool_arguments)
+            
+            if tool_name == "final_answer":
                 final_answer = tool_output_value
                 is_final = True
 
-            # On génère une observation textuelle pour la prochaine étape de l'IA
             observation = self.render_tool_result(tool_output_value)
             memory_step.observations = observation
             
             yield {"tool_call_id": tool_call.id, "output": observation}
 
-        # 5. Renvoyer le résultat final de l'étape
         yield ActionOutput(output=final_answer, is_final_answer=is_final)
 
 
     def execute_tool_call(self, tool_name: str, arguments: dict) -> any:
         """
-        Exécute un outil avec les arguments fournis. C'est la fonction qui manquait.
+        Exécute un outil avec les arguments fournis.
         """
         if tool_name not in self.tools:
             raise AgentToolExecutionError(f"Outil inconnu '{tool_name}'. Les outils disponibles sont : {', '.join(self.tools.keys())}", self.logger)
@@ -114,11 +114,12 @@ If you have the final answer, respond with an empty plan and the answer in the '
         tool = self.tools[tool_name]
         
         try:
-            # La plupart des outils attendent les arguments directement
+            if arguments is None:
+                arguments = {}
             return tool(**arguments)
         except TypeError:
-            # Certains outils plus simples attendent un seul argument positionnel
             try:
+                # Gère le cas où l'outil attend un seul argument
                 return tool(list(arguments.values())[0])
             except Exception as e:
                 raise AgentToolExecutionError(f"Erreur lors de l'exécution de l'outil '{tool_name}' avec les arguments {arguments}: {e}", self.logger)