File size: 1,246 Bytes
11be797
9b5b26a
 
 
c19d193
6aae614
9b5b26a
 
bd125ca
8c01ffb
6aae614
e121372
11be797
 
 
 
13d500a
8c01ffb
11be797
 
8c01ffb
bd125ca
 
8c01ffb
861422e
 
11be797
bd125ca
8fe992b
11be797
 
bd125ca
11be797
 
 
 
 
 
 
 
9b5b26a
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
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

# ... (Your existing my_custom_tool and get_current_time_in_timezone definitions)

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

# Initialize the DuckDuckGo Search Tool
duckduckgo_search_tool = DuckDuckGoSearchTool()

# Removed the attempt to load the Wikipedia tool
# The previous code to load the Wikipedia tool was here.

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    tools_list = [final_answer, duckduckgo_search_tool]
    # The conditional block to add the Wikipedia tool has been removed.

    agent = CodeAgent(
        model=model,
        tools=tools_list,  # Only includes final_answer and duckduckgo_search_tool now
        max_steps=6,
        verbosity_level=1,
        grammar=None,
        planning_interval=None,
        name=None,
        description=None,
        prompt_templates=prompt_templates
    )

GradioUI(agent).launch()