tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 08:22:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

(test) s3 cache with setting s3_bucket_name

Browse Source
This commit is contained in:
ishaan-jaff
2024-01-03 15:42:23 +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
litellm/tests/test_caching.py
+5 -3
View File
@@ -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()