| from smolagents import CodeAgent, HfApiModel, FinalAnswerTool | |
| # Basic inference model | |
| 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, | |
| ) | |
| # Code Agent | |
| agent = CodeAgent( | |
| model=model, | |
| tools=[FinalAnswerTool()], | |
| max_steps=2 | |
| ) | |