mlflow.bedrock

mlflow.bedrock.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.33.0 <= boto3 <= 1.35.97. Autologging may not succeed when used with package versions outside of this range.

Enables (or disables) and configures autologging from Amazon Bedrock to MLflow. Only synchronous calls are supported. Asynchnorous APIs and streaming are not recorded.

Parameters

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

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

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