label-studio-sdk 1.0.3__py3-none-any.whl → 1.0.5__py3-none-any.whl

label_studio_sdk/__init__.py

@@ -54,6 +54,9 @@ from .types import (
     S3ExportStorageStatus,
     S3ImportStorage,
     S3ImportStorageStatus,
+    S3SExportStorage,
+    S3SImportStorage,
+    S3SImportStorageStatus,
     SerializationOption,
     SerializationOptions,
     Task,
@@ -65,6 +68,7 @@ from .types import (
     WebhookActionsItem,
     WebhookSerializerForUpdate,
     WebhookSerializerForUpdateActionsItem,
+    Workspace,
 )
 from .errors import BadRequestError, InternalServerError
 from . import (
@@ -80,6 +84,7 @@ from . import (
     users,
     views,
     webhooks,
+    workspaces,
 )
 from ._legacy import Client
 from .actions import (
@@ -211,6 +216,9 @@ __all__ = [
     "S3ExportStorageStatus",
     "S3ImportStorage",
     "S3ImportStorageStatus",
+    "S3SExportStorage",
+    "S3SImportStorage",
+    "S3SImportStorageStatus",
     "SerializationOption",
     "SerializationOptions",
     "Task",
@@ -243,6 +251,7 @@ __all__ = [
     "WebhookSerializerForUpdate",
     "WebhookSerializerForUpdateActionsItem",
     "WebhooksUpdateRequestActionsItem",
+    "Workspace",
     "__version__",
     "actions",
     "annotations",
@@ -256,4 +265,5 @@ __all__ = [
     "users",
     "views",
     "webhooks",
+    "workspaces",
 ]

label_studio_sdk/actions/client.py

@@ -44,9 +44,9 @@ class ActionsClient:
         _response = self._client_wrapper.httpx_client.request(
             "api/dm/actions/", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -134,9 +134,9 @@ class ActionsClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -172,9 +172,9 @@ class AsyncActionsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "api/dm/actions/", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -262,9 +262,9 @@ class AsyncActionsClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

label_studio_sdk/annotations/client.py

@@ -51,9 +51,9 @@ class AnnotationsClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/annotations/{jsonable_encoder(id)}/", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -93,9 +93,9 @@ class AnnotationsClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/annotations/{jsonable_encoder(id)}/", method="DELETE", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -182,7 +182,7 @@ class AnnotationsClient:
                         "width": 50,
                         "height": 60,
                         "rotation": 0,
-                        "values": {"rectanglelabels": {"0": "Person"}},
+                        "values": {"rectanglelabels": ["Person"]},
                     },
                 }
             ],
@@ -206,9 +206,9 @@ class AnnotationsClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -247,9 +247,9 @@ class AnnotationsClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/tasks/{jsonable_encoder(id)}/annotations/", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(typing.List[Annotation], _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[Annotation], _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -348,7 +348,7 @@ class AnnotationsClient:
                         "width": 50,
                         "height": 60,
                         "rotation": 0,
-                        "values": {"rectanglelabels": {"0": "Person"}},
+                        "values": {"rectanglelabels": ["Person"]},
                     },
                 }
             ],
@@ -372,9 +372,9 @@ class AnnotationsClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -418,9 +418,9 @@ class AsyncAnnotationsClient:
         _response = await self._client_wrapper.httpx_client.request(
             f"api/annotations/{jsonable_encoder(id)}/", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -460,9 +460,9 @@ class AsyncAnnotationsClient:
         _response = await self._client_wrapper.httpx_client.request(
             f"api/annotations/{jsonable_encoder(id)}/", method="DELETE", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -549,7 +549,7 @@ class AsyncAnnotationsClient:
                         "width": 50,
                         "height": 60,
                         "rotation": 0,
-                        "values": {"rectanglelabels": {"0": "Person"}},
+                        "values": {"rectanglelabels": ["Person"]},
                     },
                 }
             ],
@@ -573,9 +573,9 @@ class AsyncAnnotationsClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -616,9 +616,9 @@ class AsyncAnnotationsClient:
         _response = await self._client_wrapper.httpx_client.request(
             f"api/tasks/{jsonable_encoder(id)}/annotations/", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(typing.List[Annotation], _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[Annotation], _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -717,7 +717,7 @@ class AsyncAnnotationsClient:
                         "width": 50,
                         "height": 60,
                         "rotation": 0,
-                        "values": {"rectanglelabels": {"0": "Person"}},
+                        "values": {"rectanglelabels": ["Person"]},
                     },
                 }
             ],
@@ -741,9 +741,9 @@ class AsyncAnnotationsClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Annotation, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

label_studio_sdk/base_client.py

@@ -20,6 +20,7 @@ from .tasks.client import AsyncTasksClient, TasksClient
 from .users.client import AsyncUsersClient, UsersClient
 from .views.client import AsyncViewsClient, ViewsClient
 from .webhooks.client import AsyncWebhooksClient, WebhooksClient
+from .workspaces.client import AsyncWorkspacesClient, WorkspacesClient
 
 
 class LabelStudioBase:
@@ -96,6 +97,7 @@ class LabelStudioBase:
         self.import_storage = ImportStorageClient(client_wrapper=self._client_wrapper)
         self.export_storage = ExportStorageClient(client_wrapper=self._client_wrapper)
         self.webhooks = WebhooksClient(client_wrapper=self._client_wrapper)
+        self.workspaces = WorkspacesClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncLabelStudioBase:
@@ -172,6 +174,7 @@ class AsyncLabelStudioBase:
         self.import_storage = AsyncImportStorageClient(client_wrapper=self._client_wrapper)
         self.export_storage = AsyncExportStorageClient(client_wrapper=self._client_wrapper)
         self.webhooks = AsyncWebhooksClient(client_wrapper=self._client_wrapper)
+        self.workspaces = AsyncWorkspacesClient(client_wrapper=self._client_wrapper)
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LabelStudioEnvironment) -> str:

label_studio_sdk/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "label-studio-sdk",
-            "X-Fern-SDK-Version": "1.0.3",
+            "X-Fern-SDK-Version": "1.0.4",
         }
         headers["Authorization"] = f"Token  {self.api_key}"
         return headers

label_studio_sdk/core/http_client.py

@@ -2,6 +2,7 @@
 
 import asyncio
 import email.utils
+import json
 import re
 import time
 import typing
@@ -125,6 +126,25 @@ def maybe_filter_request_body(
     return data_content
 
 
+# Abstracted out for testing purposes
+def get_request_body(
+    *,
+    json: typing.Optional[typing.Any],
+    data: typing.Optional[typing.Any],
+    request_options: typing.Optional[RequestOptions],
+    omit: typing.Optional[typing.Any],
+) -> typing.Tuple[typing.Optional[typing.Any], typing.Optional[typing.Any]]:
+    json_body = None
+    data_body = None
+    if data is not None:
+        data_body = maybe_filter_request_body(data, request_options, omit)
+    else:
+        # If both data and json are None, we send json data in the event extra properties are specified
+        json_body = maybe_filter_request_body(json, request_options, omit)
+
+    return json_body, data_body
+
+
 class HttpClient:
     def __init__(
         self,
@@ -168,6 +188,8 @@ class HttpClient:
             else self.base_timeout
         )
 
+        json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
+
         response = self.httpx_client.request(
             method=method,
             url=urllib.parse.urljoin(f"{base_url}/", path),
@@ -197,8 +219,8 @@ class HttpClient:
                     )
                 )
             ),
-            json=maybe_filter_request_body(json, request_options, omit),
-            data=maybe_filter_request_body(data, request_options, omit),
+            json=json_body,
+            data=data_body,
             content=content,
             files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files)) if files is not None else None,
             timeout=timeout,
@@ -248,6 +270,8 @@ class HttpClient:
             else self.base_timeout
         )
 
+        json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
+
         with self.httpx_client.stream(
             method=method,
             url=urllib.parse.urljoin(f"{base_url}/", path),
@@ -277,8 +301,8 @@ class HttpClient:
                     )
                 )
             ),
-            json=maybe_filter_request_body(json, request_options, omit),
-            data=maybe_filter_request_body(data, request_options, omit),
+            json=json_body,
+            data=data_body,
             content=content,
             files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files)) if files is not None else None,
             timeout=timeout,
@@ -329,6 +353,8 @@ class AsyncHttpClient:
             else self.base_timeout
         )
 
+        json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
+
         # Add the input to each of these and do None-safety checks
         response = await self.httpx_client.request(
             method=method,
@@ -359,8 +385,8 @@ class AsyncHttpClient:
                     )
                 )
             ),
-            json=maybe_filter_request_body(json, request_options, omit),
-            data=maybe_filter_request_body(data, request_options, omit),
+            json=json_body,
+            data=data_body,
             content=content,
             files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files)) if files is not None else None,
             timeout=timeout,
@@ -409,6 +435,8 @@ class AsyncHttpClient:
             else self.base_timeout
         )
 
+        json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
+
         async with self.httpx_client.stream(
             method=method,
             url=urllib.parse.urljoin(f"{base_url}/", path),
@@ -438,8 +466,8 @@ class AsyncHttpClient:
                     )
                 )
             ),
-            json=maybe_filter_request_body(json, request_options, omit),
-            data=maybe_filter_request_body(data, request_options, omit),
+            json=json_body,
+            data=data_body,
             content=content,
             files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files)) if files is not None else None,
             timeout=timeout,

label_studio_sdk/core/request_options.py

@@ -5,10 +5,10 @@ import typing
 try:
     from typing import NotRequired  # type: ignore
 except ImportError:
-    from typing_extensions import NotRequired  # type: ignore
+    from typing_extensions import NotRequired
 
 
-class RequestOptions(typing.TypedDict):
+class RequestOptions(typing.TypedDict, total=False):
     """
     Additional options for request-specific configuration when calling APIs via the SDK.
     This is used primarily as an optional final parameter for service functions.

label_studio_sdk/export_storage/__init__.py

@@ -1,7 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import ExportStorageListTypesResponseItem
-from . import azure, gcs, local, redis, s3
+from . import azure, gcs, local, redis, s3, s3s
 from .azure import AzureCreateResponse, AzureUpdateResponse
 from .gcs import GcsCreateResponse, GcsUpdateResponse
 from .local import LocalCreateResponse, LocalUpdateResponse
@@ -25,4 +25,5 @@ __all__ = [
     "local",
     "redis",
     "s3",
+    "s3s",
 ]

label_studio_sdk/export_storage/azure/client.py

@@ -55,9 +55,9 @@ class AzureClient:
         _response = self._client_wrapper.httpx_client.request(
             "api/storages/export/azure", method="GET", params={"project": project}, request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(typing.List[AzureBlobExportStorage], _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[AzureBlobExportStorage], _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -142,9 +142,9 @@ class AzureClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureCreateResponse, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureCreateResponse, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -229,9 +229,9 @@ class AzureClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -270,9 +270,9 @@ class AzureClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/storages/export/azure/{jsonable_encoder(id)}", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -310,9 +310,9 @@ class AzureClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/storages/export/azure/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -401,9 +401,9 @@ class AzureClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureUpdateResponse, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureUpdateResponse, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -443,9 +443,9 @@ class AzureClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/storages/export/azure/{jsonable_encoder(id)}/sync", method="POST", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -491,9 +491,9 @@ class AsyncAzureClient:
         _response = await self._client_wrapper.httpx_client.request(
             "api/storages/export/azure", method="GET", params={"project": project}, request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(typing.List[AzureBlobExportStorage], _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[AzureBlobExportStorage], _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -578,9 +578,9 @@ class AsyncAzureClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureCreateResponse, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureCreateResponse, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -665,9 +665,9 @@ class AsyncAzureClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -706,9 +706,9 @@ class AsyncAzureClient:
         _response = await self._client_wrapper.httpx_client.request(
             f"api/storages/export/azure/{jsonable_encoder(id)}", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -746,9 +746,9 @@ class AsyncAzureClient:
         _response = await self._client_wrapper.httpx_client.request(
             f"api/storages/export/azure/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return
         try:
+            if 200 <= _response.status_code < 300:
+                return
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -837,9 +837,9 @@ class AsyncAzureClient:
             request_options=request_options,
             omit=OMIT,
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureUpdateResponse, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureUpdateResponse, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -879,9 +879,9 @@ class AsyncAzureClient:
         _response = await self._client_wrapper.httpx_client.request(
             f"api/storages/export/azure/{jsonable_encoder(id)}/sync", method="POST", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

label_studio_sdk/export_storage/client.py

@@ -12,6 +12,7 @@ from .gcs.client import AsyncGcsClient, GcsClient
 from .local.client import AsyncLocalClient, LocalClient
 from .redis.client import AsyncRedisClient, RedisClient
 from .s3.client import AsyncS3Client, S3Client
+from .s3s.client import AsyncS3SClient, S3SClient
 from .types.export_storage_list_types_response_item import ExportStorageListTypesResponseItem
 
 
@@ -23,6 +24,7 @@ class ExportStorageClient:
         self.local = LocalClient(client_wrapper=self._client_wrapper)
         self.redis = RedisClient(client_wrapper=self._client_wrapper)
         self.s3 = S3Client(client_wrapper=self._client_wrapper)
+        self.s3s = S3SClient(client_wrapper=self._client_wrapper)
 
     def list_types(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -52,9 +54,9 @@ class ExportStorageClient:
         _response = self._client_wrapper.httpx_client.request(
             "api/storages/export/types", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(typing.List[ExportStorageListTypesResponseItem], _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[ExportStorageListTypesResponseItem], _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -69,6 +71,7 @@ class AsyncExportStorageClient:
         self.local = AsyncLocalClient(client_wrapper=self._client_wrapper)
         self.redis = AsyncRedisClient(client_wrapper=self._client_wrapper)
         self.s3 = AsyncS3Client(client_wrapper=self._client_wrapper)
+        self.s3s = AsyncS3SClient(client_wrapper=self._client_wrapper)
 
     async def list_types(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -98,9 +101,9 @@ class AsyncExportStorageClient:
         _response = await self._client_wrapper.httpx_client.request(
             "api/storages/export/types", method="GET", request_options=request_options
         )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(typing.List[ExportStorageListTypesResponseItem], _response.json())  # type: ignore
         try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[ExportStorageListTypesResponseItem], _response.json())  # type: ignore
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)