mlflow.ag2

mlflow.ag2.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 ag2 to MLflow. Currently, MLflow only supports tracing for ag2 agents.

Parameters

  • log_traces – If True, traces are logged for AG2 agents by using runtime logging. If False, no traces are collected during inference. Default to True.

  • disable – If True, disables the AG2 autologging. Default to False.

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