mlflow.mistral

mlflow.mistral.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.0.0 <= mistralai <= 1.9.3. Autologging may not succeed when used with package versions outside of this range.

Enables (or disables) and configures autologging from Mistral AI to MLflow. Only synchronous calls to the Text generation API are supported. Asynchronous APIs and streaming are not recorded.

Parameters

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

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

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