phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) s3 cache with setting s3_bucket_name

Browse source
This commit is contained in:
ishaan-jaff 2024-01-03 15:42:09 +05:30
parent 58ce5d44ae
commit d14a41863f
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_caching.py
View file

@ -549,8 +549,10 @@ def test_s3_cache_acompletion_stream_bedrock():
"content": f"write a one sentence poem about: {random_word}", "content": f"write a one sentence poem about: {random_word}",
} }
] ]
litellm.cache = Cache(type="s3") litellm.cache = Cache(
print("test for caching, streaming + completion") 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_1_content = ""
response_2_content = "" response_2_content = ""
@ -600,7 +602,7 @@ def test_s3_cache_acompletion_stream_bedrock():
raise e raise e
# test_s3_cache_acompletion_stream_bedrock() test_s3_cache_acompletion_stream_bedrock()
# test_redis_cache_acompletion_stream_bedrock() # test_redis_cache_acompletion_stream_bedrock()