From 080dfe2bd59931027aeedaf8707f7662ee2d2b29 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 4 Dec 2023 11:29:26 -0800 Subject: [PATCH] (fix) proxy: custom callbacks --- litellm/proxy/proxy_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 1238fd8e26..ae3ec5298e 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -564,7 +564,6 @@ def load_router_config(router: Optional[litellm.Router], config_file_path: str): ) elif key == "callbacks": print(f"{blue_color_code}\nSetting custom callbacks on Proxy") - print() passed_module, instance_name = value.split(".") # Dynamically import the module @@ -574,11 +573,12 @@ def load_router_config(router: Optional[litellm.Router], config_file_path: str): methods = [method for method in dir(instance) if callable(getattr(instance, method))] # Print the methods - print("Methods in the instance:") + print("Methods in the custom callbacks instance:") for method in methods: print(method) litellm.callbacks = [instance] + print() else: setattr(litellm, key, value)