Spaces:
Sleeping
Sleeping
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() |