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

label_studio_sdk/prompts/client.py

@@ -1,26 +1,31 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
 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 .versions.client import VersionsClient
+from .runs.client import RunsClient
+from .indicators.client import IndicatorsClient
 from ..core.request_options import RequestOptions
 from ..types.prompt import Prompt
+from ..core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ..core.api_error import ApiError
 from ..types.prompt_created_by import PromptCreatedBy
+import datetime as dt
 from ..types.prompt_organization import PromptOrganization
-from .indicators.client import AsyncIndicatorsClient, IndicatorsClient
-from .runs.client import AsyncRunsClient, RunsClient
+from ..types.prompt_associated_projects_item import PromptAssociatedProjectsItem
+from ..core.serialization import convert_and_respect_annotation_metadata
+from ..core.jsonable_encoder import jsonable_encoder
+from .types.prompts_batch_predictions_request_results_item import PromptsBatchPredictionsRequestResultsItem
+from .types.prompts_batch_predictions_response import PromptsBatchPredictionsResponse
 from .types.prompts_batch_failed_predictions_request_failed_predictions_item import (
     PromptsBatchFailedPredictionsRequestFailedPredictionsItem,
 )
 from .types.prompts_batch_failed_predictions_response import PromptsBatchFailedPredictionsResponse
-from .types.prompts_batch_predictions_request_results_item import PromptsBatchPredictionsRequestResultsItem
-from .types.prompts_batch_predictions_response import PromptsBatchPredictionsResponse
-from .versions.client import AsyncVersionsClient, VersionsClient
+from ..core.client_wrapper import AsyncClientWrapper
+from .versions.client import AsyncVersionsClient
+from .runs.client import AsyncRunsClient
+from .indicators.client import AsyncIndicatorsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -49,7 +54,7 @@ class PromptsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -57,11 +62,19 @@ class PromptsClient:
         client.prompts.list()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/prompts/", method="GET", request_options=request_options
+            "api/prompts/",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[Prompt], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[Prompt],
+                    parse_obj_as(
+                        type_=typing.List[Prompt],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -78,7 +91,7 @@ class PromptsClient:
         created_at: typing.Optional[dt.datetime] = OMIT,
         updated_at: typing.Optional[dt.datetime] = OMIT,
         organization: typing.Optional[PromptOrganization] = OMIT,
-        associated_projects: typing.Optional[typing.Sequence[int]] = OMIT,
+        associated_projects: typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]] = OMIT,
         skill_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Prompt:
@@ -111,8 +124,8 @@ class PromptsClient:
         organization : typing.Optional[PromptOrganization]
             Organization ID of the prompt
 
-        associated_projects : typing.Optional[typing.Sequence[int]]
-            List of associated projects IDs
+        associated_projects : typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]]
+            List of associated projects IDs or objects
 
         skill_name : typing.Optional[str]
             Name of the skill
@@ -127,7 +140,7 @@ class PromptsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -144,13 +157,21 @@ class PromptsClient:
             json={
                 "title": title,
                 "description": description,
-                "created_by": created_by,
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=PromptCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
-                "organization": organization,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=PromptOrganization, direction="write"
+                ),
                 "input_fields": input_fields,
                 "output_classes": output_classes,
-                "associated_projects": associated_projects,
+                "associated_projects": convert_and_respect_annotation_metadata(
+                    object_=associated_projects,
+                    annotation=typing.Sequence[PromptAssociatedProjectsItem],
+                    direction="write",
+                ),
                 "skill_name": skill_name,
             },
             request_options=request_options,
@@ -158,7 +179,13 @@ class PromptsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(Prompt, _response.json())  # type: ignore
+                return typing.cast(
+                    Prompt,
+                    parse_obj_as(
+                        type_=Prompt,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -183,7 +210,7 @@ class PromptsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -193,11 +220,19 @@ class PromptsClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}", method="GET", request_options=request_options
+            f"api/prompts/{jsonable_encoder(id)}",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(Prompt, _response.json())  # type: ignore
+                return typing.cast(
+                    Prompt,
+                    parse_obj_as(
+                        type_=Prompt,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -221,7 +256,7 @@ class PromptsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -231,7 +266,9 @@ class PromptsClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+            f"api/prompts/{jsonable_encoder(id)}",
+            method="DELETE",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -253,7 +290,7 @@ class PromptsClient:
         created_at: typing.Optional[dt.datetime] = OMIT,
         updated_at: typing.Optional[dt.datetime] = OMIT,
         organization: typing.Optional[PromptOrganization] = OMIT,
-        associated_projects: typing.Optional[typing.Sequence[int]] = OMIT,
+        associated_projects: typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]] = OMIT,
         skill_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Prompt:
@@ -289,8 +326,8 @@ class PromptsClient:
         organization : typing.Optional[PromptOrganization]
             Organization ID of the prompt
 
-        associated_projects : typing.Optional[typing.Sequence[int]]
-            List of associated projects IDs
+        associated_projects : typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]]
+            List of associated projects IDs or objects
 
         skill_name : typing.Optional[str]
             Name of the skill
@@ -305,7 +342,7 @@ class PromptsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -323,13 +360,21 @@ class PromptsClient:
             json={
                 "title": title,
                 "description": description,
-                "created_by": created_by,
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=PromptCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
-                "organization": organization,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=PromptOrganization, direction="write"
+                ),
                 "input_fields": input_fields,
                 "output_classes": output_classes,
-                "associated_projects": associated_projects,
+                "associated_projects": convert_and_respect_annotation_metadata(
+                    object_=associated_projects,
+                    annotation=typing.Sequence[PromptAssociatedProjectsItem],
+                    direction="write",
+                ),
                 "skill_name": skill_name,
             },
             request_options=request_options,
@@ -337,7 +382,13 @@ class PromptsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(Prompt, _response.json())  # type: ignore
+                return typing.cast(
+                    Prompt,
+                    parse_obj_as(
+                        type_=Prompt,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -370,7 +421,7 @@ class PromptsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -380,13 +431,29 @@ class PromptsClient:
         _response = self._client_wrapper.httpx_client.request(
             "api/model-run/batch-predictions",
             method="POST",
-            json={"modelrun_id": modelrun_id, "results": results},
+            json={
+                "modelrun_id": modelrun_id,
+                "results": convert_and_respect_annotation_metadata(
+                    object_=results,
+                    annotation=typing.Sequence[PromptsBatchPredictionsRequestResultsItem],
+                    direction="write",
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PromptsBatchPredictionsResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    PromptsBatchPredictionsResponse,
+                    parse_obj_as(
+                        type_=PromptsBatchPredictionsResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -421,7 +488,7 @@ class PromptsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -431,13 +498,29 @@ class PromptsClient:
         _response = self._client_wrapper.httpx_client.request(
             "api/model-run/batch-failed-predictions",
             method="POST",
-            json={"modelrun_id": modelrun_id, "failed_predictions": failed_predictions},
+            json={
+                "modelrun_id": modelrun_id,
+                "failed_predictions": convert_and_respect_annotation_metadata(
+                    object_=failed_predictions,
+                    annotation=typing.Sequence[PromptsBatchFailedPredictionsRequestFailedPredictionsItem],
+                    direction="write",
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PromptsBatchFailedPredictionsResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    PromptsBatchFailedPredictionsResponse,
+                    parse_obj_as(
+                        type_=PromptsBatchFailedPredictionsResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -467,19 +550,35 @@ class AsyncPromptsClient:
 
         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.prompts.list()
+
+
+        async def main() -> None:
+            await client.prompts.list()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/prompts/", method="GET", request_options=request_options
+            "api/prompts/",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[Prompt], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[Prompt],
+                    parse_obj_as(
+                        type_=typing.List[Prompt],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -496,7 +595,7 @@ class AsyncPromptsClient:
         created_at: typing.Optional[dt.datetime] = OMIT,
         updated_at: typing.Optional[dt.datetime] = OMIT,
         organization: typing.Optional[PromptOrganization] = OMIT,
-        associated_projects: typing.Optional[typing.Sequence[int]] = OMIT,
+        associated_projects: typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]] = OMIT,
         skill_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Prompt:
@@ -529,8 +628,8 @@ class AsyncPromptsClient:
         organization : typing.Optional[PromptOrganization]
             Organization ID of the prompt
 
-        associated_projects : typing.Optional[typing.Sequence[int]]
-            List of associated projects IDs
+        associated_projects : typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]]
+            List of associated projects IDs or objects
 
         skill_name : typing.Optional[str]
             Name of the skill
@@ -545,16 +644,24 @@ class AsyncPromptsClient:
 
         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.prompts.create(
-            title="title",
-            input_fields=["input_fields"],
-            output_classes=["output_classes"],
-        )
+
+
+        async def main() -> None:
+            await client.prompts.create(
+                title="title",
+                input_fields=["input_fields"],
+                output_classes=["output_classes"],
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/prompts/",
@@ -562,13 +669,21 @@ class AsyncPromptsClient:
             json={
                 "title": title,
                 "description": description,
-                "created_by": created_by,
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=PromptCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
-                "organization": organization,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=PromptOrganization, direction="write"
+                ),
                 "input_fields": input_fields,
                 "output_classes": output_classes,
-                "associated_projects": associated_projects,
+                "associated_projects": convert_and_respect_annotation_metadata(
+                    object_=associated_projects,
+                    annotation=typing.Sequence[PromptAssociatedProjectsItem],
+                    direction="write",
+                ),
                 "skill_name": skill_name,
             },
             request_options=request_options,
@@ -576,7 +691,13 @@ class AsyncPromptsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(Prompt, _response.json())  # type: ignore
+                return typing.cast(
+                    Prompt,
+                    parse_obj_as(
+                        type_=Prompt,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -601,21 +722,37 @@ class AsyncPromptsClient:
 
         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.prompts.get(
-            id=1,
-        )
+
+
+        async def main() -> None:
+            await client.prompts.get(
+                id=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}", method="GET", request_options=request_options
+            f"api/prompts/{jsonable_encoder(id)}",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(Prompt, _response.json())  # type: ignore
+                return typing.cast(
+                    Prompt,
+                    parse_obj_as(
+                        type_=Prompt,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -639,17 +776,27 @@ class AsyncPromptsClient:
 
         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.prompts.delete(
-            id=1,
-        )
+
+
+        async def main() -> None:
+            await client.prompts.delete(
+                id=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/prompts/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+            f"api/prompts/{jsonable_encoder(id)}",
+            method="DELETE",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -671,7 +818,7 @@ class AsyncPromptsClient:
         created_at: typing.Optional[dt.datetime] = OMIT,
         updated_at: typing.Optional[dt.datetime] = OMIT,
         organization: typing.Optional[PromptOrganization] = OMIT,
-        associated_projects: typing.Optional[typing.Sequence[int]] = OMIT,
+        associated_projects: typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]] = OMIT,
         skill_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Prompt:
@@ -707,8 +854,8 @@ class AsyncPromptsClient:
         organization : typing.Optional[PromptOrganization]
             Organization ID of the prompt
 
-        associated_projects : typing.Optional[typing.Sequence[int]]
-            List of associated projects IDs
+        associated_projects : typing.Optional[typing.Sequence[PromptAssociatedProjectsItem]]
+            List of associated projects IDs or objects
 
         skill_name : typing.Optional[str]
             Name of the skill
@@ -723,17 +870,25 @@ class AsyncPromptsClient:
 
         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.prompts.update(
-            id=1,
-            title="title",
-            input_fields=["input_fields"],
-            output_classes=["output_classes"],
-        )
+
+
+        async def main() -> None:
+            await client.prompts.update(
+                id=1,
+                title="title",
+                input_fields=["input_fields"],
+                output_classes=["output_classes"],
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(id)}",
@@ -741,13 +896,21 @@ class AsyncPromptsClient:
             json={
                 "title": title,
                 "description": description,
-                "created_by": created_by,
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=PromptCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
-                "organization": organization,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=PromptOrganization, direction="write"
+                ),
                 "input_fields": input_fields,
                 "output_classes": output_classes,
-                "associated_projects": associated_projects,
+                "associated_projects": convert_and_respect_annotation_metadata(
+                    object_=associated_projects,
+                    annotation=typing.Sequence[PromptAssociatedProjectsItem],
+                    direction="write",
+                ),
                 "skill_name": skill_name,
             },
             request_options=request_options,
@@ -755,7 +918,13 @@ class AsyncPromptsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(Prompt, _response.json())  # type: ignore
+                return typing.cast(
+                    Prompt,
+                    parse_obj_as(
+                        type_=Prompt,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -788,23 +957,47 @@ class AsyncPromptsClient:
 
         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.prompts.batch_predictions()
+
+
+        async def main() -> None:
+            await client.prompts.batch_predictions()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/model-run/batch-predictions",
             method="POST",
-            json={"modelrun_id": modelrun_id, "results": results},
+            json={
+                "modelrun_id": modelrun_id,
+                "results": convert_and_respect_annotation_metadata(
+                    object_=results,
+                    annotation=typing.Sequence[PromptsBatchPredictionsRequestResultsItem],
+                    direction="write",
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PromptsBatchPredictionsResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    PromptsBatchPredictionsResponse,
+                    parse_obj_as(
+                        type_=PromptsBatchPredictionsResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -839,23 +1032,47 @@ class AsyncPromptsClient:
 
         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.prompts.batch_failed_predictions()
+
+
+        async def main() -> None:
+            await client.prompts.batch_failed_predictions()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/model-run/batch-failed-predictions",
             method="POST",
-            json={"modelrun_id": modelrun_id, "failed_predictions": failed_predictions},
+            json={
+                "modelrun_id": modelrun_id,
+                "failed_predictions": convert_and_respect_annotation_metadata(
+                    object_=failed_predictions,
+                    annotation=typing.Sequence[PromptsBatchFailedPredictionsRequestFailedPredictionsItem],
+                    direction="write",
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PromptsBatchFailedPredictionsResponse, _response.json())  # type: ignore
+                return typing.cast(
+                    PromptsBatchFailedPredictionsResponse,
+                    parse_obj_as(
+                        type_=PromptsBatchFailedPredictionsResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)