(fix) proxy setting success callbacks

This commit is contained in:
ishaan-jaff 2024-03-08 16:27:53 -08:00
parent 4ff68c8562
commit d6dc28f0ed
2 changed files with 5 additions and 8 deletions

View file

@ -5,12 +5,9 @@ model_list:
api_base: os.environ/AZURE_API_BASE
api_key: os.environ/AZURE_API_KEY
api_version: "2023-07-01-preview"
- model_name: azure-gpt-3.5
litellm_params:
model: gpt-3.5-turbo
api_key: os.environ/OPENAI_API_KEY
model_info:
access_groups: ["public"]
litellm_settings:
set_verbose: True
success_callback: ["langfuse"]
router_settings:
set_verbose: True
debug_level: "DEBUG"

View file

@ -1675,9 +1675,9 @@ class ProxyConfig:
# these are litellm callbacks - "langfuse", "sentry", "wandb"
else:
litellm.success_callback.append(callback)
verbose_proxy_logger.debug(
print( # noqa
f"{blue_color_code} Initialized Success Callbacks - {litellm.success_callback} {reset_color_code}"
)
) # noqa
elif key == "failure_callback":
litellm.failure_callback = []