mlflow.gemini

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

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

注意

自动日志记录已知与以下软件包版本兼容：1.0.0 <= google-genai <= 1.55.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，则显示所有事件和警告。