fix maintain older usage of logfire

This commit is contained in:
Ishaan Jaff 2024-06-17 19:42:39 -07:00
parent 61ff712a08
commit c93463b535
2 changed files with 48 additions and 8 deletions

View file

@ -341,14 +341,6 @@ def function_setup(
global callback_list, add_breadcrumb, user_logger_fn, Logging
function_id = kwargs["id"] if "id" in kwargs else None
# Note: maintain backwards compatibility with logfire. some users use litellm.success_callback=["logfire"]
if (
litellm.success_callback is not None
and "logfire" in litellm.success_callback
):
litellm.success_callback.remove("logfire")
litellm.callbacks.append("logfire")
if len(litellm.callbacks) > 0:
for callback in litellm.callbacks:
# check if callback is a string - e.g. "lago", "openmeter"