phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-02 03:24: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

13
tests/integration/datasets/test_datasets.py
View file

@ -19,6 +19,15 @@ import pytest
def test_register_dataset(llama_stack_client):
dataset = llama_stack_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",
},
)
print(dataset)
assert dataset.identifier is not None
assert dataset.provider_id == "huggingface"
iterrow_response = llama_stack_client.datasets.iterrows(
dataset.identifier, limit=10
)
assert len(iterrow_response.data) == 10
assert iterrow_response.next_index is not None

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__":