Spaces:
Sleeping
Sleeping
File size: 1,030 Bytes
25d91b2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import datetime
import pytz
from smolagents.tools import Tool
class GetCurrentTimeTool(Tool):
name = "get_current_time_in_timezone"
description = "A tool that fetches the current local time in a specified timezone."
inputs = {'timezone': {'type': 'string', 'description': 'A string representing a valid timezone (e.g., \'America/New_York\').'}}
output_type = "string"
def forward(self, timezone: str) -> str:
"""A tool that fetches the current local time in a specified timezone.
Args:
timezone: A string representing a valid timezone (e.g., 'America/New_York').
"""
try:
# Create timezone object
tz = pytz.timezone(timezone)
# Get current time in that timezone
local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
return f"The current local time in {timezone} is: {local_time}"
except Exception as e:
return f"Error fetching time for timezone '{timezone}': {str(e)}"
|