From 66b76200e79df54d8b9e96d31f2bfd04fa578f94 Mon Sep 17 00:00:00 2001
From: Albert Villanova del Moral
 <8515462+albertvillanova@users.noreply.github.com>
Date: Wed, 5 Feb 2025 11:55:37 +0100
Subject: [PATCH] Initialize step_number attribute (#497)

* Test step_number attribute is initialized

* Initialize step_number attribute

* Fix style
---
 src/smolagents/agents.py | 1 +
 tests/test_agents.py     | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/smolagents/agents.py b/src/smolagents/agents.py
index 5db3d78..89f686e 100644
--- a/src/smolagents/agents.py
+++ b/src/smolagents/agents.py
@@ -176,6 +176,7 @@ class MultiStepAgent:
             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
diff --git a/tests/test_agents.py b/tests/test_agents.py
index 9caa221..80976d0 100644
--- a/tests/test_agents.py
+++ b/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"