phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_batch_completions.py): handle anthropic overloaded error

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-26 13:55:03 -07:00
parent 7d38c62717
commit 4d7f4550e2
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_batch_completions.py
View file

@ -74,6 +74,8 @@ def test_batch_completion_models_all_responses():
assert len(responses) == 2 assert len(responses) == 2
except Timeout as e: except Timeout as e:
pass pass
except litellm.APIError as e:
pass
except Exception as e: except Exception as e:
pytest.fail(f"An error occurred: {e}") pytest.fail(f"An error occurred: {e}")

2
litellm/utils.py
View file

@ -7231,7 +7231,7 @@ def exception_type(
exception_mapping_worked = True exception_mapping_worked = True
raise APIError( raise APIError(
status_code=original_exception.status_code, status_code=original_exception.status_code,
message=f"AnthropicException - {original_exception.message}", message=f"AnthropicException - {original_exception.message}. Handle with `litellm.APIError`.",
llm_provider="anthropic", llm_provider="anthropic",
model=model, model=model,
request=original_exception.request, request=original_exception.request,