phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_get_exception_class_name

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-04 21:32:55 -07:00
parent df4593d58b
commit b7cd4cef07
1 changed files with 30 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

30
tests/logging_callback_tests/test_prometheus_unit_tests.py
View file

@ -1500,3 +1500,33 @@ def test_set_team_budget_metrics_with_custom_labels(prometheus_logger, monkeypat
"metadata_organization": None,
"metadata_environment": None,
}
def test_get_exception_class_name(prometheus_logger):
"""
Test that _get_exception_class_name correctly formats the exception class name
"""
# Test case 1: Exception with llm_provider
rate_limit_error = litellm.RateLimitError(
message="Rate limit exceeded",
llm_provider="openai",
model="gpt-3.5-turbo"
)
assert prometheus_logger._get_exception_class_name(rate_limit_error) == "Openai.RateLimitError"
# Test case 2: Exception with empty llm_provider
auth_error = litellm.AuthenticationError(
message="Invalid API key",
llm_provider="",
model="gpt-4"
)
assert prometheus_logger._get_exception_class_name(auth_error) == "AuthenticationError"
# Test case 3: Exception with None llm_provider
context_window_error = litellm.ContextWindowExceededError(
message="Context length exceeded",
llm_provider=None,
model="gpt-4"
)
assert prometheus_logger._get_exception_class_name(context_window_error) == "ContextWindowExceededError"