Spaces:
Sleeping
Sleeping
File size: 1,388 Bytes
db4b897 9b5b26a c19d193 6aae614 8fe992b 9b5b26a db4b897 9b5b26a db4b897 8c01ffb 6aae614 ae7a494 e121372 db4b897 13d500a 8c01ffb db4b897 8c01ffb db4b897 9b5b26a 8c01ffb 8fe992b db4b897 8c01ffb db4b897 8c01ffb db4b897 8fe992b db4b897 |
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 |
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
# Optional custom tool
@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').
"""
...
final_answer = FinalAnswerTool()
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
custom_role_conversions=None
)
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
# Instantiate the built-in DuckDuckGoSearchTool
search_tool = DuckDuckGoSearchTool()
# Load the HF image generation tool
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
agent = CodeAgent(
model=model,
tools=[
final_answer,
search_tool, # for web searches
image_generation_tool, # for text-to-image generation
get_current_time_in_timezone
],
max_steps=6,
verbosity_level=1,
prompt_templates=prompt_templates,
grammar=None,
planning_interval=None,
name=None,
description=None
)
GradioUI(agent).launch()
|