mlflow.smolagents

mlflow.smolagents 模块提供了一个用于跟踪 Smolagents AI 代理的 API。

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

注意

已知自动日志记录与以下软件包版本兼容：1.14.0 <= smolagents <= 1.23.0。 在此范围之外的软件包版本使用自动日志记录时，可能会失败。

启用（或禁用）和配置从 Smolagents 到 MLflow 的自动日志记录。请注意，异步 API 和工具调用目前未被记录。

参数

  • log_traces – 如果为 True，则为 Smolagents 代理记录跟踪。如果为 False，则在推理期间不收集任何跟踪。默认为 True。

  • disable – 如果为 True，则禁用 Smolagents 自动日志记录。默认为 False。

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