Update tools/final_answer.py

tools/final_answer.py

@@ -1,10 +1,18 @@
-from typing import Any, Optional
+from typing import Any
 from smolagents.tools import Tool
 
+
 class FinalAnswerTool(Tool):
     name = "final_answer"
-    description = "Provides a final answer to the given problem."
-    inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
+    description = "Provides the final answer to the user. This can be text or an image."
+    
+    inputs = {
+        "answer": {
+            "type": "any",
+            "description": "The final answer to the user, which can be text or an image."
+        }
+    }
+    
     output_type = "any"
 
     def forward(self, answer: Any) -> Any: