from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool import datetime import requests import pytz import yaml import os from tools.final_answer import FinalAnswerTool from Gradio_UI import GradioUI # Below is an example of a tool that does nothing. Amaze us with your creativity ! @tool def my_custom_tool(arg1:int, arg2:int, arg3:str)-> str: #it's import to specify the return type """A tool that performs the specified mathematical operation on the specified integer arguments Args: arg1: the first integer arg2: the second integer arg3: the operator as string. Supported operations are "+","-","*","/" """ match arg3: case "*": result = arg1 * arg2 return f"The result is: '{result}'" case "/": if(arg2!=0): result = arg1 / arg2 return f"The result is: '{result}'" else: return f"Error: Division by zero not possible" case "+": result = arg1 + arg2 return f"The result is: '{result}'" case "-": result = arg1 - arg2 return f"The result is: '{result}'" case _: return f"Error: '{arg3}' is an unsupported operation, Supported operations are +,-,*,/" @tool def get_current_time_in_timezone(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)}" 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 #model_id='https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud', # worked!! #model_id='Qwen/Qwen2.5-Coder-3B-Instruct', #model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B', model_id='https://i3krvakf2arisixm.us-east-1.aws.endpoints.huggingface.cloud', token=os.getenv('HF_TOKEN_WRITE'), 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') as stream: prompt_templates = yaml.safe_load(stream) agent = CodeAgent( model=model, tools=[final_answer, my_custom_tool, get_current_time_in_timezone], ## 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 ) GradioUI(agent).launch()