prepare for autogen llama_stack/templates/open-benchmark/run.yaml instead

This commit is contained in:
Yang Yang 2025-03-12 11:49:57 -07:00
parent 5034011e62
commit 5ec1881c85

View file

@ -226,6 +226,22 @@ def get_distribution_template() -> DistributionTemplate:
"chat_completion_input": {"type": "string"},
},
),
DatasetInput(
dataset_id="bfcl",
provider_id="huggingface",
url={"uri": "https://huggingface.co/datasets/llamastack/bfcl_v3"},
metadata={
"path": "llamastack/bfcl_v3",
"split": "train",
},
dataset_schema={
"function": {"type": "string"},
"language": {"type": "string"},
"ground_truth": {"type": "string"},
"id": {"type": "string"},
"chat_completion_input": {"type": "string"},
},
),
]
default_benchmarks = [
@ -249,6 +265,11 @@ def get_distribution_template() -> DistributionTemplate:
dataset_id="math_500",
scoring_functions=["basic::regex_parser_math_response"],
),
BenchmarkInput(
benchmark_id="meta-reference-bfcl",
dataset_id="bfcl",
scoring_functions=["basic::bfcl"],
),
]
return DistributionTemplate(
name=name,