This commit is contained in:
Mrunmayee Daware 2026-10-21 03:38:09 +00:00 • committed by GitHub
commit 9d7822f0e5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 31 additions and 1 deletions

View file

@ -92,7 +92,13 @@ class AzureOpenAIResponsesAPIConfig(OpenAIResponsesAPIConfig):
return BaseAzureLLM.base_validate_azure_environment(headers=headers, litellm_params=litellm_params)
def get_stripped_model_name(self, model: str) -> str:
return model.replace("responses/", "").replace("o_series/", "").replace("azure_ai/", "")
# gpt5_series/ is a litellm routing marker, not part of the Azure deployment name
return (
model.replace("responses/", "")
.replace("o_series/", "")
.replace("azure_ai/", "")
.replace(AzureOpenAIGPT5Config.GPT5_SERIES_ROUTE, "")
)
def _handle_reasoning_item(self, item: dict[str, Any]) -> dict[str, Any]:
"""

View file

@ -23,6 +23,30 @@ from litellm.types.utils import LlmProviders
from litellm.utils import ProviderConfigManager
@pytest.mark.serial
def test_get_stripped_model_name_strips_gpt5_series_route():
"""The gpt5_series/ routing marker must not be sent to Azure as part of the deployment name."""
config = AzureOpenAIResponsesAPIConfig()
assert (
config.get_stripped_model_name("azure/gpt5_series/my-gpt5-deployment")
== "azure/my-gpt5-deployment"
)
assert (
config.get_stripped_model_name("azure/responses/gpt5_series/my-gpt5-deployment")
== "azure/my-gpt5-deployment"
)
# existing markers still strip, and plain deployments pass through untouched
assert (
config.get_stripped_model_name("azure/o_series/my-o3-deployment")
== "azure/my-o3-deployment"
)
assert (
config.get_stripped_model_name("azure/my-gpt5-deployment")
== "azure/my-gpt5-deployment"
)
@pytest.mark.serial
def test_validate_environment_api_key_within_litellm_params():
azure_openai_responses_apiconfig = AzureOpenAIResponsesAPIConfig()