mlflow.groq

mlflow.groq 模块提供用于记录和加载 Groq 模型的 API。

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

注意

已知 Autologging 与以下包版本兼容: 0.13.0 <= groq <= 0.31.0。当与此范围之外的包版本一起使用时,Autologging 可能无法成功。

启用(或禁用)并配置从 Groq 到 MLflow 的自动记录。仅支持同步调用。异步 API 和流式不会被记录。

Parameters
  • log_traces – 如果 True,则为 Groq 模型记录跟踪。如果 False,则在推理期间不收集任何跟踪。默认值为 True

  • disable – 如果 True,则禁用 Groq 的自动记录。默认值为 False

  • silent – 如果 True,在 Groq autologging 期间抑制来自 MLflow 的所有事件日志和警告。 如果 False,显示所有事件和警告。