scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 178

fix test_bedrock_httpx_streaming

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-10 17:14:53 -07:00
parent de0dacc42d
commit 7efe9beac5
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/bedrock_httpx.py
View file

@ -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)

5
litellm/tests/test_streaming.py
View file

@ -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}")