VAKRA
收藏资源简介:
VAKRA(eValuating API and Knowledge Retrieval Agents)是一个由IBM研究院创建的基准数据集,旨在评估AI代理在多源异构环境下的组合推理能力。该数据集包含超过8000个可执行API,覆盖62个领域,总计9484个任务样本,按难度分为三类:多样API交互风格、结构化API多跳推理、以及结合自然语言工具使用策略的多源推理。数据集构建基于BIRD-SQL数据库扩展API,并融合ClapNQ和Wikidata5M文档构建检索索引,通过四阶段流水线生成2-5跳推理链,同时经人工质量评估确保语义连贯性。VAKRA主要用于测试代理在实体消歧、跨源信息对齐和策略遵守等方面的语言中介推理瓶颈,弥补了现有基准孤立评估API调用、检索或策略能力的不足,为真实企业场景(如客户支持、业务智能)中的智能体部署提供可靠评估框架。
VAKRA (eValuating API and Knowledge Retrieval Agents) is a benchmark dataset created by IBM Research, aiming to evaluate the compositional reasoning capability of AI Agents in multi-source heterogeneous environments. This dataset contains over 8000 executable APIs covering 62 domains, with a total of 9484 task samples, which are divided into three categories based on difficulty: diverse API interaction styles, structured API multi-hop reasoning, and multi-source reasoning integrated with natural language tool use strategies. The dataset is constructed by extending APIs based on the BIRD-SQL database, fusing ClapNQ and Wikidata5M documents to build retrieval indexes. A four-stage pipeline is employed to generate 2-5 hop reasoning chains, and manual quality evaluation is carried out to ensure semantic coherence. VAKRA is mainly used to test the language-mediated reasoning bottlenecks of AI Agents in entity disambiguation, cross-source information alignment, policy compliance and other aspects, filling the gap that existing benchmarks only evaluate API calling, retrieval or policy capabilities in isolation, and providing a reliable evaluation framework for the deployment of intelligent agents in real enterprise scenarios such as customer support and business intelligence.
VAKRA 数据集概述
基本信息
- 数据集名称:VAKRA(eValuating API and Knowledge Retrieval Agents using multi-hop, multi-source dialogues)
- 许可协议:cc-by-nc-sa-4.0
- 任务类型:问答、文本检索、文本生成
- 语言:英语
- 数据规模:1K < n < 10K 样本
- 标签:LLM Agent、工具调用、多跳推理、多源检索、RAG
数据集简介
VAKRA 是一个面向 AI 代理(Agent)的可执行基准测试,用于评估其在企业场景中端到端的多跳、多源工具调用能力。它不测试孤立的技能,而是衡量代理跨 API 和文档的组合推理能力,并通过完整的执行轨迹来验证代理能否可靠完成多步骤工作流。
该数据集提供可执行环境,代理可交互的 API 超过 8,000 个(源自 LiveAPIBench[1]),这些 API 由覆盖 62 个领域的真实数据库(源自 BIRD-SQL[2])支撑,并配有领域对齐的文档集(源自 CLAPnq[3] 和 Wikidata5M[4])。
核心特性
- 8,000+ 可执行 API,由覆盖 62 个领域的真实数据库支撑
- 多跳推理(3–7 步),结合 API 调用和文档检索
- 跨源基础,通过结构化 API + 非结构化文档实现
- 轨迹级验证,支持可重放的执行路径
- 确定性评估,使用本地托管的工具
基准结构
数据集按四种能力组织评估,对应三种复杂度递增的设置:
1. 多样化的 API 交互风格
- capability_1_bi_apis(API 链式调用):嵌套和组合式 API 链
- capability_2_dashboard_apis(工具选择):大规模的查询对齐端点工具选择
2. 结构化 API 上的多跳推理
- capability_3_multihop_reasoning(多跳 API 推理):需要 1–3 个 API 的依赖推理链,前序输出需被解释和转换以用于后续调用
3. 带工具使用策略的多跳、多源推理
- capability_4_multiturn(多跳多源策略遵守):结合 API 和文档检索的多轮对话,并包含关于工具使用的自然语言约束。推理链 1–4 个工具,是难度最高的设置。
数据统计
训练集
| 能力 | 领域数 | 样本数 | 平均工具调用 | 最大工具调用 | 平均轮次 | 最大轮次 |
|---|---|---|---|---|---|---|
| Capability_1(API 链式) | 33 | 1,324 | 4.05 | 12 | — | — |
| Capability_2(工具选择) | 40 | 1,860 | 1.00 | 1 | — | — |
| Capability_3(多跳推理) | 28 | 346 | 2.05 | 3 | — | — |
| Capability_4(多源策略) | 36 | 898 | 1.05 | 3 | 2.06 | 5 |
测试集
| 能力 | 领域数 | 样本数 | 平均工具调用 | 最大工具调用 | 平均轮次 | 最大轮次 |
|---|---|---|---|---|---|---|
| Capability_1(API 链式) | 54 | 2,077 | 3.96 | 10 | — | — |
| Capability_2(工具选择) | 17 | 1,597 | 1.00 | 1 | — | — |
| Capability_3(多跳推理) | 38 | 869 | 2.04 | 5 | — | — |
| Capability_4(多源策略) | 41 | 644 | 1.34 | 4 | 2.01 | 7 |
目录结构
<base_path>/ ├── indexed_documents/ ├── databases/ │ └── <domain>/ │ ├── database_description/ │ └── domain.sqlite ├── test/ │ └── capability-X/ │ └── input/ └── train/ └── capability-X/ ├── input/ └── output/
数据格式
输入示例
json { "uuid": "str", "domain": "str", "num_turns": 2, "dialogue": { "turns": [ {"turn_id": 0, "query": "str", "answer": "str"}, {"turn_id": 1, "query": "str"} ] }, "additional_instructions": "str" }
注:additional_instructions 字段仅存在于 capability_4_multiturn 中;非多轮输入文件中不会出现历史轮次和该字段。
输出 Schema
json { "uuid": "str", "domain": "str", "output": [ { "turn_id": 0, "query": "str", "answer": "str", "sequence": { "tool_call": [{"name": "str", "arguments": {"key": "value"}}], "tool_response": [{"name": "str", "response": {"key": "value"}}] } } ], "additional_instructions": "str" }
评估与评分
评估代码、评分脚本和字段排除列表维护在 GitHub 仓库(github.com/ibm/vakra)中。基准使用瀑布式评判器,包含三个组件:
- PolicyJudge:程序化评估策略遵守情况
- ExactMatchJudge:程序化评估预测工具响应与真实工具响应之间的顺序不变精确匹配(基于子集)
- GroundednessJudge:LLM 作为评判者,评估答案对工具响应和查询的忠实度
资源链接
- 排行榜:https://ibm-research-vakra.hf.space/
- 环境与设置:https://github.com/IBM/vakra/blob/main/setup.md
- 评估与评分:https://github.com/IBM/vakra
- 提交结果:https://github.com/IBM/vakra/issues/new?template=leaderboard_submission.yml
参考文献
- [1] Elder, Benjamin, et al. "Live API-Bench: 2500+ Live APIs for Testing Multi-Step Tool Calling."
- [2] Li, Jinyang, et al. "Can llm already serve as a database interface? a big bench for large-scale database grounded text-to-sqls."
- [3] Rosenthal, Sara, et al. "CLAPnq: Cohesive Long-form Answers from Passages in Natural Questions for RAG systems."
- [4] Wang, Xiaozhi, et al. "KEPLER: A unified model for knowledge embedding and pre-trained language representation."
- [5] Shlomov, Segev, et al. "From Benchmarks to Business Impact: Deploying IBM Generalist Agent in Enterprise Production."




