label-studio-sdk 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

label_studio_sdk/actions/types/actions_create_request_selected_items_excluded.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ...core.datetime_utils import serialize_datetime
+from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class ActionsCreateRequestSelectedItemsExcluded(pydantic_v1.BaseModel):
+    all_: bool = pydantic_v1.Field(alias="all")
+    """
+    All tasks are selected
+    """
+
+    excluded: typing.Optional[typing.List[int]] = pydantic_v1.Field(default=None)
+    """
+    List of excluded task IDs
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/actions/types/actions_create_request_selected_items_included.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ...core.datetime_utils import serialize_datetime
+from ...core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class ActionsCreateRequestSelectedItemsIncluded(pydantic_v1.BaseModel):
+    all_: bool = pydantic_v1.Field(alias="all")
+    """
+    No tasks are selected
+    """
+
+    included: typing.Optional[typing.List[int]] = pydantic_v1.Field(default=None)
+    """
+    List of included task IDs
+    """
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/base_client.py

@@ -0,0 +1,183 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import os
+import typing
+
+import httpx
+
+from .actions.client import ActionsClient, AsyncActionsClient
+from .annotations.client import AnnotationsClient, AsyncAnnotationsClient
+from .core.api_error import ApiError
+from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .environment import LabelStudioEnvironment
+from .export_storage.client import AsyncExportStorageClient, ExportStorageClient
+from .files.client import AsyncFilesClient, FilesClient
+from .import_storage.client import AsyncImportStorageClient, ImportStorageClient
+from .ml.client import AsyncMlClient, MlClient
+from .predictions.client import AsyncPredictionsClient, PredictionsClient
+from .projects.client import AsyncProjectsClient, ProjectsClient
+from .tasks.client import AsyncTasksClient, TasksClient
+from .users.client import AsyncUsersClient, UsersClient
+from .views.client import AsyncViewsClient, ViewsClient
+from .webhooks.client import AsyncWebhooksClient, WebhooksClient
+
+
+class LabelStudioBase:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : LabelStudioEnvironment
+        The environment to use for requests from the client. from .environment import LabelStudioEnvironment
+
+
+
+        Defaults to LabelStudioEnvironment.DEFAULT
+
+
+
+    api_key : typing.Optional[str]
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.Client]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    Examples
+    --------
+    from label_studio_sdk.client import LabelStudio
+
+    client = LabelStudio(
+        api_key="YOUR_API_KEY",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: LabelStudioEnvironment = LabelStudioEnvironment.DEFAULT,
+        api_key: typing.Optional[str] = os.getenv("LABEL_STUDIO_API_KEY"),
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.Client] = None
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        if api_key is None:
+            raise ApiError(
+                body="The client must be instantiated be either passing in api_key or setting LABEL_STUDIO_API_KEY"
+            )
+        self._client_wrapper = SyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            api_key=api_key,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.Client(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self.annotations = AnnotationsClient(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.projects = ProjectsClient(client_wrapper=self._client_wrapper)
+        self.ml = MlClient(client_wrapper=self._client_wrapper)
+        self.predictions = PredictionsClient(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)
+
+
+class AsyncLabelStudioBase:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : LabelStudioEnvironment
+        The environment to use for requests from the client. from .environment import LabelStudioEnvironment
+
+
+
+        Defaults to LabelStudioEnvironment.DEFAULT
+
+
+
+    api_key : typing.Optional[str]
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.AsyncClient]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    Examples
+    --------
+    from label_studio_sdk.client import AsyncLabelStudio
+
+    client = AsyncLabelStudio(
+        api_key="YOUR_API_KEY",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: LabelStudioEnvironment = LabelStudioEnvironment.DEFAULT,
+        api_key: typing.Optional[str] = os.getenv("LABEL_STUDIO_API_KEY"),
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        if api_key is None:
+            raise ApiError(
+                body="The client must be instantiated be either passing in api_key or setting LABEL_STUDIO_API_KEY"
+            )
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            api_key=api_key,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self.annotations = AsyncAnnotationsClient(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.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
+        self.ml = AsyncMlClient(client_wrapper=self._client_wrapper)
+        self.predictions = AsyncPredictionsClient(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)
+
+
+def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LabelStudioEnvironment) -> str:
+    if base_url is not None:
+        return base_url
+    elif environment is not None:
+        return environment.value
+    else:
+        raise Exception("Please pass in either base_url or environment to construct the client")

label_studio_sdk/client.py

@@ -1,183 +1,25 @@
-# This file was auto-generated by Fern from our API Definition.
+from .base_client import LabelStudioBase, AsyncLabelStudioBase
+from .tasks.client_ext import TasksClientExt, AsyncTasksClientExt
+from .projects.client_ext import ProjectsClientExt, AsyncProjectsClientExt
 
-import os
-import typing
 
-import httpx
+class LabelStudio(LabelStudioBase):
+    """"""
+    __doc__ += LabelStudioBase.__doc__
 
-from .actions.client import ActionsClient, AsyncActionsClient
-from .annotations.client import AnnotationsClient, AsyncAnnotationsClient
-from .core.api_error import ApiError
-from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from .environment import LabelStudioEnvironment
-from .export_storage.client import AsyncExportStorageClient, ExportStorageClient
-from .files.client import AsyncFilesClient, FilesClient
-from .import_storage.client import AsyncImportStorageClient, ImportStorageClient
-from .ml.client import AsyncMlClient, MlClient
-from .predictions.client import AsyncPredictionsClient, PredictionsClient
-from .projects.client import AsyncProjectsClient, ProjectsClient
-from .tasks.client import AsyncTasksClient, TasksClient
-from .users.client import AsyncUsersClient, UsersClient
-from .views.client import AsyncViewsClient, ViewsClient
-from .webhooks.client import AsyncWebhooksClient, WebhooksClient
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
 
+        self.tasks = TasksClientExt(client_wrapper=self._client_wrapper)
+        self.projects = ProjectsClientExt(client_wrapper=self._client_wrapper)
 
-class LabelStudio:
-    """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
 
-    Parameters
-    ----------
-    base_url : typing.Optional[str]
-        The base url to use for requests from the client.
+class AsyncLabelStudio(AsyncLabelStudioBase):
+    """"""
+    __doc__ += AsyncLabelStudioBase.__doc__
 
-    environment : LabelStudioEnvironment
-        The environment to use for requests from the client. from .environment import LabelStudioEnvironment
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
 
-
-
-        Defaults to LabelStudioEnvironment.DEFAULT
-
-
-
-    api_key : typing.Optional[str]
-    timeout : typing.Optional[float]
-        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
-
-    follow_redirects : typing.Optional[bool]
-        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
-
-    httpx_client : typing.Optional[httpx.Client]
-        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
-
-    Examples
-    --------
-    from label_studio_sdk.client import LabelStudio
-
-    client = LabelStudio(
-        api_key="YOUR_API_KEY",
-    )
-    """
-
-    def __init__(
-        self,
-        *,
-        base_url: typing.Optional[str] = None,
-        environment: LabelStudioEnvironment = LabelStudioEnvironment.DEFAULT,
-        api_key: typing.Optional[str] = os.getenv("LABEL_STUDIO_API_KEY"),
-        timeout: typing.Optional[float] = None,
-        follow_redirects: typing.Optional[bool] = True,
-        httpx_client: typing.Optional[httpx.Client] = None
-    ):
-        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
-        if api_key is None:
-            raise ApiError(
-                body="The client must be instantiated be either passing in api_key or setting LABEL_STUDIO_API_KEY"
-            )
-        self._client_wrapper = SyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
-            api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
-            if follow_redirects is not None
-            else httpx.Client(timeout=_defaulted_timeout),
-            timeout=_defaulted_timeout,
-        )
-        self.annotations = AnnotationsClient(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.projects = ProjectsClient(client_wrapper=self._client_wrapper)
-        self.ml = MlClient(client_wrapper=self._client_wrapper)
-        self.predictions = PredictionsClient(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)
-
-
-class AsyncLabelStudio:
-    """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
-
-    Parameters
-    ----------
-    base_url : typing.Optional[str]
-        The base url to use for requests from the client.
-
-    environment : LabelStudioEnvironment
-        The environment to use for requests from the client. from .environment import LabelStudioEnvironment
-
-
-
-        Defaults to LabelStudioEnvironment.DEFAULT
-
-
-
-    api_key : typing.Optional[str]
-    timeout : typing.Optional[float]
-        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
-
-    follow_redirects : typing.Optional[bool]
-        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
-
-    httpx_client : typing.Optional[httpx.AsyncClient]
-        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
-
-    Examples
-    --------
-    from label_studio_sdk.client import AsyncLabelStudio
-
-    client = AsyncLabelStudio(
-        api_key="YOUR_API_KEY",
-    )
-    """
-
-    def __init__(
-        self,
-        *,
-        base_url: typing.Optional[str] = None,
-        environment: LabelStudioEnvironment = LabelStudioEnvironment.DEFAULT,
-        api_key: typing.Optional[str] = os.getenv("LABEL_STUDIO_API_KEY"),
-        timeout: typing.Optional[float] = None,
-        follow_redirects: typing.Optional[bool] = True,
-        httpx_client: typing.Optional[httpx.AsyncClient] = None
-    ):
-        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
-        if api_key is None:
-            raise ApiError(
-                body="The client must be instantiated be either passing in api_key or setting LABEL_STUDIO_API_KEY"
-            )
-        self._client_wrapper = AsyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
-            api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
-            if follow_redirects is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout),
-            timeout=_defaulted_timeout,
-        )
-        self.annotations = AsyncAnnotationsClient(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.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
-        self.ml = AsyncMlClient(client_wrapper=self._client_wrapper)
-        self.predictions = AsyncPredictionsClient(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)
-
-
-def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LabelStudioEnvironment) -> str:
-    if base_url is not None:
-        return base_url
-    elif environment is not None:
-        return environment.value
-    else:
-        raise Exception("Please pass in either base_url or environment to construct the client")
+        self.tasks = AsyncTasksClientExt(client_wrapper=self._client_wrapper)
+        self.projects = AsyncProjectsClientExt(client_wrapper=self._client_wrapper)

label_studio_sdk/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "label-studio-sdk",
-            "X-Fern-SDK-Version": "0.0.93",
+            "X-Fern-SDK-Version": "0.0.121",
         }
         headers["Authorization"] = f"Token  {self.api_key}"
         return headers

label_studio_sdk/core/http_client.py

@@ -106,7 +106,11 @@ def maybe_filter_request_body(
     omit: typing.Optional[typing.Any],
 ) -> typing.Optional[typing.Any]:
     if data is None:
-        return None
+        return (
+            jsonable_encoder(request_options.get("additional_body_parameters", {}))
+            if request_options is not None
+            else None
+        )
     elif not isinstance(data, typing.Mapping):
         data_content = jsonable_encoder(data)
     else: