Upload tool

tool.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import pytz
 import datetime
+import pytz
 
 class GetTime(Tool):
     name = "get_time"
@@ -9,7 +9,7 @@ class GetTime(Tool):
     inputs = {'tz': {'type': 'string', 'description': 'the timezone (such as Asia/Shanghai, Europe/Amsterdam, etc.)'}}
     output_type = "string"
 
-    def forward(self, tz: str) -> str:
+    def forward(self, tz: str) -> datetime:
         """Get the current time in a specific timezone.
 
         Args:
@@ -18,7 +18,7 @@ class GetTime(Tool):
         import pytz
         from datetime import datetime
 
-        return datetime.now(pytz.timezone(tz)).strftime("%Y-%m-%d %H:%M:%S")
+        return datetime.now(pytz.timezone(tz))
 
     def __init__(self, *args, **kwargs):
         self.is_initialized = False