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

fix test_completion_bedrock_httpx_models

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-10 17:42:40 -07:00
parent 9590d63a38
commit ca76d2fd72
1 changed files with 26 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_completion.py
View file

@ -3065,6 +3065,7 @@ def response_format_tests(response: litellm.ModelResponse):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_completion_bedrock_httpx_models(sync_mode, model): async def test_completion_bedrock_httpx_models(sync_mode, model):
litellm.set_verbose = True litellm.set_verbose = True
try:
if sync_mode: if sync_mode:
response = completion( response = completion(
@ -3091,6 +3092,11 @@ async def test_completion_bedrock_httpx_models(sync_mode, model):
response_format_tests(response=response) response_format_tests(response=response)
print(f"response: {response}") print(f"response: {response}")
except litellm.RateLimitError as e:
print("got rate limit error=", e)
pass
except Exception as e:
pytest.fail(f"An error occurred - {str(e)}")
def test_completion_bedrock_titan_null_response(): def test_completion_bedrock_titan_null_response():