label-studio-sdk 2.0.6__py3-none-any.whl → 2.0.8__py3-none-any.whl

label_studio_sdk/projects/__init__.py

@@ -6,6 +6,7 @@ from .types import (
     PatchedLseProjectUpdateRequestSampling,
     PatchedLseProjectUpdateRequestSkipQueue,
     ProjectsDuplicateResponse,
+    ProjectsImportPredictionsResponse,
     ProjectsImportTasksResponse,
     ProjectsListRequestFilter,
 )
@@ -28,13 +29,27 @@ from .assignments import (
 )
 from .exports import ExportsConvertResponse
 from .stats import (
+    StatsAgreementAnnotatorResponse,
+    StatsDataFiltersResponse,
+    StatsDataFiltersResponseUserFilters,
+    StatsDataFiltersResponseUserFiltersStatsItem,
+    StatsFinishedTasksResponse,
     StatsIaaResponse,
     StatsIaaResponseCommonTasks,
     StatsIaaResponseIaa,
     StatsIaaResponseStd,
+    StatsLeadTimeResponse,
+    StatsLeadTimeResponseLeadTimeStatsItem,
     StatsTotalAgreementResponse,
     StatsTotalAgreementResponseOne,
     StatsTotalAgreementResponseZero,
+    StatsUserGroundTruthAgreementResponse,
+    StatsUserGroundTruthAgreementResponseAgreement,
+    StatsUserPredictionAgreementResponse,
+    StatsUserPredictionAgreementResponseAveragePredictionAgreementPerUser,
+    StatsUserReviewScoreResponse,
+    StatsUserReviewScoreResponsePerformanceScore,
+    StatsUserReviewScoreResponseReviewScore,
 )
 
 __all__ = [
@@ -58,15 +73,30 @@ __all__ = [
     "PatchedLseProjectUpdateRequestSampling",
     "PatchedLseProjectUpdateRequestSkipQueue",
     "ProjectsDuplicateResponse",
+    "ProjectsImportPredictionsResponse",
     "ProjectsImportTasksResponse",
     "ProjectsListRequestFilter",
+    "StatsAgreementAnnotatorResponse",
+    "StatsDataFiltersResponse",
+    "StatsDataFiltersResponseUserFilters",
+    "StatsDataFiltersResponseUserFiltersStatsItem",
+    "StatsFinishedTasksResponse",
     "StatsIaaResponse",
     "StatsIaaResponseCommonTasks",
     "StatsIaaResponseIaa",
     "StatsIaaResponseStd",
+    "StatsLeadTimeResponse",
+    "StatsLeadTimeResponseLeadTimeStatsItem",
     "StatsTotalAgreementResponse",
     "StatsTotalAgreementResponseOne",
     "StatsTotalAgreementResponseZero",
+    "StatsUserGroundTruthAgreementResponse",
+    "StatsUserGroundTruthAgreementResponseAgreement",
+    "StatsUserPredictionAgreementResponse",
+    "StatsUserPredictionAgreementResponseAveragePredictionAgreementPerUser",
+    "StatsUserReviewScoreResponse",
+    "StatsUserReviewScoreResponsePerformanceScore",
+    "StatsUserReviewScoreResponseReviewScore",
     "assignments",
     "exports",
     "members",

label_studio_sdk/projects/client.py

@@ -22,6 +22,7 @@ from .types.lse_project_create_request_sampling import LseProjectCreateRequestSa
 from .types.lse_project_create_request_skip_queue import LseProjectCreateRequestSkipQueue
 from ..types.lse_project_create import LseProjectCreate
 from ..core.serialization import convert_and_respect_annotation_metadata
+from ..types.paginated_lse_project_counts_list import PaginatedLseProjectCountsList
 from ..types.project import Project
 from ..core.jsonable_encoder import jsonable_encoder
 from ..types.assignment_settings_request import AssignmentSettingsRequest
@@ -34,6 +35,8 @@ from .types.projects_duplicate_response import ProjectsDuplicateResponse
 from ..types.import_api_request import ImportApiRequest
 from .types.projects_import_tasks_response import ProjectsImportTasksResponse
 from ..errors.bad_request_error import BadRequestError
+from ..types.prediction_request import PredictionRequest
+from .types.projects_import_predictions_response import ProjectsImportPredictionsResponse
 from ..types.project_label_config import ProjectLabelConfig
 from ..core.client_wrapper import AsyncClientWrapper
 from .exports.client import AsyncExportsClient
@@ -366,6 +369,99 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_counts(
+        self,
+        *,
+        filter: typing.Optional[str] = None,
+        ids: typing.Optional[str] = None,
+        include: typing.Optional[str] = None,
+        ordering: typing.Optional[str] = None,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        search: typing.Optional[str] = None,
+        title: typing.Optional[str] = None,
+        workspaces: typing.Optional[float] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedLseProjectCountsList:
+        """
+        Returns a list of projects with their counts. For example, task_number which is the total task number in project
+
+        Parameters
+        ----------
+        filter : typing.Optional[str]
+            Project filter setting. One of 'all', 'pinned_only', 'exclude_pinned'.
+
+        ids : typing.Optional[str]
+            Filter id by in list
+
+        include : typing.Optional[str]
+            Comma-separated list of fields to include
+
+        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]
+            Search term for project title and description
+
+        title : typing.Optional[str]
+            Filter title by contains (case-insensitive)
+
+        workspaces : typing.Optional[float]
+            Filter workspaces by exact match
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedLseProjectCountsList
+
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.projects.list_counts()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/projects/counts/",
+            method="GET",
+            params={
+                "filter": filter,
+                "ids": ids,
+                "include": include,
+                "ordering": ordering,
+                "page": page,
+                "page_size": page_size,
+                "search": search,
+                "title": title,
+                "workspaces": workspaces,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedLseProjectCountsList,
+                    construct_type(
+                        type_=PaginatedLseProjectCountsList,  # 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, *, request_options: typing.Optional[RequestOptions] = None) -> Project:
         """
         Retrieve information about a project by project ID.
@@ -924,6 +1020,81 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def import_predictions(
+        self,
+        id: int,
+        *,
+        request: typing.Sequence[PredictionRequest],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ProjectsImportPredictionsResponse:
+        """
+        Import model predictions for tasks in the specified project.
+
+        Parameters
+        ----------
+        id : int
+            A unique integer value identifying this project.
+
+        request : typing.Sequence[PredictionRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ProjectsImportPredictionsResponse
+            Predictions successfully imported
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio, PredictionRequest
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.projects.import_predictions(
+            id=1,
+            request=[
+                PredictionRequest(
+                    result=[{"key": "value"}],
+                    task=1,
+                )
+            ],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/projects/{jsonable_encoder(id)}/import/predictions",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=typing.Sequence[PredictionRequest], direction="write"
+            ),
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    ProjectsImportPredictionsResponse,
+                    construct_type(
+                        type_=ProjectsImportPredictionsResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        construct_type(
+                            type_=typing.Optional[typing.Any],  # 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 validate_label_config(
         self, id: int, *, label_config: str, request_options: typing.Optional[RequestOptions] = None
     ) -> ProjectLabelConfig:
@@ -1316,6 +1487,107 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def list_counts(
+        self,
+        *,
+        filter: typing.Optional[str] = None,
+        ids: typing.Optional[str] = None,
+        include: typing.Optional[str] = None,
+        ordering: typing.Optional[str] = None,
+        page: typing.Optional[int] = None,
+        page_size: typing.Optional[int] = None,
+        search: typing.Optional[str] = None,
+        title: typing.Optional[str] = None,
+        workspaces: typing.Optional[float] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedLseProjectCountsList:
+        """
+        Returns a list of projects with their counts. For example, task_number which is the total task number in project
+
+        Parameters
+        ----------
+        filter : typing.Optional[str]
+            Project filter setting. One of 'all', 'pinned_only', 'exclude_pinned'.
+
+        ids : typing.Optional[str]
+            Filter id by in list
+
+        include : typing.Optional[str]
+            Comma-separated list of fields to include
+
+        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]
+            Search term for project title and description
+
+        title : typing.Optional[str]
+            Filter title by contains (case-insensitive)
+
+        workspaces : typing.Optional[float]
+            Filter workspaces by exact match
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedLseProjectCountsList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.projects.list_counts()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/projects/counts/",
+            method="GET",
+            params={
+                "filter": filter,
+                "ids": ids,
+                "include": include,
+                "ordering": ordering,
+                "page": page,
+                "page_size": page_size,
+                "search": search,
+                "title": title,
+                "workspaces": workspaces,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedLseProjectCountsList,
+                    construct_type(
+                        type_=PaginatedLseProjectCountsList,  # 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, *, request_options: typing.Optional[RequestOptions] = None) -> Project:
         """
         Retrieve information about a project by project ID.
@@ -1914,6 +2186,89 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def import_predictions(
+        self,
+        id: int,
+        *,
+        request: typing.Sequence[PredictionRequest],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ProjectsImportPredictionsResponse:
+        """
+        Import model predictions for tasks in the specified project.
+
+        Parameters
+        ----------
+        id : int
+            A unique integer value identifying this project.
+
+        request : typing.Sequence[PredictionRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ProjectsImportPredictionsResponse
+            Predictions successfully imported
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio, PredictionRequest
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.projects.import_predictions(
+                id=1,
+                request=[
+                    PredictionRequest(
+                        result=[{"key": "value"}],
+                        task=1,
+                    )
+                ],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/projects/{jsonable_encoder(id)}/import/predictions",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=typing.Sequence[PredictionRequest], direction="write"
+            ),
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    ProjectsImportPredictionsResponse,
+                    construct_type(
+                        type_=ProjectsImportPredictionsResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        construct_type(
+                            type_=typing.Optional[typing.Any],  # 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 validate_label_config(
         self, id: int, *, label_config: str, request_options: typing.Optional[RequestOptions] = None
     ) -> ProjectLabelConfig:

label_studio_sdk/projects/stats/__init__.py

@@ -1,21 +1,49 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    StatsAgreementAnnotatorResponse,
+    StatsDataFiltersResponse,
+    StatsDataFiltersResponseUserFilters,
+    StatsDataFiltersResponseUserFiltersStatsItem,
+    StatsFinishedTasksResponse,
     StatsIaaResponse,
     StatsIaaResponseCommonTasks,
     StatsIaaResponseIaa,
     StatsIaaResponseStd,
+    StatsLeadTimeResponse,
+    StatsLeadTimeResponseLeadTimeStatsItem,
     StatsTotalAgreementResponse,
     StatsTotalAgreementResponseOne,
     StatsTotalAgreementResponseZero,
+    StatsUserGroundTruthAgreementResponse,
+    StatsUserGroundTruthAgreementResponseAgreement,
+    StatsUserPredictionAgreementResponse,
+    StatsUserPredictionAgreementResponseAveragePredictionAgreementPerUser,
+    StatsUserReviewScoreResponse,
+    StatsUserReviewScoreResponsePerformanceScore,
+    StatsUserReviewScoreResponseReviewScore,
 )
 
 __all__ = [
+    "StatsAgreementAnnotatorResponse",
+    "StatsDataFiltersResponse",
+    "StatsDataFiltersResponseUserFilters",
+    "StatsDataFiltersResponseUserFiltersStatsItem",
+    "StatsFinishedTasksResponse",
     "StatsIaaResponse",
     "StatsIaaResponseCommonTasks",
     "StatsIaaResponseIaa",
     "StatsIaaResponseStd",
+    "StatsLeadTimeResponse",
+    "StatsLeadTimeResponseLeadTimeStatsItem",
     "StatsTotalAgreementResponse",
     "StatsTotalAgreementResponseOne",
     "StatsTotalAgreementResponseZero",
+    "StatsUserGroundTruthAgreementResponse",
+    "StatsUserGroundTruthAgreementResponseAgreement",
+    "StatsUserPredictionAgreementResponse",
+    "StatsUserPredictionAgreementResponseAveragePredictionAgreementPerUser",
+    "StatsUserReviewScoreResponse",
+    "StatsUserReviewScoreResponsePerformanceScore",
+    "StatsUserReviewScoreResponseReviewScore",
 ]