Initialize step_number attribute (#497)

* Test step_number attribute is initialized

* Initialize step_number attribute

* Fix style

Author: albertvillanova

src/smolagents/agents.py

@@ -176,6 +176,7 @@ def __init__(
             tool_description_template if tool_description_template else DEFAULT_TOOL_DESCRIPTION_TEMPLATE
         )
         self.max_steps = max_steps
+        self.step_number: int = 0
         self.tool_parser = tool_parser
         self.grammar = grammar
         self.planning_interval = planning_interval

tests/test_agents.py

@@ -658,6 +658,8 @@ class TestMultiStepAgent:
     def test_step_number(self):
         fake_model = MagicMock()
         agent = MultiStepAgent(tools=[], model=fake_model)
+        assert hasattr(agent, "step_number"), "step_number attribute should be defined"
+        assert agent.step_number == 0, "step_number should be initialized to 0"
         agent.run("Test task", single_step=True)
         assert hasattr(agent, "step_number"), "step_number attribute should be defined"
         assert agent.step_number == 1, "step_number should be set to 1 after run method is called"