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

precommit

Browse source
This commit is contained in:
Xi Yan 2025-03-15 16:20:58 -07:00
parent 9da092ff2d
commit cf225c9710
6 changed files with 56 additions and 160 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -4,10 +4,6 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
import base64
import mimetypes
import os
from pathlib import Path
import pytest
@ -36,8 +32,6 @@ def test_register_dataset(llama_stack_client, purpose, source, provider_id):
)
assert dataset.identifier is not None
assert dataset.provider_id == provider_id
iterrow_response = llama_stack_client.datasets.iterrows(
dataset.identifier, limit=10
)
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

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

@ -1,3 +1,9 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from llama_stack_client import LlamaStackClient
from rich.pretty import pprint