diff --git a/tests/test_prevent_infinite_tool_loop.py b/tests/test_tool_choice_reset.py similarity index 99% rename from tests/test_prevent_infinite_tool_loop.py rename to tests/test_tool_choice_reset.py index c60a951..e01a5f0 100644 --- a/tests/test_prevent_infinite_tool_loop.py +++ b/tests/test_tool_choice_reset.py @@ -54,7 +54,7 @@ class MockModel: ) -class TestPreventInfiniteToolLoop: +class TestToolChoiceReset: async def test_tool_choice_resets_after_call(self): """Test that tool_choice is reset to 'auto' after tool call when set to 'required'""" # Create an agent with tool_choice="required"