Javier-DlaP/First_agent_template

1from smolagents import CodeAgent, HfApiModel ,tool2import datetime3import requests4import pytz5import yaml6from tools.final_answer import FinalAnswerTool7 8from Gradio_UI import GradioUI9import wikipedia10 11@tool12def get_wikipedia_page(page_name: str) -> str:13    """A tool that fetches the Wikipedia page for a given page name.14    Args:15        page_name: A string representing the name of the Wikipedia page to fetch.16    """17 18    try:19        page = wikipedia.page(page_name, auto_suggest=True)20        return page.content21    except Exception as e:22        return f"Error fetching Wikipedia page.\n{e}"23 24@tool25def get_current_time_in_timezone(timezone: str) -> str:26    """A tool that fetches the current local time in a specified timezone.27    Args:28        timezone: A string representing a valid timezone (e.g., 'America/New_York').29    """30    try:31        # Create timezone object32        tz = pytz.timezone(timezone)33        # Get current time in that timezone34        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")35        return f"The current local time in {timezone} is: {local_time}"36    except Exception as e:37        return f"Error fetching time for timezone '{timezone}': {str(e)}"38 39 40final_answer = FinalAnswerTool()41 42# 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:43# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 44 45model = HfApiModel(46max_tokens=2096,47temperature=0.5,48model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded49custom_role_conversions=None,50)51 52with open("prompts.yaml", 'r') as stream:53    prompt_templates = yaml.safe_load(stream)54    55agent = CodeAgent(56    model=model,57    tools=[get_wikipedia_page, 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()