From 80c349965f1e434976d8425c41636ce0d4713a98 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9bastien=20Han?= Date: Mon, 12 May 2025 19:56:14 +0200 Subject: [PATCH] chore(refact): move paginate_records fn outside of datasetio (#2137) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit # What does this PR do? Move under utils. Signed-off-by: Sébastien Han --- llama_stack/providers/inline/agents/meta_reference/agents.py | 2 +- llama_stack/providers/inline/datasetio/localfs/datasetio.py | 2 +- .../providers/remote/datasetio/huggingface/huggingface.py | 2 +- llama_stack/providers/utils/{datasetio => }/pagination.py | 0 4 files changed, 3 insertions(+), 3 deletions(-) rename llama_stack/providers/utils/{datasetio => }/pagination.py (100%) diff --git a/llama_stack/providers/inline/agents/meta_reference/agents.py b/llama_stack/providers/inline/agents/meta_reference/agents.py index 19d60c816..db10972de 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agents.py +++ b/llama_stack/providers/inline/agents/meta_reference/agents.py @@ -37,8 +37,8 @@ from llama_stack.apis.inference import ( from llama_stack.apis.safety import Safety from llama_stack.apis.tools import ToolGroups, ToolRuntime from llama_stack.apis.vector_io import VectorIO -from llama_stack.providers.utils.datasetio.pagination import paginate_records from llama_stack.providers.utils.kvstore import InmemoryKVStoreImpl, kvstore_impl +from llama_stack.providers.utils.pagination import paginate_records from .agent_instance import ChatAgent from .config import MetaReferenceAgentsImplConfig diff --git a/llama_stack/providers/inline/datasetio/localfs/datasetio.py b/llama_stack/providers/inline/datasetio/localfs/datasetio.py index 5640df6ab..da71ecb17 100644 --- a/llama_stack/providers/inline/datasetio/localfs/datasetio.py +++ b/llama_stack/providers/inline/datasetio/localfs/datasetio.py @@ -11,9 +11,9 @@ from llama_stack.apis.common.responses import PaginatedResponse from llama_stack.apis.datasetio import DatasetIO from llama_stack.apis.datasets import Dataset from llama_stack.providers.datatypes import DatasetsProtocolPrivate -from llama_stack.providers.utils.datasetio.pagination import paginate_records from llama_stack.providers.utils.datasetio.url_utils import get_dataframe_from_uri from llama_stack.providers.utils.kvstore import kvstore_impl +from llama_stack.providers.utils.pagination import paginate_records from .config import LocalFSDatasetIOConfig diff --git a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py index e329c88a7..fafd1d8ff 100644 --- a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py +++ b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py @@ -12,8 +12,8 @@ from llama_stack.apis.common.responses import PaginatedResponse from llama_stack.apis.datasetio import DatasetIO from llama_stack.apis.datasets import Dataset from llama_stack.providers.datatypes import DatasetsProtocolPrivate -from llama_stack.providers.utils.datasetio.pagination import paginate_records from llama_stack.providers.utils.kvstore import kvstore_impl +from llama_stack.providers.utils.pagination import paginate_records from .config import HuggingfaceDatasetIOConfig diff --git a/llama_stack/providers/utils/datasetio/pagination.py b/llama_stack/providers/utils/pagination.py similarity index 100% rename from llama_stack/providers/utils/datasetio/pagination.py rename to llama_stack/providers/utils/pagination.py