mlflow.smolagents

The mlflow.smolagents module provides an API for tracing Smolagents AI agents.

mlflow.smolagents.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.

Note

Autologging is known to be compatible with the following package versions: 1.14.0 <= smolagents <= 1.14.0. Autologging may not succeed when used with package versions outside of this range.

Enables (or disables) and configures autologging from Smolagents to MLflow. Note that asynchnorous APIs and Tool calling are not recorded now.

Parameters

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

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

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