跳转到主要内容

get_data

签名
@app.tool(output="benchmark->q_ls,gt_ls")
def get_data(benchmark: Dict[str, Any]) -> Dict[str, List[Any]]
功能
  • 从本地文件加载评测样本,支持 .jsonl / .json / .parquet
  • key_map 将原始字段映射为统一输出键(如 q_ls, gt_ls)。
  • 支持样本打乱(shuffle)与采样上限(limit)。
输出格式(JSON)
{
  "q_ls": ["Question 1", "Question 2"],
  "gt_ls": [["Answer A1", "Answer A2"], ["Answer B"]]
}

参数配置

https://mintcdn.com/ultrarag/T7GffHzZitf6TThi/images/yaml.svg?fit=max&auto=format&n=T7GffHzZitf6TThi&q=85&s=69b41e79144bc908039c2ee3abbb1c3bservers/benchmark/parameter.yaml
benchmark:
  name: nq
  path: data/sample_nq_10.jsonl
  key_map:
    q_ls: question
    gt_ls: golden_answers
  shuffle: false
  seed: 42
  limit: -1
参数说明:
参数类型说明
namestr评测集名称,仅用于日志与标识(示例:nq
pathstr数据文件路径,支持 .jsonl.json.parquet
key_mapdict字段映射表,将原始字段映射为工具输出键
key_map.q_lsstr问题字段名(如 question
key_map.gt_lsstr标准答案字段名(如 golden_answers,允许列表)
shufflebool是否打乱样本顺序(默认 false
seedint随机种子(shuffle=true 时生效)
limitint采样上限,-1 表示全部,正整数表示前 limit