diff --git a/tests/ra_aid/test_agent_utils.py b/tests/ra_aid/test_agent_utils.py index a0ad065..de91244 100644 --- a/tests/ra_aid/test_agent_utils.py +++ b/tests/ra_aid/test_agent_utils.py @@ -151,8 +151,7 @@ def test_create_agent_anthropic(mock_model, mock_config_repository): mock_react.assert_called_once_with( mock_model, [], - interrupt_before=['agent', 'tools'], - interrupt_after=['agent', 'tools'], + interrupt_after=['tools'], version="v2", state_modifier=mock_react.call_args[1]["state_modifier"], )