AutoGen 多智能体教程：代码执行器与 Agent 实例化

DEFAULT_USER_PROXY_AGENT_DESCRIPTIONS = {
    "ALWAYS": "An attentive HUMAN user who can answer questions about the task, and can perform tasks such as running Python code or inputting command line commands at a Linux terminal and reporting back the execution results.",
    "TERMINATE": "A user that can run Python code or input command line commands at a Linux terminal and report back the execution results.",
    "NEVER": "A computer terminal that performs no other action than running Python scripts...",
}

2.2 AssistantAgent

代表 AI 助手，负责编写代码和解决问题。默认系统提示词 (DEFAULT_SYSTEM_MESSAGE) 包含以下要求：

1. 分步骤完成任务，先解释计划再执行代码。
2. 代码块需标注类型（Python 或 Shell）。
3. 如需保存文件，第一行必须包含 # filename: <filename>。
4. 遇到错误需自行分析并修正，而非请求用户修改。
5. 任务完成时输出 TERMINATE。

示例代码

from autogen import AssistantAgent, UserProxyAgent

assistant = AssistantAgent(
    name="assistant",
    llm_config=llm_config,  # 需预先配置好 LLM 连接信息
    code_execution_config=False,  # 助手本身不执行代码，由 UserProxy 执行
)

user = UserProxyAgent(
    name="executor",
    code_execution_config={"executor": executor},
    human_input_mode="ALWAYS",
    default_auto_reply="continue"  # 避免空回复导致某些 API 报错
)

注意: 设置 default_auto_reply="continue" 是为了兼容部分大模型接口（如通义千问），防止因空内容返回 400 错误。

3. 任务执行流程

通过 initiate_chat 启动对话，描述具体任务。

3.1 场景示例

假设需要读取远程文件，提取特定格式的车厂名称并去重保存。

user.initiate_chat(
    assistant, 
    max_turns=5, 
    message="""
    读取一个位于 https://github.com/xxx/vmmr.txt 文件，该文件每一行类似 acura_cdx_2016 格式，
    要求提取第一个 _ 之前的字符串，它是车厂名字，提取所有的车厂后将其写到一个 vm.txt 文件中
    """
)

3.2 交互过程

1. Assistant 生成代码: 根据任务编写 Python 脚本，包含文件名注释。
2. User 审查: 检查逻辑是否正确（例如是否需要去重）。
3. 执行与反馈: User 将代码发送给执行器运行。若报错（如缺少依赖），Assistant 会自动安装依赖并重新运行。
4. 结果确认: 代码执行成功后，Assistant 总结结果并终止对话。

生成的代码和输出文件均存储在 work_dir 指定的目录下（如 code 文件夹），Docker 容器会将该目录挂载至内部路径，确保数据持久化。

4. 安全与最佳实践

4.1 安全性

• 容器隔离: 始终优先使用 Docker 执行器，避免 LLM 误操作宿主机文件系统。
• 权限控制: 限制容器的网络访问权限，仅允许必要的出站请求。
• 命令过滤: 利用 LocalCommandlineCodeExecutor 中的白名单机制过滤危险命令。

4.2 常见问题排查

• 依赖缺失: 若代码运行时报错 ModuleNotFoundError，Assistant 通常会自动尝试 pip install，也可手动在 code_execution_config 中预置依赖。
• 文件路径: 确保 work_dir 在宿主机和容器内均有写入权限。
• API 限流: 长时间对话可能导致 Token 超限，建议设置 max_turns 限制轮数。

5. 总结

通过配置 AutoGen 的代码执行器和 Agent 角色，开发者可以构建高度自动化的编程辅助系统。这不仅减少了重复性编码工作，还实现了从需求到代码落地的闭环。在实际应用中，结合 Docker 容器化和严格的 Prompt 工程，可以显著提升系统的稳定性和安全性。