mlflow.bedrock

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

注意

Autologging 已知与以下包版本兼容: 1.33.0 <= boto3 <= 1.40.6。Autologging 在与此范围之外的包版本一起使用时可能无法成功。

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

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

  • disable – 如果 True,则禁用 Bedrock 的 autologging。默认为 False

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