label-studio-sdk 1.0.8__py3-none-any.whl → 1.0.10__py3-none-any.whl

label_studio_sdk/export_storage/azure/types/azure_create_response.py

@@ -1,67 +1,57 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ....core.pydantic_utilities import UniversalBaseModel
 import typing
+import pydantic
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
 
-from ....core.datetime_utils import serialize_datetime
-from ....core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class AzureCreateResponse(pydantic_v1.BaseModel):
-    can_delete_objects: typing.Optional[bool] = pydantic_v1.Field(default=None)
+class AzureCreateResponse(UniversalBaseModel):
+    can_delete_objects: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Deletion from storage enabled
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Storage title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Storage description
     """
 
-    project: typing.Optional[int] = pydantic_v1.Field(default=None)
+    project: typing.Optional[int] = pydantic.Field(default=None)
     """
     Project ID
     """
 
-    container: typing.Optional[str] = pydantic_v1.Field(default=None)
+    container: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure blob container
     """
 
-    prefix: typing.Optional[str] = pydantic_v1.Field(default=None)
+    prefix: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure blob prefix name
     """
 
-    account_name: typing.Optional[str] = pydantic_v1.Field(default=None)
+    account_name: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure Blob account name
     """
 
-    account_key: typing.Optional[str] = pydantic_v1.Field(default=None)
+    account_key: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure Blob account key
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/export_storage/azure/types/azure_update_response.py

@@ -1,67 +1,57 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ....core.pydantic_utilities import UniversalBaseModel
 import typing
+import pydantic
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
 
-from ....core.datetime_utils import serialize_datetime
-from ....core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class AzureUpdateResponse(pydantic_v1.BaseModel):
-    can_delete_objects: typing.Optional[bool] = pydantic_v1.Field(default=None)
+class AzureUpdateResponse(UniversalBaseModel):
+    can_delete_objects: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Deletion from storage enabled
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Storage title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Storage description
     """
 
-    project: typing.Optional[int] = pydantic_v1.Field(default=None)
+    project: typing.Optional[int] = pydantic.Field(default=None)
     """
     Project ID
     """
 
-    container: typing.Optional[str] = pydantic_v1.Field(default=None)
+    container: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure blob container
     """
 
-    prefix: typing.Optional[str] = pydantic_v1.Field(default=None)
+    prefix: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure blob prefix name
     """
 
-    account_name: typing.Optional[str] = pydantic_v1.Field(default=None)
+    account_name: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure Blob account name
     """
 
-    account_key: typing.Optional[str] = pydantic_v1.Field(default=None)
+    account_key: typing.Optional[str] = pydantic.Field(default=None)
     """
     Azure Blob account key
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/export_storage/client.py

@@ -1,19 +1,25 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from ..core.client_wrapper import SyncClientWrapper
+from .azure.client import AzureClient
+from .gcs.client import GcsClient
+from .local.client import LocalClient
+from .redis.client import RedisClient
+from .s3.client import S3Client
+from .s3s.client import S3SClient
 import typing
-from json.decoder import JSONDecodeError
-
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.pydantic_utilities import pydantic_v1
 from ..core.request_options import RequestOptions
-from .azure.client import AsyncAzureClient, AzureClient
-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
+from ..core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper
+from .azure.client import AsyncAzureClient
+from .gcs.client import AsyncGcsClient
+from .local.client import AsyncLocalClient
+from .redis.client import AsyncRedisClient
+from .s3.client import AsyncS3Client
+from .s3s.client import AsyncS3SClient
 
 
 class ExportStorageClient:
@@ -44,7 +50,7 @@ class ExportStorageClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -52,11 +58,19 @@ class ExportStorageClient:
         client.export_storage.list_types()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/storages/export/types", method="GET", request_options=request_options
+            "api/storages/export/types",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[ExportStorageListTypesResponseItem], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[ExportStorageListTypesResponseItem],
+                    parse_obj_as(
+                        type_=typing.List[ExportStorageListTypesResponseItem],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -91,19 +105,35 @@ class AsyncExportStorageClient:
 
         Examples
         --------
-        from label_studio_sdk.client import AsyncLabelStudio
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
 
         client = AsyncLabelStudio(
             api_key="YOUR_API_KEY",
         )
-        await client.export_storage.list_types()
+
+
+        async def main() -> None:
+            await client.export_storage.list_types()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/storages/export/types", method="GET", request_options=request_options
+            "api/storages/export/types",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[ExportStorageListTypesResponseItem], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[ExportStorageListTypesResponseItem],
+                    parse_obj_as(
+                        type_=typing.List[ExportStorageListTypesResponseItem],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)