diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index ababe7cd6e3..b9e9d196eac 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -448,6 +448,8 @@ def test_completion_gemini_stream(): except litellm.APIError as e: pass except Exception as e: + if "429 Resource has been exhausted": + return pytest.fail(f"Error occurred: {e}") @@ -798,10 +800,10 @@ def test_vertex_ai_stream(): def test_bedrock_claude_3_streaming(): try: litellm.set_verbose = True - response: ModelResponse = completion( + response: ModelResponse = completion( # type: ignore model="bedrock/anthropic.claude-3-sonnet-20240229-v1:0", messages=messages, - max_tokens=10, + max_tokens=10, # type: ignore stream=True, ) complete_response = "" @@ -831,7 +833,7 @@ def test_claude_3_streaming_finish_reason(): {"role": "system", "content": "Be helpful"}, {"role": "user", "content": "What do you know?"}, ] - response: ModelResponse = completion( + response: ModelResponse = completion( # type: ignore model="claude-3-opus-20240229", messages=messages, stream=True,