From 317c290b694ccb233d4d7e3bb384ce7a679bb24a Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 1 Nov 2023 11:42:40 -0700 Subject: [PATCH] (test) add num retries to caching testing --- litellm/tests/test_caching.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_caching.py b/litellm/tests/test_caching.py index 49d20a2259..1ba43195fc 100644 --- a/litellm/tests/test_caching.py +++ b/litellm/tests/test_caching.py @@ -316,9 +316,9 @@ def test_custom_redis_cache_with_key(): # patch this redis cache get and set call - response1 = completion(model="gpt-3.5-turbo", messages=messages, temperature=1, caching=True) - response2 = completion(model="gpt-3.5-turbo", messages=messages, temperature=1, caching=True) - response3 = completion(model="gpt-3.5-turbo", messages=messages, temperature=1, caching=False) + response1 = completion(model="gpt-3.5-turbo", messages=messages, temperature=1, caching=True, num_retries=3) + response2 = completion(model="gpt-3.5-turbo", messages=messages, temperature=1, caching=True, num_retries=3) + response3 = completion(model="gpt-3.5-turbo", messages=messages, temperature=1, caching=False, num_retries=3) print(f"response1: {response1}") print(f"response2: {response2}")