phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #3685 from BerriAI/litellm_lago_integration

Browse source 
feat(lago.py): Enable Usage-based billing with lago
This commit is contained in:
Krish Dholakia 2024-05-16 13:09:48 -07:00 • committed by GitHub
commit 4c8b620352
6 changed files with 198 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/utils.py
View file

@ -140,6 +140,8 @@ class ProxyLogging:
self.slack_alerting_instance.response_taking_too_long_callback
)
for callback in litellm.callbacks:
if isinstance(callback, str):
callback = litellm.utils._init_custom_logger_compatible_class(callback)
if callback not in litellm.input_callback:
litellm.input_callback.append(callback)
if callback not in litellm.success_callback: