label-studio-sdk 1.0.20__py3-none-any.whl → 2.0.0__py3-none-any.whl

label_studio_sdk/import_storage/types/import_storage_list_types_response_item.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.unchecked_base_model import UncheckedBaseModel
 import typing
 from ...core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class ImportStorageListTypesResponseItem(UniversalBaseModel):
+class ImportStorageListTypesResponseItem(UncheckedBaseModel):
     name: typing.Optional[str] = None
     title: typing.Optional[str] = None
 

label_studio_sdk/jwt_settings/client.py

@@ -3,8 +3,8 @@
 import typing
 from ..core.client_wrapper import SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..types.jwt_settings_response import JwtSettingsResponse
-from ..core.pydantic_utilities import parse_obj_as
+from ..types.lsejwt_settings import LsejwtSettings
+from ..core.unchecked_base_model import construct_type
 from json.decoder import JSONDecodeError
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper
@@ -17,9 +17,9 @@ class JwtSettingsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> JwtSettingsResponse:
+    def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> LsejwtSettings:
         """
-        Retrieve JWT settings for the currently-active organization.
+        Retrieve JWT settings for the currently active organization.
 
         Parameters
         ----------
@@ -28,8 +28,8 @@ class JwtSettingsClient:
 
         Returns
         -------
-        JwtSettingsResponse
-            JWT settings retrieved successfully
+        LsejwtSettings
+
 
         Examples
         --------
@@ -48,9 +48,9 @@ class JwtSettingsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    JwtSettingsResponse,
-                    parse_obj_as(
-                        type_=JwtSettingsResponse,  # type: ignore
+                    LsejwtSettings,
+                    construct_type(
+                        type_=LsejwtSettings,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -59,35 +59,35 @@ class JwtSettingsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create(
+    def update(
         self,
         *,
-        api_tokens_enabled: bool,
-        legacy_api_tokens_enabled: bool,
-        api_token_ttl_days: int,
+        api_tokens_enabled: typing.Optional[bool] = OMIT,
+        legacy_api_tokens_enabled: typing.Optional[bool] = OMIT,
+        api_token_ttl_days: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> JwtSettingsResponse:
+    ) -> LsejwtSettings:
         """
         Update JWT settings for the currently active organization.
 
         Parameters
         ----------
-        api_tokens_enabled : bool
-            Whether JWT API tokens are enabled
+        api_tokens_enabled : typing.Optional[bool]
+            Enable JWT API token authentication for this organization
 
-        legacy_api_tokens_enabled : bool
-            Whether legacy API tokens are enabled
+        legacy_api_tokens_enabled : typing.Optional[bool]
+            Enable legacy API token authentication for this organization
 
-        api_token_ttl_days : int
-            Number of days before API tokens expire
+        api_token_ttl_days : typing.Optional[int]
+            Number of days before JWT API tokens expire
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        JwtSettingsResponse
-            JWT settings updated successfully
+        LsejwtSettings
+
 
         Examples
         --------
@@ -96,11 +96,7 @@ class JwtSettingsClient:
         client = LabelStudio(
             api_key="YOUR_API_KEY",
         )
-        client.jwt_settings.create(
-            api_tokens_enabled=True,
-            legacy_api_tokens_enabled=True,
-            api_token_ttl_days=1,
-        )
+        client.jwt_settings.update()
         """
         _response = self._client_wrapper.httpx_client.request(
             "api/jwt/settings",
@@ -110,15 +106,18 @@ class JwtSettingsClient:
                 "legacy_api_tokens_enabled": legacy_api_tokens_enabled,
                 "api_token_ttl_days": api_token_ttl_days,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    JwtSettingsResponse,
-                    parse_obj_as(
-                        type_=JwtSettingsResponse,  # type: ignore
+                    LsejwtSettings,
+                    construct_type(
+                        type_=LsejwtSettings,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -132,9 +131,9 @@ class AsyncJwtSettingsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> JwtSettingsResponse:
+    async def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> LsejwtSettings:
         """
-        Retrieve JWT settings for the currently-active organization.
+        Retrieve JWT settings for the currently active organization.
 
         Parameters
         ----------
@@ -143,8 +142,8 @@ class AsyncJwtSettingsClient:
 
         Returns
         -------
-        JwtSettingsResponse
-            JWT settings retrieved successfully
+        LsejwtSettings
+
 
         Examples
         --------
@@ -171,9 +170,9 @@ class AsyncJwtSettingsClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    JwtSettingsResponse,
-                    parse_obj_as(
-                        type_=JwtSettingsResponse,  # type: ignore
+                    LsejwtSettings,
+                    construct_type(
+                        type_=LsejwtSettings,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -182,35 +181,35 @@ class AsyncJwtSettingsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def create(
+    async def update(
         self,
         *,
-        api_tokens_enabled: bool,
-        legacy_api_tokens_enabled: bool,
-        api_token_ttl_days: int,
+        api_tokens_enabled: typing.Optional[bool] = OMIT,
+        legacy_api_tokens_enabled: typing.Optional[bool] = OMIT,
+        api_token_ttl_days: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> JwtSettingsResponse:
+    ) -> LsejwtSettings:
         """
         Update JWT settings for the currently active organization.
 
         Parameters
         ----------
-        api_tokens_enabled : bool
-            Whether JWT API tokens are enabled
+        api_tokens_enabled : typing.Optional[bool]
+            Enable JWT API token authentication for this organization
 
-        legacy_api_tokens_enabled : bool
-            Whether legacy API tokens are enabled
+        legacy_api_tokens_enabled : typing.Optional[bool]
+            Enable legacy API token authentication for this organization
 
-        api_token_ttl_days : int
-            Number of days before API tokens expire
+        api_token_ttl_days : typing.Optional[int]
+            Number of days before JWT API tokens expire
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        JwtSettingsResponse
-            JWT settings updated successfully
+        LsejwtSettings
+
 
         Examples
         --------
@@ -224,11 +223,7 @@ class AsyncJwtSettingsClient:
 
 
         async def main() -> None:
-            await client.jwt_settings.create(
-                api_tokens_enabled=True,
-                legacy_api_tokens_enabled=True,
-                api_token_ttl_days=1,
-            )
+            await client.jwt_settings.update()
 
 
         asyncio.run(main())
@@ -241,15 +236,18 @@ class AsyncJwtSettingsClient:
                 "legacy_api_tokens_enabled": legacy_api_tokens_enabled,
                 "api_token_ttl_days": api_token_ttl_days,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    JwtSettingsResponse,
-                    parse_obj_as(
-                        type_=JwtSettingsResponse,  # type: ignore
+                    LsejwtSettings,
+                    construct_type(
+                        type_=LsejwtSettings,  # type: ignore
                         object_=_response.json(),
                     ),
                 )

label_studio_sdk/label_interface/control_tags.py

@@ -287,10 +287,9 @@ class ControlTag(LabelStudioTag):
 
     def _validate_value_labels(self, value):
         """ """
-        if self._label_attr_name not in value:
-            return False
-
-        return self._validate_labels(value.get(self._label_attr_name))
+        if hasattr(self, "_label_attr_name") and self._label_attr_name in value:
+            return self._validate_labels(value.get(self._label_attr_name))
+        return False
 
     def validate_value(self, value: dict) -> bool:
         """
@@ -312,6 +311,10 @@ class ControlTag(LabelStudioTag):
             True if the value is valid, False otherwise
         """
         
+        # Accept inputs that nest the payload under the control name, e.g. {"ranker": "rank": ["a", "b", "c"]}
+        if isinstance(value, dict) and self.name in value and isinstance(value[self.name], dict):
+            value = value[self.name]
+
         if hasattr(self, "_label_attr_name"):
             if not self._validate_value_labels(value):
                 return False
@@ -499,13 +502,13 @@ class ControlTag(LabelStudioTag):
 
 
 class SpanSelection(BaseModel):
-    start: int = Field(..., ge=0)
-    end: int = Field(..., ge=0)
+    start: Union[int, str]
+    end: Union[int, str]
 
 
 class SpanSelectionOffsets(SpanSelection):
-    startOffset: int = Field(..., ge=0)
-    endOffset: int = Field(..., ge=0)
+    startOffset: Optional[int] = Field(None, ge=0)
+    endOffset: Optional[int] = Field(None, ge=0)
 
 
 class ChoicesValue(BaseModel):
@@ -912,6 +915,25 @@ class RatingTag(ControlTag):
     _value_class: Type[RatingValue] = RatingValue
     _label_attr_name: str = "rating"
 
+    def _validate_value_labels(self, value):
+        """Override to handle rating values correctly - ratings are not labels"""
+        if self._label_attr_name not in value:
+            return False
+
+        # For ratings, we just check that the rating value exists and is valid
+        # The actual validation is done by the RatingValue model
+        rating_value = value.get(self._label_attr_name)
+        if rating_value is None:
+            return False
+
+        # Check if rating is within valid range (0 to maxRating)
+        max_rating = int(self.attr.get('maxRating', 5))
+        try:
+            rating_int = int(rating_value)
+            return 0 <= rating_int <= max_rating
+        except (ValueError, TypeError):
+            return False
+
     def to_json_schema(self):
         """
         Converts the current RatingTag instance into a JSON Schema.
@@ -953,6 +975,24 @@ class TaxonomyTag(ControlTag):
     _value_class: Type[TaxonomyValue] = TaxonomyValue
     _label_attr_name: str = "taxonomy"
 
+    def _validate_value_labels(self, value):
+        """Validate taxonomy labels by flattening selected paths before subset check.
+
+        Expected value format:
+          { "taxonomy": [["A"], ["A", "B"], ...] }
+        """
+        paths = value.get("taxonomy")
+        if paths is None:
+            return False
+        if not isinstance(paths, list):
+            return False
+        flat_labels: List[str] = []
+        for path in paths:
+            if not isinstance(path, list):
+                return False
+            flat_labels.extend(path)
+        return self._validate_labels(flat_labels)
+
     def to_json_schema(self):
         """
         Converts the current TaxonomyTag instance into a JSON Schema.