From 00e57d693fa318a3e37d4dd05fd90b9680e44a49 Mon Sep 17 00:00:00 2001 From: Rashmi Pawar Date: Wed, 16 Apr 2025 14:55:21 +0530 Subject: [PATCH] update provider to provider_id --- llama_stack/distribution/routers/routing_tables.py | 4 ++-- tests/integration/providers/nvidia/test_datastore.py | 2 +- tests/unit/providers/nvidia/test_datastore.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index fe52a0abb..68ee837bf 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -439,8 +439,8 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets): # infer provider from source if metadata: - if metadata.get("provider"): - provider_id = metadata.get("provider") # pass through from nvidia datasetio + if metadata.get("provider_id"): + provider_id = metadata.get("provider_id") # pass through from nvidia datasetio elif source.type == DatasetType.rows.value: provider_id = "localfs" elif source.type == DatasetType.uri.value: diff --git a/tests/integration/providers/nvidia/test_datastore.py b/tests/integration/providers/nvidia/test_datastore.py index f6ccd55a6..5f96dee9f 100644 --- a/tests/integration/providers/nvidia/test_datastore.py +++ b/tests/integration/providers/nvidia/test_datastore.py @@ -30,7 +30,7 @@ def test_register_and_unregister(llama_stack_client, provider_id): dataset_id=dataset_id, purpose=purpose, source=source, - metadata={"provider": provider_id, "format": "json", "description": "Test dataset description"}, + metadata={"provider_id": provider_id, "format": "json", "description": "Test dataset description"}, ) assert dataset.identifier is not None assert dataset.provider_id == provider_id diff --git a/tests/unit/providers/nvidia/test_datastore.py b/tests/unit/providers/nvidia/test_datastore.py index 504f1af5d..a17e51a9c 100644 --- a/tests/unit/providers/nvidia/test_datastore.py +++ b/tests/unit/providers/nvidia/test_datastore.py @@ -60,7 +60,7 @@ class TestNvidiaDatastore(unittest.TestCase): provider_id="", purpose=DatasetPurpose.post_training_messages, source=URIDataSource(uri="https://example.com/data.jsonl"), - metadata={"format": "jsonl", "description": "Test dataset description"}, + metadata={"provider_id": "nvidia", "format": "jsonl", "description": "Test dataset description"}, ) self.run_async(self.adapter.register_dataset(dataset_def))