mlflow.gemini

The mlflow.gemini 模块提供了一个 API，用于跟踪与 Gemini 模型的交互。

mlflow.gemini.autolog(log_traces: bool = True, disable: bool = False, silent: bool = False)

注意

已知自动日志与以下软件包版本兼容：1.0.0 <= google-genai <= 1.57.0。在超出此范围的软件包版本中使用时，自动日志记录可能不会成功。

启用（或禁用）并配置从 Gemini 到 MLflow 的自动日志记录。目前，支持旧版 SDK google-generativeai 和新版 SDK google-genai。新版 SDK 支持同步和异步调用。

参数

• log_traces – 如果为 True，则为 Gemini 模型记录跟踪信息。如果为 False，则在推理过程中不收集任何跟踪信息。默认为 True。

• disable – 如果为 True，则禁用 Gemini 自动日志记录。默认为 False。

• silent – 如果为 True，则在 Gemini 自动日志记录期间抑制来自 MLflow 的所有事件日志和警告。如果为 False，则显示所有事件和警告。