| import yaml | |
| from smolagents import GradioUI, CodeAgent, HfApiModel | |
| from tools.final_answer import FinalAnswerTool | |
| from web_agent.app import agent_web_agent | |
| from useless.app import agent_useless | |
| model = HfApiModel( | |
| max_tokens=2096, | |
| temperature=0.5, | |
| model_id='Qwen/Qwen2.5-Coder-32B-Instruct', | |
| provider=None, | |
| ) | |
| final_answer = FinalAnswerTool() | |
| with open("prompts.yaml", 'r') as stream: | |
| prompt_templates = yaml.safe_load(stream) | |
| agent = CodeAgent( | |
| model=model, | |
| tools=[], | |
| managed_agents=[agent_web_agent, agent_useless], | |
| 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() | |