superb-ai-onprem 0.5.19__py3-none-any.whl → 0.5.21__py3-none-any.whl

spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.19'
-__version_tuple__ = version_tuple = (0, 5, 19)
+__version__ = version = '0.5.21'
+__version_tuple__ = version_tuple = (0, 5, 21)
 
 __commit_id__ = commit_id = None

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

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

spb_onprem/models/service.py

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

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
     }

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,

spb_onprem/slices/service.py

@@ -39,15 +39,30 @@ class SliceService(BaseService):
         Returns:
             Slice: The created slice object.
         """
+        # Debug logging: 파라미터 확인
+        print(f"[DEBUG] create_slice called with:")
+        print(f"  dataset_id: {dataset_id} (type: {type(dataset_id)})")
+        print(f"  name: {name} (type: {type(name)})")
+        print(f"  description: {description} (type: {type(description)})")
+
+        variables = Queries.CREATE_SLICE["variables"](
+            dataset_id=dataset_id,
+            slice_name=name,
+            slice_description=description
+        )
+        print(f"[DEBUG] GraphQL variables: {variables}")
+
         response = self.request_gql(
             Queries.CREATE_SLICE,
-            Queries.CREATE_SLICE["variables"](
-                dataset_id=dataset_id,
-                slice_name=name,
-                slice_description=description
-            )
+            variables
         )
+        # Debug logging: 실제 GraphQL 응답 확인
+        print(f"[DEBUG] GraphQL create_slice response: {response}")
+        print(f"[DEBUG] Response type: {type(response)}")
+        print(f"[DEBUG] Response keys: {response.keys() if isinstance(response, dict) else 'Not a dict'}")
+
         slice_dict = response.get("createSlice", {})
+        print(f"[DEBUG] Extracted slice_dict: {slice_dict}")
         return Slice.model_validate(slice_dict)
 
     def get_slices(
@@ -85,7 +100,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.19.dist-info → superb_ai_onprem-0.5.21.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 spb_onprem/__init__.py,sha256=ZldbV-bTUi0Vb3fj_PXSZZEDRsBYn-5UBfWb-uVff6A,2896
-spb_onprem/_version.py,sha256=KptclRJuFjZ74Te6eYPUhLNqlAq3tgEZ2wkF6kzPYkg,706
+spb_onprem/_version.py,sha256=PhPxRTxzvodKjo7czONYqVudjCWggAdKCgJAXsE4PIY,706
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=_6aLM_zn4Ycd7nAc_BA395KmxH5lcSxd7gmgG9DdOf4,6518
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
@@ -23,7 +23,7 @@ spb_onprem/activities/params/update_activity.py,sha256=QU3h2YkPFkqHEHXv6078V2PWE
 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=DuBlNTvgbo1gYfYIodScu2G5FngerZL6DzteBejV0-g,1188
-spb_onprem/contents/service.py,sha256=Waf3Kx3r4JXL7K8JUeg-LsJ5jEI3OqyoxgM3JWM78QI,5521
+spb_onprem/contents/service.py,sha256=8vKsPD-3Kx7vl8KRZQjJQgzVhlTojZaX_hFQtcWvpjU,5493
 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
@@ -78,7 +78,7 @@ spb_onprem/data/params/update_slice_annotation_version.py,sha256=M2STViLDSQDZ0WN
 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=-qsAaD19ESZ-qsbBEao19FM4b0LgLWYHQQXlpP5Lbos,2207
-spb_onprem/datasets/service.py,sha256=r5dim9mQcyb_gwBQ9oG4je0K8iHs1elvcWFLW9wvzYA,4439
+spb_onprem/datasets/service.py,sha256=1qVX3oBtM4ZQ6HWxOsdJQjcgwAwqdKKpPVK7L7xKPWs,4407
 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=uqy7QijSgvOzrRY0xIg9GxRUo5Bquscs3dTiboLeTCM,370
@@ -103,24 +103,24 @@ spb_onprem/inferences/service.py,sha256=ZkORci7Yt8Ruo0aY2KPsvynZG_uuitMKJO3FR22N
 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/service.py,sha256=XyupxjYmhPDOIEMNnTQhEV_V1w_viQwjWhQ8M4Ln2k4,2473
 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/service.py,sha256=_PlSjwJMkmxGMyyHBQ_vPHMCihWPnfmlbYe7Jn7Myl8,8030
 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_from_data.py,sha256=f3NmS_BSIEX6Fg-TbPV88SfU7jCyopTvQfcgkeAlIpU,564
 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
+spb_onprem/slices/service.py,sha256=oKlCQPxgc-dSjftqQtY3yicrcOdDMe3uXioFDVkc9hk,6331
 spb_onprem/slices/entities/__init__.py,sha256=fXBFWw9NI0DkTORkHczs_oFPqE9MVISOsAqnN4-euh0,52
 spb_onprem/slices/entities/slice.py,sha256=Kcn3G8NLQhpRID9xkwe6_CvS5kaJOMDPUStHM-BYkhY,623
 spb_onprem/slices/params/__init__.py,sha256=NFimX_7wDtqFda5q5W5PtiBCgDwH3iEQHijz2BY7Loc,476
@@ -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.19.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.5.21.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=Zsxj17-NwjZjxwL2PME29D-jMiPspD8m4h8qRajbz68,16901
+tests/data/test_data_service.py,sha256=LDfqXXca1FUDMsW3akgrzRLtZLXs0bVo57dLrDKoIu4,16832
 tests/datasets/__init__.py,sha256=nMsjhPXe9rU87gI_wasqCtz7SsTae2LzpUUnXXzS5zI,23
-tests/datasets/test_dataset_service.py,sha256=0BLmg_jslfzA_plm9dElws6eVfeOEQKuOq-nAmKNnqc,4899
+tests/datasets/test_dataset_service.py,sha256=RYY60U2TwKNv5JWzzkgUfP8ZgT5UYvgTYOl_fyp-Gek,4807
 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
@@ -150,10 +150,10 @@ tests/exports/test_integration.py,sha256=cCcEgwIIHyQRlc04EAXSKz7RcblQvhI2GBR3uVa
 tests/exports/test_params.py,sha256=oRRa6nEru_FImlB3TrmFiBidz6ZstCx4rVaCK-EMGfQ,11070
 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/models/test_model_service.py,sha256=lQKFPHlfxhqcnrKHz6l2sGp53lpmC_kCHgQ-sXBbgMw,7556
 tests/predictions/__init__.py,sha256=mFAXy0xOEXVYM7quYfIhwSih3kcEDdD4rLPq5G91Igg,26
-tests/predictions/test_prediction_service.py,sha256=9maX0JGxfUJ1QO_YEdS9OI5-o1tXlMWyeEbvZB8LB48,12679
-superb_ai_onprem-0.5.19.dist-info/METADATA,sha256=ihdW0S6cPSuqkaVm_uXnx442FRkNUuWDBu_OQ9Mo-l8,5818
-superb_ai_onprem-0.5.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.5.19.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.5.19.dist-info/RECORD,,
+tests/predictions/test_prediction_service.py,sha256=u2MYXNXSXKRAbv-Rz7sV0roitbD2h5U5ZIAaqeFMA1M,12632
+superb_ai_onprem-0.5.21.dist-info/METADATA,sha256=VxXqBTs7NBpsC5iX0_OxWCNqTV4f1O5T75LOWeE3ds0,5818
+superb_ai_onprem-0.5.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.5.21.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.5.21.dist-info/RECORD,,

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

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

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

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