fix: UTC-310: Fix prediction validation for per-region labels (#629)

Co-authored-by: hakan458 <hakan@heartex.com>

Author: hakan458

src/label_studio_sdk/label_interface/control_tags.py

@@ -503,8 +503,8 @@ def as_tuple(self):
 
 
 class SpanSelection(BaseModel):
-    start: Union[int, str]
-    end: Union[int, str]
+    start: Union[float, int, str]
+    end: Union[float, int, str]
 
 
 class SpanSelectionOffsets(SpanSelection):
@@ -561,6 +561,7 @@ class LabelsTag(ControlTag):
     _label_attr_name: str = "labels"
     _value_class: Type[LabelsValue] = LabelsValue
 
+
     def to_json_schema(self):
         """
         Converts the current LabelsTag instance into a JSON Schema.
@@ -572,16 +573,20 @@ def to_json_schema(self):
             "type": "array",
             "items": {
                 "type": "object",
-                "required": ["start", "end", "labels"],
+                "required": ["labels"],
                 "properties": {
                     "start": {
-                        "type": "integer",
+                        "oneOf": [
+                            {"type": "integer", "minimum": 0},
+                            {"type": "number", "minimum": 0}
+                        ],
                         # TODO: this is incompatible with the OpenAI API using PredictedOutputs
-                        "minimum": 0
                     },
                     "end": {
-                        "type": "integer",
-                        "minimum": 0
+                        "oneOf": [
+                            {"type": "integer", "minimum": 0},
+                            {"type": "number", "minimum": 0}
+                        ]
                     },
                     "labels": {
                         "type": "array",

tests/custom/test_interface/test_json_schema.py

@@ -86,15 +86,19 @@
                     "description": "Labels and span indices for txt",
                     "items": {
                         "type": "object",
-                        "required": ["start", "end", "labels"],
+                        "required": ["labels"],
                         "properties": {
                             "start": {
-                                "type": "integer",
-                                "minimum": 0
+                                "oneOf": [
+                                    {"type": "integer", "minimum": 0},
+                                    {"type": "number", "minimum": 0}
+                                ]
                             },
                             "end": {
-                                "type": "integer",
-                                "minimum": 0
+                                "oneOf": [
+                                    {"type": "integer", "minimum": 0},
+                                    {"type": "number", "minimum": 0}
+                                ]
                             },
                             "labels": {
                                 "type": "array",
@@ -397,4 +401,4 @@ def test_concurrent_json_schema_to_pydantic_threaded():
         assert None not in results
 
         # Check if execution time is reasonable (adjust as needed)
-        assert end_time - start_time < 1.0, f"Execution took too long: {end_time - start_time} seconds"
+        assert end_time - start_time < 1.0, f"Execution took too long: {end_time - start_time} seconds"