mlflow.smolagents

mlflow.smolagents 模块提供用于跟踪 Smolagents AI 智能体的 API。

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

注意

实验性:此功能可能在将来的发布中更改或在不另行通知的情况下被移除。

注意

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

启用(或禁用)并配置从 Smolagents 到 MLflow 的自动记录。请注意,目前异步 API 和工具调用不会被记录。

Parameters
  • log_traces – 如果 True,会为 Smolagents 智能体记录跟踪信息。 如果 False,推理期间不会收集跟踪信息。默认值为 True

  • disable – 如果 True,将禁用 Smolagents 的 autologging。默认值为 False

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