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

label_studio_sdk/prompts/indicators/client.py

@@ -1,15 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from ...core.client_wrapper import SyncClientWrapper
 import typing
+from ...core.request_options import RequestOptions
+from ...types.key_indicators import KeyIndicators
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
-
 from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import pydantic_v1
-from ...core.request_options import RequestOptions
 from ...types.key_indicator_value import KeyIndicatorValue
-from ...types.key_indicators import KeyIndicators
+from ...core.client_wrapper import AsyncClientWrapper
 
 
 class IndicatorsClient:
@@ -35,7 +35,7 @@ class IndicatorsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -45,11 +45,19 @@ class IndicatorsClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/inference-runs/{jsonable_encoder(pk)}/indicators", method="GET", request_options=request_options
+            f"api/inference-runs/{jsonable_encoder(pk)}/indicators",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(KeyIndicators, _response.json())  # type: ignore
+                return typing.cast(
+                    KeyIndicators,
+                    parse_obj_as(
+                        type_=KeyIndicators,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -79,7 +87,7 @@ class IndicatorsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -96,7 +104,13 @@ class IndicatorsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(KeyIndicatorValue, _response.json())  # type: ignore
+                return typing.cast(
+                    KeyIndicatorValue,
+                    parse_obj_as(
+                        type_=KeyIndicatorValue,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -126,21 +140,37 @@ class AsyncIndicatorsClient:
 
         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.indicators.list(
-            pk=1,
-        )
+
+
+        async def main() -> None:
+            await client.prompts.indicators.list(
+                pk=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/inference-runs/{jsonable_encoder(pk)}/indicators", method="GET", request_options=request_options
+            f"api/inference-runs/{jsonable_encoder(pk)}/indicators",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(KeyIndicators, _response.json())  # type: ignore
+                return typing.cast(
+                    KeyIndicators,
+                    parse_obj_as(
+                        type_=KeyIndicators,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -170,15 +200,23 @@ class AsyncIndicatorsClient:
 
         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.indicators.get(
-            indicator_key="indicator_key",
-            pk=1,
-        )
+
+
+        async def main() -> None:
+            await client.prompts.indicators.get(
+                indicator_key="indicator_key",
+                pk=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/inference-runs/{jsonable_encoder(pk)}/indicators/{jsonable_encoder(indicator_key)}",
@@ -187,7 +225,13 @@ class AsyncIndicatorsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(KeyIndicatorValue, _response.json())  # type: ignore
+                return typing.cast(
+                    KeyIndicatorValue,
+                    parse_obj_as(
+                        type_=KeyIndicatorValue,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

label_studio_sdk/prompts/runs/client.py

@@ -1,20 +1,21 @@
 # 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 .types.runs_list_request_project_subset import RunsListRequestProjectSubset
 from ...core.request_options import RequestOptions
 from ...types.inference_run import InferenceRun
-from ...types.inference_run_created_by import InferenceRunCreatedBy
-from ...types.inference_run_organization import InferenceRunOrganization
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ...core.api_error import ApiError
 from ...types.inference_run_project_subset import InferenceRunProjectSubset
+from ...types.inference_run_organization import InferenceRunOrganization
+from ...types.inference_run_created_by import InferenceRunCreatedBy
 from ...types.inference_run_status import InferenceRunStatus
-from .types.runs_list_request_project_subset import RunsListRequestProjectSubset
+import datetime as dt
+from ...core.serialization import convert_and_respect_annotation_metadata
+from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -60,7 +61,7 @@ class RunsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -75,12 +76,21 @@ class RunsClient:
         _response = self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}/inference-runs",
             method="GET",
-            params={"project": project, "project_subset": project_subset},
+            params={
+                "project": project,
+                "project_subset": project_subset,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(InferenceRun, _response.json())  # type: ignore
+                return typing.cast(
+                    InferenceRun,
+                    parse_obj_as(
+                        type_=InferenceRun,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -94,7 +104,7 @@ class RunsClient:
         project: int,
         project_subset: InferenceRunProjectSubset,
         organization: typing.Optional[InferenceRunOrganization] = OMIT,
-        model_version: typing.Optional[str] = OMIT,
+        model_version: typing.Optional[int] = OMIT,
         created_by: typing.Optional[InferenceRunCreatedBy] = OMIT,
         status: typing.Optional[InferenceRunStatus] = OMIT,
         job_id: typing.Optional[str] = OMIT,
@@ -121,7 +131,7 @@ class RunsClient:
 
         organization : typing.Optional[InferenceRunOrganization]
 
-        model_version : typing.Optional[str]
+        model_version : typing.Optional[int]
 
         created_by : typing.Optional[InferenceRunCreatedBy]
 
@@ -147,7 +157,7 @@ class RunsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -163,10 +173,14 @@ class RunsClient:
             f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}/inference-runs",
             method="POST",
             json={
-                "organization": organization,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=InferenceRunOrganization, direction="write"
+                ),
                 "project": project,
                 "model_version": model_version,
-                "created_by": created_by,
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=InferenceRunCreatedBy, direction="write"
+                ),
                 "project_subset": project_subset,
                 "status": status,
                 "job_id": job_id,
@@ -180,7 +194,13 @@ class RunsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(InferenceRun, _response.json())  # type: ignore
+                return typing.cast(
+                    InferenceRun,
+                    parse_obj_as(
+                        type_=InferenceRun,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -227,27 +247,44 @@ class AsyncRunsClient:
 
         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.runs.list(
-            id=1,
-            version_id=1,
-            project=1,
-            project_subset="All",
-        )
+
+
+        async def main() -> None:
+            await client.prompts.runs.list(
+                id=1,
+                version_id=1,
+                project=1,
+                project_subset="All",
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}/inference-runs",
             method="GET",
-            params={"project": project, "project_subset": project_subset},
+            params={
+                "project": project,
+                "project_subset": project_subset,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(InferenceRun, _response.json())  # type: ignore
+                return typing.cast(
+                    InferenceRun,
+                    parse_obj_as(
+                        type_=InferenceRun,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -261,7 +298,7 @@ class AsyncRunsClient:
         project: int,
         project_subset: InferenceRunProjectSubset,
         organization: typing.Optional[InferenceRunOrganization] = OMIT,
-        model_version: typing.Optional[str] = OMIT,
+        model_version: typing.Optional[int] = OMIT,
         created_by: typing.Optional[InferenceRunCreatedBy] = OMIT,
         status: typing.Optional[InferenceRunStatus] = OMIT,
         job_id: typing.Optional[str] = OMIT,
@@ -288,7 +325,7 @@ class AsyncRunsClient:
 
         organization : typing.Optional[InferenceRunOrganization]
 
-        model_version : typing.Optional[str]
+        model_version : typing.Optional[int]
 
         created_by : typing.Optional[InferenceRunCreatedBy]
 
@@ -314,26 +351,38 @@ class AsyncRunsClient:
 
         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.runs.create(
-            id=1,
-            version_id=1,
-            project=1,
-            project_subset="All",
-        )
+
+
+        async def main() -> None:
+            await client.prompts.runs.create(
+                id=1,
+                version_id=1,
+                project=1,
+                project_subset="All",
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/prompts/{jsonable_encoder(id)}/versions/{jsonable_encoder(version_id)}/inference-runs",
             method="POST",
             json={
-                "organization": organization,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=InferenceRunOrganization, direction="write"
+                ),
                 "project": project,
                 "model_version": model_version,
-                "created_by": created_by,
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=InferenceRunCreatedBy, direction="write"
+                ),
                 "project_subset": project_subset,
                 "status": status,
                 "job_id": job_id,
@@ -347,7 +396,13 @@ class AsyncRunsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(InferenceRun, _response.json())  # type: ignore
+                return typing.cast(
+                    InferenceRun,
+                    parse_obj_as(
+                        type_=InferenceRun,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

label_studio_sdk/prompts/types/prompts_batch_failed_predictions_request_failed_predictions_item.py

@@ -1,42 +1,32 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ...core.pydantic_utilities import UniversalBaseModel
 import typing
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
 
-from ...core.datetime_utils import serialize_datetime
-from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class PromptsBatchFailedPredictionsRequestFailedPredictionsItem(pydantic_v1.BaseModel):
-    task_id: typing.Optional[int] = pydantic_v1.Field(default=None)
+class PromptsBatchFailedPredictionsRequestFailedPredictionsItem(UniversalBaseModel):
+    task_id: typing.Optional[int] = pydantic.Field(default=None)
     """
     Task ID to associate the prediction with
     """
 
-    error_type: typing.Optional[str] = pydantic_v1.Field(default=None)
+    error_type: typing.Optional[str] = pydantic.Field(default=None)
     """
     Type of error (e.g. "Timeout", "Rate Limit", etc)
     """
 
-    message: typing.Optional[str] = pydantic_v1.Field(default=None)
+    message: typing.Optional[str] = pydantic.Field(default=None)
     """
     Error message details
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/prompts/types/prompts_batch_failed_predictions_response.py

@@ -1,29 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ...core.pydantic_utilities import UniversalBaseModel
 import typing
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
-from ...core.datetime_utils import serialize_datetime
-from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class PromptsBatchFailedPredictionsResponse(pydantic_v1.BaseModel):
+class PromptsBatchFailedPredictionsResponse(UniversalBaseModel):
     detail: typing.Optional[str] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/prompts/types/prompts_batch_predictions_request_results_item.py

@@ -1,62 +1,59 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ...core.pydantic_utilities import UniversalBaseModel
 import typing
+import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
 
-from ...core.datetime_utils import serialize_datetime
-from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class PromptsBatchPredictionsRequestResultsItem(pydantic_v1.BaseModel):
-    task_id: typing.Optional[int] = pydantic_v1.Field(default=None)
+class PromptsBatchPredictionsRequestResultsItem(UniversalBaseModel):
+    task_id: typing.Optional[int] = pydantic.Field(default=None)
     """
     Task ID to associate the prediction with
     """
 
-    output: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    output: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
-    Prediction output that contains keys from labeling config. Each key must be a valid control tag name from the labeling config. For example, given the output: `json {"sentiment": "positive"} ` it will be converted to the internal LS annotation format: `json { "value": { "choices": ["positive"] }, "from_name": "label", "to_name": "", ... } `
+    Prediction output that contains keys from labeling config. Each key must be a valid control tag name from the labeling config. For example, given the output: ```json {"sentiment": "positive"} ``` it will be converted to the internal LS annotation format: ```json {
+      "value": {
+        "choices": ["positive"]
+      },
+      "from_name": "label",
+      "to_name": "",
+      ...
+    } ```
     """
 
-    prompt_tokens: typing.Optional[int] = pydantic_v1.Field(default=None)
+    prompt_tokens: typing.Optional[int] = pydantic.Field(default=None)
     """
     Number of tokens in the prompt
     """
 
-    completion_tokens: typing.Optional[int] = pydantic_v1.Field(default=None)
+    completion_tokens: typing.Optional[int] = pydantic.Field(default=None)
     """
     Number of tokens in the completion
     """
 
-    prompt_cost_usd: typing.Optional[float] = pydantic_v1.Field(default=None)
+    prompt_cost_usd: typing.Optional[float] = pydantic.Field(default=None)
     """
     Cost of the prompt (in USD)
     """
 
-    completion_cost_usd: typing.Optional[float] = pydantic_v1.Field(default=None)
+    completion_cost_usd: typing.Optional[float] = pydantic.Field(default=None)
     """
     Cost of the completion (in USD)
     """
 
-    total_cost_usd: typing.Optional[float] = pydantic_v1.Field(default=None)
+    total_cost_usd: typing.Optional[float] = pydantic.Field(default=None)
     """
     Total cost of the inference (in USD)
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/prompts/types/prompts_batch_predictions_response.py

@@ -1,29 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ...core.pydantic_utilities import UniversalBaseModel
 import typing
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
-from ...core.datetime_utils import serialize_datetime
-from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class PromptsBatchPredictionsResponse(pydantic_v1.BaseModel):
+class PromptsBatchPredictionsResponse(UniversalBaseModel):
     detail: typing.Optional[str] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow