fix(utils.py): support prometheus failed call metrics

This commit is contained in:
Krrish Dholakia 2024-04-18 12:29:15 -07:00
parent 05d6d9e45f
commit 28edb77350
4 changed files with 38 additions and 4 deletions

View file

@ -2205,9 +2205,9 @@ class ProxyConfig:
# these are litellm callbacks - "langfuse", "sentry", "wandb"
else:
litellm.failure_callback.append(callback)
verbose_proxy_logger.debug(
f"{blue_color_code} Initialized Success Callbacks - {litellm.failure_callback} {reset_color_code}"
)
print( # noqa
f"{blue_color_code} Initialized Failure Callbacks - {litellm.failure_callback} {reset_color_code}"
) # noqa
elif key == "cache_params":
# this is set in the cache branch
# see usage here: https://docs.litellm.ai/docs/proxy/caching