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

test(test_streaming.py): handle predibase instability

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-30 15:32:36 -07:00
parent ec6db03c41
commit e1cbb397fe
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_streaming.py
View file

@ -571,6 +571,8 @@ async def test_completion_predibase_streaming(sync_mode):
pass pass
except litellm.InternalServerError as e: except litellm.InternalServerError as e:
pass pass
except litellm.ServiceUnavailableError as e:
pass
except Exception as e: except Exception as e:
print("ERROR class", e.__class__) print("ERROR class", e.__class__)
print("ERROR message", e) print("ERROR message", e)