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

fix all occurrence

Browse source
This commit is contained in:
Xi Yan 2025-03-15 14:20:17 -07:00
parent 917679cc2f
commit a9c662d68b
9 changed files with 177 additions and 70 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/distribution/routers/routers.py
View file

@ -537,7 +537,7 @@ class DatasetIORouter(DatasetIO):
logger.debug("DatasetIORouter.shutdown")
pass
async def get_rows_paginated(
async def iterrows(
self,
dataset_id: str,
rows_in_page: int,
@ -545,11 +545,9 @@ class DatasetIORouter(DatasetIO):
filter_condition: Optional[str] = None,
) -> IterrowsResponse:
logger.debug(
f"DatasetIORouter.get_rows_paginated: {dataset_id}, rows_in_page={rows_in_page}",
f"DatasetIORouter.iterrows: {dataset_id}, rows_in_page={rows_in_page}",
)
return await self.routing_table.get_provider_impl(
dataset_id
).get_rows_paginated(
return await self.routing_table.get_provider_impl(dataset_id).iterrows(
dataset_id=dataset_id,
rows_in_page=rows_in_page,
page_token=page_token,