File size: 2,837 Bytes
92c9ccd 9b5b26a c19d193 92c9ccd 6aae614 9b5b26a 92c9ccd 9b5b26a 92c9ccd 9b5b26a 92c9ccd 9b5b26a 92c9ccd 9b5b26a 92c9ccd 9b5b26a 92c9ccd 9b5b26a 8c01ffb 92c9ccd 8c01ffb 92c9ccd 6aae614 e121372 92c9ccd 13d500a 8c01ffb 92c9ccd 9b5b26a 8c01ffb 861422e 92c9ccd 8c01ffb 8fe992b 92c9ccd 8c01ffb 92c9ccd 8fe992b 92c9ccd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
import random # Import for random.choice
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
# Example of a tool that does nothing (placeholder for creativity)
@tool
def my_custom_tool(arg1: str, arg2: int) -> str: # Fix typo in function name
"""A tool that does nothing yet.
Args:
arg1: The first argument.
arg2: The second argument.
Returns:
A string indicating this tool is currently a placeholder.
"""
return "What magic will you build?"
# Tool to fetch the current time in a specific timezone
@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').
Returns:
A string with the current local time in the given timezone or an error message.
"""
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)}"
# Creative tool to share a random fun fact
@tool
def get_fun_fact() -> str:
"""A tool that shares a random fun fact.
Returns:
A string containing a fun fact.
"""
fun_facts = [
"Honey never spoils. Archaeologists have found pots of honey in ancient Egyptian tombs that are over 3,000 years old!",
"Octopuses have three hearts!",
"Bananas are berries, but strawberries aren't.",
"Your heartbeat changes with the music you listen to."
]
return random.choice(fun_facts)
# Final answer and agent setup
final_answer = FinalAnswerTool()
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
custom_role_conversions=None,
)
# Load image generation tool
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)
# Create the agent and add tools
agent = CodeAgent(
model=model,
tools=[
final_answer,
get_current_time_in_timezone,
get_fun_fact, # Add fun fact tool
image_generation_tool, # Add image generation tool
],
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates,
)
# Launch Gradio UI
GradioUI(agent).launch()
|