label-studio-sdk 2.0.8__py3-none-any.whl → 2.0.9__py3-none-any.whl

label_studio_sdk/export_storage/databricks/client.py

@@ -0,0 +1,1406 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from ...core.client_wrapper import SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.databricks_export_storage import DatabricksExportStorage
+from ...core.unchecked_base_model import construct_type
+from json.decoder import JSONDecodeError
+from ...core.api_error import ApiError
+import datetime as dt
+from ...types.status_c5a_enum import StatusC5AEnum
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.client_wrapper import AsyncClientWrapper
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class DatabricksClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list(
+        self,
+        *,
+        ordering: typing.Optional[str] = None,
+        project: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[DatabricksExportStorage]:
+        """
+        Get a list of all Databricks Files export storage connections.
+
+        Parameters
+        ----------
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        project : typing.Optional[int]
+            Project ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[DatabricksExportStorage]
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.export_storage.databricks.list()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/storages/export/databricks",
+            method="GET",
+            params={
+                "ordering": ordering,
+                "project": project,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[DatabricksExportStorage],
+                    construct_type(
+                        type_=typing.List[DatabricksExportStorage],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def create(
+        self,
+        *,
+        catalog: str,
+        host: str,
+        project: int,
+        schema: str,
+        volume: str,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        last_sync: typing.Optional[dt.datetime] = OMIT,
+        last_sync_count: typing.Optional[int] = OMIT,
+        last_sync_job: typing.Optional[str] = OMIT,
+        meta: typing.Optional[typing.Optional[typing.Any]] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        regex_filter: typing.Optional[str] = OMIT,
+        request_timeout_s: typing.Optional[int] = OMIT,
+        status: typing.Optional[StatusC5AEnum] = OMIT,
+        stream_chunk_bytes: typing.Optional[int] = OMIT,
+        synchronizable: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        token: typing.Optional[str] = OMIT,
+        traceback: typing.Optional[str] = OMIT,
+        use_blob_urls: typing.Optional[bool] = OMIT,
+        verify_tls: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatabricksExportStorage:
+        """
+        Create a Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        catalog : str
+            UC catalog name
+
+        host : str
+            Databricks workspace base URL (https://...)
+
+        project : int
+            A unique integer value identifying this project.
+
+        schema : str
+            UC schema name
+
+        volume : str
+            UC volume name
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        description : typing.Optional[str]
+            Cloud storage description
+
+        last_sync : typing.Optional[dt.datetime]
+            Last sync finished time
+
+        last_sync_count : typing.Optional[int]
+            Count of tasks synced last time
+
+        last_sync_job : typing.Optional[str]
+            Last sync job ID
+
+        meta : typing.Optional[typing.Optional[typing.Any]]
+
+        prefix : typing.Optional[str]
+            Export path prefix under the volume
+
+        regex_filter : typing.Optional[str]
+            Regex for filtering objects
+
+        request_timeout_s : typing.Optional[int]
+
+        status : typing.Optional[StatusC5AEnum]
+
+        stream_chunk_bytes : typing.Optional[int]
+
+        synchronizable : typing.Optional[bool]
+
+        title : typing.Optional[str]
+            Cloud storage title
+
+        token : typing.Optional[str]
+
+        traceback : typing.Optional[str]
+            Traceback report for the last failed sync
+
+        use_blob_urls : typing.Optional[bool]
+            Generate blob URLs in tasks
+
+        verify_tls : typing.Optional[bool]
+            Verify TLS certificates
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.export_storage.databricks.create(
+            catalog="catalog",
+            host="host",
+            project=1,
+            schema="schema",
+            volume="volume",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/storages/export/databricks",
+            method="POST",
+            json={
+                "can_delete_objects": can_delete_objects,
+                "catalog": catalog,
+                "description": description,
+                "host": host,
+                "last_sync": last_sync,
+                "last_sync_count": last_sync_count,
+                "last_sync_job": last_sync_job,
+                "meta": meta,
+                "prefix": prefix,
+                "project": project,
+                "regex_filter": regex_filter,
+                "request_timeout_s": request_timeout_s,
+                "schema": schema,
+                "status": status,
+                "stream_chunk_bytes": stream_chunk_bytes,
+                "synchronizable": synchronizable,
+                "title": title,
+                "token": token,
+                "traceback": traceback,
+                "use_blob_urls": use_blob_urls,
+                "verify_tls": verify_tls,
+                "volume": volume,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def validate(
+        self,
+        *,
+        catalog: str,
+        host: str,
+        project: int,
+        schema: str,
+        volume: str,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        last_sync: typing.Optional[dt.datetime] = OMIT,
+        last_sync_count: typing.Optional[int] = OMIT,
+        last_sync_job: typing.Optional[str] = OMIT,
+        meta: typing.Optional[typing.Optional[typing.Any]] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        regex_filter: typing.Optional[str] = OMIT,
+        request_timeout_s: typing.Optional[int] = OMIT,
+        status: typing.Optional[StatusC5AEnum] = OMIT,
+        stream_chunk_bytes: typing.Optional[int] = OMIT,
+        synchronizable: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        token: typing.Optional[str] = OMIT,
+        traceback: typing.Optional[str] = OMIT,
+        use_blob_urls: typing.Optional[bool] = OMIT,
+        verify_tls: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Validate a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        catalog : str
+            UC catalog name
+
+        host : str
+            Databricks workspace base URL (https://...)
+
+        project : int
+            A unique integer value identifying this project.
+
+        schema : str
+            UC schema name
+
+        volume : str
+            UC volume name
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        description : typing.Optional[str]
+            Cloud storage description
+
+        last_sync : typing.Optional[dt.datetime]
+            Last sync finished time
+
+        last_sync_count : typing.Optional[int]
+            Count of tasks synced last time
+
+        last_sync_job : typing.Optional[str]
+            Last sync job ID
+
+        meta : typing.Optional[typing.Optional[typing.Any]]
+
+        prefix : typing.Optional[str]
+            Export path prefix under the volume
+
+        regex_filter : typing.Optional[str]
+            Regex for filtering objects
+
+        request_timeout_s : typing.Optional[int]
+
+        status : typing.Optional[StatusC5AEnum]
+
+        stream_chunk_bytes : typing.Optional[int]
+
+        synchronizable : typing.Optional[bool]
+
+        title : typing.Optional[str]
+            Cloud storage title
+
+        token : typing.Optional[str]
+
+        traceback : typing.Optional[str]
+            Traceback report for the last failed sync
+
+        use_blob_urls : typing.Optional[bool]
+            Generate blob URLs in tasks
+
+        verify_tls : typing.Optional[bool]
+            Verify TLS certificates
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.export_storage.databricks.validate(
+            catalog="catalog",
+            host="host",
+            project=1,
+            schema="schema",
+            volume="volume",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/storages/export/databricks/validate",
+            method="POST",
+            json={
+                "can_delete_objects": can_delete_objects,
+                "catalog": catalog,
+                "description": description,
+                "host": host,
+                "last_sync": last_sync,
+                "last_sync_count": last_sync_count,
+                "last_sync_job": last_sync_job,
+                "meta": meta,
+                "prefix": prefix,
+                "project": project,
+                "regex_filter": regex_filter,
+                "request_timeout_s": request_timeout_s,
+                "schema": schema,
+                "status": status,
+                "stream_chunk_bytes": stream_chunk_bytes,
+                "synchronizable": synchronizable,
+                "title": title,
+                "token": token,
+                "traceback": traceback,
+                "use_blob_urls": use_blob_urls,
+                "verify_tls": verify_tls,
+                "volume": volume,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> DatabricksExportStorage:
+        """
+        Get a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        id : int
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.export_storage.databricks.get(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Delete a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        id : int
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.export_storage.databricks.delete(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}",
+            method="DELETE",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def update(
+        self,
+        id: int,
+        *,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        catalog: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        host: typing.Optional[str] = OMIT,
+        last_sync: typing.Optional[dt.datetime] = OMIT,
+        last_sync_count: typing.Optional[int] = OMIT,
+        last_sync_job: typing.Optional[str] = OMIT,
+        meta: typing.Optional[typing.Optional[typing.Any]] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        regex_filter: typing.Optional[str] = OMIT,
+        request_timeout_s: typing.Optional[int] = OMIT,
+        schema: typing.Optional[str] = OMIT,
+        status: typing.Optional[StatusC5AEnum] = OMIT,
+        stream_chunk_bytes: typing.Optional[int] = OMIT,
+        synchronizable: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        token: typing.Optional[str] = OMIT,
+        traceback: typing.Optional[str] = OMIT,
+        use_blob_urls: typing.Optional[bool] = OMIT,
+        verify_tls: typing.Optional[bool] = OMIT,
+        volume: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatabricksExportStorage:
+        """
+        Update a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        id : int
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        catalog : typing.Optional[str]
+            UC catalog name
+
+        description : typing.Optional[str]
+            Cloud storage description
+
+        host : typing.Optional[str]
+            Databricks workspace base URL (https://...)
+
+        last_sync : typing.Optional[dt.datetime]
+            Last sync finished time
+
+        last_sync_count : typing.Optional[int]
+            Count of tasks synced last time
+
+        last_sync_job : typing.Optional[str]
+            Last sync job ID
+
+        meta : typing.Optional[typing.Optional[typing.Any]]
+
+        prefix : typing.Optional[str]
+            Export path prefix under the volume
+
+        project : typing.Optional[int]
+            A unique integer value identifying this project.
+
+        regex_filter : typing.Optional[str]
+            Regex for filtering objects
+
+        request_timeout_s : typing.Optional[int]
+
+        schema : typing.Optional[str]
+            UC schema name
+
+        status : typing.Optional[StatusC5AEnum]
+
+        stream_chunk_bytes : typing.Optional[int]
+
+        synchronizable : typing.Optional[bool]
+
+        title : typing.Optional[str]
+            Cloud storage title
+
+        token : typing.Optional[str]
+
+        traceback : typing.Optional[str]
+            Traceback report for the last failed sync
+
+        use_blob_urls : typing.Optional[bool]
+            Generate blob URLs in tasks
+
+        verify_tls : typing.Optional[bool]
+            Verify TLS certificates
+
+        volume : typing.Optional[str]
+            UC volume name
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.export_storage.databricks.update(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}",
+            method="PATCH",
+            json={
+                "can_delete_objects": can_delete_objects,
+                "catalog": catalog,
+                "description": description,
+                "host": host,
+                "last_sync": last_sync,
+                "last_sync_count": last_sync_count,
+                "last_sync_job": last_sync_job,
+                "meta": meta,
+                "prefix": prefix,
+                "project": project,
+                "regex_filter": regex_filter,
+                "request_timeout_s": request_timeout_s,
+                "schema": schema,
+                "status": status,
+                "stream_chunk_bytes": stream_chunk_bytes,
+                "synchronizable": synchronizable,
+                "title": title,
+                "token": token,
+                "traceback": traceback,
+                "use_blob_urls": use_blob_urls,
+                "verify_tls": verify_tls,
+                "volume": volume,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def sync(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> DatabricksExportStorage:
+        """
+        Export annotations to a Databricks Files storage.
+
+        Parameters
+        ----------
+        id : int
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.export_storage.databricks.sync(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}/sync",
+            method="POST",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncDatabricksClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list(
+        self,
+        *,
+        ordering: typing.Optional[str] = None,
+        project: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[DatabricksExportStorage]:
+        """
+        Get a list of all Databricks Files export storage connections.
+
+        Parameters
+        ----------
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        project : typing.Optional[int]
+            Project ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[DatabricksExportStorage]
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.export_storage.databricks.list()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/storages/export/databricks",
+            method="GET",
+            params={
+                "ordering": ordering,
+                "project": project,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[DatabricksExportStorage],
+                    construct_type(
+                        type_=typing.List[DatabricksExportStorage],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create(
+        self,
+        *,
+        catalog: str,
+        host: str,
+        project: int,
+        schema: str,
+        volume: str,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        last_sync: typing.Optional[dt.datetime] = OMIT,
+        last_sync_count: typing.Optional[int] = OMIT,
+        last_sync_job: typing.Optional[str] = OMIT,
+        meta: typing.Optional[typing.Optional[typing.Any]] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        regex_filter: typing.Optional[str] = OMIT,
+        request_timeout_s: typing.Optional[int] = OMIT,
+        status: typing.Optional[StatusC5AEnum] = OMIT,
+        stream_chunk_bytes: typing.Optional[int] = OMIT,
+        synchronizable: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        token: typing.Optional[str] = OMIT,
+        traceback: typing.Optional[str] = OMIT,
+        use_blob_urls: typing.Optional[bool] = OMIT,
+        verify_tls: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatabricksExportStorage:
+        """
+        Create a Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        catalog : str
+            UC catalog name
+
+        host : str
+            Databricks workspace base URL (https://...)
+
+        project : int
+            A unique integer value identifying this project.
+
+        schema : str
+            UC schema name
+
+        volume : str
+            UC volume name
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        description : typing.Optional[str]
+            Cloud storage description
+
+        last_sync : typing.Optional[dt.datetime]
+            Last sync finished time
+
+        last_sync_count : typing.Optional[int]
+            Count of tasks synced last time
+
+        last_sync_job : typing.Optional[str]
+            Last sync job ID
+
+        meta : typing.Optional[typing.Optional[typing.Any]]
+
+        prefix : typing.Optional[str]
+            Export path prefix under the volume
+
+        regex_filter : typing.Optional[str]
+            Regex for filtering objects
+
+        request_timeout_s : typing.Optional[int]
+
+        status : typing.Optional[StatusC5AEnum]
+
+        stream_chunk_bytes : typing.Optional[int]
+
+        synchronizable : typing.Optional[bool]
+
+        title : typing.Optional[str]
+            Cloud storage title
+
+        token : typing.Optional[str]
+
+        traceback : typing.Optional[str]
+            Traceback report for the last failed sync
+
+        use_blob_urls : typing.Optional[bool]
+            Generate blob URLs in tasks
+
+        verify_tls : typing.Optional[bool]
+            Verify TLS certificates
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.export_storage.databricks.create(
+                catalog="catalog",
+                host="host",
+                project=1,
+                schema="schema",
+                volume="volume",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/storages/export/databricks",
+            method="POST",
+            json={
+                "can_delete_objects": can_delete_objects,
+                "catalog": catalog,
+                "description": description,
+                "host": host,
+                "last_sync": last_sync,
+                "last_sync_count": last_sync_count,
+                "last_sync_job": last_sync_job,
+                "meta": meta,
+                "prefix": prefix,
+                "project": project,
+                "regex_filter": regex_filter,
+                "request_timeout_s": request_timeout_s,
+                "schema": schema,
+                "status": status,
+                "stream_chunk_bytes": stream_chunk_bytes,
+                "synchronizable": synchronizable,
+                "title": title,
+                "token": token,
+                "traceback": traceback,
+                "use_blob_urls": use_blob_urls,
+                "verify_tls": verify_tls,
+                "volume": volume,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def validate(
+        self,
+        *,
+        catalog: str,
+        host: str,
+        project: int,
+        schema: str,
+        volume: str,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        last_sync: typing.Optional[dt.datetime] = OMIT,
+        last_sync_count: typing.Optional[int] = OMIT,
+        last_sync_job: typing.Optional[str] = OMIT,
+        meta: typing.Optional[typing.Optional[typing.Any]] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        regex_filter: typing.Optional[str] = OMIT,
+        request_timeout_s: typing.Optional[int] = OMIT,
+        status: typing.Optional[StatusC5AEnum] = OMIT,
+        stream_chunk_bytes: typing.Optional[int] = OMIT,
+        synchronizable: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        token: typing.Optional[str] = OMIT,
+        traceback: typing.Optional[str] = OMIT,
+        use_blob_urls: typing.Optional[bool] = OMIT,
+        verify_tls: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Validate a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        catalog : str
+            UC catalog name
+
+        host : str
+            Databricks workspace base URL (https://...)
+
+        project : int
+            A unique integer value identifying this project.
+
+        schema : str
+            UC schema name
+
+        volume : str
+            UC volume name
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        description : typing.Optional[str]
+            Cloud storage description
+
+        last_sync : typing.Optional[dt.datetime]
+            Last sync finished time
+
+        last_sync_count : typing.Optional[int]
+            Count of tasks synced last time
+
+        last_sync_job : typing.Optional[str]
+            Last sync job ID
+
+        meta : typing.Optional[typing.Optional[typing.Any]]
+
+        prefix : typing.Optional[str]
+            Export path prefix under the volume
+
+        regex_filter : typing.Optional[str]
+            Regex for filtering objects
+
+        request_timeout_s : typing.Optional[int]
+
+        status : typing.Optional[StatusC5AEnum]
+
+        stream_chunk_bytes : typing.Optional[int]
+
+        synchronizable : typing.Optional[bool]
+
+        title : typing.Optional[str]
+            Cloud storage title
+
+        token : typing.Optional[str]
+
+        traceback : typing.Optional[str]
+            Traceback report for the last failed sync
+
+        use_blob_urls : typing.Optional[bool]
+            Generate blob URLs in tasks
+
+        verify_tls : typing.Optional[bool]
+            Verify TLS certificates
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.export_storage.databricks.validate(
+                catalog="catalog",
+                host="host",
+                project=1,
+                schema="schema",
+                volume="volume",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/storages/export/databricks/validate",
+            method="POST",
+            json={
+                "can_delete_objects": can_delete_objects,
+                "catalog": catalog,
+                "description": description,
+                "host": host,
+                "last_sync": last_sync,
+                "last_sync_count": last_sync_count,
+                "last_sync_job": last_sync_job,
+                "meta": meta,
+                "prefix": prefix,
+                "project": project,
+                "regex_filter": regex_filter,
+                "request_timeout_s": request_timeout_s,
+                "schema": schema,
+                "status": status,
+                "stream_chunk_bytes": stream_chunk_bytes,
+                "synchronizable": synchronizable,
+                "title": title,
+                "token": token,
+                "traceback": traceback,
+                "use_blob_urls": use_blob_urls,
+                "verify_tls": verify_tls,
+                "volume": volume,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> DatabricksExportStorage:
+        """
+        Get a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        id : int
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.export_storage.databricks.get(
+                id=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Delete a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        id : int
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.export_storage.databricks.delete(
+                id=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}",
+            method="DELETE",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def update(
+        self,
+        id: int,
+        *,
+        can_delete_objects: typing.Optional[bool] = OMIT,
+        catalog: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        host: typing.Optional[str] = OMIT,
+        last_sync: typing.Optional[dt.datetime] = OMIT,
+        last_sync_count: typing.Optional[int] = OMIT,
+        last_sync_job: typing.Optional[str] = OMIT,
+        meta: typing.Optional[typing.Optional[typing.Any]] = OMIT,
+        prefix: typing.Optional[str] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        regex_filter: typing.Optional[str] = OMIT,
+        request_timeout_s: typing.Optional[int] = OMIT,
+        schema: typing.Optional[str] = OMIT,
+        status: typing.Optional[StatusC5AEnum] = OMIT,
+        stream_chunk_bytes: typing.Optional[int] = OMIT,
+        synchronizable: typing.Optional[bool] = OMIT,
+        title: typing.Optional[str] = OMIT,
+        token: typing.Optional[str] = OMIT,
+        traceback: typing.Optional[str] = OMIT,
+        use_blob_urls: typing.Optional[bool] = OMIT,
+        verify_tls: typing.Optional[bool] = OMIT,
+        volume: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatabricksExportStorage:
+        """
+        Update a specific Databricks Files export storage connection.
+
+        Parameters
+        ----------
+        id : int
+
+        can_delete_objects : typing.Optional[bool]
+            Deletion from storage enabled
+
+        catalog : typing.Optional[str]
+            UC catalog name
+
+        description : typing.Optional[str]
+            Cloud storage description
+
+        host : typing.Optional[str]
+            Databricks workspace base URL (https://...)
+
+        last_sync : typing.Optional[dt.datetime]
+            Last sync finished time
+
+        last_sync_count : typing.Optional[int]
+            Count of tasks synced last time
+
+        last_sync_job : typing.Optional[str]
+            Last sync job ID
+
+        meta : typing.Optional[typing.Optional[typing.Any]]
+
+        prefix : typing.Optional[str]
+            Export path prefix under the volume
+
+        project : typing.Optional[int]
+            A unique integer value identifying this project.
+
+        regex_filter : typing.Optional[str]
+            Regex for filtering objects
+
+        request_timeout_s : typing.Optional[int]
+
+        schema : typing.Optional[str]
+            UC schema name
+
+        status : typing.Optional[StatusC5AEnum]
+
+        stream_chunk_bytes : typing.Optional[int]
+
+        synchronizable : typing.Optional[bool]
+
+        title : typing.Optional[str]
+            Cloud storage title
+
+        token : typing.Optional[str]
+
+        traceback : typing.Optional[str]
+            Traceback report for the last failed sync
+
+        use_blob_urls : typing.Optional[bool]
+            Generate blob URLs in tasks
+
+        verify_tls : typing.Optional[bool]
+            Verify TLS certificates
+
+        volume : typing.Optional[str]
+            UC volume name
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.export_storage.databricks.update(
+                id=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}",
+            method="PATCH",
+            json={
+                "can_delete_objects": can_delete_objects,
+                "catalog": catalog,
+                "description": description,
+                "host": host,
+                "last_sync": last_sync,
+                "last_sync_count": last_sync_count,
+                "last_sync_job": last_sync_job,
+                "meta": meta,
+                "prefix": prefix,
+                "project": project,
+                "regex_filter": regex_filter,
+                "request_timeout_s": request_timeout_s,
+                "schema": schema,
+                "status": status,
+                "stream_chunk_bytes": stream_chunk_bytes,
+                "synchronizable": synchronizable,
+                "title": title,
+                "token": token,
+                "traceback": traceback,
+                "use_blob_urls": use_blob_urls,
+                "verify_tls": verify_tls,
+                "volume": volume,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def sync(
+        self, id: int, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatabricksExportStorage:
+        """
+        Export annotations to a Databricks Files storage.
+
+        Parameters
+        ----------
+        id : int
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatabricksExportStorage
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.export_storage.databricks.sync(
+                id=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/storages/export/databricks/{jsonable_encoder(id)}/sync",
+            method="POST",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DatabricksExportStorage,
+                    construct_type(
+                        type_=DatabricksExportStorage,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)