From fd296fe7b21c62d430485de3fedfda51821aafad Mon Sep 17 00:00:00 2001 From: jake-valsamis Date: Tue, 31 Dec 2024 12:33:47 -0500 Subject: [PATCH 1/3] add next_token to SearchParams and SearchResults --- src/codeocean/data_asset.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/codeocean/data_asset.py b/src/codeocean/data_asset.py index 7b8e10c..839fe5c 100644 --- a/src/codeocean/data_asset.py +++ b/src/codeocean/data_asset.py @@ -204,6 +204,7 @@ class DataAssetSearchParams: archived: bool favorite: bool query: Optional[str] = None + next_token: Optional[str] = None sort_field: Optional[DataAssetSortBy] = None sort_order: Optional[SortOrder] = None type: Optional[DataAssetType] = None @@ -217,6 +218,7 @@ class DataAssetSearchParams: class DataAssetSearchResults: has_more: bool results: list[DataAsset] + next_token: Optional[str] = None @dataclass_json From f92ba17aa2aba26048c794c4b5f204e23dfb5122 Mon Sep 17 00:00:00 2001 From: jake-valsamis Date: Tue, 31 Dec 2024 15:16:11 -0500 Subject: [PATCH 2/3] Make all DataAssetSearchParams optional --- src/codeocean/data_asset.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/codeocean/data_asset.py b/src/codeocean/data_asset.py index 839fe5c..f2b6530 100644 --- a/src/codeocean/data_asset.py +++ b/src/codeocean/data_asset.py @@ -199,10 +199,10 @@ class DataAssetSearchOrigin(StrEnum): @dataclass_json @dataclass(frozen=True) class DataAssetSearchParams: - limit: int - offset: int - archived: bool - favorite: bool + limit: Optional[int] = None + offset: Optional[int] = None + archived: Optional[bool] = None + favorite: Optional[bool] = None query: Optional[str] = None next_token: Optional[str] = None sort_field: Optional[DataAssetSortBy] = None From ebae2492b2fcb071cda07c5070e6307b9acea1e1 Mon Sep 17 00:00:00 2001 From: jake-valsamis Date: Tue, 31 Dec 2024 15:31:08 -0500 Subject: [PATCH 3/3] add model as DataAssetType --- src/codeocean/data_asset.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/codeocean/data_asset.py b/src/codeocean/data_asset.py index f2b6530..09a3054 100644 --- a/src/codeocean/data_asset.py +++ b/src/codeocean/data_asset.py @@ -15,6 +15,7 @@ class DataAssetType(StrEnum): Dataset = "dataset" Result = "result" Combined = "combined" + Model = "model" class DataAssetState(StrEnum):