mlflow.autogen

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

注意

已知 Autologging 与以下包版本兼容： 0.4.9 <= autogen-agentchat <= 0.7.2。在该范围之外的包版本中使用时，Autologging 可能不会成功。

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

Parameters

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

  • disable – 如果 True，则禁用 AutoGen autologging。默认值为 False。

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

示例：

Example 
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)