From 113b5e728481e2a362c35747814b88f05a95548c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 5 Jan 2024 12:40:53 +0530 Subject: [PATCH] (ci/cd) retry cloudflare request 3 times --- litellm/tests/test_completion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index b484c0d60..a3ee1183a 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -1933,6 +1933,7 @@ def test_completion_cloudflare(): model="cloudflare/@cf/meta/llama-2-7b-chat-int8", messages=[{"content": "what llm are you", "role": "user"}], max_tokens=15, + num_retries=3, ) print(response) @@ -1940,7 +1941,7 @@ def test_completion_cloudflare(): pytest.fail(f"Error occurred: {e}") -# test_completion_cloudflare() +test_completion_cloudflare() def test_moderation():