feat: DIA-1748: 'Generate labels with AI' button in Project, in SC

.gitignore

@@ -2,4 +2,3 @@ dist/
 .mypy_cache/
 __pycache__/
 poetry.toml
-.idea/

.mock/definition/__package__.yml

@@ -556,6 +556,7 @@ types:
         type: optional<boolean>
         docs: Show annotation history to annotator
       organization: optional<integer>
+      prompt: optional<Prompt>
       color:
         type: optional<string>
         validation:

.mock/definition/projects.yml

@@ -227,6 +227,12 @@ service:
                   enable_empty_annotation: true
                   show_annotation_history: true
                   organization: 1
+                  prompt:
+                    title: title
+                    input_fields:
+                      - input_fields
+                    output_classes:
+                      - output_classes
                   color: color
                   maximum_annotations: 1
                   is_published: true
@@ -423,6 +429,20 @@ service:
               enable_empty_annotation: true
               show_annotation_history: true
               organization: 1
+              prompt:
+                title: title
+                description: description
+                created_by: 1
+                created_at: '2024-01-15T09:30:00Z'
+                updated_at: '2024-01-15T09:30:00Z'
+                organization: 1
+                input_fields:
+                  - input_fields
+                output_classes:
+                  - output_classes
+                associated_projects:
+                  - 1
+                skill_name: skill_name
               color: '#FF0000'
               maximum_annotations: 1
               is_published: true

.mock/definition/users.yml

@@ -306,6 +306,22 @@ service:
               date_joined: '2024-01-15T09:30:00Z'
       audiences:
         - public
+    get_product_tour:
+      path: /api/current-user/product-tour
+      method: GET
+      auth: true
+      examples:
+        - {}
+      audiences:
+        - internal
+    update_product_tour:
+      path: /api/current-user/product-tour
+      method: PATCH
+      auth: true
+      examples:
+        - {}
+      audiences:
+        - internal
   source:
     openapi: openapi/openapi.yaml
 imports:

.mock/openapi/openapi.yaml

@@ -8982,6 +8982,9 @@ components:
           title: Organization
           type: integer
           nullable: true
+        prompt:
+          $ref: "#/components/schemas/Prompt"
+          nullable: true
         color:
           title: Color
           type: string

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "label-studio-sdk"
-version = "1.0.9.dev"
+version = "1.0.9"
 description = ""
 readme = "README.md"
 authors = []
@@ -37,6 +37,7 @@ appdirs = ">=1.4.3"
 datamodel-code-generator = "0.26.1"
 httpx = ">=0.21.2"
 ijson = ">=3.2.3"
+jsf = "^0.11.2"
 jsonschema = ">=4.23.0"
 lxml = ">=4.2.5"
 nltk = "^3.9.1"
@@ -49,7 +50,6 @@ typing_extensions = ">= 4.0.0"
 ujson = ">=5.8.0"
 xmljson = "0.2.1"
 
-jsf = "^0.11.2"
 [tool.poetry.dev-dependencies]
 mypy = "1.0.1"
 pytest = "^7.4.0"

src/label_studio_sdk/types/project.py

@@ -7,6 +7,7 @@
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .project_sampling import ProjectSampling
 from .project_skip_queue import ProjectSkipQueue
+from .prompt import Prompt
 from .user_simple import UserSimple
 
 
@@ -53,6 +54,7 @@ class Project(pydantic_v1.BaseModel):
     """
 
     organization: typing.Optional[int] = None
+    prompt: typing.Optional[Prompt] = None
     color: typing.Optional[str] = None
     maximum_annotations: typing.Optional[int] = pydantic_v1.Field(default=None)
     """

tests/test_projects.py

@@ -59,6 +59,18 @@ async def test_get(client: LabelStudio, async_client: AsyncLabelStudio) -> None:
         "enable_empty_annotation": True,
         "show_annotation_history": True,
         "organization": 1,
+        "prompt": {
+            "title": "title",
+            "description": "description",
+            "created_by": 1,
+            "created_at": "2024-01-15T09:30:00Z",
+            "updated_at": "2024-01-15T09:30:00Z",
+            "organization": 1,
+            "input_fields": ["input_fields"],
+            "output_classes": ["output_classes"],
+            "associated_projects": [1],
+            "skill_name": "skill_name",
+        },
         "color": "#FF0000",
         "maximum_annotations": 1,
         "is_published": True,
@@ -110,6 +122,18 @@ async def test_get(client: LabelStudio, async_client: AsyncLabelStudio) -> None:
         "enable_empty_annotation": None,
         "show_annotation_history": None,
         "organization": "integer",
+        "prompt": {
+            "title": None,
+            "description": None,
+            "created_by": "integer",
+            "created_at": "datetime",
+            "updated_at": "datetime",
+            "organization": "integer",
+            "input_fields": ("list", {0: None}),
+            "output_classes": ("list", {0: None}),
+            "associated_projects": ("list", {0: "integer"}),
+            "skill_name": None,
+        },
         "color": None,
         "maximum_annotations": "integer",
         "is_published": None,