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

label_studio_sdk/projects/exports/client_ext.py

@@ -0,0 +1,134 @@
+import json
+import time
+import asyncio
+import typing
+import pandas as pd
+from .client import ExportsClient, AsyncExportsClient
+from io import BytesIO
+from label_studio_sdk.versions.client import VersionsClient, AsyncVersionsClient
+from label_studio_sdk.core.api_error import ApiError
+
+
+class ExportTimeoutError(ApiError):
+    
+    def __init__(self, export_snapshot):
+        super().__init__(
+            status_code=500,
+            body=(
+                f"Export job timed out after {timeout} seconds: "
+                f"unable to retrieve export job {export_snapshot.id}. "
+                f"Current status: {export_snapshot.status}. "
+                f"Try manually checking the running job with "
+                f"`ls.projects.exports.get(project_id={project_id}, export_pk={export_snapshot.id})`."
+            )
+        )
+        
+        
+def _bytestream_to_fileobj(bytestream: typing.Iterable[bytes]) -> typing.BinaryIO:
+    buffer = BytesIO()
+    for chunk in bytestream:
+        buffer.write(chunk)
+    buffer.seek(0)
+    return buffer
+
+def _bytestream_to_binary(bytestream: typing.Iterable[bytes]) -> bytes:
+    fileobj = _bytestream_to_fileobj(bytestream)
+    return fileobj.getvalue()
+
+def _bytestream_to_json(bytestream: typing.Iterable[bytes]) -> dict:
+    fileobj = _bytestream_to_fileobj(bytestream)
+    return json.load(fileobj)
+
+def _bytestream_to_pandas(bytestream: typing.Iterable[bytes]) -> pd.DataFrame:
+    fileobj = _bytestream_to_fileobj(bytestream)
+    return pd.read_csv(fileobj)
+
+class ExportsClientExt(ExportsClient):
+    
+    def _get_bytestream(
+        self,
+        project_id: int,
+        export_type: str,
+        timeout: int = 60,
+        create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None,
+        convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None,
+        download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None,
+    ):
+        version = VersionsClient(client_wrapper=self._client_wrapper).get()
+        
+        if version.edition == "Enterprise":
+            # Enterprise edition exports are async, so we need to wait for the export job to complete
+            export_snapshot = self.create(project_id, **(create_kwargs or {}))
+            if export_type != "JSON":
+                self.convert(project_id, export_pk=export_snapshot.id, export_type=export_type, **(convert_kwargs or {}))
+            start_time = time.time()
+            while export_snapshot.status != "completed":
+                export_snapshot = self.get(project_id, export_pk=export_snapshot.id)
+                if time.time() - start_time > timeout:
+                    raise ExportTimeoutError(export_snapshot)
+                time.sleep(1)
+            bytestream = self.download(project_id, export_pk=export_snapshot.id, export_type=export_type, request_options={'chunk_size': 1024}, **(download_kwargs or {}))
+        else:
+            # Community edition exports are sync, so we can download the file immediately
+            bytestream = self.download_sync(project_id, export_type=export_type, download_all_tasks=True, download_resources=True)
+        return bytestream
+    
+    def as_file(self, project_id: int, export_type: str = "JSON", timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = self._get_bytestream(project_id, export_type, timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_fileobj(bytestream)
+    
+    def as_binary(self, project_id: int, export_type: str = "JSON", timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = self._get_bytestream(project_id, export_type, timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_binary(bytestream)
+    
+    def as_json(self, project_id: int, timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = self._get_bytestream(project_id, "JSON", timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_json(bytestream)
+    
+    def as_pandas(self, project_id: int, timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = self._get_bytestream(project_id, "CSV", timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_pandas(bytestream)
+    
+class AsyncExportsClientExt(AsyncExportsClient):
+
+    async def _get_bytestream(
+        self,
+        project_id: int,
+        export_type: str,
+        timeout: int = 60,
+        create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None,
+        convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None,
+        download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None,
+    ):
+        version = await AsyncVersionsClient(client_wrapper=self._client_wrapper).get()
+        if version.edition == "Enterprise":
+            # Enterprise edition exports are async, so we need to wait for the export job to complete
+            export_snapshot = await self.create(project_id, **(create_kwargs or {}))
+            if export_type != "JSON":
+                await self.convert(project_id, export_pk=export_snapshot.id, export_type=export_type, **(convert_kwargs or {}))
+            start_time = time.time()
+            while export_snapshot.status != "completed":
+                export_snapshot = await self.get(project_id, export_pk=export_snapshot.id)
+                if time.time() - start_time > timeout:
+                    raise ExportTimeoutError(export_snapshot)
+                await asyncio.sleep(1)
+            bytestream = await self.download(project_id, export_pk=export_snapshot.id, export_type=export_type, request_options={'chunk_size': 1024}, **(download_kwargs or {}))
+        else:
+            bytestream = await self.download_sync(project_id, export_type=export_type, download_all_tasks=True, download_resources=True)
+        return bytestream
+    
+    async def as_file(self, project_id: int, export_type: str = "JSON", timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = await self._get_bytestream(project_id, export_type, timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_fileobj(bytestream)
+    
+    async def as_binary(self, project_id: int, export_type: str = "JSON", timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = await self._get_bytestream(project_id, export_type, timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_binary(bytestream)
+    
+    async def as_json(self, project_id: int, timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = await self._get_bytestream(project_id, "JSON", timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_json(bytestream)
+    
+    async def as_pandas(self, project_id: int, timeout: int = 60, create_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, convert_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None, download_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None):
+        bytestream = await self._get_bytestream(project_id, "CSV", timeout, create_kwargs, convert_kwargs, download_kwargs)
+        return _bytestream_to_pandas(bytestream)

label_studio_sdk/projects/exports/types/__init__.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .exports_convert_response import ExportsConvertResponse
+from .exports_list_formats_response_item import ExportsListFormatsResponseItem
+
+__all__ = ["ExportsConvertResponse", "ExportsListFormatsResponseItem"]

label_studio_sdk/projects/exports/types/exports_convert_response.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ....core.pydantic_utilities import UniversalBaseModel
+import typing
+from ....types.export_format import ExportFormat
+import pydantic
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class ExportsConvertResponse(UniversalBaseModel):
+    export_type: typing.Optional[ExportFormat] = None
+    converted_format: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    ID of the converted format
+    """
+
+    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.Extra.allow

label_studio_sdk/projects/exports/types/exports_list_formats_response_item.py

@@ -0,0 +1,44 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ....core.pydantic_utilities import UniversalBaseModel
+import typing
+from ....types.export_format import ExportFormat
+import pydantic
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class ExportsListFormatsResponseItem(UniversalBaseModel):
+    name: typing.Optional[ExportFormat] = None
+    title: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Export format title
+    """
+
+    description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Export format description
+    """
+
+    link: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Export format documentation link
+    """
+
+    tags: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Export format tags
+    """
+
+    disabled: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    If true, the export format is not supported by the project.
+    """
+
+    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.Extra.allow

label_studio_sdk/projects/types/projects_create_response.py

@@ -1,96 +1,91 @@
 # 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 ProjectsCreateResponse(pydantic_v1.BaseModel):
+class ProjectsCreateResponse(UniversalBaseModel):
     """
     Project
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    id: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Project ID
+    """
+
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Project title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Project description
     """
 
-    label_config: typing.Optional[str] = pydantic_v1.Field(default=None)
+    label_config: typing.Optional[str] = pydantic.Field(default=None)
     """
     Label config in XML format
     """
 
-    expert_instruction: typing.Optional[str] = pydantic_v1.Field(default=None)
+    expert_instruction: typing.Optional[str] = pydantic.Field(default=None)
     """
     Labeling instructions to show to the user
     """
 
-    show_instruction: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_instruction: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show labeling instructions
     """
 
-    show_skip_button: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_skip_button: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show skip button
     """
 
-    enable_empty_annotation: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    enable_empty_annotation: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Allow empty annotations
     """
 
-    show_annotation_history: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_annotation_history: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show annotation history
     """
 
-    reveal_preannotations_interactively: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    reveal_preannotations_interactively: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Reveal preannotations interactively. If set to True, predictions will be shown to the user only after selecting the area of interest
     """
 
-    show_collab_predictions: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_collab_predictions: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show predictions to annotators
     """
 
-    maximum_annotations: typing.Optional[int] = pydantic_v1.Field(default=None)
+    maximum_annotations: typing.Optional[int] = pydantic.Field(default=None)
     """
     Maximum annotations per task
     """
 
-    color: typing.Optional[str] = pydantic_v1.Field(default=None)
+    color: typing.Optional[str] = pydantic.Field(default=None)
     """
     Project color in HEX format
     """
 
-    control_weights: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    control_weights: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Dict of weights for each control tag in metric calculation. Each control tag (e.g. label or choice) will have its own key in control weight dict with weight for each label and overall weight. For example, if a bounding box annotation with a control tag named my_bbox should be included with 0.33 weight in agreement calculation, and the first label Car should be twice as important as Airplane, then you need to specify: {'my_bbox': {'type': 'RectangleLabels', 'labels': {'Car': 1.0, 'Airplane': 0.5}, 'overall': 0.33}
     """
 
-    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/projects/types/projects_import_tasks_response.py

@@ -1,71 +1,61 @@
 # 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 ProjectsImportTasksResponse(pydantic_v1.BaseModel):
+class ProjectsImportTasksResponse(UniversalBaseModel):
     """
     Task creation response
     """
 
-    task_count: typing.Optional[int] = pydantic_v1.Field(default=None)
+    task_count: typing.Optional[int] = pydantic.Field(default=None)
     """
     Number of tasks added
     """
 
-    annotation_count: typing.Optional[int] = pydantic_v1.Field(default=None)
+    annotation_count: typing.Optional[int] = pydantic.Field(default=None)
     """
     Number of annotations added
     """
 
-    predictions_count: typing.Optional[int] = pydantic_v1.Field(default=None)
+    predictions_count: typing.Optional[int] = pydantic.Field(default=None)
     """
     Number of predictions added
     """
 
-    duration: typing.Optional[float] = pydantic_v1.Field(default=None)
+    duration: typing.Optional[float] = pydantic.Field(default=None)
     """
     Time in seconds to create
     """
 
-    file_upload_ids: typing.Optional[typing.List[int]] = pydantic_v1.Field(default=None)
+    file_upload_ids: typing.Optional[typing.List[int]] = pydantic.Field(default=None)
     """
     Database IDs of uploaded files
     """
 
-    could_be_tasks_list: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    could_be_tasks_list: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Whether uploaded files can contain lists of tasks, like CSV/TSV files
     """
 
-    found_formats: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
+    found_formats: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
     The list of found file formats
     """
 
-    data_columns: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
+    data_columns: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
     The list of found data columns
     """
 
-    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/projects/types/projects_list_response.py

@@ -1,33 +1,23 @@
 # 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.datetime_utils import serialize_datetime
-from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from ...types.project import Project
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class ProjectsListResponse(pydantic_v1.BaseModel):
+class ProjectsListResponse(UniversalBaseModel):
     count: int
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
     results: typing.List[Project]
 
-    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/projects/types/projects_update_response.py

@@ -1,96 +1,86 @@
 # 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 ProjectsUpdateResponse(pydantic_v1.BaseModel):
+class ProjectsUpdateResponse(UniversalBaseModel):
     """
     Project
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Project title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Project description
     """
 
-    label_config: typing.Optional[str] = pydantic_v1.Field(default=None)
+    label_config: typing.Optional[str] = pydantic.Field(default=None)
     """
     Label config in XML format
     """
 
-    expert_instruction: typing.Optional[str] = pydantic_v1.Field(default=None)
+    expert_instruction: typing.Optional[str] = pydantic.Field(default=None)
     """
     Labeling instructions to show to the user
     """
 
-    show_instruction: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_instruction: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show labeling instructions
     """
 
-    show_skip_button: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_skip_button: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show skip button
     """
 
-    enable_empty_annotation: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    enable_empty_annotation: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Allow empty annotations
     """
 
-    show_annotation_history: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_annotation_history: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show annotation history
     """
 
-    reveal_preannotations_interactively: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    reveal_preannotations_interactively: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Reveal preannotations interactively. If set to True, predictions will be shown to the user only after selecting the area of interest
     """
 
-    show_collab_predictions: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    show_collab_predictions: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Show predictions to annotators
     """
 
-    maximum_annotations: typing.Optional[int] = pydantic_v1.Field(default=None)
+    maximum_annotations: typing.Optional[int] = pydantic.Field(default=None)
     """
     Maximum annotations per task
     """
 
-    color: typing.Optional[str] = pydantic_v1.Field(default=None)
+    color: typing.Optional[str] = pydantic.Field(default=None)
     """
     Project color in HEX format
     """
 
-    control_weights: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    control_weights: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Dict of weights for each control tag in metric calculation. Each control tag (e.g. label or choice) will have its own key in control weight dict with weight for each label and overall weight. For example, if a bounding box annotation with a control tag named my_bbox should be included with 0.33 weight in agreement calculation, and the first label Car should be twice as important as Airplane, then you need to specify: {'my_bbox': {'type': 'RectangleLabels', 'labels': {'Car': 1.0, 'Airplane': 0.5}, 'overall': 0.33}
     """
 
-    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