mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-12 01:05:43 +00:00
fix test_bedrock_httpx_streaming
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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}")
|
||||
|
||||
Reference in New Issue
Block a user