m-ric/agent_toolcall2

1import yaml2import os3from smolagents import GradioUI, ToolCallingAgent, HfApiModel4 5# Get current directory path6CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))7 8from tools.final_answer import FinalAnswerTool as FinalAnswer9 10from managed_agents.web_agent.app import agent_web_agent11from managed_agents.useless.app import agent_useless12 13 14model = HfApiModel(15max_tokens=2096,16temperature=0.5,17model_id='Qwen/Qwen2.5-Coder-32B-Instruct',18provider=None,19)20 21final_answer = FinalAnswer()22 23 24with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:25    prompt_templates = yaml.safe_load(stream)26 27agent = ToolCallingAgent(28    model=model,29    tools=[],30    managed_agents=[agent_web_agent, agent_useless],31    max_steps=6,32    verbosity_level=1,33    grammar=None,34    planning_interval=None,35    name=None,36    description=None,37    prompt_templates=prompt_templates38)39if __name__ == "__main__":40    GradioUI(agent).launch()41