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