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

fix(proxy/utils.py): support redis caching for alerting

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-01 16:13:59 -07:00
parent 91120df3f7
commit 8d35e659ad
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/utils.py
View file

@ -63,7 +63,7 @@ class ProxyLogging:
self.cache_control_check = _PROXY_CacheControlCheck()
self.alerting: Optional[List] = None
self.alerting_threshold: float = 300 # default to 5 min. threshold
self.redis_usage_cache = redis_usage_cache
self.internal_usage_cache = DualCache(redis_cache=redis_usage_cache)
def update_values(
self, alerting: Optional[List], alerting_threshold: Optional[float]
@ -332,7 +332,7 @@ class ProxyLogging:
# - Alert once within 28d period
# - Cache this information
# - Don't re-alert, if alert already sent
_cache: DualCache = self.call_details["user_api_key_cache"]
_cache: DualCache = self.internal_usage_cache
# check if 5% of max budget is left
if percent_left <= 0.05: