phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: dataset metadata without provider_id

Browse source 
Signed-off-by: Juanma Barea <juanmabareamartinez@gmail.com>
This commit is contained in:
Juanma Barea 2025-06-26 15:13:02 +02:00
parent dbdc811d16
commit 6b10020fb7
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/distribution/routing_tables/datasets.py
View file

@ -57,9 +57,8 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
provider_dataset_id = dataset_id provider_dataset_id = dataset_id
# infer provider from source # infer provider from source
if metadata: if metadata and metadata.get("provider_id"):
if metadata.get("provider_id"): provider_id = metadata.get("provider_id") # 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: