chpusch/First_agent_template

1from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool2import datetime3import requests4import pytz5import yaml6import random7from tools.final_answer import FinalAnswerTool8 9from Gradio_UI import GradioUI10 11# Below is an example of a tool that does nothing. Amaze us with your creativity !12@tool13def randomecity (current_timezone: str, random_city: str)-> str: 14    #Keep this format for the description / args / args description but feel free to modify the tool15    """Provides a city within the current timezone 16    Args:17        current_timezone: A string representing the timezone of the user18        random_city: A string representing a valid city within the timezone of the user19    """20    current_timezone = time.tzname[time.daylight]21    22    # Zuordnung von Zeitzonen zu Städten23    timezone_to_cities = {24    'Europe/Paris': ['Paris', 'Berlin', 'Madrid', 'Rome'],25    'America/New_York': ['New York', 'Boston', 'Philadelphia'],26    'Asia/Tokyo': ['Tokyo', 'Osaka', 'Kyoto'],27    }28 29    # Wähle eine zufällige Stadt aus der gegebenen Zeitzone30    cities_in_timezone = timezone_to_cities.get(current_timezone, [])31    random_city = random.choice(cities_in_timezone) if cities_in_timezone else "Keine Stadt gefunden"32 33    return f"Eine Stadt in der Zeitzone {timezone} ist {random_city}."34 35 36final_answer = FinalAnswerTool()37 38# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:39# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 40 41model = HfApiModel(42max_tokens=2096,43temperature=0.5,44model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded45custom_role_conversions=None,46)47 48 49# Import tool from Hub50image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)51 52with open("prompts.yaml", 'r') as stream:53    prompt_templates = yaml.safe_load(stream)54    55agent = CodeAgent(56    model=model,57    tools=[final_answer], ## add your tools here (don't remove final answer)58    max_steps=6,59    verbosity_level=1,60    grammar=None,61    planning_interval=None,62    name=None,63    description=None,64    prompt_templates=prompt_templates65)66 67 68GradioUI(agent).launch()