File size: 2,780 Bytes
9b5b26a c19d193 6aae614 9b5b26a 3747792 9b5b26a e55e980 3747792 e55e980 9b5b26a 3747792 9b5b26a 3747792 9b5b26a 8c01ffb 6aae614 e121372 bf6d34c 3747792 fe328e0 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e 9b5b26a 8c01ffb 8fe992b 3747792 8c01ffb 861422e 8fe992b 9b5b26a 3747792 8c01ffb |
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 |
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 xkcd_tool(id:int = 0)-> str:
"""A tool that fetches XKCD comics. For the latest comic do not pass an id argument.
This tool will return the requested comic title, alt text, and a url which points to the comic image.
Create a Markdown version of the comic with the following requirements:
- The title should be a level 1 header.
- The alt text should be a blcokquote.
- The image should be displayed. using the following syntax: 
The Final Answer should be This markdown string
Args:
id: a number representing a valid XKCD comic ID, or none for the latest comic.
"""
try:
url = f"https://xkcd.com/{id}/info.0.json" if id else "https://xkcd.com/info.0.json"
response = requests.get(url)
if response.status_code == 200:
data = response.json()
return {
"title": data["title"],
"alt": data["alt"],
"img_url": data["img"]
}
else:
return "Failed to fetch XKCD comic"
except Exception as e:
return f"Error fetching XKCD comic: {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:
# 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)}"
final_answer = FinalAnswerTool()
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud',# 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_current_time_in_timezone,xkcd_tool], ## add your tools here (don't remove final answer)
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch() |