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)}"