feat: UTC-288: Add bulk import predictions api to sdk (#595)

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

Author: robot-ci-heartex

.gitignore

@@ -3,4 +3,3 @@ dist/
 __pycache__/
 poetry.toml
 .ruff_cache/
-CLAUDE.local.md

.mock/definition/projects.yml

@@ -74,6 +74,14 @@ types:
         docs: Number of tasks added
     source:
       openapi: openapi/openapi.yaml
+  ProjectsImportPredictionsResponse:
+    docs: Import result
+    properties:
+      created:
+        type: optional<integer>
+        docs: Number of predictions created
+    source:
+      openapi: openapi/openapi.yaml
   ApiProjectsProjectExtraParamsRetrieveResponse:
     docs: Create or change annotator weights for statistics
     properties:
@@ -995,6 +1003,38 @@ service:
               task_count: 1
       audiences:
         - public
+    import_predictions:
+      path: /api/projects/{id}/import/predictions
+      method: POST
+      auth: true
+      docs: Import model predictions for tasks in the specified project.
+      source:
+        openapi: openapi/openapi.yaml
+      path-parameters:
+        id:
+          type: integer
+          docs: A unique integer value identifying this project.
+      display-name: Import predictions
+      request:
+        body: list<root.PredictionRequest>
+        content-type: application/json
+      response:
+        docs: Predictions successfully imported
+        type: ProjectsImportPredictionsResponse
+      errors:
+        - root.BadRequestError
+      examples:
+        - path-parameters:
+            id: 1
+          request:
+            - result:
+                - key: value
+              task: 1
+          response:
+            body:
+              created: 1
+      audiences:
+        - public
     api_projects_members_create:
       path: /api/projects/{id}/members/
       method: POST

.mock/openapi/openapi.yaml

@@ -7486,6 +7486,60 @@ paths:
         - public
       x-fern-sdk-group-name: projects
       x-fern-sdk-method-name: import_tasks
+  /api/projects/{id}/import/predictions:
+    post:
+      description: Import model predictions for tasks in the specified project.
+      operationId: api_projects_import_predictions_create
+      parameters:
+        - description: A unique integer value identifying this project.
+          in: path
+          name: id
+          required: true
+          schema:
+            type: integer
+      requestBody:
+        content:
+          application/json:
+            schema:
+              items:
+                $ref: '#/components/schemas/PredictionRequest'
+              type: array
+          application/x-www-form-urlencoded:
+            schema:
+              items:
+                $ref: '#/components/schemas/PredictionRequest'
+              type: array
+          multipart/form-data:
+            schema:
+              items:
+                $ref: '#/components/schemas/PredictionRequest'
+              type: array
+        required: true
+      responses:
+        '201':
+          content:
+            application/json:
+              schema:
+                description: Import result
+                properties:
+                  created:
+                    description: Number of predictions created
+                    title: created
+                    type: integer
+                title: Predictions import response
+                type: object
+          description: Predictions successfully imported
+        '400':
+          description: Bad Request
+      security:
+        - Token: []
+      summary: Import predictions
+      tags:
+        - Import
+      x-fern-audiences:
+        - public
+      x-fern-sdk-group-name: projects
+      x-fern-sdk-method-name: import_predictions
   /api/projects/{id}/imports/{import_pk}/:
     get:
       description: Return data related to async project import operation

reference.md

@@ -9865,6 +9865,90 @@ client.projects.import_tasks(
 </dl>
 
 
+</dd>
+</dl>
+</details>
+
+<details><summary><code>client.projects.<a href="src/label_studio_sdk/projects/client.py">import_predictions</a>(...)</code></summary>
+<dl>
+<dd>
+
+#### 📝 Description
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+Import model predictions for tasks in the specified project.
+</dd>
+</dl>
+</dd>
+</dl>
+
+#### 🔌 Usage
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+```python
+from label_studio_sdk import LabelStudio, PredictionRequest
+
+client = LabelStudio(
+    api_key="YOUR_API_KEY",
+)
+client.projects.import_predictions(
+    id=1,
+    request=[
+        PredictionRequest(
+            result=[{"key": "value"}],
+            task=1,
+        )
+    ],
+)
+
+```
+</dd>
+</dl>
+</dd>
+</dl>
+
+#### ⚙️ Parameters
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+**id:** `int` — A unique integer value identifying this project.
+    
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**request:** `typing.Sequence[PredictionRequest]` 
+    
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**request_options:** `typing.Optional[RequestOptions]` — Request-specific configuration.
+    
+</dd>
+</dl>
+</dd>
+</dl>
+
+
 </dd>
 </dl>
 </details>

src/label_studio_sdk/__init__.py

@@ -292,6 +292,7 @@
     PatchedLseProjectUpdateRequestSampling,
     PatchedLseProjectUpdateRequestSkipQueue,
     ProjectsDuplicateResponse,
+    ProjectsImportPredictionsResponse,
     ProjectsImportTasksResponse,
     ProjectsListRequestFilter,
 )
@@ -537,6 +538,7 @@
     "ProjectTemplate",
     "ProjectTemplateRequest",
     "ProjectsDuplicateResponse",
+    "ProjectsImportPredictionsResponse",
     "ProjectsImportTasksResponse",
     "ProjectsListRequestFilter",
     "PromptsCompatibleProjectsRequestProjectType",

src/label_studio_sdk/projects/__init__.py

@@ -6,6 +6,7 @@
     PatchedLseProjectUpdateRequestSampling,
     PatchedLseProjectUpdateRequestSkipQueue,
     ProjectsDuplicateResponse,
+    ProjectsImportPredictionsResponse,
     ProjectsImportTasksResponse,
     ProjectsListRequestFilter,
 )
@@ -58,6 +59,7 @@
     "PatchedLseProjectUpdateRequestSampling",
     "PatchedLseProjectUpdateRequestSkipQueue",
     "ProjectsDuplicateResponse",
+    "ProjectsImportPredictionsResponse",
     "ProjectsImportTasksResponse",
     "ProjectsListRequestFilter",
     "StatsIaaResponse",