自定义Agent
DevKit AI支持在YAML文件中填写Agent名称和功能,来自定义子Agent。
操作步骤
- 使用
SSH 远程登录工具,登录DevKit AI的安装节点。 - 进入工具注册的配置文件所在目录。
cd DevKit-AI-x.x.x-Linux-Kunpeng/subagent_config
默认已存在cpp_migrate.yaml和sql_migrate.yaml两个配置文件,分别用于C/C++源码迁移与SQL迁移。
- 新增工具,以新增文件内容总结工具为例。
- 新增YAML文件。
vi summary.yaml
- 按“i”进入编辑模式,新增如下内容。
agent_name: 文件内容总结工具 description: 可以通过读取文件内容并进行总结。 custom_tools: - name: 文件读取工具 description: 读取目标文件内容。 command_tool: "cat" parameters: - name: file_path type: string description: 文件路径 required: true system_prompt: | 文件内容总结工具。通过读取目标文件内容,对文本进行解析,精准提炼核心观点和关键信息,为您生成简介明了的内容摘要,大幅度提升信息获取与处理效率。表1 参数说明 参数
说明
agent_name
Agent名称,会显示在CLI的启动界面。
description
Agent功能描述,会显示在CLI的启动界面。
custom_tools
name
Agent自定义工具名称。
description
工具功能描述。
command_tool
命令行工具。
fixed_args
工具的固定参数。
parameters
name
参数名。
flag
参数名对应的flag。
type
参数类型。
description
参数描述。
required
是否为必选参数。
enum
参数范围。
system_prompt
工具的总结信息。
- 按“Esc”键退出编辑模式,输入:wq!,按“Enter”键保存并退出文件。
- 新增YAML文件。
- 执行命令行工具。
cd DevKit-AI-x.x.x-Linux-Kunpeng ./devkitai_cli
返回信息如下。
================================================================================ DevKit AI 命令行助手 ================================================================================ 您好!我是 DevKit AI 助手,可以帮助您完成以下任务: 1. 文件内容总结工具 - 可以通过读取文件内容并进行总结。 可用命令: /help - 显示帮助信息 /quit 或 /exit - 退出程序 ================================================================================ [会话ID] 20260519_120417_e01c2cd9 - 询问工具支持的能力。
>>> 你有什么工具
返回信息如下,可以看到新增的工具。
根据当前系统配置,我支持以下工具: 文件内容总结工具:可以通过读取文件内容并进行总结。 使用方法: 读取目标文件内容并进行总结。 如果您需要使用这个工具,请告诉我,我会帮您执行相应的操作。
父主题: 高阶功能