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

label_studio_sdk/__init__.py

@@ -1,11 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AccessTokenResponse,
     Annotation,
     AnnotationFilterOptions,
     AnnotationLastAction,
     AnnotationsDmField,
     AnnotationsDmFieldLastAction,
+    ApiTokenResponse,
     AzureBlobExportStorage,
     AzureBlobExportStorageStatus,
     AzureBlobImportStorage,
@@ -23,9 +25,9 @@ from .types import (
     DataManagerTaskSerializerDraftsItem,
     DataManagerTaskSerializerPredictionsItem,
     Export,
-    ExportConvert,
-    ExportCreate,
-    ExportCreateStatus,
+    ExportFormat,
+    ExportSnapshot,
+    ExportSnapshotStatus,
     ExportStatus,
     FileUpload,
     Filter,
@@ -40,6 +42,7 @@ from .types import (
     InferenceRunOrganization,
     InferenceRunProjectSubset,
     InferenceRunStatus,
+    JwtSettingsResponse,
     KeyIndicatorValue,
     KeyIndicators,
     KeyIndicatorsItem,
@@ -58,6 +61,8 @@ from .types import (
     ModelProviderConnectionOrganization,
     ModelProviderConnectionProvider,
     ModelProviderConnectionScope,
+    Pause,
+    PausePausedBy,
     Prediction,
     Project,
     ProjectImport,
@@ -66,6 +71,8 @@ from .types import (
     ProjectSampling,
     ProjectSkipQueue,
     Prompt,
+    PromptAssociatedProjectsItem,
+    PromptAssociatedProjectsItemId,
     PromptCreatedBy,
     PromptOrganization,
     PromptVersion,
@@ -99,7 +106,7 @@ from .types import (
     WebhookSerializerForUpdateActionsItem,
     Workspace,
 )
-from .errors import BadRequestError, InternalServerError
+from .errors import BadRequestError, InternalServerError, NotFoundError, UnauthorizedError
 from . import (
     actions,
     annotations,
@@ -107,13 +114,16 @@ from . import (
     export_storage,
     files,
     import_storage,
+    jwt_settings,
     ml,
     model_providers,
     predictions,
     projects,
     prompts,
     tasks,
+    tokens,
     users,
+    versions,
     views,
     webhooks,
     workspaces,
@@ -132,7 +142,8 @@ from .actions import (
     ActionsCreateRequestSelectedItemsExcluded,
     ActionsCreateRequestSelectedItemsIncluded,
 )
-from .annotations import AnnotationsCreateBulkResponseItem
+from .annotations import AnnotationsCreateBulkRequestSelectedItems, AnnotationsCreateBulkResponseItem
+from .client import AsyncLabelStudio, LabelStudio
 from .environment import LabelStudioEnvironment
 from .export_storage import ExportStorageListTypesResponseItem
 from .import_storage import ImportStorageListTypesResponseItem
@@ -154,6 +165,7 @@ from .prompts import (
 from .tasks import TasksListRequestFields, TasksListResponse
 from .users import UsersGetTokenResponse, UsersResetTokenResponse
 from .version import __version__
+from .versions import VersionsGetResponse, VersionsGetResponseEdition
 from .views import (
     ViewsCreateRequestData,
     ViewsCreateRequestDataFilters,
@@ -175,6 +187,7 @@ from .views import (
 from .webhooks import WebhooksUpdateRequestActionsItem
 
 __all__ = [
+    "AccessTokenResponse",
     "ActionsCreateRequestFilters",
     "ActionsCreateRequestFiltersConjunction",
     "ActionsCreateRequestFiltersItemsItem",
@@ -189,9 +202,12 @@ __all__ = [
     "Annotation",
     "AnnotationFilterOptions",
     "AnnotationLastAction",
+    "AnnotationsCreateBulkRequestSelectedItems",
     "AnnotationsCreateBulkResponseItem",
     "AnnotationsDmField",
     "AnnotationsDmFieldLastAction",
+    "ApiTokenResponse",
+    "AsyncLabelStudio",
     "AzureBlobExportStorage",
     "AzureBlobExportStorageStatus",
     "AzureBlobImportStorage",
@@ -211,9 +227,9 @@ __all__ = [
     "DataManagerTaskSerializerDraftsItem",
     "DataManagerTaskSerializerPredictionsItem",
     "Export",
-    "ExportConvert",
-    "ExportCreate",
-    "ExportCreateStatus",
+    "ExportFormat",
+    "ExportSnapshot",
+    "ExportSnapshotStatus",
     "ExportStatus",
     "ExportStorageListTypesResponseItem",
     "FileUpload",
@@ -231,11 +247,13 @@ __all__ = [
     "InferenceRunProjectSubset",
     "InferenceRunStatus",
     "InternalServerError",
+    "JwtSettingsResponse",
     "KeyIndicatorValue",
     "KeyIndicators",
     "KeyIndicatorsItem",
     "KeyIndicatorsItemAdditionalKpisItem",
     "KeyIndicatorsItemExtraKpisItem",
+    "LabelStudio",
     "LabelStudioEnvironment",
     "LocalFilesExportStorage",
     "LocalFilesExportStorageStatus",
@@ -256,6 +274,9 @@ __all__ = [
     "ModelProviderConnectionOrganization",
     "ModelProviderConnectionProvider",
     "ModelProviderConnectionScope",
+    "NotFoundError",
+    "Pause",
+    "PausePausedBy",
     "Prediction",
     "Project",
     "ProjectImport",
@@ -268,6 +289,8 @@ __all__ = [
     "ProjectsListResponse",
     "ProjectsUpdateResponse",
     "Prompt",
+    "PromptAssociatedProjectsItem",
+    "PromptAssociatedProjectsItemId",
     "PromptCreatedBy",
     "PromptOrganization",
     "PromptVersion",
@@ -299,9 +322,12 @@ __all__ = [
     "TaskFilterOptions",
     "TasksListRequestFields",
     "TasksListResponse",
+    "UnauthorizedError",
     "UserSimple",
     "UsersGetTokenResponse",
     "UsersResetTokenResponse",
+    "VersionsGetResponse",
+    "VersionsGetResponseEdition",
     "View",
     "ViewsCreateRequestData",
     "ViewsCreateRequestDataFilters",
@@ -332,13 +358,16 @@ __all__ = [
     "export_storage",
     "files",
     "import_storage",
+    "jwt_settings",
     "ml",
     "model_providers",
     "predictions",
     "projects",
     "prompts",
     "tasks",
+    "tokens",
     "users",
+    "versions",
     "views",
     "webhooks",
     "workspaces",

label_studio_sdk/_extensions/label_studio_tools/core/utils/io.py

@@ -37,6 +37,17 @@ def get_cache_dir():
     return cache_dir
 
 
+def safe_build_path(base_dir: str, user_path: str) -> str:
+    combined_path = os.path.join(base_dir, user_path)
+    absolute_path = os.path.abspath(combined_path)
+    base_dir_abs = os.path.abspath(base_dir)
+
+    if os.path.commonpath([absolute_path, base_dir_abs]) != base_dir_abs:
+        raise ValueError(f"Invalid path: {user_path}")
+
+    return absolute_path
+
+
 def get_local_path(
     url,
     cache_dir=None,
@@ -103,7 +114,7 @@ def get_local_path(
     # instead of downloading them from LS instance
     if is_local_storage_file:
         filepath = url.split("?d=")[1]
-        filepath = os.path.join(LOCAL_FILES_DOCUMENT_ROOT, filepath)
+        filepath = safe_build_path(LOCAL_FILES_DOCUMENT_ROOT, filepath)
         if os.path.exists(filepath):
             logger.debug(
                 f"Local Storage file path exists locally, use it as a local file: {filepath}"
@@ -124,9 +135,9 @@ def get_local_path(
     if is_uploaded_file and os.path.exists(image_dir):
         project_id = url.split("/")[-2]  # To retrieve project_id
         filepath = os.path.join(image_dir, project_id, os.path.basename(url))
-        if cache_dir and download_resources:
-            shutil.copy(filepath, cache_dir)
         if os.path.exists(filepath):
+            if cache_dir and download_resources:
+                shutil.copy(filepath, cache_dir)
             logger.debug(f"Uploaded file: Path exists in image_dir: {filepath}")
             return filepath
 
@@ -202,7 +213,7 @@ def download_and_cache(
     filepath = os.path.join(cache_dir, url_hash + "__" + url_filename)
 
     if not os.path.exists(filepath):
-        logger.info("Download {url} to {filepath}".format(url=url, filepath=filepath))
+        logger.info("Download {url} to {filepath}. download_resources: {download_resources}".format(url=url, filepath=filepath, download_resources=download_resources))
         if download_resources:
             headers = {
                 # avoid requests.exceptions.HTTPError: 403 Client Error: Forbidden. Please comply with the User-Agent policy:
@@ -227,6 +238,7 @@ def download_and_cache(
                 raise e
             with io.open(filepath, mode="wb") as fout:
                 fout.write(r.content)
+                logger.info(f"File downloaded to {filepath}")
     return filepath
 
 

label_studio_sdk/_extensions/label_studio_tools/core/utils/json_schema.py

@@ -2,6 +2,7 @@ import json
 import types
 import sys
 import functools
+import logging
 from typing import Type, Dict, Any, Tuple, Generator
 from pathlib import Path
 from tempfile import TemporaryDirectory
@@ -11,6 +12,8 @@ from datamodel_code_generator.parser.jsonschema import JsonSchemaParser
 from pydantic import BaseModel
 from contextlib import contextmanager
 
+logger = logging.getLogger(__name__)
+
 
 @functools.lru_cache(maxsize=128)
 def _generate_model_code(json_schema_str: str, class_name: str = 'MyModel') -> str:
@@ -65,6 +68,7 @@ def json_schema_to_pydantic(json_schema: dict, class_name: str = 'MyModel') -> G
     json_schema_str = json.dumps(json_schema)
     
     # Generate Pydantic model code from the JSON schema string
+    logger.debug(f"Generating Pydantic model code from json schema: {json_schema_str}")
     model_code: str = _generate_model_code(json_schema_str, class_name)
     
     # Create a unique module name using the id of the JSON schema string
@@ -79,6 +83,7 @@ def json_schema_to_pydantic(json_schema: dict, class_name: str = 'MyModel') -> G
         # Add the new module to sys.modules to make it importable
         # This is necessary to avoid Pydantic errors related to undefined models
         sys.modules[module_name] = mod
+        logger.debug(f"Generated Pydantic model: {model_class}")
         yield model_class
     finally:
         if module_name in sys.modules:

label_studio_sdk/_extensions/pager_ext.py

@@ -47,3 +47,11 @@ class AsyncPagerExt(AsyncPager, typing.Generic[T]):
             if exc.status_code == 404:
                 return
             raise
+        
+    async def __anext__(self) -> T:
+        try:
+            return await super().__anext__()
+        except ApiError as exc:
+            if exc.status_code == 404:
+                raise StopAsyncIteration
+            raise

label_studio_sdk/actions/client.py

@@ -1,15 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+from ..core.client_wrapper import SyncClientWrapper
+from ..core.request_options import RequestOptions
 from json.decoder import JSONDecodeError
-
 from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.request_options import RequestOptions
-from .types.actions_create_request_filters import ActionsCreateRequestFilters
 from .types.actions_create_request_id import ActionsCreateRequestId
-from .types.actions_create_request_ordering_item import ActionsCreateRequestOrderingItem
+from .types.actions_create_request_filters import ActionsCreateRequestFilters
 from .types.actions_create_request_selected_items import ActionsCreateRequestSelectedItems
+from .types.actions_create_request_ordering_item import ActionsCreateRequestOrderingItem
+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, ...)
@@ -34,7 +35,7 @@ class ActionsClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -42,7 +43,9 @@ class ActionsClient:
         client.actions.list()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/dm/actions/", method="GET", request_options=request_options
+            "api/dm/actions/",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -61,7 +64,7 @@ class ActionsClient:
         filters: typing.Optional[ActionsCreateRequestFilters] = OMIT,
         selected_items: typing.Optional[ActionsCreateRequestSelectedItems] = OMIT,
         ordering: typing.Optional[typing.Sequence[ActionsCreateRequestOrderingItem]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
         Perform a Data Manager action with the selected tasks and filters. Note: More complex actions require additional parameters in the request body. Call `GET api/actions?project=<id>` to explore them. <br>Example: `GET api/actions?id=delete_tasks&project=1`
@@ -95,12 +98,12 @@ class ActionsClient:
 
         Examples
         --------
-        from label_studio_sdk import (
+        from label_studio_sdk import LabelStudio
+        from label_studio_sdk.actions import (
             ActionsCreateRequestFilters,
             ActionsCreateRequestFiltersItemsItem,
             ActionsCreateRequestSelectedItemsExcluded,
         )
-        from label_studio_sdk.client import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -129,8 +132,23 @@ class ActionsClient:
         _response = self._client_wrapper.httpx_client.request(
             "api/dm/actions/",
             method="POST",
-            params={"id": id, "project": project, "view": view},
-            json={"filters": filters, "selectedItems": selected_items, "ordering": ordering},
+            params={
+                "id": id,
+                "project": project,
+                "view": view,
+            },
+            json={
+                "filters": convert_and_respect_annotation_metadata(
+                    object_=filters, annotation=ActionsCreateRequestFilters, direction="write"
+                ),
+                "selectedItems": convert_and_respect_annotation_metadata(
+                    object_=selected_items, annotation=ActionsCreateRequestSelectedItems, direction="write"
+                ),
+                "ordering": ordering,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -162,15 +180,25 @@ class AsyncActionsClient:
 
         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.actions.list()
+
+
+        async def main() -> None:
+            await client.actions.list()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/dm/actions/", method="GET", request_options=request_options
+            "api/dm/actions/",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -189,7 +217,7 @@ class AsyncActionsClient:
         filters: typing.Optional[ActionsCreateRequestFilters] = OMIT,
         selected_items: typing.Optional[ActionsCreateRequestSelectedItems] = OMIT,
         ordering: typing.Optional[typing.Sequence[ActionsCreateRequestOrderingItem]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
         Perform a Data Manager action with the selected tasks and filters. Note: More complex actions require additional parameters in the request body. Call `GET api/actions?project=<id>` to explore them. <br>Example: `GET api/actions?id=delete_tasks&project=1`
@@ -223,42 +251,65 @@ class AsyncActionsClient:
 
         Examples
         --------
-        from label_studio_sdk import (
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+        from label_studio_sdk.actions import (
             ActionsCreateRequestFilters,
             ActionsCreateRequestFiltersItemsItem,
             ActionsCreateRequestSelectedItemsExcluded,
         )
-        from label_studio_sdk.client import AsyncLabelStudio
 
         client = AsyncLabelStudio(
             api_key="YOUR_API_KEY",
         )
-        await client.actions.create(
-            id="retrieve_tasks_predictions",
-            project=1,
-            filters=ActionsCreateRequestFilters(
-                conjunction="or",
-                items=[
-                    ActionsCreateRequestFiltersItemsItem(
-                        filter="filter:tasks:id",
-                        operator="greater",
-                        type="Number",
-                        value=123,
-                    )
-                ],
-            ),
-            selected_items=ActionsCreateRequestSelectedItemsExcluded(
-                all_=True,
-                excluded=[124, 125, 126],
-            ),
-            ordering=["tasks:total_annotations"],
-        )
+
+
+        async def main() -> None:
+            await client.actions.create(
+                id="retrieve_tasks_predictions",
+                project=1,
+                filters=ActionsCreateRequestFilters(
+                    conjunction="or",
+                    items=[
+                        ActionsCreateRequestFiltersItemsItem(
+                            filter="filter:tasks:id",
+                            operator="greater",
+                            type="Number",
+                            value=123,
+                        )
+                    ],
+                ),
+                selected_items=ActionsCreateRequestSelectedItemsExcluded(
+                    all_=True,
+                    excluded=[124, 125, 126],
+                ),
+                ordering=["tasks:total_annotations"],
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/dm/actions/",
             method="POST",
-            params={"id": id, "project": project, "view": view},
-            json={"filters": filters, "selectedItems": selected_items, "ordering": ordering},
+            params={
+                "id": id,
+                "project": project,
+                "view": view,
+            },
+            json={
+                "filters": convert_and_respect_annotation_metadata(
+                    object_=filters, annotation=ActionsCreateRequestFilters, direction="write"
+                ),
+                "selectedItems": convert_and_respect_annotation_metadata(
+                    object_=selected_items, annotation=ActionsCreateRequestSelectedItems, direction="write"
+                ),
+                "ordering": ordering,
+            },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )

label_studio_sdk/actions/types/actions_create_request_filters.py

@@ -1,43 +1,33 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ...core.datetime_utils import serialize_datetime
-from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from ...core.pydantic_utilities import UniversalBaseModel
 from .actions_create_request_filters_conjunction import ActionsCreateRequestFiltersConjunction
+import pydantic
+import typing
 from .actions_create_request_filters_items_item import ActionsCreateRequestFiltersItemsItem
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class ActionsCreateRequestFilters(pydantic_v1.BaseModel):
+class ActionsCreateRequestFilters(UniversalBaseModel):
     """
     Filters to apply on tasks. You can use [the helper class `Filters` from this page](https://labelstud.io/sdk/data_manager.html) to create Data Manager Filters.<br>Example: `{"conjunction": "or", "items": [{"filter": "filter:tasks:completed_at", "operator": "greater", "type": "Datetime", "value": "2021-01-01T00:00:00.000Z"}]}`
     """
 
-    conjunction: ActionsCreateRequestFiltersConjunction = pydantic_v1.Field()
+    conjunction: ActionsCreateRequestFiltersConjunction = pydantic.Field()
     """
     Logical conjunction for the filters. This conjunction (either "or" or "and") will be applied to all items in the filters list. It is not possible to combine "or" and "and" within one list of filters. All filters will be either combined with "or" or with "and", but not a mix of both.
     """
 
-    items: typing.List[ActionsCreateRequestFiltersItemsItem] = pydantic_v1.Field()
+    items: typing.List[ActionsCreateRequestFiltersItemsItem] = pydantic.Field()
     """
     List of filter items
     """
 
-    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/actions/types/actions_create_request_filters_items_item.py

@@ -1,50 +1,40 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ...core.datetime_utils import serialize_datetime
-from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from ...core.pydantic_utilities import UniversalBaseModel
 from .actions_create_request_filters_items_item_filter import ActionsCreateRequestFiltersItemsItemFilter
+import pydantic
 from .actions_create_request_filters_items_item_operator import ActionsCreateRequestFiltersItemsItemOperator
 from .actions_create_request_filters_items_item_value import ActionsCreateRequestFiltersItemsItemValue
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
 
 
-class ActionsCreateRequestFiltersItemsItem(pydantic_v1.BaseModel):
-    filter: ActionsCreateRequestFiltersItemsItemFilter = pydantic_v1.Field()
+class ActionsCreateRequestFiltersItemsItem(UniversalBaseModel):
+    filter: ActionsCreateRequestFiltersItemsItemFilter = pydantic.Field()
     """
     Filter identifier, it should start with `filter:tasks:` prefix, e.g. `filter:tasks:agreement`. For `task.data` fields it may look like `filter:tasks:data.field_name`. If you need more info about columns, check the [Get data manager columns](#tag/Data-Manager/operation/api_dm_columns_list) API endpoint. Possible values:<br><li>`filter:tasks:agreement`<br> (Number) Agreement for annotation results for a specific task (Enterprise only)</li><br><li>`filter:tasks:annotations_results`<br> (String) Annotation results for the tasks</li><br><li>`filter:tasks:annotators`<br> (List) Annotators that completed the task (Community). Can include assigned annotators (Enterprise only). Important note: the filter `type` should be List, but the filter `value` is integer</li><br><li>`filter:tasks:cancelled_annotations`<br> (Number) Number of cancelled or skipped annotations for the task</li><br><li>`filter:tasks:comments`<br> (Number) Number of comments in a task</li><br><li>`filter:tasks:completed_at`<br> (Datetime) Time when a task was fully annotated</li><br><li>`filter:tasks:created_at`<br> (Datetime) Time the task was created at</li><br><li>`filter:tasks:file_upload`<br> (String) Name of the file uploaded to create the tasks</li><br><li>`filter:tasks:ground_truth`<br> (Boolean) Ground truth status of the tasks</li><br><li>`filter:tasks:id`<br> (Number) Task ID</li><br><li>`filter:tasks:inner_id`<br> (Number) Task Inner ID, it starts from 1 for all projects</li><br><li>`filter:tasks:predictions_model_versions`<br> (String) Model version used for the predictions</li><br><li>`filter:tasks:predictions_results`<br> (String) Prediction results for the tasks</li><br><li>`filter:tasks:predictions_score`<br> (Number) Prediction score for the task</li><br><li>`filter:tasks:reviewed`<br> (Boolean) Whether the tasks have been reviewed (Enterprise only)</li><br><li>`filter:tasks:reviewers`<br> (String) Reviewers that reviewed the task, or assigned reviewers (Enterprise only). Important note: the filter `type` should be List, but the filter `value` is integer</li><br><li>`filter:tasks:reviews_accepted`<br> (Number) Number of annotations accepted for a task in review (Enterprise only)</li><br><li>`filter:tasks:reviews_rejected`<br> (Number) Number of annotations rejected for a task in review (Enterprise only)</li><br><li>`filter:tasks:total_annotations`<br> (Number) Total number of annotations on a task</li><br><li>`filter:tasks:total_predictions`<br> (Number) Total number of predictions for the task</li><br><li>`filter:tasks:unresolved_comment_count`<br> (Number) Number of unresolved comments in a task</li><br><li>`filter:tasks:updated_at`<br> (Datetime) Time the task was updated at (e.g. new annotation was created, review added, etc)</li>
     """
 
-    operator: ActionsCreateRequestFiltersItemsItemOperator = pydantic_v1.Field()
+    operator: ActionsCreateRequestFiltersItemsItemOperator = pydantic.Field()
     """
     Filter operator. Possible values:<br><li>`contains`<br> Contains</li><br><li>`ends_with`<br> Ends with</li><br><li>`equal`<br> Equal to</li><br><li>`exists`<br> Exists</li><br><li>`greater`<br> Greater than</li><br><li>`greater_or_equal`<br> Greater than or equal to</li><br><li>`in`<br> Is between min and max values, so the filter `value` should be e.g. `{"min": 1, "max": 7}`</li><br><li>`less`<br> Less than</li><br><li>`less_or_equal`<br> Less than or equal to</li><br><li>`not_contains`<br> Does not contain</li><br><li>`not_equal`<br> Not equal to</li><br><li>`not_exists`<br> Does not exist</li><br><li>`not_in`<br> Is not between min and max values, so the filter `value` should be e.g. `{"min": 1, "max": 7}`</li><br><li>`starts_with`<br> Starts with</li>
     """
 
-    type: str = pydantic_v1.Field()
+    type: str = pydantic.Field()
     """
     Type of the filter value. Possible values:<br><li>`Boolean`<br> Boolean</li><br><li>`Datetime`<br> Datetime string in `strftime('%Y-%m-%dT%H:%M:%S.%fZ')` format</li><br><li>`List`<br> List of items</li><br><li>`Number`<br> Float or Integer</li><br><li>`String`<br> String</li><br><li>`Unknown`<br> Unknown is explicitly converted to string format</li>
     """
 
-    value: ActionsCreateRequestFiltersItemsItemValue = pydantic_v1.Field()
+    value: ActionsCreateRequestFiltersItemsItemValue = pydantic.Field()
     """
     Value to filter by
     """
 
-    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/actions/types/actions_create_request_filters_items_item_value.py

@@ -2,4 +2,6 @@
 
 import typing
 
-ActionsCreateRequestFiltersItemsItemValue = typing.Union[str, int, float, bool, typing.Dict[str, typing.Any]]
+ActionsCreateRequestFiltersItemsItemValue = typing.Union[
+    str, int, float, bool, typing.Dict[str, typing.Optional[typing.Any]]
+]

label_studio_sdk/actions/types/actions_create_request_selected_items.py

@@ -1,9 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-
-from .actions_create_request_selected_items_excluded import ActionsCreateRequestSelectedItemsExcluded
 from .actions_create_request_selected_items_included import ActionsCreateRequestSelectedItemsIncluded
+from .actions_create_request_selected_items_excluded import ActionsCreateRequestSelectedItemsExcluded
 
 ActionsCreateRequestSelectedItems = typing.Union[
     ActionsCreateRequestSelectedItemsIncluded, ActionsCreateRequestSelectedItemsExcluded