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

dataset datasetio

Browse source
This commit is contained in:
Xi Yan 2024-10-22 13:09:17 -07:00
parent e8de70fdbe
commit f8d9e4f60f
8 changed files with 249 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/apis/datasetio/datasetio.py
View file

@ -17,7 +17,7 @@ class PaginatedRowsResult(BaseModel):
# the rows obey the DatasetSchema for the given dataset
rows: List[Dict[str, Any]]
total_count: int
next_page_token: Optional[str] = None
next_page_token: Optional[int] = None
class DatasetStore(Protocol):
@ -34,6 +34,6 @@ class DatasetIO(Protocol):
self,
dataset_id: str,
rows_in_page: int,
page_token: Optional[str] = None,
page_token: Optional[int] = None,
filter_condition: Optional[str] = None,
) -> PaginatedRowsResult: ...