superb-ai-onprem 0.3.3__tar.gz → 0.3.5__tar.gz

{superb_ai_onprem-0.3.3/superb_ai_onprem.egg-info → superb_ai_onprem-0.3.5}/PKG-INFO

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

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.3'
-__version_tuple__ = version_tuple = (0, 3, 3)
+__version__ = version = '0.3.5'
+__version_tuple__ = version_tuple = (0, 3, 5)

superb_ai_onprem-0.3.5/spb_onprem/contents/queries.py

@@ -0,0 +1,61 @@
+from typing import (
+    Union,
+)
+
+from spb_onprem.base_types import (
+    Undefined,
+    UndefinedType,
+)
+
+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["contentType"] = content_type
+        return params
+
+    CREATE = {
+        "name": "createContent",
+        "query": '''
+            mutation CreateContent($key: String, $content_type: String) {
+                createContent(key: $key, contentType: $content_type) {
+                    content {
+                        id
+                        key
+                        location
+                        createdAt
+                        createdBy
+                    }
+                    uploadURL
+                }
+            }
+        ''',
+        "variables": create_variables
+    }
+    
+    @staticmethod
+    def get_download_url_params(
+        content_id: str,
+    ):
+        return {
+            "id": content_id
+        }
+    
+    GET_DOWNLOAD_URL = {
+        "name": "generateContentDownloadURL",
+        "query": '''
+            mutation GenerateContentDownloadURL($id: ID!) {
+                generateContentDownloadURL(id: $id) 
+            }
+        ''',
+        "variables": get_download_url_params
+    }

{superb_ai_onprem-0.3.3 → superb_ai_onprem-0.3.5}/spb_onprem/contents/service.py

@@ -39,7 +39,7 @@ class ContentService(BaseService):
             file = f.read()
         response = self.request_gql(
             query=Queries.CREATE,
-            variables=Queries.CREATE["variables"](key)
+            variables=Queries.CREATE["variables"](key, mimetypes.guess_type(file_path)[0])
         )
         upload_url = response['uploadURL']
         
@@ -73,7 +73,7 @@ class ContentService(BaseService):
         '''
         response = self.request_gql(
             query=Queries.CREATE,
-            variables=Queries.CREATE["variables"](key) if key else None
+            variables=Queries.CREATE["variables"](key, "application/json") if key else None
         )
         upload_url = response['uploadURL']
         self.request(
@@ -113,7 +113,7 @@ class ContentService(BaseService):
         # Request to get the upload URL
         response = self.request_gql(
             query=Queries.CREATE,
-            variables=Queries.CREATE["variables"](key)
+            variables=Queries.CREATE["variables"](key, content_type)
         )
         upload_url = response['uploadURL']
 
@@ -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,
+        content_id: str,
+    ) -> str:
+        '''
+        Gets the content from the server.
+        Args:
+            content_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"](content_id)
+        )
+        return response

{superb_ai_onprem-0.3.3 → superb_ai_onprem-0.3.5}/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.3 → superb_ai_onprem-0.3.5/superb_ai_onprem.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.3.3
+Version: 0.3.5
 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.3/spb_onprem/contents/queries.py

@@ -1,39 +0,0 @@
-from typing import (
-    Union,
-)
-
-from spb_onprem.base_types import (
-    Undefined,
-    UndefinedType,
-)
-
-class Queries:
-    """ Content Queries """
-    
-    @staticmethod
-    def create_variables(
-        key: Union[str, UndefinedType] = Undefined,
-    ):
-        if key is not Undefined:
-            return {"key": key}
-        else:
-            return {}
-
-    CREATE = {
-        "name": "createContent",
-        "query": '''
-            mutation CreateContent($key: String) {
-                createContent(key: $key) {
-                    content {
-                        id
-                        key
-                        location
-                        createdAt
-                        createdBy
-                    }
-                    uploadURL
-                }
-            }
-        ''',
-        "variables": create_variables
-    }