Spaces:
Sleeping
Sleeping
File size: 3,523 Bytes
26956d8 9b5b26a c19d193 6aae614 8fe992b 9b5b26a 26956d8 9b5b26a 26956d8 9b5b26a 8c01ffb 26956d8 8c01ffb 6aae614 ae7a494 26956d8 ae7a494 e121372 26956d8 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e 9b5b26a 8c01ffb 8fe992b 26956d8 8c01ffb 861422e 8fe992b 26956d8 |
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 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
@tool
def get_random_fact() -> str:
"""Returns a fun random fact fetched from an API."""
try:
response = requests.get("https://uselessfacts.jsph.pl/random.json?language=en")
fact = response.json().get("text", "No fact available.")
return fact
except Exception as e:
return f"Error fetching fact: {str(e)}"
@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').
"""
try:
tz = pytz.timezone(timezone)
local_time = datetime.datetime.now(tz)
formatted_time = local_time.strftime("%Y-%m-%d %H:%M:%S")
return formatted_time
except Exception as e:
return f"Error fetching time for timezone '{timezone}': {str(e)}"
@tool
def get_fact_with_greeting(timezone: str) -> str:
"""
Returns an entertaining message that includes an appropriate greeting based on the
current time in the given timezone along with a fun random fact.
Args:
timezone: A string representing a valid timezone (e.g., 'America/New_York').
"""
# Get the current time string and also a datetime object for hour extraction.
try:
tz = pytz.timezone(timezone)
local_time = datetime.datetime.now(tz)
formatted_time = local_time.strftime("%Y-%m-%d %H:%M:%S")
except Exception as e:
return f"Error fetching time for timezone '{timezone}': {str(e)}"
# Determine an appropriate greeting based on the hour of the day.
hour = local_time.hour
if hour < 12:
greeting = "Good morning"
elif 12 <= hour < 18:
greeting = "Good afternoon"
else:
greeting = "Good evening"
# Fetch a fun fact.
fact = get_random_fact()
# If get_random_fact returned an error, it will already be a message indicating the error.
if fact.startswith("Error"):
fun_fact_message = fact
else:
fun_fact_message = f"Here's a fun fact: {fact}"
# Create the final entertaining message.
message = (
f"{greeting}! The current local time in {timezone} is {formatted_time}. "
f"{fun_fact_message}"
)
return message
final_answer = FinalAnswerTool()
# 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:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct', # it is possible that this model may be overloaded
custom_role_conversions=None,
)
# Import tool from Hub
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)
agent = CodeAgent(
model=model,
tools=[final_answer, get_random_fact, get_current_time_in_timezone, get_fact_with_greeting],
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch()
|