superb-ai-onprem 0.3.2__tar.gz → 0.3.4__tar.gz

{superb_ai_onprem-0.3.2/superb_ai_onprem.egg-info → superb_ai_onprem-0.3.4}/PKG-INFO

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

@@ -9,6 +9,7 @@ from .data.service import DataService
 from .slices.service import SliceService
 from .activities.service import ActivityService
 from .exports.service import ExportService
+from .contents.service import ContentService
 
 # Core Entities and Enums
 from .entities import (
@@ -24,6 +25,7 @@ from .entities import (
     Activity,
     ActivityHistory,
     Export,
+    Content,
 
     # Enums
     DataType,
@@ -57,6 +59,7 @@ __all__ = (
     "SliceService",
     "ActivityService",
     "ExportService",
+    "ContentService",
 
     # Core Entities
     "Data",
@@ -70,6 +73,8 @@ __all__ = (
     "Activity",
     "ActivityHistory",
     "Export",
+    "Content",
+    
     # Enums
     "DataType",
     "SceneType",

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4}/spb_onprem/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.2'
-__version_tuple__ = version_tuple = (0, 3, 2)
+__version__ = version = '0.3.4'
+__version_tuple__ = version_tuple = (0, 3, 4)

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4}/spb_onprem/contents/queries.py

@@ -37,3 +37,21 @@ class Queries:
         ''',
         "variables": create_variables
     }
+    
+    @staticmethod
+    def get_download_url_params(
+        id: str,
+    ):
+        return {
+            "id": id
+        }
+    
+    GET_DOWNLOAD_URL = {
+        "name": "generateContentDownloadURL",
+        "query": '''
+            mutation GenerateContentDownloadURL($id: ID!) {
+                generateContentDownloadURL(id: $id) 
+            }
+        ''',
+        "variables": get_download_url_params
+    }

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4}/spb_onprem/contents/service.py

@@ -130,3 +130,21 @@ class ContentService(BaseService):
         # Retrieve the uploaded content details
         content = response['content']
         return BaseContent.model_validate(content)
+
+    def get_download_url(
+        self,
+        id: str,
+    ) -> str:
+        '''
+        Gets the content from the server.
+        Args:
+            id (str): The ID of the content to get.
+        Returns:
+            Content: The content object.
+        '''
+        
+        response = self.request_gql(
+            query=Queries.GET_DOWNLOAD_URL,
+            variables=Queries.GET_DOWNLOAD_URL["variables"](id)
+        )
+        return response

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4}/spb_onprem/data/enums/data_type.py

@@ -8,3 +8,4 @@ class DataType(str, Enum):
     """
     SUPERB_IMAGE = "SUPERB_IMAGE"
     MCAP = "MCAP"
+    SUPERB_3D = "SUPERB_3D"

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4}/spb_onprem/data/enums/scene_type.py

@@ -7,4 +7,5 @@ class SceneType(str, Enum):
     This is used to determine the type of the file.
     """
     IMAGE = "IMAGE"
+    MCAP = "MCAP"
     ETC = "ETC"

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4}/spb_onprem/data/params/data.py

@@ -17,8 +17,13 @@ def get_params(
         dict: The variables for the data query.
     """
     
-    return {
+    params = {
         "dataset_id": dataset_id,
-        "key": data_key,
-        "id": data_id,
     }
+    
+    if data_key is not Undefined:
+        params["key"] = data_key
+    if data_id is not Undefined:
+        params["id"] = data_id
+        
+    return params

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4}/spb_onprem/entities.py

@@ -22,6 +22,7 @@ from .activities.entities import (
     SchemaType,
 )
 from .exports.entities import Export
+from .contents.entities import Content
 
 __all__ = [
     # Core Entities
@@ -36,6 +37,7 @@ __all__ = [
     "Activity",
     "ActivityHistory",
     "Export",
+    "Content",
 
     # Enums
     "DataType",

{superb_ai_onprem-0.3.2 → superb_ai_onprem-0.3.4/superb_ai_onprem.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.3.2
+Version: 0.3.4
 Summary: Python SDK for Superb AI On-premise
 Home-page: https://github.com/Superb-AI-Suite/superb-ai-onprem-python
 Author: Superb AI