PolPC13
Renamed websearch tool and added import to app.py
4703af8
Raw
History Blame Contribute Delete
1.44 kB
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()