From 80884e9cb34fdae78c23b2bc2db72956ece4301d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 20 Nov 2023 19:04:10 -0800 Subject: [PATCH] (fix) using callbacks with router --- litellm/router.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/router.py b/litellm/router.py index 65c562c10..5cb0f89e0 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -81,7 +81,10 @@ class Router: self.cache_responses = cache_responses self.cache = litellm.Cache(cache_config) # use Redis for tracking load balancing ## USAGE TRACKING ## - litellm.success_callback = [self.deployment_callback] + if type(litellm.success_callback) == list: + litellm.success_callback.append(self.deployment_callback) + else: + litellm.success_callback = [self.deployment_callback] def _start_health_check_thread(self): """