mlflow.gemini

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

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

注意

Autologging 已知与以下软件包版本兼容： 1.0.0 <= google-genai <= 1.29.0. 当在此范围之外的包版本中使用时，Autologging 可能无法成功。

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

Parameters

  • log_traces – 如果 True，记录 Gemini 模型的追踪信息。 If False，推理期间不收集追踪信息。默认值为 True。

  • disable – 如果 True，则禁用 Gemini 的自动记录功能。默认值为 False。

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