phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 06:53:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

datasetio client

Browse source
This commit is contained in:
Xi Yan 2024-10-23 14:04:51 -07:00
parent 51d5ad67c4
commit 5930a92dc7
3 changed files with 105 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/impls/meta_reference/datasetio/datasetio.py
View file

@ -142,7 +142,7 @@ class MetaReferenceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
dataset_info = self.dataset_infos.get(dataset_id)
dataset_info.dataset_impl.load()
if page_token is None:
if page_token is None or not page_token.isnumeric():
next_page_token = 0
else:
next_page_token = int(page_token)