mlflow.autogen

mlflow.autogen.autolog(log_traces: bool = True, disable: bool = False, silent: bool = False)[源代码]

注意

已知自动日志记录与以下软件包版本兼容： 0.4.9 <= autogen-agentchat <= 0.7.5。 在此范围之外使用软件包版本时，自动日志记录可能不成功。

为 AutoGen 风格启用（或禁用）并配置自动日志记录。 由于其补丁设计，此方法需要在导入 AutoGen 类后才能调用。

参数

  • log_traces – 如果为 True，则会为 AutoGen 模型记录跟踪信息。 如果为 False，则在推理过程中不会收集任何跟踪信息。 默认为 True。

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

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

示例

示例 
import mlflow
from autogen_agentchat.agents import AssistantAgent
from autogen_ext.models.openai import OpenAIChatCompletionClient

mlflow.autogen.autolog()
agent = AssistantAgent("assistant", OpenAIChatCompletionClient(model="gpt-4o-mini"))
result = await agent.run(task="Say 'Hello World!'")
print(result)