scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

ci/cd - skip 429 errors from google gemini

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-05 13:36:29 -07:00
parent 8f45f5afe9
commit 3b2c34d587
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_streaming.py
View file

@ -448,6 +448,8 @@ def test_completion_gemini_stream():
except litellm.APIError as e:
pass
except Exception as e:
if "429 Resource has been exhausted":
return
pytest.fail(f"Error occurred: {e}")
@ -798,10 +800,10 @@ def test_vertex_ai_stream():
def test_bedrock_claude_3_streaming():
try:
litellm.set_verbose = True
response: ModelResponse = completion(
response: ModelResponse = completion( # type: ignore
model="bedrock/anthropic.claude-3-sonnet-20240229-v1:0",
messages=messages,
max_tokens=10,
max_tokens=10, # type: ignore
stream=True,
)
complete_response = ""
@ -831,7 +833,7 @@ def test_claude_3_streaming_finish_reason():
{"role": "system", "content": "Be helpful"},
{"role": "user", "content": "What do you know?"},
]
response: ModelResponse = completion(
response: ModelResponse = completion( # type: ignore
model="claude-3-opus-20240229",
messages=messages,
stream=True,