File size: 1,715 Bytes
947f2cd
 
 
9b5b26a
 
c19d193
9b5b26a
947f2cd
9b5b26a
 
947f2cd
 
9b5b26a
 
947f2cd
 
 
9b5b26a
 
 
 
 
 
 
8c01ffb
947f2cd
6aae614
947f2cd
 
e121372
947f2cd
 
 
 
13d500a
8c01ffb
947f2cd
9b5b26a
8c01ffb
947f2cd
861422e
 
947f2cd
 
8c01ffb
8fe992b
947f2cd
8c01ffb
 
 
 
 
 
861422e
8fe992b
 
947f2cd
8c01ffb
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
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

# Custom tool: Get current time in a specified timezone
@tool
def get_current_time_in_timezone(timezone: str) -> str:
    """Fetch the current local time in a specified timezone.
    
    Args:
        timezone: A string representing a valid timezone (e.g., 'America/New_York').
    
    Returns:
        A string with the current time in the specified timezone or an error message.
    """
    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)}"

# Initialize the FinalAnswerTool
final_answer = FinalAnswerTool()

# Load the model
model = HfApiModel(
    max_tokens=2096,
    temperature=0.5,
    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
    custom_role_conversions=None,
)

# Load the image generation tool from the Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)

# Load prompt templates from YAML file
with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)

# Initialize the CodeAgent
agent = CodeAgent(
    model=model,
    tools=[final_answer],  # Add your tools here (don't remove final_answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)

# Launch the Gradio UI
GradioUI(agent).launch()