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

label_studio_sdk/ml/client.py

@@ -1,19 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from json.decoder import JSONDecodeError
-
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.jsonable_encoder import jsonable_encoder
-from ..core.pydantic_utilities import pydantic_v1
+from ..core.client_wrapper import SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..errors.internal_server_error import InternalServerError
 from ..types.ml_backend import MlBackend
+from ..core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ..core.api_error import ApiError
 from .types.ml_create_request_auth_method import MlCreateRequestAuthMethod
 from .types.ml_create_response import MlCreateResponse
+from ..core.jsonable_encoder import jsonable_encoder
 from .types.ml_update_request_auth_method import MlUpdateRequestAuthMethod
 from .types.ml_update_response import MlUpdateResponse
+from ..errors.internal_server_error import InternalServerError
+from ..core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -27,8 +27,10 @@ class MlClient:
         self, *, project: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[MlBackend]:
         """
+
         List all configured Machine Learning (ML) backends for a specific project by ID. For more information about ML backends, see [Machine learning integration](https://labelstud.io/guide/ml).
 
+
         You will need to provide the project ID. This can be found in the URL when viewing the project in Label Studio, or you can retrieve all project IDs using [List all projects](../projects/list).
 
         Parameters
@@ -46,7 +48,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -54,11 +56,22 @@ class MlClient:
         client.ml.list()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/ml/", method="GET", params={"project": project}, request_options=request_options
+            "api/ml/",
+            method="GET",
+            params={
+                "project": project,
+            },
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[MlBackend], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[MlBackend],
+                    parse_obj_as(
+                        type_=typing.List[MlBackend],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -75,11 +88,12 @@ class MlClient:
         auth_method: typing.Optional[MlCreateRequestAuthMethod] = OMIT,
         basic_auth_user: typing.Optional[str] = OMIT,
         basic_auth_pass: typing.Optional[str] = OMIT,
-        extra_params: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        extra_params: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         timeout: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> MlCreateResponse:
         """
+
         Add an ML backend to a project. For more information about what you need to configure when adding an ML backend, see [Connect the model to Label studio](https://labelstud.io/guide/ml#Connect-the-model-to-Label-Studio).
 
         <Note>If you are using Docker Compose, you may need to adjust your ML backend URL. See [localhost and Docker containers](https://labelstud.io/guide/ml#localhost-and-Docker-containers).</Note>
@@ -112,7 +126,7 @@ class MlClient:
         basic_auth_pass : typing.Optional[str]
             Basic auth password
 
-        extra_params : typing.Optional[typing.Dict[str, typing.Any]]
+        extra_params : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Extra parameters
 
         timeout : typing.Optional[int]
@@ -128,7 +142,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -150,12 +164,21 @@ class MlClient:
                 "extra_params": extra_params,
                 "timeout": timeout,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlCreateResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    MlCreateResponse,
+                    parse_obj_as(
+                        type_=MlCreateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -163,6 +186,7 @@ class MlClient:
 
     def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> MlBackend:
         """
+
         Get details about a specific ML backend. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         For more information, see [Machine learning integration](https://labelstud.io/guide/ml).
@@ -182,7 +206,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -192,11 +216,19 @@ class MlClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/ml/{jsonable_encoder(id)}", method="GET", request_options=request_options
+            f"api/ml/{jsonable_encoder(id)}",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlBackend, _response.json())  # type: ignore
+                return typing.cast(
+                    MlBackend,
+                    parse_obj_as(
+                        type_=MlBackend,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -204,6 +236,7 @@ class MlClient:
 
     def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
+
         Remove an existing ML backend connection. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         For more information, see [Machine learning integration](https://labelstud.io/guide/ml).
@@ -222,7 +255,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -232,7 +265,9 @@ class MlClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/ml/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+            f"api/ml/{jsonable_encoder(id)}",
+            method="DELETE",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -254,11 +289,12 @@ class MlClient:
         auth_method: typing.Optional[MlUpdateRequestAuthMethod] = OMIT,
         basic_auth_user: typing.Optional[str] = OMIT,
         basic_auth_pass: typing.Optional[str] = OMIT,
-        extra_params: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        extra_params: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         timeout: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> MlUpdateResponse:
         """
+
         Update the ML backend parameters. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         For more information, see [Machine learning integration](https://labelstud.io/guide/ml).
@@ -292,7 +328,7 @@ class MlClient:
         basic_auth_pass : typing.Optional[str]
             Basic auth password
 
-        extra_params : typing.Optional[typing.Dict[str, typing.Any]]
+        extra_params : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Extra parameters
 
         timeout : typing.Optional[int]
@@ -308,7 +344,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -332,12 +368,21 @@ class MlClient:
                 "extra_params": extra_params,
                 "timeout": timeout,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlUpdateResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    MlUpdateResponse,
+                    parse_obj_as(
+                        type_=MlUpdateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -348,10 +393,11 @@ class MlClient:
         id: int,
         *,
         task: int,
-        context: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        context: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
+
         Enable interactive pre-annotations for a specific task.
 
         ML-assisted labeling with interactive pre-annotations works with image segmentation and object detection tasks using rectangles, ellipses, polygons, brush masks, and keypoints, as well as with HTML and text named entity recognition tasks. Your ML backend must support the type of labeling that you’re performing, recognize the input that you create, and be able to respond with the relevant output for a prediction. For more information, see [Interactive pre-annotations](https://labelstud.io/guide/ml.html#Interactive-pre-annotations).
@@ -368,7 +414,7 @@ class MlClient:
         task : int
             ID of task to annotate
 
-        context : typing.Optional[typing.Dict[str, typing.Any]]
+        context : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Context for ML model
 
         request_options : typing.Optional[RequestOptions]
@@ -380,7 +426,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -393,7 +439,13 @@ class MlClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/ml/{jsonable_encoder(id)}/interactive-annotating",
             method="POST",
-            json={"task": task, "context": context},
+            json={
+                "task": task,
+                "context": context,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -413,6 +465,7 @@ class MlClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
+
         After you connect a model to Label Studio as a machine learning backend and annotate at least one task, you can start training the model. Training logs appear in stdout and the console.
 
         For more information, see [Model training](https://labelstud.io/guide/ml.html#Model-training).
@@ -436,7 +489,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -448,7 +501,12 @@ class MlClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/ml/{jsonable_encoder(id)}/train",
             method="POST",
-            json={"use_ground_truth": use_ground_truth},
+            json={
+                "use_ground_truth": use_ground_truth,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -456,7 +514,15 @@ class MlClient:
             if 200 <= _response.status_code < 300:
                 return
             if _response.status_code == 500:
-                raise InternalServerError(pydantic_v1.parse_obj_as(str, _response.json()))  # type: ignore
+                raise InternalServerError(
+                    typing.cast(
+                        str,
+                        parse_obj_as(
+                            type_=str,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -464,6 +530,7 @@ class MlClient:
 
     def list_model_versions(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
+
         Get available versions of the model. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         Parameters
@@ -479,7 +546,7 @@ class MlClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -489,7 +556,9 @@ class MlClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/ml/{jsonable_encoder(id)}/versions", method="GET", request_options=request_options
+            f"api/ml/{jsonable_encoder(id)}/versions",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -508,8 +577,10 @@ class AsyncMlClient:
         self, *, project: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[MlBackend]:
         """
+
         List all configured Machine Learning (ML) backends for a specific project by ID. For more information about ML backends, see [Machine learning integration](https://labelstud.io/guide/ml).
 
+
         You will need to provide the project ID. This can be found in the URL when viewing the project in Label Studio, or you can retrieve all project IDs using [List all projects](../projects/list).
 
         Parameters
@@ -527,19 +598,38 @@ class AsyncMlClient:
 
         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.ml.list()
+
+
+        async def main() -> None:
+            await client.ml.list()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/ml/", method="GET", params={"project": project}, request_options=request_options
+            "api/ml/",
+            method="GET",
+            params={
+                "project": project,
+            },
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[MlBackend], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[MlBackend],
+                    parse_obj_as(
+                        type_=typing.List[MlBackend],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -556,11 +646,12 @@ class AsyncMlClient:
         auth_method: typing.Optional[MlCreateRequestAuthMethod] = OMIT,
         basic_auth_user: typing.Optional[str] = OMIT,
         basic_auth_pass: typing.Optional[str] = OMIT,
-        extra_params: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        extra_params: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         timeout: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> MlCreateResponse:
         """
+
         Add an ML backend to a project. For more information about what you need to configure when adding an ML backend, see [Connect the model to Label studio](https://labelstud.io/guide/ml#Connect-the-model-to-Label-Studio).
 
         <Note>If you are using Docker Compose, you may need to adjust your ML backend URL. See [localhost and Docker containers](https://labelstud.io/guide/ml#localhost-and-Docker-containers).</Note>
@@ -593,7 +684,7 @@ class AsyncMlClient:
         basic_auth_pass : typing.Optional[str]
             Basic auth password
 
-        extra_params : typing.Optional[typing.Dict[str, typing.Any]]
+        extra_params : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Extra parameters
 
         timeout : typing.Optional[int]
@@ -609,12 +700,20 @@ class AsyncMlClient:
 
         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.ml.create()
+
+
+        async def main() -> None:
+            await client.ml.create()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/ml/",
@@ -631,12 +730,21 @@ class AsyncMlClient:
                 "extra_params": extra_params,
                 "timeout": timeout,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlCreateResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    MlCreateResponse,
+                    parse_obj_as(
+                        type_=MlCreateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -644,6 +752,7 @@ class AsyncMlClient:
 
     async def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> MlBackend:
         """
+
         Get details about a specific ML backend. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         For more information, see [Machine learning integration](https://labelstud.io/guide/ml).
@@ -663,21 +772,37 @@ class AsyncMlClient:
 
         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.ml.get(
-            id=1,
-        )
+
+
+        async def main() -> None:
+            await client.ml.get(
+                id=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/ml/{jsonable_encoder(id)}", method="GET", request_options=request_options
+            f"api/ml/{jsonable_encoder(id)}",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlBackend, _response.json())  # type: ignore
+                return typing.cast(
+                    MlBackend,
+                    parse_obj_as(
+                        type_=MlBackend,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -685,6 +810,7 @@ class AsyncMlClient:
 
     async def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
+
         Remove an existing ML backend connection. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         For more information, see [Machine learning integration](https://labelstud.io/guide/ml).
@@ -703,17 +829,27 @@ class AsyncMlClient:
 
         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.ml.delete(
-            id=1,
-        )
+
+
+        async def main() -> None:
+            await client.ml.delete(
+                id=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/ml/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+            f"api/ml/{jsonable_encoder(id)}",
+            method="DELETE",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -735,11 +871,12 @@ class AsyncMlClient:
         auth_method: typing.Optional[MlUpdateRequestAuthMethod] = OMIT,
         basic_auth_user: typing.Optional[str] = OMIT,
         basic_auth_pass: typing.Optional[str] = OMIT,
-        extra_params: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        extra_params: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         timeout: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> MlUpdateResponse:
         """
+
         Update the ML backend parameters. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         For more information, see [Machine learning integration](https://labelstud.io/guide/ml).
@@ -773,7 +910,7 @@ class AsyncMlClient:
         basic_auth_pass : typing.Optional[str]
             Basic auth password
 
-        extra_params : typing.Optional[typing.Dict[str, typing.Any]]
+        extra_params : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Extra parameters
 
         timeout : typing.Optional[int]
@@ -789,14 +926,22 @@ class AsyncMlClient:
 
         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.ml.update(
-            id=1,
-        )
+
+
+        async def main() -> None:
+            await client.ml.update(
+                id=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/ml/{jsonable_encoder(id)}",
@@ -813,12 +958,21 @@ class AsyncMlClient:
                 "extra_params": extra_params,
                 "timeout": timeout,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlUpdateResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    MlUpdateResponse,
+                    parse_obj_as(
+                        type_=MlUpdateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -829,10 +983,11 @@ class AsyncMlClient:
         id: int,
         *,
         task: int,
-        context: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        context: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
+
         Enable interactive pre-annotations for a specific task.
 
         ML-assisted labeling with interactive pre-annotations works with image segmentation and object detection tasks using rectangles, ellipses, polygons, brush masks, and keypoints, as well as with HTML and text named entity recognition tasks. Your ML backend must support the type of labeling that you’re performing, recognize the input that you create, and be able to respond with the relevant output for a prediction. For more information, see [Interactive pre-annotations](https://labelstud.io/guide/ml.html#Interactive-pre-annotations).
@@ -849,7 +1004,7 @@ class AsyncMlClient:
         task : int
             ID of task to annotate
 
-        context : typing.Optional[typing.Dict[str, typing.Any]]
+        context : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Context for ML model
 
         request_options : typing.Optional[RequestOptions]
@@ -861,20 +1016,34 @@ class AsyncMlClient:
 
         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.ml.predict_interactive(
-            id=1,
-            task=1,
-        )
+
+
+        async def main() -> None:
+            await client.ml.predict_interactive(
+                id=1,
+                task=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/ml/{jsonable_encoder(id)}/interactive-annotating",
             method="POST",
-            json={"task": task, "context": context},
+            json={
+                "task": task,
+                "context": context,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -894,6 +1063,7 @@ class AsyncMlClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
+
         After you connect a model to Label Studio as a machine learning backend and annotate at least one task, you can start training the model. Training logs appear in stdout and the console.
 
         For more information, see [Model training](https://labelstud.io/guide/ml.html#Model-training).
@@ -917,19 +1087,32 @@ class AsyncMlClient:
 
         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.ml.train(
-            id=1,
-        )
+
+
+        async def main() -> None:
+            await client.ml.train(
+                id=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/ml/{jsonable_encoder(id)}/train",
             method="POST",
-            json={"use_ground_truth": use_ground_truth},
+            json={
+                "use_ground_truth": use_ground_truth,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -937,7 +1120,15 @@ class AsyncMlClient:
             if 200 <= _response.status_code < 300:
                 return
             if _response.status_code == 500:
-                raise InternalServerError(pydantic_v1.parse_obj_as(str, _response.json()))  # type: ignore
+                raise InternalServerError(
+                    typing.cast(
+                        str,
+                        parse_obj_as(
+                            type_=str,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -945,6 +1136,7 @@ class AsyncMlClient:
 
     async def list_model_versions(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
+
         Get available versions of the model. You will need to specify an ID for the backend connection. You can find this using [List ML backends](list).
 
         Parameters
@@ -960,17 +1152,27 @@ class AsyncMlClient:
 
         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.ml.list_model_versions(
-            id="id",
-        )
+
+
+        async def main() -> None:
+            await client.ml.list_model_versions(
+                id="id",
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/ml/{jsonable_encoder(id)}/versions", method="GET", request_options=request_options
+            f"api/ml/{jsonable_encoder(id)}/versions",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300: