From 0385309862c1115f04f4d0771d04e18c6e4d535a Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 5 Mar 2025 15:18:00 -0800 Subject: [PATCH] update provider id --- tests/integration/datasetio/test_datasetio.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/integration/datasetio/test_datasetio.py b/tests/integration/datasetio/test_datasetio.py index 5d0e7e37f..f112071a6 100644 --- a/tests/integration/datasetio/test_datasetio.py +++ b/tests/integration/datasetio/test_datasetio.py @@ -56,11 +56,14 @@ def register_dataset(llama_stack_client, for_generation=False, for_rag=False, da "generated_answer": {"type": "string"}, } + dataset_providers = [x for x in llama_stack_client.providers.list() if x.api == "datasetio"] + dataset_provider_id = dataset_providers[0].provider_id + llama_stack_client.datasets.register( dataset_id=dataset_id, dataset_schema=dataset_schema, url=dict(uri=test_url), - provider_id="localfs", + provider_id=dataset_provider_id, )