File size: 2,556 Bytes
eff64f5 9b5b26a c19d193 6aae614 9b5b26a eff64f5 9b5b26a eff64f5 9b5b26a eff64f5 9b5b26a eff64f5 9b5b26a eff64f5 9b5b26a 8c01ffb eff64f5 6aae614 ae7a494 eff64f5 e121372 eff64f5 13d500a 8c01ffb eff64f5 861422e eff64f5 8c01ffb 8fe992b eff64f5 8c01ffb 861422e 8fe992b eff64f5 |
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()
|