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_completion.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-21 07:01:02 -07:00
parent de8eae20d9
commit a6ee1e04fe
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_completion.py
View file

@ -4032,7 +4032,9 @@ def test_completion_gemini(model):
# Add any assertions,here to check the response # Add any assertions,here to check the response
print(response) print(response)
assert response.choices[0]["index"] == 0 assert response.choices[0]["index"] == 0
except litellm.APIError as e: except litellm.RateLimitError:
pass
except litellm.APIError:
pass pass
except Exception as e: except Exception as e:
if "InternalServerError" in str(e): if "InternalServerError" in str(e):