superb-ai-onprem 0.5.2__tar.gz → 0.5.4__tar.gz

{superb_ai_onprem-0.5.2/superb_ai_onprem.egg-info → superb_ai_onprem-0.5.4}/PKG-INFO

@@ -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 → superb_ai_onprem-0.5.4}/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 = 'gfef127d74'
+__commit_id__ = commit_id = 'g627b2c843'

superb_ai_onprem-0.5.4/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",
+)

superb_ai_onprem-0.5.4/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

superb_ai_onprem-0.5.4/spb_onprem/contents/params/get_download_url.py

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

{superb_ai_onprem-0.5.2 → superb_ai_onprem-0.5.4}/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",

{superb_ai_onprem-0.5.2 → superb_ai_onprem-0.5.4}/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",
 )

superb_ai_onprem-0.5.4/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,
+    }

{superb_ai_onprem-0.5.2 → superb_ai_onprem-0.5.4}/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,
+    }

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

@@ -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 → superb_ai_onprem-0.5.4}/superb_ai_onprem.egg-info/SOURCES.txt

@@ -35,6 +35,9 @@ spb_onprem/contents/service.py
 spb_onprem/contents/entities/__init__.py
 spb_onprem/contents/entities/base_content.py
 spb_onprem/contents/entities/content.py
+spb_onprem/contents/params/__init__.py
+spb_onprem/contents/params/create.py
+spb_onprem/contents/params/get_download_url.py
 spb_onprem/data/__init__.py
 spb_onprem/data/queries.py
 spb_onprem/data/service.py