mlflow.gemini

The mlflow.gemini module provides an API for tracing the interaction with Gemini models.

mlflow.gemini.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: 0.8.0 <= google-generativeai <= 0.8.3. Autologging may not succeed when used with package versions outside of this range.

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

Parameters

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

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

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