scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

test(test_streaming.py): adding retries to streaming tests

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-13 16:34:59 -08:00
parent 38ff412b9a
commit 42d0925209
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_streaming.py
View file

@ -14,7 +14,8 @@ import litellm
from litellm import completion, acompletion, AuthenticationError, BadRequestError, RateLimitError, ModelResponse
litellm.logging = False
litellm.set_verbose = False
litellm.set_verbose = True
litellm.num_retries = 3
litellm.cache = None
score = 0
@ -350,6 +351,7 @@ def test_completion_cohere_stream_bad_key():
def test_completion_azure_stream():
try:
litellm.set_verbose = True
messages = [
{"role": "system", "content": "You are a helpful assistant."},
{
@ -372,7 +374,7 @@ def test_completion_azure_stream():
print(f"completion_response: {complete_response}")
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_azure_stream()
test_completion_azure_stream()
def test_completion_claude_stream():
try: