phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) use litellm.cache for getting key

Browse source
This commit is contained in:
ishaan-jaff 2024-01-24 18:34:22 -08:00
parent 2130a61b6e
commit bf851ef19a
2 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/proxy/utils.py
View file

@ -995,15 +995,14 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time):
if api_key is not None and isinstance(api_key, str) and api_key.startswith("sk-"):
# hash the api_key
api_key = hash_token(api_key)
from litellm.caching import Cache
c = Cache()
cache_key = c.get_cache_key(**kwargs)
if "headers" in metadata and "authorization" in metadata["headers"]:
metadata["headers"].pop(
"authorization"
) # do not store the original `sk-..` api key in the db
if litellm.cache is not None:
cache_key = litellm.cache.get_cache_key(**kwargs)
else:
cache_key = "Cache OFF"
payload = {
"request_id": id,