superb-ai-onprem 0.5.18__tar.gz → 0.5.20__tar.gz

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/PKG-INFO

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

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.18'
-__version_tuple__ = version_tuple = (0, 5, 18)
+__version__ = version = '0.5.20'
+__version_tuple__ = version_tuple = (0, 5, 20)
 
-__commit_id__ = commit_id = 'g1b65e6e33'
+__commit_id__ = commit_id = 'gc6ce00d2b'

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/spb_onprem/contents/service.py

@@ -194,4 +194,4 @@ class ContentService(BaseService):
             query=Queries.DELETE,
             variables=Queries.DELETE["variables"](content_id)
         )
-        return response.get("deleteContent", False)
+        return response

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/spb_onprem/data/service.py

@@ -291,7 +291,7 @@ class DataService(BaseService):
             Queries.DELETE,
             Queries.DELETE["variables"](dataset_id=dataset_id, data_id=data_id)
         )
-        return response.get("deleteData", False)
+        return response
 
     def insert_prediction(
         self,

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/spb_onprem/datasets/service.py

@@ -44,7 +44,7 @@ class DatasetService(BaseService):
         return (
             datasets,
             response.get("next", None),
-            response.get("totalCount", False)
+            response.get("totalCount", 0)
         )
 
     def get_dataset(
@@ -147,4 +147,4 @@ class DatasetService(BaseService):
             Queries.DELETE_DATASET,
             Queries.DELETE_DATASET["variables"](dataset_id=dataset_id)
         )
-        return response.get("deleteDataset", False)
+        return response

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/spb_onprem/models/service.py

@@ -73,4 +73,4 @@ class ModelService(BaseService):
                 model_id=model_id
             )
         )
-        return response.get("deleteModel", False)
+        return response

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/spb_onprem/predictions/params/delete_prediction_from_data.py

@@ -16,5 +16,5 @@ def delete_prediction_from_data_params(
     return {
         "dataset_id": dataset_id,
         "data_id": data_id,
-        "prediction_set_id": prediction_set_id
+        "set_id": prediction_set_id
     }

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/spb_onprem/predictions/service.py

@@ -102,7 +102,7 @@ class PredictionService(BaseService):
                 prediction_set_id=prediction_set_id
             )
         )
-        return response.get("deletePredictionSet", False)
+        return response
     
     def delete_prediction_from_data(
         self,

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/spb_onprem/slices/service.py

@@ -85,7 +85,7 @@ class SliceService(BaseService):
         return (
             slices,
             response.get("next", None),
-            response.get("totalCount", False)
+            response.get("totalCount", 0)
         )
 
     def get_slice(

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/superb_ai_onprem.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.18
+Version: 0.5.20
 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.18 → superb_ai_onprem-0.5.20}/tests/contents/test_content_service.py

@@ -17,7 +17,7 @@ class TestContentService:
         """Test successful content deletion."""
         # Arrange
         content_id = "test-content-123"
-        mock_response = {"deleteContent": True}
+        mock_response = True
         self.content_service.request_gql.return_value = mock_response
 
         # Act
@@ -34,7 +34,7 @@ class TestContentService:
         """Test content deletion failure."""
         # Arrange
         content_id = "nonexistent-content"
-        mock_response = {"deleteContent": False}
+        mock_response = False
         self.content_service.request_gql.return_value = mock_response
 
         # Act
@@ -51,7 +51,7 @@ class TestContentService:
         """Test content deletion with missing response field."""
         # Arrange
         content_id = "test-content-456"
-        mock_response = {}  # No deleteContent field
+        mock_response = False
         self.content_service.request_gql.return_value = mock_response
 
         # Act

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/tests/data/test_data_service.py

@@ -422,17 +422,16 @@ class TestDataService:
         variables = Queries.DELETE["variables"](dataset_id=dataset_id, data_id=data_id)
         
         # Assert
-        assert "mutation DeleteData(" in query
+        assert "mutation (" 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 "$data_id: ID!," in query  # GraphQL allows trailing commas
         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 "id: $data_id," in query  # GraphQL allows trailing commas
         assert variables == {
             "dataset_id": dataset_id,
-            "id": data_id
+            "data_id": data_id
         }
 
     def test_delete_data_success(self):
@@ -440,7 +439,7 @@ class TestDataService:
         # Arrange
         dataset_id = "dataset-123"
         data_id = "data-456"
-        mock_response = {"deleteData": True}
+        mock_response = True
         self.data_service.request_gql.return_value = mock_response
 
         # Act
@@ -458,7 +457,7 @@ class TestDataService:
         # Arrange
         dataset_id = "dataset-123"
         data_id = "nonexistent-data"
-        mock_response = {"deleteData": False}
+        mock_response = False
         self.data_service.request_gql.return_value = mock_response
 
         # Act
@@ -476,7 +475,7 @@ class TestDataService:
         # Arrange
         dataset_id = "dataset-123"
         data_id = "data-456"
-        mock_response = {}
+        mock_response = False
         self.data_service.request_gql.return_value = mock_response
 
         # Act

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/tests/datasets/test_dataset_service.py

@@ -18,7 +18,7 @@ class TestDatasetService:
         """Test successful dataset deletion."""
         # Arrange
         dataset_id = "dataset-123"
-        mock_response = {"deleteDataset": True}
+        mock_response = True
         self.dataset_service.request_gql.return_value = mock_response
 
         # Act
@@ -35,7 +35,7 @@ class TestDatasetService:
         """Test dataset deletion failure."""
         # Arrange
         dataset_id = "nonexistent-dataset"
-        mock_response = {"deleteDataset": False}
+        mock_response = False
         self.dataset_service.request_gql.return_value = mock_response
 
         # Act
@@ -52,7 +52,7 @@ class TestDatasetService:
         """Test dataset deletion with missing response field."""
         # Arrange
         dataset_id = "dataset-123"
-        mock_response = {}
+        mock_response = False
         self.dataset_service.request_gql.return_value = mock_response
 
         # Act
@@ -74,7 +74,7 @@ class TestDatasetService:
 
         # Act & Assert
         # Empty string should be allowed by the service, but would fail at GraphQL level
-        mock_response = {"deleteDataset": False}
+        mock_response = False
         self.dataset_service.request_gql.return_value = mock_response
 
         result = self.dataset_service.delete_dataset(dataset_id)
@@ -84,7 +84,7 @@ class TestDatasetService:
         """Test that delete dataset uses correct query structure."""
         # Arrange
         dataset_id = "test-dataset"
-        mock_response = {"deleteDataset": True}
+        mock_response = True
         self.dataset_service.request_gql.return_value = mock_response
 
         # Act
@@ -120,7 +120,7 @@ class TestDatasetService:
         """Test dataset deletion with special characters in ID."""
         # Arrange
         dataset_id = "dataset-with-special-chars_123@test"
-        mock_response = {"deleteDataset": True}
+        mock_response = True
         self.dataset_service.request_gql.return_value = mock_response
 
         # Act

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/tests/models/test_model_service.py

@@ -160,7 +160,7 @@ class TestModelService:
         dataset_id = "dataset-123"
         model_id = "model-456"
         
-        mock_response = {"deleteModel": True}
+        mock_response = True
         self.model_service.request_gql.return_value = mock_response
 
         # Act
@@ -185,7 +185,7 @@ class TestModelService:
         dataset_id = "dataset-123"
         model_id = "nonexistent-model"
         
-        mock_response = {"deleteModel": False}
+        mock_response = False
         self.model_service.request_gql.return_value = mock_response
 
         # Act
@@ -203,7 +203,7 @@ class TestModelService:
         dataset_id = "dataset-123"
         model_id = "model-456"
         
-        mock_response = {}
+        mock_response = False
         self.model_service.request_gql.return_value = mock_response
 
         # Act

{superb_ai_onprem-0.5.18 → superb_ai_onprem-0.5.20}/tests/predictions/test_prediction_service.py

@@ -189,7 +189,7 @@ class TestPredictionService:
         dataset_id = "dataset-123"
         prediction_set_id = "pred-set-456"
         
-        mock_response = {"deletePredictionSet": True}
+        mock_response = True
         self.prediction_service.request_gql.return_value = mock_response
 
         # Act
@@ -214,7 +214,7 @@ class TestPredictionService:
         dataset_id = "dataset-123"
         prediction_set_id = "nonexistent-pred-set"
         
-        mock_response = {"deletePredictionSet": False}
+        mock_response = False
         self.prediction_service.request_gql.return_value = mock_response
 
         # Act
@@ -232,7 +232,7 @@ class TestPredictionService:
         dataset_id = "dataset-123"
         prediction_set_id = "pred-set-456"
         
-        mock_response = {}
+        mock_response = False
         self.prediction_service.request_gql.return_value = mock_response
 
         # Act