tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-20 00:23:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

(fix) using callbacks with router

Browse Source
This commit is contained in:
ishaan-jaff
2023-11-20 19:08:53 -08:00
parent 668d45393c
commit 80884e9cb3
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/router.py
+4 -1
View File
@@ -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):
"""