论文

调整嵌入模型以进行代理能力检索

Adapting Embedding Models for Agent Capability Retrieval

模型训练参数高效训练

摘要

开放代理市场在同一搜索界面中列出了本地代理、工具包和可重用技能包,但从业者对于如何在这个混合目录中进行检索仍然缺乏指导。我们研究针对一般文本检索进行训练的现成检索模型是否可以适应将用户查询与可执行代理功能相匹配,以及学习到的信号是否超出用于调整的基准。我们在 AgentSelect 上微调了三个开放检索主干,BGE-base、KaLM-v1.5 和 EasyRec,将市场可见的单元表示为从公共元数据派生的功能配置文件,并在训练期间未见的两个目录上进行测试传输:MuleRun 本地代理和包含 50 项技能和 1,000 个查询的 ClawHub 基准。适应对这两个目录都有帮助。代码和数据将在发布后发布。