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

label_studio_sdk/__init__.py

@@ -20,6 +20,9 @@ from .types import (
     AzureBlobImportStorage,
     BatchFailedPredictions,
     BatchPredictions,
+    BillingChecks,
+    BillingFlags,
+    BillingInfoResponse,
     BlankEnum,
     BlueprintList,
     BudgetResetPeriodEnum,
@@ -29,6 +32,7 @@ from .types import (
     CommentSerializerWithExpandedUser,
     ConvertedFormat,
     ConvertedFormatRequest,
+    CountLimit,
     CustomScriptsEditableByEnum,
     DefaultRoleEnum,
     EditionEnum,
@@ -57,6 +61,7 @@ from .types import (
     LseKeyIndicatorValue,
     LseOrganization,
     LseOrganizationCustomScriptsEditableBy,
+    LseOrganizationMemberList,
     LseProjectCreate,
     LseProjectCreateSampling,
     LseProjectCreateSkipQueue,
@@ -88,6 +93,7 @@ from .types import (
     LseTaskSerializerForReviewersPredictionsItem,
     LseUser,
     LseUserApi,
+    LseUserOrganizationMemberList,
     LseapiTokenCreate,
     LseapiTokenList,
     LsejwtSettings,
@@ -110,6 +116,7 @@ from .types import (
     OrganizationMember,
     OrganizationMembership,
     PaginatedAllRolesProjectListList,
+    PaginatedLseOrganizationMemberListList,
     PaginatedRoleBasedTaskList,
     Pause,
     PauseRequest,
@@ -122,6 +129,7 @@ from .types import (
     ProjectSampling,
     ProjectSkipQueue,
     ProjectSubsetEnum,
+    PromptsStatusEnum,
     ProviderEnum,
     ReasonEnum,
     RedisExportStorage,
@@ -136,6 +144,7 @@ from .types import (
     ReviewSettingsRequeueRejectedTasksMode,
     ReviewSettingsReviewCriteria,
     ReviewedEnum,
+    Role9E7Enum,
     RoleBasedTask,
     S3ExportStorage,
     S3ImportStorage,
@@ -178,6 +187,7 @@ from .errors import (
 from . import (
     actions,
     annotations,
+    billing,
     comments,
     export_storage,
     files,
@@ -307,6 +317,9 @@ __all__ = [
     "BadRequestError",
     "BatchFailedPredictions",
     "BatchPredictions",
+    "BillingChecks",
+    "BillingFlags",
+    "BillingInfoResponse",
     "BlankEnum",
     "BlueprintList",
     "BudgetResetPeriodEnum",
@@ -317,6 +330,7 @@ __all__ = [
     "CommentSerializerWithExpandedUser",
     "ConvertedFormat",
     "ConvertedFormatRequest",
+    "CountLimit",
     "CustomScriptsEditableByEnum",
     "DefaultRoleEnum",
     "EditionEnum",
@@ -351,6 +365,7 @@ __all__ = [
     "LseKeyIndicatorValue",
     "LseOrganization",
     "LseOrganizationCustomScriptsEditableBy",
+    "LseOrganizationMemberList",
     "LseProjectCreate",
     "LseProjectCreateRequestSampling",
     "LseProjectCreateRequestSkipQueue",
@@ -384,6 +399,7 @@ __all__ = [
     "LseTaskSerializerForReviewersPredictionsItem",
     "LseUser",
     "LseUserApi",
+    "LseUserOrganizationMemberList",
     "LseapiTokenCreate",
     "LseapiTokenList",
     "LsejwtSettings",
@@ -412,6 +428,7 @@ __all__ = [
     "OrganizationMember",
     "OrganizationMembership",
     "PaginatedAllRolesProjectListList",
+    "PaginatedLseOrganizationMemberListList",
     "PaginatedRoleBasedTaskList",
     "PatchedLseProjectUpdateRequestSampling",
     "PatchedLseProjectUpdateRequestSkipQueue",
@@ -430,6 +447,7 @@ __all__ = [
     "ProjectsImportTasksResponse",
     "ProjectsListRequestFilter",
     "PromptsCompatibleProjectsRequestProjectType",
+    "PromptsStatusEnum",
     "ProviderEnum",
     "ReasonEnum",
     "RedisExportStorage",
@@ -444,6 +462,7 @@ __all__ = [
     "ReviewSettingsRequeueRejectedTasksMode",
     "ReviewSettingsReviewCriteria",
     "ReviewedEnum",
+    "Role9E7Enum",
     "RoleBasedTask",
     "S3ExportStorage",
     "S3ImportStorage",
@@ -513,6 +532,7 @@ __all__ = [
     "__version__",
     "actions",
     "annotations",
+    "billing",
     "comments",
     "export_storage",
     "files",

label_studio_sdk/base_client.py

@@ -7,6 +7,7 @@ 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
@@ -28,6 +29,7 @@ from .webhooks.client import WebhooksClient
 from .workspaces.client import WorkspacesClient
 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
@@ -112,6 +114,7 @@ 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)
@@ -196,6 +199,7 @@ 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)

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)

label_studio_sdk/organizations/members/client.py

@@ -1,23 +1,191 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.client_wrapper import SyncClientWrapper
 import typing
+from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
-from ...types.organization_member import OrganizationMember
+from ...types.paginated_lse_organization_member_list_list import PaginatedLseOrganizationMemberListList
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.unchecked_base_model import construct_type
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
+from ...types.role9e7enum import Role9E7Enum
+from ...types.lse_organization_member_list import LseOrganizationMemberList
+from ...types.organization_member import OrganizationMember
 from ...errors.forbidden_error import ForbiddenError
 from ...errors.not_found_error import NotFoundError
 from ...errors.method_not_allowed_error import MethodNotAllowedError
 from ...core.client_wrapper import AsyncClientWrapper
 
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
 
 class MembersClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    def list(
+        self,
+        id: int,
+        *,
+        exclude_project_id: typing.Optional[int] = None,
+        exclude_workspace_id: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedLseOrganizationMemberListList:
+        """
+        Retrieve a list of all users and roles in a specific organization.
+
+        Parameters
+        ----------
+        id : int
+            A unique integer value identifying this organization.
+
+        exclude_project_id : typing.Optional[int]
+            Project ID to exclude users who are already associated with this project (direct members, workspace members, or implicit admin/owner access).
+
+        exclude_workspace_id : typing.Optional[int]
+            Workspace ID to exclude users who are already associated with this workspace (direct workspace members or implicit admin/owner access).
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        page : typing.Optional[int]
+            A page number within the paginated result set.
+
+        page_size : typing.Optional[int]
+            Number of results to return per page.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedLseOrganizationMemberListList
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.organizations.members.list(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/organizations/{jsonable_encoder(id)}/memberships",
+            method="GET",
+            params={
+                "exclude_project_id": exclude_project_id,
+                "exclude_workspace_id": exclude_workspace_id,
+                "ordering": ordering,
+                "page": page,
+                "page_size": page_size,
+                "search": search,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedLseOrganizationMemberListList,
+                    construct_type(
+                        type_=PaginatedLseOrganizationMemberListList,  # 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)
+
+    def update(
+        self,
+        id: int,
+        *,
+        user_id: typing.Optional[int] = OMIT,
+        role: typing.Optional[Role9E7Enum] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> LseOrganizationMemberList:
+        """
+        Update organization membership or role for a specific user ID.
+
+        **User Rotation Best Practices for API Usage**
+
+        To maintain compliance with our licensing terms and ensure optimal performance of HumanSignal's APIs, please consider the following guidelines when managing user assignments:
+
+        * **Maintain a 7-Day Minimum Assignment**: Once a licensed seat is assigned to a user, maintain that assignment for at least seven consecutive days before rotating it to another user.
+
+        * **Automate, Monitor, and Log Rotations**: Implement automated scheduling and logging mechanisms to track the timing of user rotations. This helps ensure that rotations adhere to the seven-day minimum period.
+
+        * **Adhere to API Update Frequency and Wait Periods**: When updating user assignments via our APIs, follow the recommended frequency and wait period guidelines provided in the HumanSignal API documentation. Avoid sending rapid, successive requests that might overload the endpoint. Instead, incorporate appropriate delays between calls as specified in the documentation.
+
+        * **Avoid Overloading the API Endpoint**: Design your integration to batch or schedule updates where possible, and implement backoff strategies if the API indicates rate limiting. This helps prevent service disruptions and ensures a smooth operation.
+
+        Parameters
+        ----------
+        id : int
+            A unique integer value identifying this organization.
+
+        user_id : typing.Optional[int]
+
+        role : typing.Optional[Role9E7Enum]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        LseOrganizationMemberList
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.organizations.members.update(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/organizations/{jsonable_encoder(id)}/memberships",
+            method="PATCH",
+            json={
+                "user_id": user_id,
+                "role": role,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    LseOrganizationMemberList,
+                    construct_type(
+                        type_=LseOrganizationMemberList,  # 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)
+
     def get(
         self, id: int, user_pk: int, *, request_options: typing.Optional[RequestOptions] = None
     ) -> OrganizationMember:
@@ -148,6 +316,184 @@ class AsyncMembersClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    async def list(
+        self,
+        id: int,
+        *,
+        exclude_project_id: typing.Optional[int] = None,
+        exclude_workspace_id: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedLseOrganizationMemberListList:
+        """
+        Retrieve a list of all users and roles in a specific organization.
+
+        Parameters
+        ----------
+        id : int
+            A unique integer value identifying this organization.
+
+        exclude_project_id : typing.Optional[int]
+            Project ID to exclude users who are already associated with this project (direct members, workspace members, or implicit admin/owner access).
+
+        exclude_workspace_id : typing.Optional[int]
+            Workspace ID to exclude users who are already associated with this workspace (direct workspace members or implicit admin/owner access).
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        page : typing.Optional[int]
+            A page number within the paginated result set.
+
+        page_size : typing.Optional[int]
+            Number of results to return per page.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedLseOrganizationMemberListList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.organizations.members.list(
+                id=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/organizations/{jsonable_encoder(id)}/memberships",
+            method="GET",
+            params={
+                "exclude_project_id": exclude_project_id,
+                "exclude_workspace_id": exclude_workspace_id,
+                "ordering": ordering,
+                "page": page,
+                "page_size": page_size,
+                "search": search,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedLseOrganizationMemberListList,
+                    construct_type(
+                        type_=PaginatedLseOrganizationMemberListList,  # 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)
+
+    async def update(
+        self,
+        id: int,
+        *,
+        user_id: typing.Optional[int] = OMIT,
+        role: typing.Optional[Role9E7Enum] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> LseOrganizationMemberList:
+        """
+        Update organization membership or role for a specific user ID.
+
+        **User Rotation Best Practices for API Usage**
+
+        To maintain compliance with our licensing terms and ensure optimal performance of HumanSignal's APIs, please consider the following guidelines when managing user assignments:
+
+        * **Maintain a 7-Day Minimum Assignment**: Once a licensed seat is assigned to a user, maintain that assignment for at least seven consecutive days before rotating it to another user.
+
+        * **Automate, Monitor, and Log Rotations**: Implement automated scheduling and logging mechanisms to track the timing of user rotations. This helps ensure that rotations adhere to the seven-day minimum period.
+
+        * **Adhere to API Update Frequency and Wait Periods**: When updating user assignments via our APIs, follow the recommended frequency and wait period guidelines provided in the HumanSignal API documentation. Avoid sending rapid, successive requests that might overload the endpoint. Instead, incorporate appropriate delays between calls as specified in the documentation.
+
+        * **Avoid Overloading the API Endpoint**: Design your integration to batch or schedule updates where possible, and implement backoff strategies if the API indicates rate limiting. This helps prevent service disruptions and ensures a smooth operation.
+
+        Parameters
+        ----------
+        id : int
+            A unique integer value identifying this organization.
+
+        user_id : typing.Optional[int]
+
+        role : typing.Optional[Role9E7Enum]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        LseOrganizationMemberList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.organizations.members.update(
+                id=1,
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/organizations/{jsonable_encoder(id)}/memberships",
+            method="PATCH",
+            json={
+                "user_id": user_id,
+                "role": role,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    LseOrganizationMemberList,
+                    construct_type(
+                        type_=LseOrganizationMemberList,  # 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)
+
     async def get(
         self, id: int, user_pk: int, *, request_options: typing.Optional[RequestOptions] = None
     ) -> OrganizationMember: