mlflow.litellm

mlflow.litellm.autolog(log_traces: bool = True, disable: bool = False, silent: bool = False)[source]

Note

Experimental: This function may change or be removed in a future release without warning.

Enables (or disables) and configures autologging from LiteLLM to MLflow. Currently, MLflow only supports autologging for tracing.

Parameters

  • log_traces – If True, traces are logged for LiteLLM calls. If False, no traces are collected during inference. Default to True.

  • disable – If True, disables the LiteLLM autologging integration. If False, enables the LiteLLM autologging integration.

  • silent – If True, suppress all event logs and warnings from MLflow during LiteLLM autologging. If False, show all events and warnings.