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

update provider id

Browse source
This commit is contained in:
Xi Yan 2025-03-05 15:18:00 -08:00
parent 2091585843
commit 0385309862
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/integration/datasetio/test_datasetio.py
View file

@ -56,11 +56,14 @@ def register_dataset(llama_stack_client, for_generation=False, for_rag=False, da
"generated_answer": {"type": "string"}, "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( llama_stack_client.datasets.register(
dataset_id=dataset_id, dataset_id=dataset_id,
dataset_schema=dataset_schema, dataset_schema=dataset_schema,
url=dict(uri=test_url), url=dict(uri=test_url),
provider_id="localfs", provider_id=dataset_provider_id,
) )