From 6a4b531057dca7d6d7fef152ad7a6b99c2bcdb1a Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 14 Feb 2026 12:29:42 -0800 Subject: [PATCH] def get_advanced_model_for_shell_tool(self) -> Optional[str]: AZURE --- tests/llm_responses_api_testing/test_azure_responses_api.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/llm_responses_api_testing/test_azure_responses_api.py b/tests/llm_responses_api_testing/test_azure_responses_api.py index 86b490994d..5e876ce084 100644 --- a/tests/llm_responses_api_testing/test_azure_responses_api.py +++ b/tests/llm_responses_api_testing/test_azure_responses_api.py @@ -29,6 +29,10 @@ class TestAzureResponsesAPITest(BaseResponsesAPITest): "api_key": os.getenv("AZURE_API_KEY"), "api_version": "2025-03-01-preview", } + + def get_advanced_model_for_shell_tool(self) -> Optional[str]: + """If specified, overrides the model used by test_responses_api_shell_tool_streaming_sees_shell_output (e.g. openai/gpt-5.2 for shell support).""" + return "azure/gpt-5-mini" @pytest.mark.asyncio