label-studio-sdk 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

label_studio_sdk/export_storage/azure/client.py

@@ -66,6 +66,9 @@ class AzureClient:
     def create(
         self,
         *,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
         project: typing.Optional[int] = OMIT,
         container: typing.Optional[str] = OMIT,
         prefix: typing.Optional[str] = OMIT,
@@ -82,6 +85,15 @@ class AzureClient:
 
         Parameters
         ----------
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        title : typing.Optional[str]
+            Storage title
+
+        description : typing.Optional[str]
+            Storage description
+
         project : typing.Optional[int]
             Project ID
 
@@ -118,6 +130,9 @@ class AzureClient:
             "api/storages/export/azure",
             method="POST",
             json={
+                "can_delete_objects": can_delete_objects,
+                "title": title,
+                "description": description,
                 "project": project,
                 "container": container,
                 "prefix": prefix,
@@ -135,19 +150,58 @@ class AzureClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def validate(self, *, request_options: typing.Optional[RequestOptions] = None) -> AzureBlobExportStorage:
+    def validate(
+        self,
+        *,
+        id: typing.Optional[int] = OMIT,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        container: typing.Optional[str] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        account_name: typing.Optional[str] = OMIT,
+        account_key: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
         """
         Validate a specific Azure export storage connection. This is useful to ensure that the storage configuration settings are correct and operational before attempting to export data.
 
         Parameters
         ----------
+        id : typing.Optional[int]
+            Storage ID. If set, storage with specified ID will be updated
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        title : typing.Optional[str]
+            Storage title
+
+        description : typing.Optional[str]
+            Storage description
+
+        project : typing.Optional[int]
+            Project ID
+
+        container : typing.Optional[str]
+            Azure blob container
+
+        prefix : typing.Optional[str]
+            Azure blob prefix name
+
+        account_name : typing.Optional[str]
+            Azure Blob account name
+
+        account_key : typing.Optional[str]
+            Azure Blob account key
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AzureBlobExportStorage
-
+        None
 
         Examples
         --------
@@ -159,10 +213,24 @@ class AzureClient:
         client.export_storage.azure.validate()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/storages/export/azure/validate", method="POST", request_options=request_options
+            "api/storages/export/azure/validate",
+            method="POST",
+            json={
+                "id": id,
+                "can_delete_objects": can_delete_objects,
+                "title": title,
+                "description": description,
+                "project": project,
+                "container": container,
+                "prefix": prefix,
+                "account_name": account_name,
+                "account_key": account_key,
+            },
+            request_options=request_options,
+            omit=OMIT,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
+            return
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -254,6 +322,9 @@ class AzureClient:
         self,
         id: int,
         *,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
         project: typing.Optional[int] = OMIT,
         container: typing.Optional[str] = OMIT,
         prefix: typing.Optional[str] = OMIT,
@@ -271,6 +342,15 @@ class AzureClient:
         id : int
             A unique integer value identifying this azure blob export storage.
 
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        title : typing.Optional[str]
+            Storage title
+
+        description : typing.Optional[str]
+            Storage description
+
         project : typing.Optional[int]
             Project ID
 
@@ -309,6 +389,9 @@ class AzureClient:
             f"api/storages/export/azure/{jsonable_encoder(id)}",
             method="PATCH",
             json={
+                "can_delete_objects": can_delete_objects,
+                "title": title,
+                "description": description,
                 "project": project,
                 "container": container,
                 "prefix": prefix,
@@ -419,6 +502,9 @@ class AsyncAzureClient:
     async def create(
         self,
         *,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
         project: typing.Optional[int] = OMIT,
         container: typing.Optional[str] = OMIT,
         prefix: typing.Optional[str] = OMIT,
@@ -435,6 +521,15 @@ class AsyncAzureClient:
 
         Parameters
         ----------
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        title : typing.Optional[str]
+            Storage title
+
+        description : typing.Optional[str]
+            Storage description
+
         project : typing.Optional[int]
             Project ID
 
@@ -471,6 +566,9 @@ class AsyncAzureClient:
             "api/storages/export/azure",
             method="POST",
             json={
+                "can_delete_objects": can_delete_objects,
+                "title": title,
+                "description": description,
                 "project": project,
                 "container": container,
                 "prefix": prefix,
@@ -488,19 +586,58 @@ class AsyncAzureClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def validate(self, *, request_options: typing.Optional[RequestOptions] = None) -> AzureBlobExportStorage:
+    async def validate(
+        self,
+        *,
+        id: typing.Optional[int] = OMIT,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        container: typing.Optional[str] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        account_name: typing.Optional[str] = OMIT,
+        account_key: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
         """
         Validate a specific Azure export storage connection. This is useful to ensure that the storage configuration settings are correct and operational before attempting to export data.
 
         Parameters
         ----------
+        id : typing.Optional[int]
+            Storage ID. If set, storage with specified ID will be updated
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        title : typing.Optional[str]
+            Storage title
+
+        description : typing.Optional[str]
+            Storage description
+
+        project : typing.Optional[int]
+            Project ID
+
+        container : typing.Optional[str]
+            Azure blob container
+
+        prefix : typing.Optional[str]
+            Azure blob prefix name
+
+        account_name : typing.Optional[str]
+            Azure Blob account name
+
+        account_key : typing.Optional[str]
+            Azure Blob account key
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AzureBlobExportStorage
-
+        None
 
         Examples
         --------
@@ -512,10 +649,24 @@ class AsyncAzureClient:
         await client.export_storage.azure.validate()
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/storages/export/azure/validate", method="POST", request_options=request_options
+            "api/storages/export/azure/validate",
+            method="POST",
+            json={
+                "id": id,
+                "can_delete_objects": can_delete_objects,
+                "title": title,
+                "description": description,
+                "project": project,
+                "container": container,
+                "prefix": prefix,
+                "account_name": account_name,
+                "account_key": account_key,
+            },
+            request_options=request_options,
+            omit=OMIT,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(AzureBlobExportStorage, _response.json())  # type: ignore
+            return
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -607,6 +758,9 @@ class AsyncAzureClient:
         self,
         id: int,
         *,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
         project: typing.Optional[int] = OMIT,
         container: typing.Optional[str] = OMIT,
         prefix: typing.Optional[str] = OMIT,
@@ -624,6 +778,15 @@ class AsyncAzureClient:
         id : int
             A unique integer value identifying this azure blob export storage.
 
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        title : typing.Optional[str]
+            Storage title
+
+        description : typing.Optional[str]
+            Storage description
+
         project : typing.Optional[int]
             Project ID
 
@@ -662,6 +825,9 @@ class AsyncAzureClient:
             f"api/storages/export/azure/{jsonable_encoder(id)}",
             method="PATCH",
             json={
+                "can_delete_objects": can_delete_objects,
+                "title": title,
+                "description": description,
                 "project": project,
                 "container": container,
                 "prefix": prefix,

label_studio_sdk/export_storage/azure/types/azure_create_response.py

@@ -8,6 +8,21 @@ 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)
+    """
+    Deletion from storage enabled
+    """
+
+    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Storage title
+    """
+
+    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Storage description
+    """
+
     project: typing.Optional[int] = pydantic_v1.Field(default=None)
     """
     Project ID

label_studio_sdk/export_storage/azure/types/azure_update_response.py

@@ -8,6 +8,21 @@ 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)
+    """
+    Deletion from storage enabled
+    """
+
+    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Storage title
+    """
+
+    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    """
+    Storage description
+    """
+
     project: typing.Optional[int] = pydantic_v1.Field(default=None)
     """
     Project ID