From 7efe9beac57e5ffe86eccb41019b5157e157df46 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 10 Jul 2024 17:14:53 -0700 Subject: [PATCH] fix test_bedrock_httpx_streaming --- litellm/llms/bedrock_httpx.py | 2 +- litellm/tests/test_streaming.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/llms/bedrock_httpx.py b/litellm/llms/bedrock_httpx.py index 4e737d1d24..33d9203381 100644 --- a/litellm/llms/bedrock_httpx.py +++ b/litellm/llms/bedrock_httpx.py @@ -1066,7 +1066,7 @@ class BedrockLLM(BaseLLM): if response.status_code != 200: raise BedrockError( - status_code=response.status_code, message=response.text + status_code=response.status_code, message=response.read() ) decoder = AWSEventStreamDecoder(model=model) diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index a9540b20a1..1b0f91fe14 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -21,6 +21,8 @@ sys.path.insert( from dotenv import load_dotenv load_dotenv() +import random + import litellm from litellm import ( AuthenticationError, @@ -1373,7 +1375,8 @@ async def test_bedrock_httpx_streaming(sync_mode, model): if complete_response.strip() == "": raise Exception("Empty response received") print(f"completion_response: {complete_response}\n\nFinalChunk: {final_chunk}") - except RateLimitError: + except RateLimitError as e: + print("got rate limit error=", e) pass except Exception as e: pytest.fail(f"Error occurred: {e}")