"""Run hf agent.""" # pylint: disable=line-too-long, broad-exception-caught import datetime # import requests import pytz import yaml from Gradio_UI import GradioUI from loguru import logger from smolagents import ( CodeAgent, # DuckDuckGoSearchTool, HfApiModel, load_tool, tool, ) from tools.final_answer import FinalAnswerTool # Below is an example of a tool that does nothing. Amaze us with your creativity ! @tool def my_cutom_tool(arg1: str, arg2: int) -> str: # it's import to specify the return type # Keep this format for the description / args / args description but feel free to modify the tool """ Run a tool that does nothing yet. Args: arg1: the first argument arg2: the second argument """ return f"What magic will you build? {arg1=}, {arg2=}" @tool def get_current_time_in_timezone(timezone: str) -> str: """ Run 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'). """ logger.debug(f" get_current_time_in_timezone {timezone=}") 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: logger.error(e) return f"Error fetching time for timezone '{timezone}': {str(e)}" final_answer = FinalAnswerTool() model = HfApiModel( max_tokens=2096, temperature=0.5, model_id="https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud", # it is possible that this model may be overloaded custom_role_conversions=None, ) # Import tool from Hub image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) with open("prompts.yaml", "r", encoding="utf8") as stream: prompt_templates = yaml.safe_load(stream) agent = CodeAgent( model=model, tools=[ my_cutom_tool, get_current_time_in_timezone, final_answer, ], ## add your tools here (don't remove final answer) max_steps=6, verbosity_level=1, grammar=None, planning_interval=None, name=None, description=None, prompt_templates=prompt_templates, ) # debug = True # share = True ssr = True GradioUI(agent).launch( # debug=debug, # already set in GradioUI # share=share, # ssr=ssr, )