Spaces:
Runtime error
Runtime error
File size: 4,322 Bytes
bc0e097 9b5b26a c19d193 6aae614 8fe992b 9b5b26a cd19b00 9b5b26a cd19b00 9b5b26a cd19b00 9b5b26a cd19b00 9b5b26a cd19b00 8c01ffb cd19b00 8c01ffb 6aae614 ae7a494 cd19b00 ae7a494 2ec0a72 e121372 b1237e8 2ec0a72 cd19b00 13d500a 8c01ffb 2ec0a72 cd19b00 bc0e097 cd19b00 bc0e097 8c01ffb cd19b00 8c01ffb cd19b00 861422e cd19b00 8c01ffb 8fe992b cd19b00 bc0e097 cd19b00 8c01ffb cd19b00 8fe992b cd19b00 a663e9d 9b5b26a cd19b00 |
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 108 109 110 |
from smolagents import (
CodeAgent,
DuckDuckGoSearchTool,
HfApiModel,
load_tool,
tool
)
import datetime
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
# —————————————————————————————————————————————
# 1) Define your custom tools
# —————————————————————————————————————————————
@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
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}': {e}"
# —————————————————————————————————————————————
# 2) Instantiate the built-in final_answer tool
# —————————————————————————————————————————————
final_answer = FinalAnswerTool()
# —————————————————————————————————————————————
# 3) Load & instantiate your LLModel
# —————————————————————————————————————————————
# model = HfApiModel(
# model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
# max_tokens=2096,
# temperature=0.5,
# custom_role_conversions=None,
# )
model = HfApiModel(
model_id="HuggingFaceH4/zephyr-7b-beta",
max_tokens=2048,
temperature=0.5,
custom_role_conversions=None,
)
# —————————————————————————————————————————————
# 4) Load external tools
# * **Do not** call the returned object!
# —————————————————————————————————————————————
duck_search = DuckDuckGoSearchTool()
image_generation_tool = load_tool(
"agents-course/text-to-image",
trust_remote_code=True
)
# —————————————————————————————————————————————
# 5) Load your custom prompt templates
# —————————————————————————————————————————————
with open("prompts.yaml", "r") as stream:
prompt_templates = yaml.safe_load(stream)
# —————————————————————————————————————————————
# 6) Build your agent with **instances** of each tool
# —————————————————————————————————————————————
agent = CodeAgent(
model=model,
tools=[
final_answer, # instance of FinalAnswerTool
my_custom_tool, # your @tool
get_current_time_in_timezone, # your @tool
duck_search, # instance of DuckDuckGoSearchTool
image_generation_tool, # the load_tool() instance
],
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates,
)
# —————————————————————————————————————————————
# 7) Launch the Gradio UI
# ————————————————————————————————————————————-
GradioUI(agent).launch()
|