phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-01 21:34:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix endpoint, only sdk change

Browse source
This commit is contained in:
Xi Yan 2025-03-15 16:15:45 -07:00
parent 13c7c5b6a1
commit 9e6d99f7b1
8 changed files with 161 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/integration/datasets/test_script.py
View file

@ -6,9 +6,15 @@ def test_register_dataset():
client = LlamaStackClient(base_url="http://localhost:8321")
dataset = client.datasets.register(
purpose="eval/messages-answer",
source={"type": "uri", "uri": "huggingface://llamastack/simpleqa?split=train"},
source={
"type": "uri",
"uri": "huggingface://datasets/llamastack/simpleqa?split=train",
},
)
dataset_id = dataset.identifier
pprint(dataset)
rows = client.datasets.iterrows(dataset_id=dataset_id, limit=10)
pprint(rows)
if __name__ == "__main__":