superb-ai-onprem 0.5.2__py3-none-any.whl → 0.5.4__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.2'
-__version_tuple__ = version_tuple = (0, 5, 2)
+__version__ = version = '0.5.4'
+__version_tuple__ = version_tuple = (0, 5, 4)
 
 __commit_id__ = commit_id = None

spb_onprem/contents/params/__init__.py

@@ -0,0 +1,8 @@
+
+from .create import create_variables
+from .get_download_url import get_download_url_params
+
+__all__ = (
+    "create_variables",
+    "get_download_url_params",
+)

spb_onprem/contents/params/create.py

@@ -0,0 +1,18 @@
+from typing import Union
+from spb_onprem.base_types import (
+    Undefined,
+    UndefinedType,
+)
+
+
+def create_variables(
+    key: Union[str, UndefinedType] = Undefined,
+    content_type: Union[str, UndefinedType] = Undefined,
+):
+    params = {}
+    
+    if key is not Undefined:
+        params["key"] = key
+    if content_type is not Undefined:
+        params["content_type"] = content_type
+    return params

spb_onprem/contents/params/get_download_url.py

@@ -0,0 +1,8 @@
+
+
+def get_download_url_params(
+    content_id: str,
+):
+    return {
+        "id": content_id
+    }

spb_onprem/contents/queries.py

@@ -1,28 +1,9 @@
-from typing import (
-    Union,
-)
-
-from spb_onprem.base_types import (
-    Undefined,
-    UndefinedType,
+from .params import (
+    create_variables,
+    get_download_url_params,
 )
 
 class Queries:
-    """ Content Queries """
-    
-    @staticmethod
-    def create_variables(
-        key: Union[str, UndefinedType] = Undefined,
-        content_type: Union[str, UndefinedType] = Undefined,
-    ):
-        params = {}
-        
-        if key is not Undefined:
-            params["key"] = key
-        if content_type is not Undefined:
-            params["content_type"] = content_type
-        return params
-
     CREATE = {
         "name": "createContent",
         "query": '''
@@ -42,13 +23,6 @@ class Queries:
         "variables": create_variables
     }
     
-    @staticmethod
-    def get_download_url_params(
-        content_id: str,
-    ):
-        return {
-            "id": content_id
-        }
     
     GET_DOWNLOAD_URL = {
         "name": "generateContentDownloadURL",

spb_onprem/slices/params/__init__.py

@@ -12,6 +12,9 @@ from .create_slice import (
 from .update_slice import (
     update_slice_params,
 )
+from .delete_slice import (
+    delete_slice_params,
+)
 
 __all__ = (
     "SlicesFilter",
@@ -20,4 +23,5 @@ __all__ = (
     "slice_params",
     "create_slice_params",
     "update_slice_params",
+    "delete_slice_params",
 )

spb_onprem/slices/params/delete_slice.py

@@ -0,0 +1,25 @@
+from spb_onprem.exceptions import BadParameterError
+
+
+def delete_slice_params(
+    dataset_id: str,
+    slice_id: str,
+):
+    """Create parameters for slice deletion.
+    
+    Args:
+        dataset_id (str): The dataset ID the slice belongs to.
+        slice_id (str): The ID of the slice to delete.
+        
+    Returns:
+        dict: Parameters for slice deletion
+    """
+    if dataset_id is None:
+        raise BadParameterError("Dataset ID is required")
+    if slice_id is None:
+        raise BadParameterError("Slice ID is required")
+
+    return {
+        "dataset_id": dataset_id,
+        "id": slice_id,
+    }

spb_onprem/slices/queries.py

@@ -3,6 +3,7 @@ from spb_onprem.slices.params import (
     slice_params,
     create_slice_params,
     update_slice_params,
+    delete_slice_params,
 )
 
 
@@ -119,3 +120,16 @@ class Queries:
         ''',
         "variables": update_slice_params,
     }
+
+    DELETE_SLICE = {
+        "name": "deleteSlice",
+        "query": '''
+            mutation deleteSlice(
+                $dataset_id: String!,
+                $id: ID!
+            ) {
+                deleteSlice(datasetId: $dataset_id, id: $id)
+            }
+        ''',
+        "variables": delete_slice_params,
+    }

spb_onprem/slices/service.py

@@ -171,3 +171,26 @@ class SliceService(BaseService):
         )
         slice_dict = response.get("updateSlice", {})
         return Slice.model_validate(slice_dict)
+
+    def delete_slice(
+        self,
+        dataset_id: str,
+        slice_id: str,
+    ) -> bool:
+        """Delete a slice.
+        
+        Args:
+            dataset_id (str): The ID of the dataset the slice belongs to.
+            slice_id (str): The ID of the slice to delete.
+        
+        Returns:
+            bool: True if the slice was deleted, False otherwise.
+        """
+        response = self.request_gql(
+            Queries.DELETE_SLICE,
+            Queries.DELETE_SLICE["variables"](
+                dataset_id=dataset_id,
+                slice_id=slice_id,
+            )
+        )
+        return response

{superb_ai_onprem-0.5.2.dist-info → superb_ai_onprem-0.5.4.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 spb_onprem/__init__.py,sha256=lGZRdgpRdBFM97SawPJIC-a5uRzrd5jdb9i5yHDeveQ,2649
-spb_onprem/_version.py,sha256=LGYtjQ6cyPZC_N0AovMIeSYYDK21050nm3HYgDanQBM,704
+spb_onprem/_version.py,sha256=OrfVZdCDQ-QC6dUnxdROooJjwvLfeDMedTBstpAdSBU,704
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=dPfr3mGXYlqadOXycu6RBFX1HcZ1qzEsskLoOxERLOU,5737
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
@@ -22,11 +22,14 @@ 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=NwQYf0wNlNT5s5MTEMaAEHFPsXFE7_qsws2UPT8vOXs,1502
+spb_onprem/contents/queries.py,sha256=hk7JHfUBinIDp4xswYxmTMQwxsRbKWTxFeZQtUVl9K4,925
 spb_onprem/contents/service.py,sha256=a_Y_l18sULozh14obwAsEJG9x2BLSAgerqCNCf5TH2Q,5024
 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/create.py,sha256=JLMbNiIecOsVvVSYY6DfUpWjZxZKuO9-NSpuRlSIxS0,408
+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=tGMRRBJQ0Un8jWoDesjYq_jLF1pn_Mliq7rjyVRx23w,15606
 spb_onprem/data/service.py,sha256=zVtcWJqzSBKuuY_ZbRHVsy3_F-_FmTxsiTGYuSO_Ato,24799
@@ -92,20 +95,20 @@ spb_onprem/exports/params/export.py,sha256=0EP6nkQc6vFI-f8218Yq4NxfFcw8MQtHMNkYl
 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/slices/__init__.py,sha256=xgpNGYzqgwQ8C-Bgw9AZWMAgBW38UU-U4Wube8hkodI,69
-spb_onprem/slices/queries.py,sha256=dl_q6Uc2_oeuItgSD6gUL7a3H5VrOW9Ig5Epte7sl78,2732
-spb_onprem/slices/service.py,sha256=byhB9CdxNKV7uLIiL9yI674UVSlgaAwLaYoGQGBCawE,4988
+spb_onprem/slices/queries.py,sha256=13JcjACvCg7AbdGfUODDJeq77gGZlNjngFB-9UmgLZk,3080
+spb_onprem/slices/service.py,sha256=TnWGMVflesdaUXFKOBZTpdHSIKkQwj93Q3q9aGts5xI,5615
 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=dEUDlOK-iw3Sx7gpkDMnwMqTFE16-856ZdbYuwppeqs,394
+spb_onprem/slices/params/__init__.py,sha256=NFimX_7wDtqFda5q5W5PtiBCgDwH3iEQHijz2BY7Loc,476
 spb_onprem/slices/params/create_slice.py,sha256=qUpX60A72Uht0SzN7b2-QSKvd_MSEV5T9kYIVk_td8A,1009
-spb_onprem/slices/params/delete_slice.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+spb_onprem/slices/params/delete_slice.py,sha256=IDtHOopLuNli6lAkDU6oQk5xx3LMmy4mjXGCMbVeKFQ,618
 spb_onprem/slices/params/slice.py,sha256=R8U_RadZLWPeQu6ZWGIvXH6Dxi4ikzoHyDKWGewmUjw,1035
 spb_onprem/slices/params/slices.py,sha256=SIe_JfsFb8ULwpRYsQkAO3aIWqqarp3m6uePmUj10tw,1935
 spb_onprem/slices/params/update_slice.py,sha256=kryOmCnRTQ_OU0qDKgugppLrpeUpuLwmn_87M5zKqIA,1209
 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.2.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.5.4.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
@@ -118,7 +121,7 @@ tests/exports/test_entities.py,sha256=hG7G4kVkyHKT3mv4lvrzUqOW8ILeHiYj87QZjQcmg9
 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.2.dist-info/METADATA,sha256=6V-Waf9aiRroI_08aL3OPx6LQg_d0qojiz8xGOCiad8,5817
-superb_ai_onprem-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.5.2.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.5.2.dist-info/RECORD,,
+superb_ai_onprem-0.5.4.dist-info/METADATA,sha256=-Gfn_a4L9Ds8ilLUxNdXq5QHI00GYfdH30CI0NObv7Y,5817
+superb_ai_onprem-0.5.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.5.4.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.5.4.dist-info/RECORD,,