scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

address feedback on PR

Browse source
This commit is contained in:
Brian Caswell 2026-03-12 17:15:55 +00:00
parent bf073f489c
commit 2f5ecb16bc
1 changed files with 33 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
tests/test_litellm/llms/azure/test_azure_exception_mapping.py
View file

@ -422,7 +422,38 @@ class TestAzureExceptionMapping:
def test_azure_context_length_exceeded_via_error_code(self):
"""Azure Responses API returns context_length_exceeded in the error code
with a different message format than the chat completions API. This
should still be mapped to ContextWindowExceededError."""
should still be mapped to ContextWindowExceededError.
The exception message intentionally avoids the string-match trigger
substrings so that only the azure_error_code == "context_length_exceeded"
branch can cause the mapping."""
mock_exception = Exception(
"An unexpected error occurred (code: context_length_exceeded)."
)
mock_exception.body = {
"error": {
"message": "An unexpected error occurred (code: context_length_exceeded).",
"type": "invalid_request_error",
"param": "input",
"code": "context_length_exceeded",
}
}
mock_response = MagicMock()
mock_response.status_code = 400
mock_exception.response = mock_response
with pytest.raises(ContextWindowExceededError):
exception_type(
model="azure/gpt-4o",
original_exception=mock_exception,
custom_llm_provider="azure",
)
def test_azure_context_length_exceeded_via_message_string(self):
"""When Azure returns a message containing 'input exceeds the context
window' but without a structured error code, the string-match branch
should still map to ContextWindowExceededError."""
mock_exception = Exception(
"Your input exceeds the context window of this model. "
@ -436,7 +467,7 @@ class TestAzureExceptionMapping:
),
"type": "invalid_request_error",
"param": "input",
"code": "context_length_exceeded",
"code": None,
}
}
mock_response = MagicMock()