fix: BROS-32: Task retrieval SDK bugs (#471)

Co-authored-by: fern-api <115122769+fern-api[bot]@users.noreply.github.com>

Author: robot-ci-heartex

.mock/definition/__package__.yml

@@ -1763,6 +1763,14 @@ types:
         docs: User who created the last annotation history item
     source:
       openapi: openapi/openapi.yaml
+  DataManagerTaskSerializerPredictionsItemModelRun:
+    discriminated: false
+    union:
+      - map<string, unknown>
+      - integer
+    source:
+      openapi: openapi/openapi.yaml
+    inline: true
   DataManagerTaskSerializerPredictionsItem:
     properties:
       result:
@@ -1774,7 +1782,7 @@ types:
       model:
         type: optional<map<string, unknown>>
       model_run:
-        type: optional<map<string, unknown>>
+        type: optional<DataManagerTaskSerializerPredictionsItemModelRun>
       task:
         type: optional<integer>
       project:
@@ -1805,6 +1813,15 @@ types:
     source:
       openapi: openapi/openapi.yaml
     inline: true
+  DataManagerTaskSerializerCommentAuthorsItem:
+    discriminated: false
+    docs: Users who wrote comments
+    union:
+      - integer
+      - map<string, unknown>
+    source:
+      openapi: openapi/openapi.yaml
+    inline: true
   DataManagerTaskSerializer:
     properties:
       id: optional<integer>
@@ -1874,7 +1891,7 @@ types:
         type: optional<integer>
         docs: Project ID for this task
       comment_authors:
-        type: optional<list<integer>>
+        type: optional<list<DataManagerTaskSerializerCommentAuthorsItem>>
         docs: Users who wrote comments
     source:
       openapi: openapi/openapi.yaml

.mock/openapi/openapi.yaml

@@ -10283,7 +10283,9 @@ components:
                 type: object
                 title: ML Backend instance
               model_run:
-                type: object
+                oneOf:
+                  - type: object
+                  - type: integer
                 title: Model Run instance
               task:
                 type: integer
@@ -10452,7 +10454,9 @@ components:
           type: array
           items:
             description: Users who wrote comments
-            type: integer
+            oneOf:
+              - type: integer
+              - type: object
           uniqueItems: true
     Webhook:
       required:

poetry.lock

@@ -3,7 +3,7 @@ name = "label-studio-sdk"
 
 [tool.poetry]
 name = "label-studio-sdk"
-version = "1.0.17.dev"
+version = "1.0.17"
 description = ""
 readme = "README.md"
 authors = []

pyproject.toml

@@ -23,8 +23,10 @@
     ConvertedFormatStatus,
     DataManagerTaskSerializer,
     DataManagerTaskSerializerAnnotatorsItem,
+    DataManagerTaskSerializerCommentAuthorsItem,
     DataManagerTaskSerializerDraftsItem,
     DataManagerTaskSerializerPredictionsItem,
+    DataManagerTaskSerializerPredictionsItemModelRun,
     Export,
     ExportFormat,
     ExportSnapshot,
@@ -227,8 +229,10 @@
     "ConvertedFormatStatus",
     "DataManagerTaskSerializer",
     "DataManagerTaskSerializerAnnotatorsItem",
+    "DataManagerTaskSerializerCommentAuthorsItem",
     "DataManagerTaskSerializerDraftsItem",
     "DataManagerTaskSerializerPredictionsItem",
+    "DataManagerTaskSerializerPredictionsItemModelRun",
     "Export",
     "ExportFormat",
     "ExportSnapshot",

src/label_studio_sdk/__init__.py

@@ -22,8 +22,10 @@
 from .converted_format_status import ConvertedFormatStatus
 from .data_manager_task_serializer import DataManagerTaskSerializer
 from .data_manager_task_serializer_annotators_item import DataManagerTaskSerializerAnnotatorsItem
+from .data_manager_task_serializer_comment_authors_item import DataManagerTaskSerializerCommentAuthorsItem
 from .data_manager_task_serializer_drafts_item import DataManagerTaskSerializerDraftsItem
 from .data_manager_task_serializer_predictions_item import DataManagerTaskSerializerPredictionsItem
+from .data_manager_task_serializer_predictions_item_model_run import DataManagerTaskSerializerPredictionsItemModelRun
 from .export import Export
 from .export_format import ExportFormat
 from .export_snapshot import ExportSnapshot
@@ -130,8 +132,10 @@
     "ConvertedFormatStatus",
     "DataManagerTaskSerializer",
     "DataManagerTaskSerializerAnnotatorsItem",
+    "DataManagerTaskSerializerCommentAuthorsItem",
     "DataManagerTaskSerializerDraftsItem",
     "DataManagerTaskSerializerPredictionsItem",
+    "DataManagerTaskSerializerPredictionsItemModelRun",
     "Export",
     "ExportFormat",
     "ExportSnapshot",

src/label_studio_sdk/types/__init__.py

@@ -8,6 +8,7 @@
 from .data_manager_task_serializer_drafts_item import DataManagerTaskSerializerDraftsItem
 from .data_manager_task_serializer_annotators_item import DataManagerTaskSerializerAnnotatorsItem
 import datetime as dt
+from .data_manager_task_serializer_comment_authors_item import DataManagerTaskSerializerCommentAuthorsItem
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
@@ -100,7 +101,9 @@ class DataManagerTaskSerializer(UniversalBaseModel):
     Project ID for this task
     """
 
-    comment_authors: typing.Optional[typing.List[int]] = pydantic.Field(default=None)
+    comment_authors: typing.Optional[typing.List[DataManagerTaskSerializerCommentAuthorsItem]] = pydantic.Field(
+        default=None
+    )
     """
     Users who wrote comments
     """

src/label_studio_sdk/types/data_manager_task_serializer.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+DataManagerTaskSerializerCommentAuthorsItem = typing.Union[int, typing.Dict[str, typing.Optional[typing.Any]]]

src/label_studio_sdk/types/data_manager_task_serializer_comment_authors_item.py

@@ -2,6 +2,7 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+from .data_manager_task_serializer_predictions_item_model_run import DataManagerTaskSerializerPredictionsItemModelRun
 import datetime as dt
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
@@ -12,7 +13,7 @@ class DataManagerTaskSerializerPredictionsItem(UniversalBaseModel):
     score: typing.Optional[float] = None
     model_version: typing.Optional[str] = None
     model: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
-    model_run: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    model_run: typing.Optional[DataManagerTaskSerializerPredictionsItemModelRun] = None
     task: typing.Optional[int] = None
     project: typing.Optional[float] = None
     created_at: typing.Optional[dt.datetime] = None

src/label_studio_sdk/types/data_manager_task_serializer_predictions_item.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+DataManagerTaskSerializerPredictionsItemModelRun = typing.Union[typing.Dict[str, typing.Optional[typing.Any]], int]