phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 01:03:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update provider to provider_id

Browse source
This commit is contained in:
Rashmi Pawar 2025-04-16 14:55:21 +05:30
parent 4491a51149
commit 00e57d693f
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/distribution/routers/routing_tables.py
View file

@ -439,8 +439,8 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
# infer provider from source # infer provider from source
if metadata: if metadata:
if metadata.get("provider"): if metadata.get("provider_id"):
provider_id = metadata.get("provider") # pass through from nvidia datasetio provider_id = metadata.get("provider_id") # pass through from nvidia datasetio
elif source.type == DatasetType.rows.value: elif source.type == DatasetType.rows.value:
provider_id = "localfs" provider_id = "localfs"
elif source.type == DatasetType.uri.value: elif source.type == DatasetType.uri.value:

2
tests/integration/providers/nvidia/test_datastore.py
View file

@ -30,7 +30,7 @@ def test_register_and_unregister(llama_stack_client, provider_id):
dataset_id=dataset_id, dataset_id=dataset_id,
purpose=purpose, purpose=purpose,
source=source, 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.identifier is not None
assert dataset.provider_id == provider_id assert dataset.provider_id == provider_id

2
tests/unit/providers/nvidia/test_datastore.py
View file

@ -60,7 +60,7 @@ class TestNvidiaDatastore(unittest.TestCase):
provider_id="", provider_id="",
purpose=DatasetPurpose.post_training_messages, purpose=DatasetPurpose.post_training_messages,
source=URIDataSource(uri="https://example.com/data.jsonl"), 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)) self.run_async(self.adapter.register_dataset(dataset_def))