diff --git a/lib/crewai/src/crewai/llm.py b/lib/crewai/src/crewai/llm.py index 8a4ac2edde..9afa2b84c7 100644 --- a/lib/crewai/src/crewai/llm.py +++ b/lib/crewai/src/crewai/llm.py @@ -2243,12 +2243,15 @@ def set_env_callbacks() -> None: ] failure_callbacks_str = os.environ.get("LITELLM_FAILURE_CALLBACKS", "") + failure_callbacks: list[str | Callable[..., Any] | CustomLogger] = [] if failure_callbacks_str: - failure_callbacks: list[str | Callable[..., Any] | CustomLogger] = [ + failure_callbacks = [ cb.strip() for cb in failure_callbacks_str.split(",") if cb.strip() ] + if success_callbacks: litellm.success_callback = success_callbacks + if failure_callbacks: litellm.failure_callback = failure_callbacks def __copy__(self) -> LLM: