Fix json schema for final answer

src/smolagents/agents.py

@@ -539,6 +539,7 @@ class MultiStepAgent:
             final_step_log = ActionStep(error=AgentMaxIterationsError(error_message))
             self.logs.append(final_step_log)
             final_answer = self.provide_final_answer(task)
+            console.print(Text(f"Final answer: {final_answer}"))
             final_step_log.action_output = final_answer
             final_step_log.end_time = time.time()
             final_step_log.duration = step_log.end_time - step_start_time
@@ -588,6 +589,7 @@ class MultiStepAgent:
             final_step_log = ActionStep(error=AgentMaxIterationsError(error_message))
             self.logs.append(final_step_log)
             final_answer = self.provide_final_answer(task)
+            console.print(Text(f"Final answer: {final_answer}"))
             final_step_log.action_output = final_answer
             final_step_log.duration = 0
             for callback in self.step_callbacks:

src/smolagents/default_tools.py

@@ -105,6 +105,7 @@ class PythonInterpreterTool(Tool):
 
     def forward(self, code: str) -> str:
         state = {}
+        try:
             output = str(
                 self.python_evaluator(
                     code,
@@ -114,6 +115,8 @@ class PythonInterpreterTool(Tool):
                 )
             )
             return f"Stdout:\n{state['print_outputs']}\nOutput: {output}"
+        except Exception as e:
+            return f"Error: {str(e)}"
 
 
 class FinalAnswerTool(Tool):

src/smolagents/models.py

@@ -66,6 +66,10 @@ tool_role_conversions = {
 
 
 def get_json_schema(tool: Tool) -> Dict:
+    properties = deepcopy(tool.inputs)
+    for value in properties.values():
+        if value["type"] == "any":
+            value["type"] = "string"
     return {
         "type": "function",
         "function": {
@@ -73,10 +77,7 @@ def get_json_schema(tool: Tool) -> Dict:
             "description": tool.description,
             "parameters": {
                 "type": "object",
-                "properties": {
-                    k: {k2: v2.replace("any", "object") for k2, v2 in v.items()}
-                    for k, v in tool.inputs.items()
-                },
+                "properties": properties,
                 "required": list(tool.inputs.keys()),
             },
         },