superb-ai-onprem 0.3.4__py3-none-any.whl → 0.3.6__py3-none-any.whl

spb_onprem/_version.py

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

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["content_type"] = 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
@@ -40,10 +44,10 @@ class Queries:
     
     @staticmethod
     def get_download_url_params(
-        id: str,
+        content_id: str,
     ):
         return {
-            "id": id
+            "id": content_id
         }
     
     GET_DOWNLOAD_URL = {

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']
 
@@ -133,18 +133,18 @@ class ContentService(BaseService):
 
     def get_download_url(
         self,
-        id: str,
+        content_id: str,
     ) -> str:
         '''
         Gets the content from the server.
         Args:
-            id (str): The ID of the content to get.
+            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"](id)
+            variables=Queries.GET_DOWNLOAD_URL["variables"](content_id)
         )
         return response

{superb_ai_onprem-0.3.4.dist-info → superb_ai_onprem-0.3.6.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 spb_onprem/__init__.py,sha256=luroRpOs-soHW_qKNeRzkpRR64KXxKPfcpC-wQO4h60,1861
-spb_onprem/_version.py,sha256=Mxups7YfGBY2vvCok_hLJKCtU6O1WHQJfsMY-bAJ0Yg,511
+spb_onprem/_version.py,sha256=5ZRqa3YWFznn7rzKg8d9Vi2wEKohJwIFxhrpOUL3pYQ,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
@@ -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=idNVcib6Db95n9wkB-tPLX2WKLnvcxMEm08IrOyuxNU,1267
-spb_onprem/contents/service.py,sha256=lYcMfHkEqW4_82ZJQS73YTMvcIgMXv6iCYQi-1V287g,4052
+spb_onprem/contents/queries.py,sha256=NwQYf0wNlNT5s5MTEMaAEHFPsXFE7_qsws2UPT8vOXs,1502
+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.4.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.3.6.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.4.dist-info/METADATA,sha256=JNqfTTOQWsEYEuRd3PAVSm4txEoHjELQNhdMWuDgfQw,5817
-superb_ai_onprem-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.3.4.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.3.4.dist-info/RECORD,,
+superb_ai_onprem-0.3.6.dist-info/METADATA,sha256=tlo8bi4HcQFoUgPgE4LLwCp6KF97bv1PrX5Kw1eTws4,5817
+superb_ai_onprem-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.3.6.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.3.6.dist-info/RECORD,,