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

spb_onprem/__init__.py

@@ -31,6 +31,7 @@ from .entities import (
     Export,
     Content,
     Frame,
+    Model,
 
     # Enums
     DataType,

spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.11'
-__version_tuple__ = version_tuple = (0, 5, 11)
+__version__ = version = '0.5.12'
+__version_tuple__ = version_tuple = (0, 5, 12)
 
 __commit_id__ = commit_id = None

spb_onprem/data/queries.py

@@ -624,7 +624,7 @@ class Queries():
     GET_DETAIL = {
         "name": "getDataDetail",
         "query": f'''
-            query GetDataDetail($datasetId: String!, $id: String!) {{
+            query GetDataDetail($datasetId: ID!, $id: ID!) {{
                 data(datasetId: $datasetId, id: $id) {{
                     {Schemas.DATA_DETAIL}
                 }}

spb_onprem/data/service.py

@@ -269,7 +269,7 @@ class DataService(BaseService):
         self,
         dataset_id: str,
         data_id: str,
-    ):
+    ) -> bool:
         """Delete a data.
 
         Args:
@@ -277,14 +277,21 @@ class DataService(BaseService):
             data_id (str): The data id.
 
         Returns:
-            Data: The deleted data.
+            bool: True if deletion was successful.
+        
+        Raises:
+            BadParameterError: If required parameters are missing.
         """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+        if data_id is None:
+            raise BadParameterError("data_id is required.")
+
         response = self.request_gql(
             Queries.DELETE,
             Queries.DELETE["variables"](dataset_id=dataset_id, data_id=data_id)
         )
-        data = Data.model_validate(response)
-        return data
+        return response.get("deleteData", False)
 
     def insert_prediction(
         self,

spb_onprem/datasets/queries.py

@@ -82,7 +82,7 @@ class Queries():
     DELETE_DATASET = {
         "name": "deleteDataset",
         "query": '''
-            mutation DeleteDataset($dataset_id: String!) {
+            mutation DeleteDataset($dataset_id: ID!) {
                 deleteDataset(datasetId: $dataset_id)
             }
         ''',

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.11
+Version: 0.5.12
 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.11.dist-info → superb_ai_onprem-0.5.12.dist-info}/RECORD

@@ -1,5 +1,5 @@
-spb_onprem/__init__.py,sha256=ugmA-tqXYbVB8OoRYW8SjxRitLTIwHvWQZ-U2KyKjdo,2885
-spb_onprem/_version.py,sha256=Sc1qEPGqm9lq7B7gOJTSc6KB6ru5lGdWZqZtirZLNXQ,706
+spb_onprem/__init__.py,sha256=ZldbV-bTUi0Vb3fj_PXSZZEDRsBYn-5UBfWb-uVff6A,2896
+spb_onprem/_version.py,sha256=WXjz7JY5pHmakepyBAnMz7K6J4B5aS4Bqj2ku8SvDoo,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
@@ -32,8 +32,8 @@ spb_onprem/contents/params/create.py,sha256=JLMbNiIecOsVvVSYY6DfUpWjZxZKuO9-NSpu
 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=mGnBBDuFNNVMpATid-uNpqVEzeu3R3u0wZnQDe8wR-o,17154
-spb_onprem/data/service.py,sha256=nUGw9Scn5FLFLURAXOid5sGrlfQLGH7r59uOCzTHVjk,25799
+spb_onprem/data/queries.py,sha256=qQSSbMsPDSR_q_teEEGHrqEsvLtk3ecbS9xvoxyNPGQ,17146
+spb_onprem/data/service.py,sha256=1YkAvtpqVweUOba79zXk13nx3Q8E3cUrUJm-K8R54AE,26081
 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=TOdnete4bfwROQ7C7oew7vV8bZyewQvFdjccyx4mec8,1217
@@ -77,7 +77,7 @@ spb_onprem/data/params/update_slice_annotation.py,sha256=7eu2DFvznOv0HuvKiZTHDOT
 spb_onprem/data/params/update_slice_annotation_version.py,sha256=M2STViLDSQDZ0WNYLcenum-NHpKQXUOfGh1XE6An_7c,1318
 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=uJKFD8awmmO6wiltX7fN8ovR0_yewarKkpHAwR7LRXY,2211
+spb_onprem/datasets/queries.py,sha256=-qsAaD19ESZ-qsbBEao19FM4b0LgLWYHQQXlpP5Lbos,2207
 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
@@ -132,7 +132,7 @@ 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.11.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.5.12.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
@@ -140,9 +140,9 @@ tests/activities/test_get_activity_history.py,sha256=gNLc4OQLgw9wqToRxV0y5REAHPu
 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/data/test_data_service.py,sha256=Zsxj17-NwjZjxwL2PME29D-jMiPspD8m4h8qRajbz68,16901
 tests/datasets/__init__.py,sha256=nMsjhPXe9rU87gI_wasqCtz7SsTae2LzpUUnXXzS5zI,23
-tests/datasets/test_dataset_service.py,sha256=Sgw0HWZZJaPWVsbEq6w8tG4WlRBA98b1EoXLT8hRz_E,4881
+tests/datasets/test_dataset_service.py,sha256=0BLmg_jslfzA_plm9dElws6eVfeOEQKuOq-nAmKNnqc,4899
 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
@@ -153,7 +153,7 @@ 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,,
+superb_ai_onprem-0.5.12.dist-info/METADATA,sha256=upjQk-uEG-jYRUQ1dip1LGz0WrKmnmreaJp4JWmf9aE,5818
+superb_ai_onprem-0.5.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.5.12.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.5.12.dist-info/RECORD,,

tests/data/test_data_service.py

@@ -126,7 +126,7 @@ class TestDataService:
 
         # Assert
         assert query_structure["name"] == "getDataDetail"
-        assert "query GetDataDetail($datasetId: String!, $id: String!)" in query_structure["query"]
+        assert "query GetDataDetail($datasetId: ID!, $id: ID!)" in query_structure["query"]
         assert "data(datasetId: $datasetId, id: $id)" in query_structure["query"]
         assert callable(query_structure["variables"])
 
@@ -409,4 +409,96 @@ class TestDataService:
         )
 
         # Assert
-        assert result == 0
+        assert result == 0
+
+    def test_delete_data_query_structure(self):
+        """Test DELETE query structure and parameters."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = "data-456"
+        
+        # Act
+        query = Queries.DELETE["query"]
+        variables = Queries.DELETE["variables"](dataset_id=dataset_id, data_id=data_id)
+        
+        # Assert
+        assert "mutation DeleteData(" in query
+        assert "$dataset_id: ID!" in query
+        assert "$data_id: ID!" in query
+        assert "$data_id: ID!," not in query  # Ensure no trailing comma
+        assert "deleteData(" in query
+        assert "datasetId: $dataset_id" in query
+        assert "id: $data_id" in query
+        assert "id: $data_id," not in query  # Ensure no trailing comma
+        assert variables == {
+            "dataset_id": dataset_id,
+            "id": data_id
+        }
+
+    def test_delete_data_success(self):
+        """Test successful data deletion."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = "data-456"
+        mock_response = {"deleteData": True}
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.delete_data(dataset_id=dataset_id, data_id=data_id)
+
+        # Assert
+        assert result is True
+        self.data_service.request_gql.assert_called_once_with(
+            Queries.DELETE,
+            Queries.DELETE["variables"](dataset_id=dataset_id, data_id=data_id)
+        )
+
+    def test_delete_data_failure(self):
+        """Test data deletion failure."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = "nonexistent-data"
+        mock_response = {"deleteData": False}
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.delete_data(dataset_id=dataset_id, data_id=data_id)
+
+        # Assert
+        assert result is False
+        self.data_service.request_gql.assert_called_once_with(
+            Queries.DELETE,
+            Queries.DELETE["variables"](dataset_id=dataset_id, data_id=data_id)
+        )
+
+    def test_delete_data_missing_response(self):
+        """Test data deletion with missing response field."""
+        # Arrange
+        dataset_id = "dataset-123"
+        data_id = "data-456"
+        mock_response = {}
+        self.data_service.request_gql.return_value = mock_response
+
+        # Act
+        result = self.data_service.delete_data(dataset_id=dataset_id, data_id=data_id)
+
+        # Assert
+        assert result is False
+
+    def test_delete_data_missing_dataset_id(self):
+        """Test delete_data with missing dataset_id."""
+        # Arrange
+        data_id = "data-456"
+
+        # Act & Assert
+        with pytest.raises(BadParameterError, match="dataset_id is required"):
+            self.data_service.delete_data(dataset_id=None, data_id=data_id)
+
+    def test_delete_data_missing_data_id(self):
+        """Test delete_data with missing data_id."""
+        # Arrange
+        dataset_id = "dataset-123"
+
+        # Act & Assert
+        with pytest.raises(BadParameterError, match="data_id is required"):
+            self.data_service.delete_data(dataset_id=dataset_id, data_id=None)

tests/datasets/test_dataset_service.py

@@ -97,7 +97,7 @@ class TestDatasetService:
 
         # Check query contains expected mutation
         assert query["name"] == "deleteDataset"
-        assert "mutation DeleteDataset" in query["query"]
+        assert "mutation DeleteDataset($dataset_id: ID!)" in query["query"]
         assert "deleteDataset(datasetId: $dataset_id)" in query["query"]
 
         # Check variables function is called correctly