Update tools/final_answer.py

tools/final_answer.py

@@ -1,6 +1,8 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
 
+from smolagents.agent_types import AgentImage
+
 class FinalAnswerTool(Tool):
     name = "final_answer"
     description = "Provides a final answer to the given problem."
@@ -8,6 +10,9 @@ class FinalAnswerTool(Tool):
     output_type = "any"
 
     def forward(self, answer: Any) -> Any:
+        # unwrap AgentImage to raw PIL image
+        if isinstance(answer, AgentImage):
+            return answer.image
         return answer
 
     def __init__(self, *args, **kwargs):