mlflow.groq

The mlflow.groq module provides an API for logging and loading Groq models.

mlflow.groq.autolog(log_traces: bool = True, disable: bool = False, silent: bool = False)[source]

Note

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

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

Parameters

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

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

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