KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Initialize step_number attribute (#497) 

* Test step_number attribute is initialized

* Initialize step_number attribute

* Fix style
This commit is contained in:
Albert Villanova del Moral 2025-02-05 11:55:37 +01:00 committed by GitHub
parent 90c30fa62b
commit 66b76200e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/smolagents/agents.py
View File

@ -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

2
tests/test_agents.py
View File

@ -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"