YannSADOWSKI/First_agent_template

1from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool2import datetime3import requests4import pytz5import yaml6from tools.final_answer import FinalAnswerTool7 8from Gradio_UI import GradioUI9 10# Below is an example of a tool that does nothing. Amaze us with your creativity !11@tool12def my_cutom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type13    #Keep this format for the description / args / args description but feel free to modify the tool14    """A tool that does nothing yet 15    Args:16        arg1: the first argument17        arg2: the second argument18    """19    return "What magic will you build ?"20 21@tool22def get_current_time_in_timezone(timezone: str) -> str:23    """A tool that fetches the current local time in a specified timezone.24    Args:25        timezone: A string representing a valid timezone (e.g., 'America/New_York').26    """27    try:28        # Create timezone object29        tz = pytz.timezone(timezone)30        # Get current time in that timezone31        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")32        return f"The current local time in {timezone} is: {local_time}"33    except Exception as e:34        return f"Error fetching time for timezone '{timezone}': {str(e)}"35 36 37@tool38def calculator(arg1: int, arg2: int) -> int: 39    """A tool that multiplies two integer arguments to return a final integer.40    41    Args:42        arg1: The first integer argument.43        arg2: The second integer argument.44    45    Returns:46        The product of arg1 and arg2.47    """48    return arg1 * arg249 50@tool51def generate_random_number(min_val: int, max_val: int) -> int:52    """A tool that generates a random integer between a given range.53    54    Args:55        min_val: The minimum value (inclusive).56        max_val: The maximum value (inclusive).57    58    Returns:59        A randomly generated integer.60    """61    import random62    return random.randint(min_val, max_val)63 64 65final_answer = FinalAnswerTool()66model = HfApiModel(67max_tokens=2096,68temperature=0.5,69model_id='Qwen/Qwen2.5-Coder-32B-Instruct',70custom_role_conversions=None,71)72 73 74# Import tool from Hub75image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)76 77with open("prompts.yaml", 'r') as stream:78    prompt_templates = yaml.safe_load(stream)79    80agent = CodeAgent(81    model=model,82    tools=[final_answer, calculator, get_current_time_in_timezone,generate_random_number], ## add your tools here (don't remove final answer)83    max_steps=6,84    verbosity_level=1,85    grammar=None,86    planning_interval=None,87    name=None,88    description=None,89    prompt_templates=prompt_templates90)91 92 93GradioUI(agent).launch()