zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(test_agent_utils.py): add name parameter to mock_react calls to ensure consistency in agent creation tests

This commit is contained in:
Ariel Frischer 2025-03-11 23:55:43 -07:00
parent 8d2d273c6b
commit d15d249929
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/ra_aid/test_agent_utils.py
View File

@ -84,6 +84,7 @@ def test_create_agent_anthropic(mock_model, mock_config_repository):
interrupt_after=['tools'],
version="v2",
state_modifier=mock_react.call_args[1]["state_modifier"],
name="React",
)
@ -213,7 +214,7 @@ def test_create_agent_anthropic_token_limiting_disabled(mock_model, mock_config_
agent = create_agent(mock_model, [])
assert agent == "react_agent"
mock_react.assert_called_once_with(mock_model, [], interrupt_after=['tools'], version="v2")
mock_react.assert_called_once_with(mock_model, [], interrupt_after=['tools'], version="v2", name="React")
# These tests have been moved to test_anthropic_token_limiter.py