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

label_studio_sdk/import_storage/azure/types/azure_create_response.py

@@ -1,82 +1,72 @@
 # 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):
-    regex_filter: typing.Optional[str] = pydantic_v1.Field(default=None)
+class AzureCreateResponse(UniversalBaseModel):
+    regex_filter: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported.
     """
 
-    use_blob_urls: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    use_blob_urls: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio.
     """
 
-    presign: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    presign: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Presign URLs for direct download
     """
 
-    presign_ttl: typing.Optional[int] = pydantic_v1.Field(default=None)
+    presign_ttl: typing.Optional[int] = pydantic.Field(default=None)
     """
     Presign TTL in minutes
     """
 
-    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/import_storage/azure/types/azure_update_response.py

@@ -1,82 +1,72 @@
 # 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):
-    regex_filter: typing.Optional[str] = pydantic_v1.Field(default=None)
+class AzureUpdateResponse(UniversalBaseModel):
+    regex_filter: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage regex for filtering objects. You must specify it otherwise no objects will be imported.
     """
 
-    use_blob_urls: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    use_blob_urls: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Interpret objects as BLOBs and generate URLs. For example, if your bucket contains images, you can use this option to generate URLs for these images. If set to False, it will read the content of the file and load it into Label Studio.
     """
 
-    presign: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    presign: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Presign URLs for direct download
     """
 
-    presign_ttl: typing.Optional[int] = pydantic_v1.Field(default=None)
+    presign_ttl: typing.Optional[int] = pydantic.Field(default=None)
     """
     Presign TTL in minutes
     """
 
-    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/import_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.import_storage_list_types_response_item import ImportStorageListTypesResponseItem
+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 ImportStorageClient:
@@ -44,7 +50,7 @@ class ImportStorageClient:
 
         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 ImportStorageClient:
         client.import_storage.list_types()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/storages/types", method="GET", request_options=request_options
+            "api/storages/types",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[ImportStorageListTypesResponseItem], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[ImportStorageListTypesResponseItem],
+                    parse_obj_as(
+                        type_=typing.List[ImportStorageListTypesResponseItem],  # 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 AsyncImportStorageClient:
 
         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.import_storage.list_types()
+
+
+        async def main() -> None:
+            await client.import_storage.list_types()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/storages/types", method="GET", request_options=request_options
+            "api/storages/types",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[ImportStorageListTypesResponseItem], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[ImportStorageListTypesResponseItem],
+                    parse_obj_as(
+                        type_=typing.List[ImportStorageListTypesResponseItem],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)