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

fix register

Browse source
This commit is contained in:
Xi Yan 2025-03-15 15:15:52 -07:00
parent 2c9d624910
commit 13c7c5b6a1
3 changed files with 80 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/integration/datasets/test_script.py Normal file
View file

@ -0,0 +1,15 @@
from llama_stack_client import LlamaStackClient
from rich.pretty import pprint
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"},
)
pprint(dataset)
if __name__ == "__main__":
test_register_dataset()