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
)