File size: 1,856 Bytes
f02eff2 9b5b26a c19d193 8fe992b f02eff2 9b5b26a f02eff2 9b5b26a f02eff2 9b5b26a f02eff2 9b5b26a f02eff2 9b5b26a f02eff2 9b5b26a f02eff2 9b5b26a 8c01ffb f02eff2 8c01ffb f02eff2 e121372 f02eff2 13d500a 8c01ffb f02eff2 861422e f02eff2 8c01ffb 8fe992b f02eff2 8c01ffb 861422e 8fe992b f02eff2 9b5b26a |
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 |
from smolagents import CodeAgent, HfApiModel, load_tool, tool
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
import datetime
import pytz
import yaml
# 🛠️ Required final answer tool
final_answer = FinalAnswerTool()
# 🧪 Example custom tool
@tool
def my_custom_tool(arg1: str, arg2: int) -> str:
"""A tool that does nothing yet.
Args:
arg1: the first argument
arg2: the second argument
"""
return "What magic will you build?"
# 🕒 Tool to fetch current time in a specific timezone
@tool
def get_current_time_in_timezone(timezone: str) -> str:
"""Fetches the current local time in a specified timezone.
Args:
timezone: A valid timezone string (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)}"
# 🎨 Load tool from Hugging Face Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
# 🧠 Load Hugging Face hosted model
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
custom_role_conversions=None,
)
# 📜 Load system prompt templates
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
# 🤖 Initialize the agent with tools
agent = CodeAgent(
model=model,
tools=[
final_answer,
my_custom_tool,
get_current_time_in_timezone,
image_generation_tool
],
max_steps=6,
verbosity_level=1,
prompt_templates=prompt_templates
)
# 🚀 Launch the agent in Gradio UI
GradioUI(agent).launch()
|