From d14a41863faeabcc28b838e448bf2a18b958cbde Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 3 Jan 2024 15:42:09 +0530 Subject: [PATCH] (test) s3 cache with setting s3_bucket_name --- litellm/tests/test_caching.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_caching.py b/litellm/tests/test_caching.py index 05b397c57..c894331ba 100644 --- a/litellm/tests/test_caching.py +++ b/litellm/tests/test_caching.py @@ -549,8 +549,10 @@ def test_s3_cache_acompletion_stream_bedrock(): "content": f"write a one sentence poem about: {random_word}", } ] - litellm.cache = Cache(type="s3") - print("test for caching, streaming + completion") + litellm.cache = Cache( + type="s3", s3_bucket_name="cache-bucket-litellm", s3_region_name="us-west-2" + ) + print("s3 Cache: test for caching, streaming + completion") response_1_content = "" response_2_content = "" @@ -600,7 +602,7 @@ def test_s3_cache_acompletion_stream_bedrock(): raise e -# test_s3_cache_acompletion_stream_bedrock() +test_s3_cache_acompletion_stream_bedrock() # test_redis_cache_acompletion_stream_bedrock()