From a8919a12a8b41881e4af16b497e69b6e002b32eb Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 20 Nov 2023 16:57:36 -0800 Subject: [PATCH] (test) passing max_retries --- litellm/tests/test_completion_with_retries.py | 23 ++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_completion_with_retries.py b/litellm/tests/test_completion_with_retries.py index a06d8c44c0..416051e383 100644 --- a/litellm/tests/test_completion_with_retries.py +++ b/litellm/tests/test_completion_with_retries.py @@ -49,4 +49,25 @@ def test_completion_with_num_retries(): except Exception as e: pass -# test_completion_with_num_retries() \ No newline at end of file +# 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() \ No newline at end of file