label-studio-sdk 2.0.6__tar.gz → 2.0.8__tar.gz

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: label-studio-sdk
-Version: 2.0.6
+Version: 2.0.8
 Summary: 
 Requires-Python: >=3.9,<4
 Classifier: Intended Audience :: Developers

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/pyproject.toml

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

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/__init__.py

@@ -42,11 +42,14 @@ from .types import (
     Comment,
     CommentRequest,
     CommentSerializerWithExpandedUser,
+    ConfigurablePermissionOption,
+    ConfigurablePermissionOptionDefault,
     ConvertedFormat,
     ConvertedFormatRequest,
     CountLimit,
     CustomScriptsEditableByEnum,
-    DefaultRoleEnum,
+    DefaultRole,
+    DefaultRoleCustomScriptsEditableBy,
     EditionEnum,
     Export,
     FileUpload,
@@ -78,9 +81,13 @@ from .types import (
     LseOrganization,
     LseOrganizationCustomScriptsEditableBy,
     LseOrganizationMemberList,
+    LseProject,
+    LseProjectCounts,
     LseProjectCreate,
     LseProjectCreateSampling,
     LseProjectCreateSkipQueue,
+    LseProjectSampling,
+    LseProjectSkipQueue,
     LseProjectUpdate,
     LseProjectUpdateSampling,
     LseProjectUpdateSkipQueue,
@@ -132,9 +139,12 @@ from .types import (
     OrganizationInvite,
     OrganizationMember,
     OrganizationMembership,
+    OrganizationPermission,
+    OrganizationPermissionRequest,
     PaginatedAllRolesProjectListList,
     PaginatedAnnotationHistoryList,
     PaginatedLseOrganizationMemberListList,
+    PaginatedLseProjectCountsList,
     PaginatedLseUserList,
     PaginatedPaginatedProjectMemberList,
     PaginatedProjectMember,
@@ -154,6 +164,8 @@ from .types import (
     ProjectSampling,
     ProjectSkipQueue,
     ProjectSubsetEnum,
+    ProjectTemplate,
+    ProjectTemplateRequest,
     PromptsStatusEnum,
     ProviderEnum,
     ReasonEnum,
@@ -238,6 +250,7 @@ from . import (
     model_providers,
     organizations,
     predictions,
+    project_templates,
     projects,
     prompts,
     session_policy,
@@ -275,12 +288,14 @@ from .export_storage import ExportStorageListTypesResponseItem
 from .import_storage import ImportStorageListTypesResponseItem
 from .ml import MlCreateRequestAuthMethod, MlListModelVersionsResponse, MlUpdateRequestAuthMethod
 from .model_providers import ModelProvidersListModelProviderChoicesResponse
+from .organizations import PatchedDefaultRoleRequestCustomScriptsEditableBy
 from .projects import (
     LseProjectCreateRequestSampling,
     LseProjectCreateRequestSkipQueue,
     PatchedLseProjectUpdateRequestSampling,
     PatchedLseProjectUpdateRequestSkipQueue,
     ProjectsDuplicateResponse,
+    ProjectsImportPredictionsResponse,
     ProjectsImportTasksResponse,
     ProjectsListRequestFilter,
 )
@@ -387,11 +402,14 @@ __all__ = [
     "Comment",
     "CommentRequest",
     "CommentSerializerWithExpandedUser",
+    "ConfigurablePermissionOption",
+    "ConfigurablePermissionOptionDefault",
     "ConvertedFormat",
     "ConvertedFormatRequest",
     "CountLimit",
     "CustomScriptsEditableByEnum",
-    "DefaultRoleEnum",
+    "DefaultRole",
+    "DefaultRoleCustomScriptsEditableBy",
     "EditionEnum",
     "Export",
     "ExportStorageListTypesResponseItem",
@@ -429,11 +447,15 @@ __all__ = [
     "LseOrganization",
     "LseOrganizationCustomScriptsEditableBy",
     "LseOrganizationMemberList",
+    "LseProject",
+    "LseProjectCounts",
     "LseProjectCreate",
     "LseProjectCreateRequestSampling",
     "LseProjectCreateRequestSkipQueue",
     "LseProjectCreateSampling",
     "LseProjectCreateSkipQueue",
+    "LseProjectSampling",
+    "LseProjectSkipQueue",
     "LseProjectUpdate",
     "LseProjectUpdateSampling",
     "LseProjectUpdateSkipQueue",
@@ -491,13 +513,17 @@ __all__ = [
     "OrganizationInvite",
     "OrganizationMember",
     "OrganizationMembership",
+    "OrganizationPermission",
+    "OrganizationPermissionRequest",
     "PaginatedAllRolesProjectListList",
     "PaginatedAnnotationHistoryList",
     "PaginatedLseOrganizationMemberListList",
+    "PaginatedLseProjectCountsList",
     "PaginatedLseUserList",
     "PaginatedPaginatedProjectMemberList",
     "PaginatedProjectMember",
     "PaginatedRoleBasedTaskList",
+    "PatchedDefaultRoleRequestCustomScriptsEditableBy",
     "PatchedLseProjectUpdateRequestSampling",
     "PatchedLseProjectUpdateRequestSkipQueue",
     "Pause",
@@ -515,7 +541,10 @@ __all__ = [
     "ProjectSampling",
     "ProjectSkipQueue",
     "ProjectSubsetEnum",
+    "ProjectTemplate",
+    "ProjectTemplateRequest",
     "ProjectsDuplicateResponse",
+    "ProjectsImportPredictionsResponse",
     "ProjectsImportTasksResponse",
     "ProjectsListRequestFilter",
     "PromptsCompatibleProjectsRequestProjectType",
@@ -630,6 +659,7 @@ __all__ = [
     "model_providers",
     "organizations",
     "predictions",
+    "project_templates",
     "projects",
     "prompts",
     "session_policy",

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/base_client.py

@@ -23,6 +23,7 @@ from .ml.client import MlClient
 from .model_providers.client import ModelProvidersClient
 from .prompts.client import PromptsClient
 from .predictions.client import PredictionsClient
+from .project_templates.client import ProjectTemplatesClient
 from .projects.client import ProjectsClient
 from .tasks.client import TasksClient
 from .session_policy.client import SessionPolicyClient
@@ -51,6 +52,7 @@ from .ml.client import AsyncMlClient
 from .model_providers.client import AsyncModelProvidersClient
 from .prompts.client import AsyncPromptsClient
 from .predictions.client import AsyncPredictionsClient
+from .project_templates.client import AsyncProjectTemplatesClient
 from .projects.client import AsyncProjectsClient
 from .tasks.client import AsyncTasksClient
 from .session_policy.client import AsyncSessionPolicyClient
@@ -142,6 +144,7 @@ class LabelStudioBase:
         self.model_providers = ModelProvidersClient(client_wrapper=self._client_wrapper)
         self.prompts = PromptsClient(client_wrapper=self._client_wrapper)
         self.predictions = PredictionsClient(client_wrapper=self._client_wrapper)
+        self.project_templates = ProjectTemplatesClient(client_wrapper=self._client_wrapper)
         self.projects = ProjectsClient(client_wrapper=self._client_wrapper)
         self.tasks = TasksClient(client_wrapper=self._client_wrapper)
         self.session_policy = SessionPolicyClient(client_wrapper=self._client_wrapper)
@@ -233,6 +236,7 @@ class AsyncLabelStudioBase:
         self.model_providers = AsyncModelProvidersClient(client_wrapper=self._client_wrapper)
         self.prompts = AsyncPromptsClient(client_wrapper=self._client_wrapper)
         self.predictions = AsyncPredictionsClient(client_wrapper=self._client_wrapper)
+        self.project_templates = AsyncProjectTemplatesClient(client_wrapper=self._client_wrapper)
         self.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
         self.tasks = AsyncTasksClient(client_wrapper=self._client_wrapper)
         self.session_policy = AsyncSessionPolicyClient(client_wrapper=self._client_wrapper)

label_studio_sdk-2.0.8/src/label_studio_sdk/converter/exports/yolo.py

@@ -0,0 +1,164 @@
+import logging
+import os
+from label_studio_sdk.converter.utils import convert_annotation_to_yolo, convert_annotation_to_yolo_obb
+from label_studio_sdk.converter.keypoints import build_kp_order
+
+logger = logging.getLogger(__name__)
+
+def process_keypoints_for_yolo(labels, label_path,
+                               category_name_to_id, categories,
+                               is_obb, kp_order):
+    class_map = {c['name']: c['id'] for c in categories}
+
+    rectangles = {}
+    for item in labels:
+        if item['type'].lower() == 'rectanglelabels':
+            bbox_id = item['id']
+            cls_name = item['rectanglelabels'][0]
+            cls_idx = class_map.get(cls_name)
+            if cls_idx is None:
+                continue
+
+            x      = item['x'] / 100.0
+            y      = item['y'] / 100.0
+            width  = item['width']  / 100.0
+            height = item['height'] / 100.0
+            x_c    = x + width  / 2.0
+            y_c    = y + height / 2.0
+
+            rectangles[bbox_id] = {
+                'class_idx': cls_idx,
+                'x_center':  x_c,
+                'y_center':  y_c,
+                'width':     width,
+                'height':    height,
+                'kp_dict':   {}
+            }
+
+    for item in labels:
+        if item['type'].lower() == 'keypointlabels':
+            parent_id = item.get('parentID')
+            if parent_id not in rectangles:
+                continue
+            label_name = item['keypointlabels'][0]
+            kp_x = item['x'] / 100.0
+            kp_y = item['y'] / 100.0
+            rectangles[parent_id]['kp_dict'][label_name] = (kp_x, kp_y, 2)  # 2 = visible
+
+    lines = []
+    for rect in rectangles.values():
+        base = [
+            rect['class_idx'],
+            rect['x_center'],
+            rect['y_center'],
+            rect['width'],
+            rect['height']
+        ]
+        keypoints = []
+        for k in kp_order:
+            keypoints.extend(rect['kp_dict'].get(k, (0.0, 0.0, 0)))
+        line = ' '.join(map(str, base + keypoints))
+        lines.append(line)
+
+    with open(label_path, 'w', encoding='utf-8') as f:
+        f.write('\n'.join(lines))
+
+
+def process_and_save_yolo_annotations(labels, label_path, category_name_to_id, categories, is_obb, is_keypoints, label_config):
+    if is_keypoints:
+        kp_order = build_kp_order(label_config)
+        process_keypoints_for_yolo(labels, label_path, category_name_to_id, categories, is_obb, kp_order)
+        return categories, category_name_to_id
+
+    # Stream annotations directly to a temporary file to avoid
+    # accumulating them in memory and to preserve atomic writes.
+    tmp_path = f"{label_path}.tmp"
+
+    try:
+        with open(tmp_path, "w") as f:
+            for label in labels:
+                category_name = None
+                category_names = []  # considering multi-label
+                for key in ["rectanglelabels", "polygonlabels", "labels"]:
+                    if key in label and len(label[key]) > 0:
+                        # change to save multi-label
+                        for category_name in label[key]:
+                            category_names.append(category_name)
+
+                if len(category_names) == 0:
+                    logger.debug(
+                        "Unknown label type or labels are empty: " + str(label)
+                    )
+                    continue
+
+                for category_name in category_names:
+                    if category_name not in category_name_to_id:
+                        category_id = len(categories)
+                        category_name_to_id[category_name] = category_id
+                        categories.append({"id": category_id, "name": category_name})
+                    category_id = category_name_to_id[category_name]
+
+                    if (
+                        "rectanglelabels" in label
+                        or "rectangle" in label
+                        or "labels" in label
+                    ):
+                        # yolo obb
+                        if is_obb:
+                            obb_annotation = convert_annotation_to_yolo_obb(label)
+                            if obb_annotation is None:
+                                continue
+
+                            top_left, top_right, bottom_right, bottom_left = (
+                                obb_annotation
+                            )
+                            x1, y1 = top_left
+                            x2, y2 = top_right
+                            x3, y3 = bottom_right
+                            x4, y4 = bottom_left
+                            annotation_values = [category_id, x1, y1, x2, y2, x3, y3, x4, y4]
+
+                        # simple yolo
+                        else:
+                            annotation = convert_annotation_to_yolo(label)
+                            if annotation is None:
+                                continue
+
+                            (
+                                x,
+                                y,
+                                w,
+                                h,
+                            ) = annotation
+                            annotation_values = [category_id, x, y, w, h]
+
+                    elif "polygonlabels" in label or "polygon" in label:
+                        if not ('points' in label):
+                            continue
+                        points_abs = [(x / 100, y / 100) for x, y in label["points"]]
+                        annotation_values = (
+                            [category_id]
+                            + [coord for point in points_abs for coord in point]
+                        )
+                    else:
+                        raise ValueError(f"Unknown label type {label}")
+
+                    # Write the annotation line immediately
+                    for idx, val in enumerate(annotation_values):
+                        if idx == len(annotation_values) - 1:
+                            f.write(f"{val}\n")
+                        else:
+                            f.write(f"{val} ")
+
+        # Replace the target file atomically after successful write
+        os.replace(tmp_path, label_path)
+
+    finally:
+        # Clean up the temp file in case of exceptions before replace
+        try:
+            if os.path.exists(tmp_path):
+                os.remove(tmp_path)
+        except Exception:
+            pass
+
+    return categories, category_name_to_id

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/core/client_wrapper.py

@@ -4,7 +4,7 @@ import httpx
 
 from .http_client import AsyncHttpClient, HttpClient
 
-VERSION = "2.0.6"
+VERSION = "2.0.8"
 
 class BaseClientWrapper:
     def __init__(

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/import_storage/azure_spi/client.py

@@ -94,6 +94,7 @@ class AzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -146,6 +147,9 @@ class AzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -204,6 +208,7 @@ class AzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -246,6 +251,7 @@ class AzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -298,6 +304,9 @@ class AzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -355,6 +364,7 @@ class AzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -479,6 +489,7 @@ class AzureSpiClient:
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
         project: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -533,6 +544,9 @@ class AzureSpiClient:
         project : typing.Optional[int]
             A unique integer value identifying this project.
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -591,6 +605,7 @@ class AzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -755,6 +770,7 @@ class AsyncAzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -807,6 +823,9 @@ class AsyncAzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -873,6 +892,7 @@ class AsyncAzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -915,6 +935,7 @@ class AsyncAzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -967,6 +988,9 @@ class AsyncAzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -1032,6 +1056,7 @@ class AsyncAzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -1172,6 +1197,7 @@ class AsyncAzureSpiClient:
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
         project: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -1226,6 +1252,9 @@ class AsyncAzureSpiClient:
         project : typing.Optional[int]
             A unique integer value identifying this project.
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -1292,6 +1321,7 @@ class AsyncAzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/label_interface/control_tags.py

@@ -39,6 +39,7 @@ _TAG_TO_CLASS = {
     "taxonomy": "TaxonomyTag",
     "textarea": "TextAreaTag",
     "timeserieslabels": "TimeSeriesLabelsTag",
+    "chatmessage": "ChatMessageTag",
 }
 
 
@@ -950,6 +951,43 @@ class RatingTag(ControlTag):
         }
 
 
+class ChatMessageContent(BaseModel):
+    role: str
+    content: str
+    createdAt: Optional[int] = None
+
+
+class ChatMessageValue(BaseModel):
+    chatmessage: ChatMessageContent
+
+
+class ChatMessageTag(ControlTag):
+    """Control tag for chat messages targeting a `<Chat>` object.
+
+    This tag is a hybrid where `from_name == to_name` and `type == 'chatmessage'`.
+    """
+    tag: str = "ChatMessage"
+    _value_class: Type[ChatMessageValue] = ChatMessageValue
+
+    def to_json_schema(self):
+        return {
+            "type": "object",
+            "required": ["chatmessage"],
+            "properties": {
+                "chatmessage": {
+                    "type": "object",
+                    "required": ["role", "content"],
+                    "properties": {
+                        "role": {"type": "string"},
+                        "content": {"type": "string"},
+                        "createdAt": {"type": "number"}
+                    }
+                }
+            },
+            "description": f"Chat message for {self.to_name[0]}"
+        }
+
+
 class RelationsTag(ControlTag):
     """ """
     tag: str = "Relations"

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/label_interface/data_examples.json

@@ -10,6 +10,12 @@
     "Header": "Task header",
     "Paragraphs":  [{"author": "Alice", "text": "Hi, Bob."}, {"author": "Bob", "text": "Hello, Alice!"}, {"author": "Alice", "text": "What's up?"}, {"author": "Bob", "text": "Good. Ciao!"}, {"author": "Alice", "text": "Bye, Bob."}],
     "ParagraphsUrl": "<HOSTNAME>/samples/paragraphs.json?",
+    "Chat": [
+      {"role": "user", "content": "Hi there!", "createdAt": 1710000000000},
+      {"role": "assistant", "content": "Hello! How can I help you today?", "createdAt": 1710000005000},
+      {"role": "user", "content": "Can you summarize our onboarding checklist?", "createdAt": 1710000010000},
+      {"role": "assistant", "content": "Sure. It includes account setup, roles, labeling instructions, GT tasks, overlap and review.", "createdAt": 1710000015000}
+    ],
     "Table": {"Card number": 18799210, "First name": "Max", "Last name":  "Nobel"},
     "$videoHack": "<video src='<HOSTNAME>/static/samples/opossum_snow.mp4' width=100% controls>",
     "Video": "<HOSTNAME>/static/samples/opossum_snow.mp4",
@@ -85,6 +91,10 @@
     "AudioPlus": "<HOSTNAME>/static/samples/game.wav",
     "Header": "Task header",
     "Paragraphs":  [{"author": "Alice", "text": "Hi, Bob."}, {"author": "Bob", "text": "Hello, Alice!"}, {"author": "Alice", "text": "What's up?"}, {"author": "Bob", "text": "Good. Ciao!"}, {"author": "Alice", "text": "Bye, Bob."}],
+    "Chat": [
+      {"role": "user", "content": "Hello!", "createdAt": 1710000000000},
+      {"role": "assistant", "content": "Hi! What can I do for you?", "createdAt": 1710000004000}
+    ],
     "Table": {"Card number": 18799210, "First name": "Max", "Last name":  "Nobel"},
     "$videoHack": "<video src='static/samples/opossum_snow.mp4' width=100% controls>",
     "Video": "<HOSTNAME>/static/samples/opossum_snow.mp4",

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/label_interface/interface.py

@@ -27,6 +27,7 @@ from label_studio_sdk._legacy.exceptions import (
 
 from .base import LabelStudioTag
 from .control_tags import (
+    ChatMessageTag,
     ControlTag,
     ChoicesTag,
     LabelsTag,
@@ -623,6 +624,18 @@ class LabelInterface:
                 if lb:
                     labels[lb.parent_name][lb.value] = lb
 
+        # Special handling: auto-create ChatMessage control for each Chat object
+        chat_object_names = [name for name, obj in objects.items() if getattr(obj, 'tag', '').lower() == 'chat']
+        for name in chat_object_names:
+            if name not in controls:
+                controls[name] = ChatMessageTag(
+                    tag='ChatMessage',
+                    name=name,
+                    to_name=[name],
+                    attr={"name": name, "toName": name}
+                )
+
+
         return controls, objects, labels, xml_tree
 
     @classmethod

{label_studio_sdk-2.0.6 → label_studio_sdk-2.0.8}/src/label_studio_sdk/label_interface/object_tags.py

@@ -20,6 +20,7 @@ _TAG_TO_CLASS = {
     "list": "ListTag",
     "paragraphs": "ParagraphsTag",
     "timeseries": "TimeSeriesTag",
+    "chat": "ChatTag",
 }
 
 _DATA_EXAMPLES = None
@@ -306,3 +307,11 @@ class TimeSeriesTag(ObjectTag):
         else:
             # data is JSON
             return generate_time_series_json(time_column, value_columns, time_format)
+
+class ChatTag(ObjectTag):
+    """ """
+    tag: str = "Chat"
+
+    def _generate_example(self, examples, only_urls=False):
+        """ """
+        return examples.get("Chat")