phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_streaming.py): try-except replicate api instability

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-28 22:19:44 -07:00
parent c9a424d28d
commit 831745e710
2 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_streaming.py
View file

@ -1265,6 +1265,8 @@ async def test_completion_replicate_llama3_streaming(sync_mode):
raise Exception("finish reason not set")
if complete_response.strip() == "":
raise Exception("Empty response received")
except litellm.UnprocessableEntityError as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")