From 5925488de41fb99d984d3af384efc5d6c8ed51c6 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 13 Nov 2024 11:01:55 -0500 Subject: [PATCH] change schema -> dataset_schema in register_dataset API --- llama_stack/apis/datasets/datasets.py | 2 +- llama_stack/distribution/routers/routing_tables.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/apis/datasets/datasets.py b/llama_stack/apis/datasets/datasets.py index 8cd94442b..2ab958782 100644 --- a/llama_stack/apis/datasets/datasets.py +++ b/llama_stack/apis/datasets/datasets.py @@ -49,7 +49,7 @@ class Datasets(Protocol): async def register_dataset( self, dataset_id: str, - schema: Dict[str, ParamType], + dataset_schema: Dict[str, ParamType], url: URL, provider_dataset_id: Optional[str] = None, provider_id: Optional[str] = None, diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index c039d3cb1..3345f4c26 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -310,7 +310,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets): async def register_dataset( self, dataset_id: str, - schema: Dict[str, ParamType], + dataset_schema: Dict[str, ParamType], url: URL, provider_dataset_id: Optional[str] = None, provider_id: Optional[str] = None, @@ -332,7 +332,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets): identifier=dataset_id, provider_resource_id=provider_dataset_id, provider_id=provider_id, - dataset_schema=schema, + dataset_schema=dataset_schema, url=url, metadata=metadata, )