mlflow.autogen

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

注意

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

启用（或禁用）和配置 AutoGen flavor 的自动日志记录。由于其补丁设计，此方法需要在导入 AutoGen 类之后调用。

参数

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

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

  • silent – 如果为 True，则抑制 MLflow 在 AutoGen 自动日志记录期间的所有事件日志和警告。如果为 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)