# This file was auto-generated by Fern from our API Definition.
import typing
from json.decoder import JSONDecodeError
from ...core.api_error import ApiError
from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
from ...core.http_response import AsyncHttpResponse, HttpResponse
from ...core.request_options import RequestOptions
from ...core.serialization import convert_and_respect_annotation_metadata
from ...core.unchecked_base_model import construct_type
from ...types.project_group_request import ProjectGroupRequest
from ...types.scim_settings import ScimSettings
from ...types.scim_settings_update import ScimSettingsUpdate
# this is used as the default value for optional parameters
OMIT = typing.cast(typing.Any, ...)
class RawScimClient:
def __init__(self, *, client_wrapper: SyncClientWrapper):
self._client_wrapper = client_wrapper
def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[ScimSettings]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Retrieve SCIM settings for the currently active organization.
Parameters
----------
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
HttpResponse[ScimSettings]
"""
_response = self._client_wrapper.httpx_client.request(
"api/scim/settings",
method="GET",
request_options=request_options,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
ScimSettings,
construct_type(
type_=ScimSettings, # type: ignore
object_=_response.json(),
),
)
return HttpResponse(response=_response, data=_data)
_response_json = _response.json()
except JSONDecodeError:
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
def update(
self,
*,
projects_groups: typing.Optional[typing.Sequence[ProjectGroupRequest]] = OMIT,
roles_groups: typing.Optional[typing.Sequence[typing.Sequence[str]]] = OMIT,
workspaces_groups: typing.Optional[typing.Sequence[typing.Sequence[str]]] = OMIT,
request_options: typing.Optional[RequestOptions] = None,
) -> HttpResponse[ScimSettingsUpdate]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Update SCIM settings for the currently active organization.
Parameters
----------
projects_groups : typing.Optional[typing.Sequence[ProjectGroupRequest]]
Projects to Groups Mapping. List of objects with project_id, group, role.
roles_groups : typing.Optional[typing.Sequence[typing.Sequence[str]]]
Organization Roles to Groups Mapping. List of [role_name, group_name] pairs.
workspaces_groups : typing.Optional[typing.Sequence[typing.Sequence[str]]]
Workspaces to Groups Mapping. List of [workspace_title, group_name] pairs.
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
HttpResponse[ScimSettingsUpdate]
"""
_response = self._client_wrapper.httpx_client.request(
"api/scim/settings",
method="POST",
json={
"projects_groups": convert_and_respect_annotation_metadata(
object_=projects_groups, annotation=typing.Sequence[ProjectGroupRequest], direction="write"
),
"roles_groups": roles_groups,
"workspaces_groups": workspaces_groups,
},
headers={
"content-type": "application/json",
},
request_options=request_options,
omit=OMIT,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
ScimSettingsUpdate,
construct_type(
type_=ScimSettingsUpdate, # type: ignore
object_=_response.json(),
),
)
return HttpResponse(response=_response, data=_data)
_response_json = _response.json()
except JSONDecodeError:
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
class AsyncRawScimClient:
def __init__(self, *, client_wrapper: AsyncClientWrapper):
self._client_wrapper = client_wrapper
async def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> AsyncHttpResponse[ScimSettings]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Retrieve SCIM settings for the currently active organization.
Parameters
----------
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
AsyncHttpResponse[ScimSettings]
"""
_response = await self._client_wrapper.httpx_client.request(
"api/scim/settings",
method="GET",
request_options=request_options,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
ScimSettings,
construct_type(
type_=ScimSettings, # type: ignore
object_=_response.json(),
),
)
return AsyncHttpResponse(response=_response, data=_data)
_response_json = _response.json()
except JSONDecodeError:
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
async def update(
self,
*,
projects_groups: typing.Optional[typing.Sequence[ProjectGroupRequest]] = OMIT,
roles_groups: typing.Optional[typing.Sequence[typing.Sequence[str]]] = OMIT,
workspaces_groups: typing.Optional[typing.Sequence[typing.Sequence[str]]] = OMIT,
request_options: typing.Optional[RequestOptions] = None,
) -> AsyncHttpResponse[ScimSettingsUpdate]:
"""
This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
Update SCIM settings for the currently active organization.
Parameters
----------
projects_groups : typing.Optional[typing.Sequence[ProjectGroupRequest]]
Projects to Groups Mapping. List of objects with project_id, group, role.
roles_groups : typing.Optional[typing.Sequence[typing.Sequence[str]]]
Organization Roles to Groups Mapping. List of [role_name, group_name] pairs.
workspaces_groups : typing.Optional[typing.Sequence[typing.Sequence[str]]]
Workspaces to Groups Mapping. List of [workspace_title, group_name] pairs.
request_options : typing.Optional[RequestOptions]
Request-specific configuration.
Returns
-------
AsyncHttpResponse[ScimSettingsUpdate]
"""
_response = await self._client_wrapper.httpx_client.request(
"api/scim/settings",
method="POST",
json={
"projects_groups": convert_and_respect_annotation_metadata(
object_=projects_groups, annotation=typing.Sequence[ProjectGroupRequest], direction="write"
),
"roles_groups": roles_groups,
"workspaces_groups": workspaces_groups,
},
headers={
"content-type": "application/json",
},
request_options=request_options,
omit=OMIT,
)
try:
if 200 <= _response.status_code < 300:
_data = typing.cast(
ScimSettingsUpdate,
construct_type(
type_=ScimSettingsUpdate, # type: ignore
object_=_response.json(),
),
)
return AsyncHttpResponse(response=_response, data=_data)
_response_json = _response.json()
except JSONDecodeError:
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)