label-studio-sdk 1.0.5__py3-none-any.whl → 1.0.8__py3-none-any.whl

label_studio_sdk/annotations/client.py

@@ -9,6 +9,7 @@ from ..core.jsonable_encoder import jsonable_encoder
 from ..core.pydantic_utilities import pydantic_v1
 from ..core.request_options import RequestOptions
 from ..types.annotation import Annotation
+from .types.annotations_create_bulk_response_item import AnnotationsCreateBulkResponseItem
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -380,6 +381,60 @@ class AnnotationsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def create_bulk(
+        self,
+        *,
+        tasks: typing.Optional[typing.Sequence[int]] = OMIT,
+        lead_time: typing.Optional[float] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        result: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[AnnotationsCreateBulkResponseItem]:
+        """
+        Create multiple annotations for specific tasks in a bulk operation.
+
+        Parameters
+        ----------
+        tasks : typing.Optional[typing.Sequence[int]]
+
+        lead_time : typing.Optional[float]
+
+        project : typing.Optional[int]
+
+        result : typing.Optional[typing.Dict[str, typing.Any]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[AnnotationsCreateBulkResponseItem]
+            Annotations created successfully
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.annotations.create_bulk()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/annotations/bulk",
+            method="POST",
+            json={"tasks": tasks, "lead_time": lead_time, "project": project, "result": result},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[AnnotationsCreateBulkResponseItem], _response.json())  # type: ignore
+            _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 AsyncAnnotationsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -748,3 +803,57 @@ class AsyncAnnotationsClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create_bulk(
+        self,
+        *,
+        tasks: typing.Optional[typing.Sequence[int]] = OMIT,
+        lead_time: typing.Optional[float] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        result: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[AnnotationsCreateBulkResponseItem]:
+        """
+        Create multiple annotations for specific tasks in a bulk operation.
+
+        Parameters
+        ----------
+        tasks : typing.Optional[typing.Sequence[int]]
+
+        lead_time : typing.Optional[float]
+
+        project : typing.Optional[int]
+
+        result : typing.Optional[typing.Dict[str, typing.Any]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[AnnotationsCreateBulkResponseItem]
+            Annotations created successfully
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.annotations.create_bulk()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/annotations/bulk",
+            method="POST",
+            json={"tasks": tasks, "lead_time": lead_time, "project": project, "result": result},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[AnnotationsCreateBulkResponseItem], _response.json())  # type: ignore
+            _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/annotations/types/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .annotations_create_bulk_response_item import AnnotationsCreateBulkResponseItem
+
+__all__ = ["AnnotationsCreateBulkResponseItem"]

label_studio_sdk/annotations/types/annotations_create_bulk_response_item.py

@@ -0,0 +1,29 @@
+# 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 AnnotationsCreateBulkResponseItem(pydantic_v1.BaseModel):
+    id: typing.Optional[int] = None
+
+    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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

label_studio_sdk/base_client.py

@@ -7,6 +7,7 @@ import httpx
 
 from .actions.client import ActionsClient, AsyncActionsClient
 from .annotations.client import AnnotationsClient, AsyncAnnotationsClient
+from .comments.client import AsyncCommentsClient, CommentsClient
 from .core.api_error import ApiError
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import LabelStudioEnvironment
@@ -14,8 +15,10 @@ 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 .model_providers.client import AsyncModelProvidersClient, ModelProvidersClient
 from .predictions.client import AsyncPredictionsClient, PredictionsClient
 from .projects.client import AsyncProjectsClient, ProjectsClient
+from .prompts.client import AsyncPromptsClient, PromptsClient
 from .tasks.client import AsyncTasksClient, TasksClient
 from .users.client import AsyncUsersClient, UsersClient
 from .views.client import AsyncViewsClient, ViewsClient
@@ -97,6 +100,9 @@ class LabelStudioBase:
         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.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.workspaces = WorkspacesClient(client_wrapper=self._client_wrapper)
 
 
@@ -174,6 +180,9 @@ class AsyncLabelStudioBase:
         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.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.workspaces = AsyncWorkspacesClient(client_wrapper=self._client_wrapper)
 
 

label_studio_sdk/comments/__init__.py

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