phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 14:57:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

evals with generation

Browse source
This commit is contained in:
Xi Yan 2024-10-24 11:30:13 -07:00
parent 071dba8871
commit 737fcb795f
15 changed files with 385 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

20
llama_stack/providers/tests/datasetio/test_datasetio.py
View file

@ -61,20 +61,26 @@ def data_url_from_file(file_path: str) -> str:
return data_url
async def register_dataset(datasets_impl: Datasets):
async def register_dataset(
datasets_impl: Datasets, include_generated_answer=True, dataset_id="test_dataset"
):
test_file = Path(os.path.abspath(__file__)).parent / "test_dataset.csv"
test_url = data_url_from_file(str(test_file))
dataset_schema = {
"expected_answer": StringType(),
"input_query": StringType(),
}
if include_generated_answer:
dataset_schema["generated_answer"] = StringType()
dataset = DatasetDefWithProvider(
identifier="test_dataset",
identifier=dataset_id,
provider_id=os.environ["PROVIDER_ID"],
url=URL(
uri=test_url,
),
dataset_schema={
"generated_answer": StringType(),
"expected_answer": StringType(),
"input_query": StringType(),
},
dataset_schema=dataset_schema,
)
await datasets_impl.register_dataset(dataset)