superb-ai-onprem 0.5.9__py3-none-any.whl → 0.5.11__py3-none-any.whl

spb_onprem/predictions/service.py

@@ -0,0 +1,225 @@
+from typing import Optional, Dict, Any, List, Tuple, Union
+
+from spb_onprem.base_service import BaseService
+from spb_onprem.base_types import Undefined, UndefinedType
+from spb_onprem.exceptions import BadParameterError
+from .queries import Queries
+from .entities import PredictionSet
+
+
+class PredictionService(BaseService):
+    """Service class for handling prediction set operations."""
+    
+    def get_prediction_sets(
+        self,
+        dataset_id: str,
+        filter: Union[UndefinedType, Dict[str, Any]] = Undefined,
+        cursor: Union[UndefinedType, str] = Undefined,
+        length: int = 50
+    ) -> Tuple[List[PredictionSet], Optional[str], int]:
+        """Get paginated list of prediction sets for a dataset.
+        
+        Args:
+            dataset_id (str): The dataset ID.
+            filter (Union[UndefinedType, Dict[str, Any]]): Filter for prediction sets.
+            cursor (Union[UndefinedType, str]): Cursor for pagination.
+            length (int): Number of items to retrieve per page.
+        
+        Returns:
+            Tuple[List[PredictionSet], Optional[str], int]: A tuple containing prediction sets, next cursor, and total count.
+        """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+
+        response = self.request_gql(
+            Queries.GET_PREDICTION_SETS,
+            Queries.GET_PREDICTION_SETS["variables"](
+                dataset_id=dataset_id,
+                filter=filter,
+                cursor=cursor,
+                length=length
+            )
+        )
+        prediction_sets_list = response.get("predictionSets", [])
+        return (
+            [PredictionSet.model_validate(ps) for ps in prediction_sets_list],
+            response.get("next"),
+            response.get("totalCount", 0)
+        )
+    
+    def get_prediction_set(
+        self,
+        dataset_id: str,
+        prediction_set_id: str
+    ) -> PredictionSet:
+        """Get detailed prediction set information including content IDs.
+        
+        Args:
+            dataset_id (str): The dataset ID.
+            prediction_set_id (str): The prediction set ID.
+        
+        Returns:
+            PredictionSet: The prediction set entity.
+        """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+        if prediction_set_id is None:
+            raise BadParameterError("prediction_set_id is required.")
+
+        response = self.request_gql(
+            Queries.GET_PREDICTION_SET,
+            Queries.GET_PREDICTION_SET["variables"](
+                dataset_id=dataset_id,
+                prediction_set_id=prediction_set_id
+            )
+        )
+        prediction_set_dict = response.get("predictionSet", {})
+        return PredictionSet.model_validate(prediction_set_dict)
+    
+    def delete_prediction_set(
+        self,
+        dataset_id: str,
+        prediction_set_id: str
+    ) -> bool:
+        """Delete a prediction set from the dataset.
+        
+        Args:
+            dataset_id (str): The dataset ID.
+            prediction_set_id (str): The prediction set ID to delete.
+        
+        Returns:
+            bool: True if deletion was successful.
+        """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+        if prediction_set_id is None:
+            raise BadParameterError("prediction_set_id is required.")
+
+        response = self.request_gql(
+            Queries.DELETE_PREDICTION_SET,
+            Queries.DELETE_PREDICTION_SET["variables"](
+                dataset_id=dataset_id,
+                prediction_set_id=prediction_set_id
+            )
+        )
+        return response.get("deletePredictionSet", False)
+    
+    def delete_prediction_from_data(
+        self,
+        dataset_id: str,
+        data_id: str,
+        prediction_set_id: str
+    ) -> bool:
+        """Delete predictions from a specific data item for a given prediction set.
+        
+        Args:
+            dataset_id (str): The dataset ID.
+            data_id (str): The data ID.
+            prediction_set_id (str): The prediction set ID.
+        
+        Returns:
+            bool: True if deletion was successful.
+        """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+        if data_id is None:
+            raise BadParameterError("data_id is required.")
+        if prediction_set_id is None:
+            raise BadParameterError("prediction_set_id is required.")
+
+        response = self.request_gql(
+            Queries.DELETE_PREDICTION_FROM_DATA,
+            Queries.DELETE_PREDICTION_FROM_DATA["variables"](
+                dataset_id=dataset_id,
+                data_id=data_id,
+                prediction_set_id=prediction_set_id
+            )
+        )
+        return response.get("deletePrediction") is not None
+
+    def create_prediction_set(
+        self,
+        dataset_id: str,
+        model_id: str,
+        name: str,
+        type: str,
+        description: Optional[str] = None,
+        annotations_count: Optional[int] = None,
+        data_count: Optional[int] = None,
+    ) -> Optional[str]:
+        """Create a new prediction set.
+        
+        Args:
+            dataset_id (str): The dataset ID.
+            model_id (str): The model ID.
+            name (str): Name of the prediction set.
+            type (str): Type of the prediction set.
+            description (Optional[str]): Description of the prediction set.
+            annotations_count (Optional[int]): Number of annotations.
+            data_count (Optional[int]): Number of data items.
+        
+        Returns:
+            Optional[str]: The created prediction set ID if successful, None otherwise.
+        """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+        if model_id is None:
+            raise BadParameterError("model_id is required.")
+        if name is None:
+            raise BadParameterError("name is required.")
+        if type is None:
+            raise BadParameterError("type is required.")
+
+        response = self.request_gql(
+            Queries.CREATE_PREDICTION_SET,
+            Queries.CREATE_PREDICTION_SET["variables"](
+                dataset_id=dataset_id,
+                model_id=model_id,
+                name=name,
+                type=type,
+                description=description,
+                annotations_count=annotations_count,
+                data_count=data_count
+            )
+        )
+        prediction_set = response.get("createPredictionSet")
+        return prediction_set.get("id") if prediction_set else None
+
+    def update_prediction_set_data_info(
+        self,
+        dataset_id: str,
+        id: str,
+        annotation_count: int,
+        data_count: int
+    ) -> Optional[str]:
+        """Update the data info of a prediction set.
+        
+        Args:
+            dataset_id (str): The dataset ID.
+            id (str): The prediction set ID to update.
+            annotation_count (int): Number of annotations.
+            data_count (int): Number of data items.
+        
+        Returns:
+            Optional[str]: The updated prediction set ID if successful, None otherwise.
+        """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+        if id is None:
+            raise BadParameterError("id is required.")
+        if annotation_count is None:
+            raise BadParameterError("annotation_count is required.")
+        if data_count is None:
+            raise BadParameterError("data_count is required.")
+
+        response = self.request_gql(
+            Queries.UPDATE_PREDICTION_SET_DATA_INFO,
+            Queries.UPDATE_PREDICTION_SET_DATA_INFO["variables"](
+                dataset_id=dataset_id,
+                id=id,
+                annotation_count=annotation_count,
+                data_count=data_count
+            )
+        )
+        prediction_set = response.get("updatePredictionSet")
+        return prediction_set.get("id") if prediction_set else None

{superb_ai_onprem-0.5.9.dist-info → superb_ai_onprem-0.5.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.9
+Version: 0.5.11
 Summary: Python SDK for Superb AI On-premise
 Home-page: https://github.com/Superb-AI-Suite/superb-ai-onprem-python
 Author: Superb AI

{superb_ai_onprem-0.5.9.dist-info → superb_ai_onprem-0.5.11.dist-info}/RECORD

@@ -1,14 +1,14 @@
-spb_onprem/__init__.py,sha256=lGZRdgpRdBFM97SawPJIC-a5uRzrd5jdb9i5yHDeveQ,2649
-spb_onprem/_version.py,sha256=TwLV4XX56BGk1ZSGUjssOTkLzWKjrtgZp7MxurdVBis,704
+spb_onprem/__init__.py,sha256=ugmA-tqXYbVB8OoRYW8SjxRitLTIwHvWQZ-U2KyKjdo,2885
+spb_onprem/_version.py,sha256=Sc1qEPGqm9lq7B7gOJTSc6KB6ru5lGdWZqZtirZLNXQ,706
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=dPfr3mGXYlqadOXycu6RBFX1HcZ1qzEsskLoOxERLOU,5737
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
-spb_onprem/entities.py,sha256=8C3DiGq04fryNjktk3zthum0Sa_sQb292jJEkZhd5zg,967
+spb_onprem/entities.py,sha256=LoywYC22ErWIWcJZQYg_foO_5c25pd77_11tFIKhsZk,1084
 spb_onprem/exceptions.py,sha256=jx5rTGsVZ5shOdbgQzk8GcSyMWFtb_3xhPq6Sylwc5o,478
 spb_onprem/searches.py,sha256=-ARAdCumAa1RViwhcinh-E7VHLs4xv5gk9dPyNze6-4,1448
 spb_onprem/activities/__init__.py,sha256=iTHUza8mtiBd_QEPH-0tzrOsd_VSwSQNMbzVWfebQ6c,175
 spb_onprem/activities/queries.py,sha256=L01PRDUQdbbIGoKP_M-CmZ1ORQzuq9NOr-rdWLW0ejE,6083
-spb_onprem/activities/service.py,sha256=HX_0IgdqFp0ProgQxfsf6rJk00uLDGnlqMOfRc3cKRM,11469
+spb_onprem/activities/service.py,sha256=jWfgAZGrULR8IETdPtUoNjwJDm-Zs8ZiC23eQEYBveA,11431
 spb_onprem/activities/entities/__init__.py,sha256=lJXPl4UqurNscZOYBcCMfRZBp27JKbPEbab5iXtL3ME,237
 spb_onprem/activities/entities/activity.py,sha256=xVJCrtU47ZG-ALRxsOauT0fIxc9EzqkTyw1xt5qVqJQ,1358
 spb_onprem/activities/entities/activity_history.py,sha256=6Iyt8mUjpWvC7KpCJj56Fn73MXuL_uIQKjz90o86cso,1037
@@ -22,20 +22,21 @@ spb_onprem/activities/params/start_activity.py,sha256=uBf78FYkjGkM5g8shIoMgz8qoC
 spb_onprem/activities/params/update_activity.py,sha256=QU3h2YkPFkqHEHXv6078V2PWEa272h60ST6FYPPZLWU,2615
 spb_onprem/activities/params/update_activity_history.py,sha256=ScWxOG7pZ-7WcTzvJleZIm-TsZVTRkwNrTCyNDyyVSc,1620
 spb_onprem/contents/__init__.py,sha256=9EfIMQxbJuUZVUqsTa3Ji-yDidFPQQB5gnJI4R01YWI,74
-spb_onprem/contents/queries.py,sha256=hk7JHfUBinIDp4xswYxmTMQwxsRbKWTxFeZQtUVl9K4,925
-spb_onprem/contents/service.py,sha256=a_Y_l18sULozh14obwAsEJG9x2BLSAgerqCNCf5TH2Q,5024
+spb_onprem/contents/queries.py,sha256=DuBlNTvgbo1gYfYIodScu2G5FngerZL6DzteBejV0-g,1188
+spb_onprem/contents/service.py,sha256=Waf3Kx3r4JXL7K8JUeg-LsJ5jEI3OqyoxgM3JWM78QI,5521
 spb_onprem/contents/entities/__init__.py,sha256=HsQ9J8UDxCx4xYTdMKQto7HCVUQilNozQCkIsceWezk,117
 spb_onprem/contents/entities/base_content.py,sha256=nM7NALpeRjtUKPv7eU0-hlqT1rPtD2mwB6Bvv_2Zc1E,346
 spb_onprem/contents/entities/content.py,sha256=YhTGHE9ykiOgFjvxbnLzSLg2665jPYrDATteB9PbGPE,534
-spb_onprem/contents/params/__init__.py,sha256=TYVs27fn47eTjnXGL6SQvkbWQqWaGwY6hJjnDLiLl-U,161
+spb_onprem/contents/params/__init__.py,sha256=YCkHhN01Dkx9ytCG35oSh2FmDDZcW-Dv2FhT1bFmzio,240
 spb_onprem/contents/params/create.py,sha256=JLMbNiIecOsVvVSYY6DfUpWjZxZKuO9-NSpuRlSIxS0,408
+spb_onprem/contents/params/delete_content.py,sha256=_M9d6FJpA47_pcM5YyHbMmTZv1F2wUwPryQTL9RxZwM,298
 spb_onprem/contents/params/get_download_url.py,sha256=3oaiqmjJRxw0U59SoTxoih1pcU-cuQvcU-g89lCcqtQ,98
 spb_onprem/data/__init__.py,sha256=5XKxNm2BlKufrX5uRQreUEzJ-nerTrozKpG1RJL5wt8,68
-spb_onprem/data/queries.py,sha256=B5vd_8Q8ziwtz44GozbZ-yEN5zKAgMY8wKex-YJJJSs,15607
-spb_onprem/data/service.py,sha256=zxkJhZhBftFyd2VBnflKjbYIlC1X2gp5eFXTVYql6Ds,25851
+spb_onprem/data/queries.py,sha256=mGnBBDuFNNVMpATid-uNpqVEzeu3R3u0wZnQDe8wR-o,17154
+spb_onprem/data/service.py,sha256=nUGw9Scn5FLFLURAXOid5sGrlfQLGH7r59uOCzTHVjk,25799
 spb_onprem/data/entities/__init__.py,sha256=ZQ2YWAHIan2ZLa1z5ozPSJwW3pqdlsJCxmAT3NuG89c,376
 spb_onprem/data/entities/annotation.py,sha256=pS97Nx3TWMBrb-iiBfRct4RXl7hPMy2IOzTIpXkmBJU,751
-spb_onprem/data/entities/data.py,sha256=9wD6fk6K-GDaCesWhxA4vW8IyrlG2Tn7xqR1ZB7VN6k,1293
+spb_onprem/data/entities/data.py,sha256=TOdnete4bfwROQ7C7oew7vV8bZyewQvFdjccyx4mec8,1217
 spb_onprem/data/entities/data_meta.py,sha256=zh5Cw_5F5c9f0Ixp-rHDe6P4PceSdIHHwTqcaMoW9Yk,1064
 spb_onprem/data/entities/data_slice.py,sha256=5ElQvOJE6Ivo9mWeNS3JlUV3EUtw8LAejPkfdLRq2aE,633
 spb_onprem/data/entities/frame.py,sha256=pnJ3rVzduvi_pZdJR_IJp2wRhexZspSDMUhY6CsY5Io,566
@@ -46,11 +47,11 @@ spb_onprem/data/enums/data_meta_type.py,sha256=9rd12-7C1udbbIGvnuGURKmd5-lndtW7o
 spb_onprem/data/enums/data_status.py,sha256=uV_yLtigSmht-tKFaLwvO4MpV5j4GzwuBRDILGrxMBA,292
 spb_onprem/data/enums/data_type.py,sha256=9ErNmOK-RIOrhyt25SE2WVPSx_nTxt3ncTG5UtW92iM,225
 spb_onprem/data/enums/scene_type.py,sha256=pTibjST4nJc3ylUrjQifYBdtLIcS2D0q71bdhyDjbIk,205
-spb_onprem/data/params/__init__.py,sha256=G83X-qgnWnuOMuMpsDKO0NZ9xWqQtmk6bdaT47XUvic,2386
+spb_onprem/data/params/__init__.py,sha256=R8x9zC7VZK4R5jm41wvBvCQ8qDCTMR1GdKyV13lnYaU,2598
 spb_onprem/data/params/change_data_labeler.py,sha256=jUG6bm0uheR1jVCy-RwFY97C9K93fdDdf3KDHJIelM4,570
 spb_onprem/data/params/change_data_reviewer.py,sha256=dFix3NfS2_8lpaoWxz6anLEJWMLdmRKM34ib8XwQiVQ,577
 spb_onprem/data/params/change_data_status.py,sha256=34gGtmTJT8wiDDpHOgQYvgP0-ziuYEBDFlpfY8SAZtI,593
-spb_onprem/data/params/create_data.py,sha256=8DOfbEchf9GSybiYY1cZoNzNYwILokoAkxXRFybJUAU,2038
+spb_onprem/data/params/create_data.py,sha256=aL6Bwd9talB2rnmdHbR9QkUx1Msni07-6aoBkhGoivM,1827
 spb_onprem/data/params/data.py,sha256=ZPJRjCbkjOCV73NxlGm5DxbSaNOAztVBRZRE8FFQ9mw,779
 spb_onprem/data/params/data_list.py,sha256=sxMBaK3n2YgbUQ9yNmcRkcngbClnkEJFAeeX07umS2M,8645
 spb_onprem/data/params/delete_annotation_version.py,sha256=R_jBVujor-09VS1Aa8yGP14R8nJ2Aa9OmmKezC4gz4c,457
@@ -58,6 +59,8 @@ spb_onprem/data/params/delete_data.py,sha256=HN4xXW9o4ZFOSjlzRZJ-F0wwtd_RzgI30gV
 spb_onprem/data/params/delete_prediction.py,sha256=X3sV2mrSUqg8FOlZYRtb0o1hAwFos48ydFK2Il8UCp4,479
 spb_onprem/data/params/delete_scene.py,sha256=pD8TtS5Jm2uQC3F2EQ6-nAbQuQbJLG2pAwFqmsgucyM,478
 spb_onprem/data/params/delete_slice_annotation_version.py,sha256=5M12bHKg9HJlh6hQ8XgLwFrfhTU7CiLM__gLZVBYrMg,595
+spb_onprem/data/params/get_data_detail.py,sha256=7_aWRcHHsX0U2chl3CWrVLRUfzeoPzK8JtjwUGQdv0Y,388
+spb_onprem/data/params/get_evaluation_value_list.py,sha256=zsnk5y2FrZXHbb8jIsxQ8BTTX9-HIAVXZe4B37mxvmk,1115
 spb_onprem/data/params/insert_annotation_version.py,sha256=bxw9LndaDJ0DLBo8U8kkGctHz1ToEgD_UxWWVpmJpks,987
 spb_onprem/data/params/insert_data_to_slice.py,sha256=_hLNC80wJbxGSouOpjwTY19mAf-itfRR4sxHnlLiiqk,478
 spb_onprem/data/params/insert_prediction.py,sha256=68q5ZSSq4arAsozy1i52phkUoBfUSzNVVxCuygRYDfI,621
@@ -66,22 +69,23 @@ spb_onprem/data/params/insert_slice_annotation_version.py,sha256=KinI05UaaQWcv-1
 spb_onprem/data/params/remove_data_from_slice.py,sha256=UjoQH0gTdm1nE1DafUmq3CP1nzHiCwDUnytQ6oBcZAA,478
 spb_onprem/data/params/remove_data_meta.py,sha256=nlj2Ln9CtdS4Si8TmETZlQoAU62nCTCLbZR2_yE3yB8,1847
 spb_onprem/data/params/update_annotation.py,sha256=zEmXYlBBYeC8A9eNyGidSw2XbgLd-pgvRZK5eui2plA,787
-spb_onprem/data/params/update_data.py,sha256=zlIqvUzMrEh-6eRUZ35lKmG2Ep6pYCPEl9dWvXz2rdM,1861
+spb_onprem/data/params/update_data.py,sha256=159Mx-5gmKQZKSAZlpTNiE4_EHVNwkEbjyeOZLs9SuM,1315
 spb_onprem/data/params/update_data_slice.py,sha256=xjxUs0y4koFtW1uW84ARKsepK8FwhGVjMkSMS6A1eFU,523
 spb_onprem/data/params/update_frames.py,sha256=AluCJDUJelq_yrXR3DjkwWF-82qQukvERkIE-qNjO8I,1226
 spb_onprem/data/params/update_scene.py,sha256=zMJFUHF3vz_hVSkkVEwHwdZBjWRbRxgTQFtsBowAg-k,930
 spb_onprem/data/params/update_slice_annotation.py,sha256=7eu2DFvznOv0HuvKiZTHDOTS2UijyTjbEiqLBbUncfE,607
 spb_onprem/data/params/update_slice_annotation_version.py,sha256=M2STViLDSQDZ0WNYLcenum-NHpKQXUOfGh1XE6An_7c,1318
-spb_onprem/data/params/upsert_data_meta.py,sha256=gVj-IY4V8DDbQQI9kEIwnwz7Pg4ObwUzyBhFoDNE-v4,1507
+spb_onprem/data/params/upsert_data_meta.py,sha256=faBI9UuH9onOxmEHc4C63nFJXqZ8YMnD2q-HKj_ZMxM,1033
 spb_onprem/datasets/__init__.py,sha256=Sjrb1tewB3CoODtHjRYOe-w2HpZi9UgbCyE2p8MzHfw,74
-spb_onprem/datasets/queries.py,sha256=PSj-3tOJfWUw8YXsv682LbcZoGbGHGgSQbWQGXjLePY,1912
-spb_onprem/datasets/service.py,sha256=ehc3IMYCN72pWZXvSs-j5jeYWQ6CxkU-0AA1-OwmtIk,3872
+spb_onprem/datasets/queries.py,sha256=uJKFD8awmmO6wiltX7fN8ovR0_yewarKkpHAwR7LRXY,2211
+spb_onprem/datasets/service.py,sha256=r5dim9mQcyb_gwBQ9oG4je0K8iHs1elvcWFLW9wvzYA,4439
 spb_onprem/datasets/entities/__init__.py,sha256=yx5tsyX4GLYrjqx6-ZEoceJ7jxQzIh15OgyFFCs7vow,59
 spb_onprem/datasets/entities/dataset.py,sha256=HO7EU5vwSBnILCI9xXhgZb36akIi6AC0ApamQlQczhU,538
-spb_onprem/datasets/params/__init__.py,sha256=7wmc4xQi3GOSeREelqfAveI0buYJWn7huX1Q9_jl_50,291
+spb_onprem/datasets/params/__init__.py,sha256=uqy7QijSgvOzrRY0xIg9GxRUo5Bquscs3dTiboLeTCM,370
 spb_onprem/datasets/params/create_dataset.py,sha256=YGhLvY4arthjZwKQ28HLv7ch0Gd2lJ-vyGHBZnMuy4E,719
 spb_onprem/datasets/params/dataset.py,sha256=WTOUl5M5cc6rtTwhLw_z31Cs209LkBq8Ja4LJGzrmGE,668
 spb_onprem/datasets/params/datasets.py,sha256=Hx4YlLxfb-Qwi4Y5AFl5pyyjupvuoVcCtxLPGjIV7UY,1580
+spb_onprem/datasets/params/delete_dataset.py,sha256=hzHV66_2TCJj1UK1czTNDzYm8fbeAthlvRvlVonRySI,323
 spb_onprem/datasets/params/update_dataset.py,sha256=1oaj2qB9hvnypl4-WtcTNCa4iSuEkJjEalq2JsTm5Ro,924
 spb_onprem/exports/__init__.py,sha256=l_eUjnrFJXs-vdOeMNiWYK-UdJkxArv7hxmKDG4Dfio,181
 spb_onprem/exports/queries.py,sha256=sFT6VX2UwAYyVNkiBteQ_JtKYnhMrt64ww_NuXhUhLM,4084
@@ -94,6 +98,26 @@ spb_onprem/exports/params/delete_export.py,sha256=EusUB86HNLtFYu4gIDJqZsODRETtTY
 spb_onprem/exports/params/export.py,sha256=0EP6nkQc6vFI-f8218Yq4NxfFcw8MQtHMNkYlGOXqo4,799
 spb_onprem/exports/params/exports.py,sha256=oOQo-2Cqsm3Th3s-0gNcVGjbOyZqB6ujPRLXipulB4Y,2417
 spb_onprem/exports/params/update_export.py,sha256=iOlZoHEN2iiY83hCZWdgTn-O9J8hcSUqd5K7_gEzvP4,3057
+spb_onprem/inferences/__init__.py,sha256=5SBXFgLWbTQV-lHoP7it4AbSAAAw_7IRm4FGRPPAeYw,68
+spb_onprem/inferences/service.py,sha256=ZkORci7Yt8Ruo0aY2KPsvynZG_uuitMKJO3FR22Nplc,1719
+spb_onprem/models/__init__.py,sha256=7Z6SiJqR0CVoK44Lx_J1WIPBO72Pe0mR4ZrWyb9tqAk,109
+spb_onprem/models/entities.py,sha256=TgsNdN5zJtgZc3FXmB2yv8rFjFwRzLaqhubdK40Zupw,205
+spb_onprem/models/queries.py,sha256=Pwymn30vaxXI89Yl_glkKSJKD5xfVj3PCLes2pUdXrQ,899
+spb_onprem/models/service.py,sha256=pvAHIJFw_0S8sBlUjDb2dUVT0whqR-ZiWfbWdw1cLuw,2499
+spb_onprem/models/params/__init__.py,sha256=iwZ8TWZEt5A4cMmidslkAdsBFspTxhs-WKTNzjAWPys,154
+spb_onprem/models/params/delete_model.py,sha256=onUSod8jKNeOH3zIziK82GCv2BzLF-fiKUjSu7bX87U,403
+spb_onprem/models/params/get_models.py,sha256=IZtCDQZd4lUumDj0NEr0q_PFefJvdHSj2hqmdymyu1E,734
+spb_onprem/predictions/__init__.py,sha256=8BDi5TI_OuCT5aHffosRQVO41v3K909mY48qTWsP6JU,135
+spb_onprem/predictions/entities.py,sha256=KABfAB4L4uP65lEkuZ5TzVS3oejCS-metmhj1YIbB2M,441
+spb_onprem/predictions/queries.py,sha256=oL1d8zI5be6kUTeyYV5GDefPW-r_npPP8DwFDwS-vkY,3477
+spb_onprem/predictions/service.py,sha256=uVECUKGUMgqOw4AxlyV5cnYPhBSF3KRbKlx08tP0DPo,8064
+spb_onprem/predictions/params/__init__.py,sha256=aU5-aQosfVekHqEBBlPImUraeFsmxSThNXd_fhhCsMc,648
+spb_onprem/predictions/params/create_prediction_set.py,sha256=DNp7oHt8X1revLBH-5f__xE5suJS3eCZrTjUlCTkD8A,1374
+spb_onprem/predictions/params/delete_prediction_from_data.py,sha256=wwkivSuFybhbDVmSOlH_08U6ze78yP_BPdiyKdBtqHE,575
+spb_onprem/predictions/params/delete_prediction_set.py,sha256=KMiuc8QIWCMLce_Xn5LZjy0PwJWxuLT9qRjKrYUg0LM,457
+spb_onprem/predictions/params/get_prediction_set.py,sha256=byoUu5CkBRfvv2ucIzSBL-8E_mKjCLMo20Lb7IIP4k4,435
+spb_onprem/predictions/params/get_prediction_sets.py,sha256=w3yLmrVWiTPGtXhERe6hwWd_rIv2M-EbwgeiAWCHPBc,761
+spb_onprem/predictions/params/update_prediction_set_data_info.py,sha256=oGl2pZPjsMCeCr3gjYlg8ms0ukl811ogwpSbDhUyxDk,767
 spb_onprem/slices/__init__.py,sha256=xgpNGYzqgwQ8C-Bgw9AZWMAgBW38UU-U4Wube8hkodI,69
 spb_onprem/slices/queries.py,sha256=13JcjACvCg7AbdGfUODDJeq77gGZlNjngFB-9UmgLZk,3080
 spb_onprem/slices/service.py,sha256=TnWGMVflesdaUXFKOBZTpdHSIKkQwj93Q3q9aGts5xI,5615
@@ -108,20 +132,28 @@ spb_onprem/slices/params/update_slice.py,sha256=kryOmCnRTQ_OU0qDKgugppLrpeUpuLwm
 spb_onprem/users/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spb_onprem/users/entities/__init__.py,sha256=X8HZsCTlQnuPszok3AwI-i7bsQi0Ehul5L_2jZaol5E,57
 spb_onprem/users/entities/auth.py,sha256=_KP-7yUErBxhJMm-dE3ObprPEG6e0JI2qNg6g8aK1qM,3371
-superb_ai_onprem-0.5.9.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.5.11.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
 tests/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/real_test.py,sha256=0gQHg7rIEuZndGZyNHMWSD5nUZPMsUGigfCjWFxMthQ,1786
 tests/activities/test_get_activity_history.py,sha256=gNLc4OQLgw9wqToRxV0y5REAHPudZK58h-plj5oczoE,5292
-tests/activities/test_params.py,sha256=L3mcnrN2c8dT0AVTjnGedu6LHF3eZ4sSDzz8eaMHJmg,2369
-tests/activities/test_service.py,sha256=AbWFwjaUjQRs5oDZ2ba0C23XJxCqsFNrxH1LamMtWmU,4698
+tests/activities/test_params.py,sha256=P_k03Ll-We3gyFTeACalErP1UzwS2gB6ayIerKMMLEU,2335
+tests/activities/test_service.py,sha256=8nijwzJlAFL9S0TqyuAvOJZzBvPPjxJbrj3EyziT_Ng,4407
+tests/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/data/test_data_service.py,sha256=JgcNozcwECKN9-LR80FXkH4RslXVWAGJh-FyODcmRCk,13702
+tests/datasets/__init__.py,sha256=nMsjhPXe9rU87gI_wasqCtz7SsTae2LzpUUnXXzS5zI,23
+tests/datasets/test_dataset_service.py,sha256=Sgw0HWZZJaPWVsbEq6w8tG4WlRBA98b1EoXLT8hRz_E,4881
 tests/exports/__init__.py,sha256=iBqr781TtnT1Tm5wLTCsdftpuRgQifIibxWrB_I7tiw,23
 tests/exports/real_test.py,sha256=KeHWvQP-vHCvFWvD4_7qFmreAX9o_Dd_YowaGFogZu8,3765
 tests/exports/test_entities.py,sha256=hG7G4kVkyHKT3mv4lvrzUqOW8ILeHiYj87QZjQcmg9E,8836
 tests/exports/test_integration.py,sha256=cCcEgwIIHyQRlc04EAXSKz7RcblQvhI2GBR3uVaOOq8,6201
 tests/exports/test_params.py,sha256=oRRa6nEru_FImlB3TrmFiBidz6ZstCx4rVaCK-EMGfQ,11070
-tests/exports/test_service.py,sha256=IDcKxrmByh00jk9142P2ThuGureMoijTHNdU0rERGG8,14628
-superb_ai_onprem-0.5.9.dist-info/METADATA,sha256=gxVpdiI4alSGo9IjsgtILhIAaMTJIl-Yk2iTHv9wGFs,5817
-superb_ai_onprem-0.5.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.5.9.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.5.9.dist-info/RECORD,,
+tests/exports/test_service.py,sha256=sAoFbLWGgRzE61QjnaZ1yymIAuDz7nrbsw42G3klTCQ,14660
+tests/models/__init__.py,sha256=Z2naiUgIjI-nHGsldwze1BqpV5JfDKTaBk-vg6_h6t4,21
+tests/models/test_model_service.py,sha256=6Wm9nmY7nbqImSu62d0q4VtNiCjKpMji6eSIgSi7lfA,7587
+tests/predictions/__init__.py,sha256=mFAXy0xOEXVYM7quYfIhwSih3kcEDdD4rLPq5G91Igg,26
+tests/predictions/test_prediction_service.py,sha256=9maX0JGxfUJ1QO_YEdS9OI5-o1tXlMWyeEbvZB8LB48,12679
+superb_ai_onprem-0.5.11.dist-info/METADATA,sha256=FK0vXshMvz4-5X8q-xxbJSLXKuvxki54YnRLwrKwEFk,5818
+superb_ai_onprem-0.5.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.5.11.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.5.11.dist-info/RECORD,,

tests/activities/test_params.py

@@ -20,7 +20,7 @@ class TestUpdateActivityHistoryParams:
         )
         
         # Then
-        assert params["activity_history_id"] == activity_history_id
+        assert params["id"] == activity_history_id
         assert params["status"] == status
         assert params["meta"] == meta
         assert "progress" not in params
@@ -37,7 +37,7 @@ class TestUpdateActivityHistoryParams:
         )
         
         # Then
-        assert params["activity_history_id"] == activity_history_id
+        assert params["id"] == activity_history_id
         assert params["progress"] == progress
         assert "status" not in params
         assert "meta" not in params

tests/activities/test_service.py

@@ -15,12 +15,10 @@ class TestActivityService:
     def test_create_activity(self, activity_service):
         # Given
         mock_response = {
-            "createJob": {
-                "id": "test_id",
-                "type": "test_type",
-                "name": "test_name",
-                "description": "test_description"
-            }
+            "id": "test_id",
+            "activity_type": "test_type",
+            "name": "test_name",
+            "description": "test_description"
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -34,29 +32,27 @@ class TestActivityService:
         # Then
         assert isinstance(activity, Activity)
         assert activity.id == "test_id"
-        assert activity.type == "test_type"
+        assert activity.activity_type == "test_type"
         assert activity.name == "test_name"
         assert activity.description == "test_description"
     
     def test_get_activities(self, activity_service):
         # Given
         mock_response = {
-            "jobs": {
-                "activities": [
-                    {
-                        "id": "test_id_1",
-                        "type": "test_type",
-                        "name": "test_name_1"
-                    },
-                    {
-                        "id": "test_id_2",
-                        "type": "test_type",
-                        "name": "test_name_2"
-                    }
-                ],
-                "next": "next_cursor",
-                "totalCount": 2
-            }
+            "jobs": [
+                {
+                    "id": "test_id_1",
+                    "activity_type": "test_type",
+                    "name": "test_name_1"
+                },
+                {
+                    "id": "test_id_2",
+                    "activity_type": "test_type",
+                    "name": "test_name_2"
+                }
+            ],
+            "next": "next_cursor",
+            "totalCount": 2
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -74,11 +70,9 @@ class TestActivityService:
     def test_start_activity(self, activity_service):
         # Given
         mock_response = {
-            "startJob": {
-                "id": "test_history_id",
-                "jobId": "test_activity_id",
-                "status": ActivityStatus.RUNNING.value
-            }
+            "id": "test_history_id",
+            "jobId": "test_activity_id",
+            "status": ActivityStatus.RUNNING.value
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -97,11 +91,9 @@ class TestActivityService:
     def test_update_activity_history_status(self, activity_service):
         # Given
         mock_response = {
-            "updateJobHistory": {
-                "id": "test_history_id",
-                "jobId": "test_activity_id",
-                "status": ActivityStatus.SUCCESS.value
-            }
+            "id": "test_history_id",
+            "jobId": "test_activity_id",
+            "status": ActivityStatus.SUCCESS.value
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -119,11 +111,9 @@ class TestActivityService:
     def test_update_activity_history_progress(self, activity_service):
         # Given
         mock_response = {
-            "updateJobHistory": {
-                "id": "test_history_id",
-                "jobId": "test_activity_id",
-                "progress": {"current": 50, "total": 100}
-            }
+            "id": "test_history_id",
+            "jobId": "test_activity_id",
+            "progress": {"current": 50, "total": 100}
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)