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

test(test_completion.py): handle gemini timeout error

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-06 19:05:39 -08:00
parent 7d824225a5
commit 7e3734d037
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_completion.py
View file

@ -2139,6 +2139,8 @@ async def test_acompletion_gemini():
response = await litellm.acompletion(model=model_name, messages=messages)
# Add any assertions here to check the response
print(f"response: {response}")
except litellm.Timeout as e:
pass
except litellm.APIError as e:
pass
except Exception as e: