File size: 1,807 Bytes
e815c73 9b5b26a c19d193 6aae614 9b5b26a e815c73 9b5b26a e815c73 9b5b26a 8c01ffb e815c73 6aae614 e815c73 ae7a494 e815c73 e121372 e815c73 13d500a 8c01ffb e815c73 861422e e815c73 8c01ffb 8fe992b e815c73 8c01ffb 861422e 8fe992b e815c73 |
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 |
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
# Example custom 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 ?"
# Example real 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:
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)}"
# Define the tools
final_answer = FinalAnswerTool()
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
duckduckgo_tool = DuckDuckGoSearchTool()
# Define the model
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
custom_role_conversions=None,
)
# Load system prompt
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
# Define the agent
agent = CodeAgent(
model=model,
tools=[
final_answer,
image_generation_tool,
duckduckgo_tool,
tool(get_current_time_in_timezone),
tool(my_custom_tool),
],
max_steps=6,
verbosity_level=1,
prompt_templates=prompt_templates
)
# Launch UI
GradioUI(agent).launch()
|