File size: 2,556 Bytes
7cade2d 9b5b26a c19d193 6aae614 9b5b26a 7cade2d 9b5b26a 7cade2d 9b5b26a 7cade2d 9b5b26a 7cade2d 9b5b26a 7cade2d 9b5b26a 8c01ffb 7cade2d 6aae614 ae7a494 7cade2d e121372 7cade2d 13d500a 8c01ffb 7cade2d 861422e 7cade2d 8c01ffb 8fe992b 7cade2d 8c01ffb 861422e 8fe992b 7cade2d |
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 |
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
# Load image generation tool early
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
@tool
def searching(search: str) -> str:
"""Searches the web using DuckDuckGo.
Args:
search: A string representing a search query.
"""
duckduckgo_tool = DuckDuckGoSearchTool()
try:
results = duckduckgo_tool.run(search)
if results:
return f"Search Result using DuckDuckGo:\n{results}"
else:
return "No Result Found"
except Exception as e:
return f"Something went wrong:\n{str(e)}"
@tool
def image_generator_(prompt: str) -> str:
"""Generates an image from a prompt using a Hugging Face image generation tool.
Args:
prompt: A text prompt describing the image to generate.
Returns:
A URL pointing to the generated image.
"""
try:
image = image_generation_tool.run(prompt)
return f"Here is the link for the generated image using the prompt:\n{image}"
except Exception as e:
return f"We couldn't create an image for the prompt:\n{str(e)}. Try giving another prompt."
@tool
def get_current_time_in_timezone(timezone: str) -> str:
"""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).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 tool
final_answer = FinalAnswerTool()
# Hugging Face model setup
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
custom_role_conversions=None,
)
# Load prompt templates
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
# Create the agent
agent = CodeAgent(
model=model,
tools=[final_answer, searching, image_generator_],
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
# Launch the UI
GradioUI(agent).launch()
|