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

(test) passing max_retries

Browse source
This commit is contained in:
ishaan-jaff 2023-11-20 16:57:36 -08:00
parent 11ec2710c6
commit 4020b807f6
1 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/tests/test_completion_with_retries.py
View file

@ -49,4 +49,25 @@ def test_completion_with_num_retries():
except Exception as e:
pass
# test_completion_with_num_retries()
# test_completion_with_num_retries()
def test_completion_with_0_num_retries():
try:
litellm.set_verbose=False
print("making request")
# Use the completion function
response = completion(
model="gpt-3.5-turbo",
messages=[{"gm": "vibe", "role": "user"}],
max_retries=4
)
print(response)
# print(response)
except Exception as e:
print("exception", e)
pass
# Call the test function
test_completion_with_0_num_retries()