File size: 1,442 Bytes
4703af8
9f77c23
 
 
c19d193
6aae614
25d91b2
4703af8
9f77c23
9b5b26a
720b4df
6025aa5
4703af8
6aae614
ae7a494
 
 
 
e121372
bf6d34c
 
29ec968
fe328e0
13d500a
8c01ffb
 
861422e
 
9b5b26a
8c01ffb
8fe992b
4703af8
8c01ffb
 
 
 
 
 
861422e
8fe992b
 
9b5b26a
6025aa5
 
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
from smolagents import CodeAgent, HfApiModel
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from tools.exchange_rates import ExchangeRatesTool
from tools.web_search import web_search
from tools.visit_webpage import VisitWebpageTool
from Gradio_UI import GradioUI
from tools.new_tools import ReverseStringTool, WikiSearchTool
from tools.get_current_time import GetCurrentTimeTool
import yaml
final_answer = FinalAnswerTool()

# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 

model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)


with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[final_answer, web_search(), VisitWebpageTool(), ExchangeRatesTool(), WikiSearchTool(), ReverseStringTool(), GetCurrentTimeTool()], ## 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
)


if __name__ == "__main__":
    GradioUI(agent).launch()