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

label_studio_sdk/annotations/types/__init__.py

@@ -1,6 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .annotations_create_bulk_request_selected_items import AnnotationsCreateBulkRequestSelectedItems
+from .annotation_bulk_serializer_with_selected_items_request_last_action import (
+    AnnotationBulkSerializerWithSelectedItemsRequestLastAction,
+)
 from .annotations_create_bulk_response_item import AnnotationsCreateBulkResponseItem
 
-__all__ = ["AnnotationsCreateBulkRequestSelectedItems", "AnnotationsCreateBulkResponseItem"]
+__all__ = ["AnnotationBulkSerializerWithSelectedItemsRequestLastAction", "AnnotationsCreateBulkResponseItem"]

label_studio_sdk/annotations/types/annotation_bulk_serializer_with_selected_items_request_last_action.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from ...types.last_action_enum import LastActionEnum
+from ...types.null_enum import NullEnum
+
+AnnotationBulkSerializerWithSelectedItemsRequestLastAction = typing.Union[LastActionEnum, NullEnum]

label_studio_sdk/annotations/types/annotations_create_bulk_response_item.py

@@ -1,13 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.pydantic_utilities import UniversalBaseModel
-import typing
-from ...core.pydantic_utilities import IS_PYDANTIC_V2
+from ...core.unchecked_base_model import UncheckedBaseModel
 import pydantic
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
 
 
-class AnnotationsCreateBulkResponseItem(UniversalBaseModel):
-    id: typing.Optional[int] = None
+class AnnotationsCreateBulkResponseItem(UncheckedBaseModel):
+    id: int = pydantic.Field()
+    """
+    Annotation ID
+    """
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

label_studio_sdk/base_client.py

@@ -7,44 +7,48 @@ import httpx
 from .core.api_error import ApiError
 from .core.client_wrapper import SyncClientWrapper
 from .annotations.client import AnnotationsClient
+from .billing.client import BillingClient
+from .comments.client import CommentsClient
 from .users.client import UsersClient
 from .actions.client import ActionsClient
 from .views.client import ViewsClient
 from .files.client import FilesClient
+from .organizations.client import OrganizationsClient
+from .jwt_settings.client import JwtSettingsClient
 from .ml.client import MlClient
+from .model_providers.client import ModelProvidersClient
+from .prompts.client import PromptsClient
 from .predictions.client import PredictionsClient
 from .projects.client import ProjectsClient
 from .tasks.client import TasksClient
 from .import_storage.client import ImportStorageClient
 from .export_storage.client import ExportStorageClient
-from .webhooks.client import WebhooksClient
+from .tokens.client import TokensClient
 from .versions.client import VersionsClient
-from .prompts.client import PromptsClient
-from .model_providers.client import ModelProvidersClient
-from .comments.client import CommentsClient
+from .webhooks.client import WebhooksClient
 from .workspaces.client import WorkspacesClient
-from .tokens.client import TokensClient
-from .jwt_settings.client import JwtSettingsClient
 from .core.client_wrapper import AsyncClientWrapper
 from .annotations.client import AsyncAnnotationsClient
+from .billing.client import AsyncBillingClient
+from .comments.client import AsyncCommentsClient
 from .users.client import AsyncUsersClient
 from .actions.client import AsyncActionsClient
 from .views.client import AsyncViewsClient
 from .files.client import AsyncFilesClient
+from .organizations.client import AsyncOrganizationsClient
+from .jwt_settings.client import AsyncJwtSettingsClient
 from .ml.client import AsyncMlClient
+from .model_providers.client import AsyncModelProvidersClient
+from .prompts.client import AsyncPromptsClient
 from .predictions.client import AsyncPredictionsClient
 from .projects.client import AsyncProjectsClient
 from .tasks.client import AsyncTasksClient
 from .import_storage.client import AsyncImportStorageClient
 from .export_storage.client import AsyncExportStorageClient
-from .webhooks.client import AsyncWebhooksClient
+from .tokens.client import AsyncTokensClient
 from .versions.client import AsyncVersionsClient
-from .prompts.client import AsyncPromptsClient
-from .model_providers.client import AsyncModelProvidersClient
-from .comments.client import AsyncCommentsClient
+from .webhooks.client import AsyncWebhooksClient
 from .workspaces.client import AsyncWorkspacesClient
-from .tokens.client import AsyncTokensClient
-from .jwt_settings.client import AsyncJwtSettingsClient
 
 
 class LabelStudioBase:
@@ -110,24 +114,26 @@ class LabelStudioBase:
             timeout=_defaulted_timeout,
         )
         self.annotations = AnnotationsClient(client_wrapper=self._client_wrapper)
+        self.billing = BillingClient(client_wrapper=self._client_wrapper)
+        self.comments = CommentsClient(client_wrapper=self._client_wrapper)
         self.users = UsersClient(client_wrapper=self._client_wrapper)
         self.actions = ActionsClient(client_wrapper=self._client_wrapper)
         self.views = ViewsClient(client_wrapper=self._client_wrapper)
         self.files = FilesClient(client_wrapper=self._client_wrapper)
+        self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
+        self.jwt_settings = JwtSettingsClient(client_wrapper=self._client_wrapper)
         self.ml = MlClient(client_wrapper=self._client_wrapper)
+        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.projects = ProjectsClient(client_wrapper=self._client_wrapper)
         self.tasks = TasksClient(client_wrapper=self._client_wrapper)
         self.import_storage = ImportStorageClient(client_wrapper=self._client_wrapper)
         self.export_storage = ExportStorageClient(client_wrapper=self._client_wrapper)
-        self.webhooks = WebhooksClient(client_wrapper=self._client_wrapper)
+        self.tokens = TokensClient(client_wrapper=self._client_wrapper)
         self.versions = VersionsClient(client_wrapper=self._client_wrapper)
-        self.prompts = PromptsClient(client_wrapper=self._client_wrapper)
-        self.model_providers = ModelProvidersClient(client_wrapper=self._client_wrapper)
-        self.comments = CommentsClient(client_wrapper=self._client_wrapper)
+        self.webhooks = WebhooksClient(client_wrapper=self._client_wrapper)
         self.workspaces = WorkspacesClient(client_wrapper=self._client_wrapper)
-        self.tokens = TokensClient(client_wrapper=self._client_wrapper)
-        self.jwt_settings = JwtSettingsClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncLabelStudioBase:
@@ -193,24 +199,26 @@ class AsyncLabelStudioBase:
             timeout=_defaulted_timeout,
         )
         self.annotations = AsyncAnnotationsClient(client_wrapper=self._client_wrapper)
+        self.billing = AsyncBillingClient(client_wrapper=self._client_wrapper)
+        self.comments = AsyncCommentsClient(client_wrapper=self._client_wrapper)
         self.users = AsyncUsersClient(client_wrapper=self._client_wrapper)
         self.actions = AsyncActionsClient(client_wrapper=self._client_wrapper)
         self.views = AsyncViewsClient(client_wrapper=self._client_wrapper)
         self.files = AsyncFilesClient(client_wrapper=self._client_wrapper)
+        self.organizations = AsyncOrganizationsClient(client_wrapper=self._client_wrapper)
+        self.jwt_settings = AsyncJwtSettingsClient(client_wrapper=self._client_wrapper)
         self.ml = AsyncMlClient(client_wrapper=self._client_wrapper)
+        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.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
         self.tasks = AsyncTasksClient(client_wrapper=self._client_wrapper)
         self.import_storage = AsyncImportStorageClient(client_wrapper=self._client_wrapper)
         self.export_storage = AsyncExportStorageClient(client_wrapper=self._client_wrapper)
-        self.webhooks = AsyncWebhooksClient(client_wrapper=self._client_wrapper)
+        self.tokens = AsyncTokensClient(client_wrapper=self._client_wrapper)
         self.versions = AsyncVersionsClient(client_wrapper=self._client_wrapper)
-        self.prompts = AsyncPromptsClient(client_wrapper=self._client_wrapper)
-        self.model_providers = AsyncModelProvidersClient(client_wrapper=self._client_wrapper)
-        self.comments = AsyncCommentsClient(client_wrapper=self._client_wrapper)
+        self.webhooks = AsyncWebhooksClient(client_wrapper=self._client_wrapper)
         self.workspaces = AsyncWorkspacesClient(client_wrapper=self._client_wrapper)
-        self.tokens = AsyncTokensClient(client_wrapper=self._client_wrapper)
-        self.jwt_settings = AsyncJwtSettingsClient(client_wrapper=self._client_wrapper)
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LabelStudioEnvironment) -> str:

label_studio_sdk/billing/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

label_studio_sdk/billing/client.py

@@ -0,0 +1,112 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.client_wrapper import SyncClientWrapper
+import typing
+from ..core.request_options import RequestOptions
+from ..types.billing_info_response import BillingInfoResponse
+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
+
+
+class BillingClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def info(self, *, request_options: typing.Optional[RequestOptions] = None) -> BillingInfoResponse:
+        """
+        Retrieve billing checks and feature flags for the active organization.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        BillingInfoResponse
+            Billing information for the active organization
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.billing.info()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/billing/info",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    BillingInfoResponse,
+                    construct_type(
+                        type_=BillingInfoResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncBillingClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def info(self, *, request_options: typing.Optional[RequestOptions] = None) -> BillingInfoResponse:
+        """
+        Retrieve billing checks and feature flags for the active organization.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        BillingInfoResponse
+            Billing information for the active organization
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.billing.info()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/billing/info",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    BillingInfoResponse,
+                    construct_type(
+                        type_=BillingInfoResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)