mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-27 06:13:07 +00:00
Merge 9a13294f20 into 8d6b8dab3d
This commit is contained in:
commit
9d7822f0e5
2 changed files with 31 additions and 1 deletions
|
|
@ -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]:
|
||||
"""
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue