superb-ai-onprem 0.3.3__py3-none-any.whl → 0.3.5__py3-none-any.whl

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",

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)

spb_onprem/contents/queries.py

@@ -13,17 +13,21 @@ class Queries:
     @staticmethod
     def create_variables(
         key: Union[str, UndefinedType] = Undefined,
+        content_type: Union[str, UndefinedType] = Undefined,
     ):
+        params = {}
+        
         if key is not Undefined:
-            return {"key": key}
-        else:
-            return {}
+            params["key"] = key
+        if content_type is not Undefined:
+            params["contentType"] = content_type
+        return params
 
     CREATE = {
         "name": "createContent",
         "query": '''
-            mutation CreateContent($key: String) {
-                createContent(key: $key) {
+            mutation CreateContent($key: String, $content_type: String) {
+                createContent(key: $key, contentType: $content_type) {
                     content {
                         id
                         key
@@ -37,3 +41,21 @@ class Queries:
         ''',
         "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
+    }

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

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.dist-info → superb_ai_onprem-0.3.5.dist-info}/METADATA

@@ -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.dist-info → superb_ai_onprem-0.3.5.dist-info}/RECORD

@@ -1,9 +1,9 @@
-spb_onprem/__init__.py,sha256=Iucr68oU5pQ3lhHlyI36HUaReA7gsanGuLCSAVZh7GI,1761
-spb_onprem/_version.py,sha256=cRYgYV4ttw-FMlrA4-5pzcSpTjS7X8uVa-nRTEADKW4,511
+spb_onprem/__init__.py,sha256=luroRpOs-soHW_qKNeRzkpRR64KXxKPfcpC-wQO4h60,1861
+spb_onprem/_version.py,sha256=qmTn4w-xTKk8dVv-032b430C-ilxMTkr9Q0HieU18N8,511
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=dPfr3mGXYlqadOXycu6RBFX1HcZ1qzEsskLoOxERLOU,5737
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
-spb_onprem/entities.py,sha256=VFwaSdHIVB6NhxX5-pFRNPzCqVLNEZS-O2JETIYMopk,823
+spb_onprem/entities.py,sha256=VOFosKVcj85xk8N8SLFYGxZv4EtDAnzzmE3NBj144Tk,877
 spb_onprem/exceptions.py,sha256=jx5rTGsVZ5shOdbgQzk8GcSyMWFtb_3xhPq6Sylwc5o,478
 spb_onprem/searches.py,sha256=VRj9vpvKU8iOdBJUA6BwfvOjUbWrKBo9fPJvNlRmKr0,750
 spb_onprem/activities/__init__.py,sha256=iTHUza8mtiBd_QEPH-0tzrOsd_VSwSQNMbzVWfebQ6c,175
@@ -22,8 +22,8 @@ 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=tGMVH8ixv0CW5bJTWICCjWDM_oAN7jkfEQXdAJUVn4Q,851
-spb_onprem/contents/service.py,sha256=KoPUffr_DEGOMIOwkue0rkidHMPfmAJG0KgJyXN6N_Y,3602
+spb_onprem/contents/queries.py,sha256=lNoLTlYKouv9KIhVnM3ZQBkCx3u6DHAy3PwF31VVXj4,1501
+spb_onprem/contents/service.py,sha256=kgZmiRnJKH4nKoYs_V71JVlvzVoQe2KljAvtE-j7wZY,4146
 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
@@ -93,7 +93,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.3.3.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.3.5.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
@@ -106,7 +106,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.3.3.dist-info/METADATA,sha256=ICaffqAmypTRf7bQeOf7X-sEkBU1IoODtC5Q_LGwD_0,5817
-superb_ai_onprem-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.3.3.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.3.3.dist-info/RECORD,,
+superb_ai_onprem-0.3.5.dist-info/METADATA,sha256=S3dm3FhxXQLiiVMppXlSEvhPq79ptJ59lrF26FfX6lA,5817
+superb_ai_onprem-0.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.3.5.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.3.5.dist-info/RECORD,,