superb-ai-onprem 0.5.11__tar.gz → 0.5.13__tar.gz

{superb_ai_onprem-0.5.11 → superb_ai_onprem-0.5.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.11
+Version: 0.5.13
 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 → superb_ai_onprem-0.5.13}/spb_onprem/__init__.py

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

{superb_ai_onprem-0.5.11 → superb_ai_onprem-0.5.13}/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.13'
+__version_tuple__ = version_tuple = (0, 5, 13)
 
-__commit_id__ = commit_id = 'gf23f2712d'
+__commit_id__ = commit_id = 'g2ecac8c69'

{superb_ai_onprem-0.5.11 → superb_ai_onprem-0.5.13}/spb_onprem/data/queries.py

@@ -311,9 +311,7 @@ class Queries():
                 deleteData(
                     datasetId: $dataset_id,
                     id: $data_id,
-                ) {{
-                    {Schemas.DATA}
-                }}
+                )
             }}
         ''',
         "variables": delete_data_params,
@@ -624,7 +622,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}
                 }}

{superb_ai_onprem-0.5.11 → superb_ai_onprem-0.5.13}/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,

{superb_ai_onprem-0.5.11 → superb_ai_onprem-0.5.13}/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 → superb_ai_onprem-0.5.13}/superb_ai_onprem.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.11
+Version: 0.5.13
 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 → superb_ai_onprem-0.5.13}/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)

{superb_ai_onprem-0.5.11 → superb_ai_onprem-0.5.13}/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