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

(fix) cache control logic

Browse source
This commit is contained in:
Ishaan Jaff 2024-03-26 07:36:45 -07:00
parent 965fb6eb2c
commit 81b716d8da
1 changed files with 13 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/utils.py
View file

@ -2815,17 +2815,20 @@ def client(original_function):
# if caching is false, don't run this
final_embedding_cached_response = None
cache_controls = kwargs.get("cache", None)
if (
kwargs.get("caching", None) is None
and cache_controls is None
and litellm.cache is not None
) or (
kwargs.get("caching", False) == True
and (
cache_controls is not None
and cache_controls.get("no-cache", False) != True
# Check if user has opted out of caching
_opted_out_with_cache_controls = (
cache_controls and cache_controls.get("no-cache", False) == True
)
): # allow users to control returning cached responses from the completion function
_opted_out_with_caching_param = kwargs.get("caching", True) == False
# cache is not None and user has not opted out
if (
litellm.cache is not None
and (not _opted_out_with_cache_controls)
and (not _opted_out_with_caching_param)
):
# allow users to control returning cached responses from the completion function
# checking cache
print_verbose(f"INSIDE CHECKING CACHE")
if (