IvDrag89/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# Import tool from Hub11image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)12 13# Below is a tool that creates images of guinea pigs in different environments!14@tool15def guinea_pig_image_generation_tool(number_of_pigs: int, color_of_pigs: str, environment: str) -> str:16    """A tool that creates images of guinea pigs in different environments.17    Args:18        number_of_pigs: The number of guinea pigs to be displayed in the image.19        color_of_pigs: The color of the guinea pigs to be displayed in the image.20        environment: The environment in which the guinea pigs are displayed.21    """22    # Construct the prompt for the image generation tool23    prompt = (24        f"An illustration of {number_of_pigs} {color_of_pigs} guinea pig(s) "25        f"playing in a {environment} environment. Cute, highly detailed, and realistic."26    )27    28    # Generate the image using the image generation tool29    image_url = image_generation_tool(prompt=prompt)30    return f"Here is your image: {image_url}"31 32@tool33def get_current_time_in_timezone(timezone: str) -> str:34    """A tool that fetches the current local time in a specified timezone.35    Args:36        timezone: A string representing a valid timezone (e.g., 'America/New_York').37    """38    try:39        # Create timezone object40        tz = pytz.timezone(timezone)41        # Get current time in that timezone42        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")43        return f"The current local time in {timezone} is: {local_time}"44    except Exception as e:45        return f"Error fetching time for timezone '{timezone}': {str(e)}"46 47 48final_answer = FinalAnswerTool()49 50# 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:51# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 52 53model = HfApiModel(54max_tokens=2096,55temperature=0.5,56model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded57custom_role_conversions=None,58)59 60with open("prompts.yaml", 'r') as stream:61    prompt_templates = yaml.safe_load(stream)62    63agent = CodeAgent(64    model=model,65    tools=[guinea_pig_image_generation_tool, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)66    max_steps=6,67    verbosity_level=1,68    grammar=None,69    planning_interval=None,70    name=None,71    description=None,72    prompt_templates=prompt_templates73)74 75 76GradioUI(agent).launch()