superb-ai-onprem 0.7.3__py3-none-any.whl → 0.8.0__py3-none-any.whl

spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.7.3'
-__version_tuple__ = version_tuple = (0, 7, 3)
+__version__ = version = '0.8.0'
+__version_tuple__ = version_tuple = (0, 8, 0)
 
 __commit_id__ = commit_id = None

spb_onprem/contents/params/__init__.py

@@ -2,9 +2,11 @@
 from .create import create_variables
 from .get_download_url import get_download_url_params
 from .delete_content import delete_content_params
+from .get_upload_url import get_upload_url_params
 
 __all__ = (
     "create_variables",
     "get_download_url_params",
     "delete_content_params",
+    "get_upload_url_params"
 )

spb_onprem/contents/params/get_upload_url.py

@@ -0,0 +1,12 @@
+
+
+def get_upload_url_params(
+    content_id: str,
+    file_name: str,
+    content_type: str = None,
+):
+    return {
+        "id": content_id,
+        "file_name": file_name,
+        "content_type": content_type
+    }

spb_onprem/contents/queries.py

@@ -2,6 +2,7 @@ from .params import (
     create_variables,
     get_download_url_params,
     delete_content_params,
+    get_upload_url_params,
 )
 
 class Queries:
@@ -24,6 +25,15 @@ class Queries:
         "variables": create_variables
     }
     
+    GET_UPLOAD_URL = {
+        "name": "generateFileUploadURL",
+        "query": '''
+            mutation GenerateFileUploadURL($id: ID!, $file_name: String!, $content_type: String) {
+                generateFileUploadURL(id: $id, fileName: $file_name, contentType: $content_type) 
+            }
+        ''',
+        "variables": get_upload_url_params
+    }
     
     GET_DOWNLOAD_URL = {
         "name": "generateContentDownloadURL",

spb_onprem/contents/service.py

@@ -158,6 +158,26 @@ class ContentService(BaseService):
         # Retrieve the uploaded content details
         content = response['content']
         return BaseContent.model_validate(content)
+    
+    def get_upload_url(
+        self,
+        content_id: str,
+        file_name: str,
+        content_type: str,
+    ) -> str:
+        '''
+        Gets the upload URL for the content.
+        Args:
+            content_id (str): The ID of the content to get the upload URL for.
+            file_name (str): The name of the file to be uploaded.
+            content_type (str): The MIME type of the file being uploaded.
+        '''
+        response = self.request_gql(
+            query=Queries.GET_UPLOAD_URL,
+            variables=Queries.GET_UPLOAD_URL["variables"](content_id, file_name, content_type)
+        )
+        return response['uploadURL']
+    )
 
     def get_download_url(
         self,

{superb_ai_onprem-0.7.3.dist-info → superb_ai_onprem-0.8.0.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 spb_onprem/__init__.py,sha256=Q5aooIhNiuf0z9-S5k5FBfM_oA_c92kCzWdjLuDX4wE,3294
-spb_onprem/_version.py,sha256=gWqiHfJwvZSzPQMjJVXuWX-SvdgzetfD6jyorihjahc,704
+spb_onprem/_version.py,sha256=Rttl-BDadtcW1QzGnNffCWA_Wc9mUKDMOBPZp--Mnsc,704
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=_6aLM_zn4Ycd7nAc_BA395KmxH5lcSxd7gmgG9DdOf4,6518
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
@@ -22,15 +22,16 @@ 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=DuBlNTvgbo1gYfYIodScu2G5FngerZL6DzteBejV0-g,1188
-spb_onprem/contents/service.py,sha256=8vKsPD-3Kx7vl8KRZQjJQgzVhlTojZaX_hFQtcWvpjU,5493
+spb_onprem/contents/queries.py,sha256=I8zp7QJvvtOHJTe6-KHxzQ96J2VNW4o5nVc4gHVgkTY,1573
+spb_onprem/contents/service.py,sha256=JXB-FCcNKANDUcX33SlJJrSs2zArDeovM4eeSlgBZqg,6155
 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
-spb_onprem/contents/params/__init__.py,sha256=YCkHhN01Dkx9ytCG35oSh2FmDDZcW-Dv2FhT1bFmzio,240
+spb_onprem/contents/params/__init__.py,sha256=OBvgLAocIY9jGQVEBK8grVnjnS4gcoa5xwwOtkL3_M4,318
 spb_onprem/contents/params/create.py,sha256=JLMbNiIecOsVvVSYY6DfUpWjZxZKuO9-NSpuRlSIxS0,408
 spb_onprem/contents/params/delete_content.py,sha256=_M9d6FJpA47_pcM5YyHbMmTZv1F2wUwPryQTL9RxZwM,298
 spb_onprem/contents/params/get_download_url.py,sha256=3oaiqmjJRxw0U59SoTxoih1pcU-cuQvcU-g89lCcqtQ,98
+spb_onprem/contents/params/get_upload_url.py,sha256=xuAZNcOgRcpfjrTy5n1JxrzfpNcvhDrPRoNwI5ne0js,217
 spb_onprem/data/__init__.py,sha256=5XKxNm2BlKufrX5uRQreUEzJ-nerTrozKpG1RJL5wt8,68
 spb_onprem/data/queries.py,sha256=0R8XosMWZPidDICmhGsSpyQNGnuMVFtSHQVH7AZPqXg,17144
 spb_onprem/data/service.py,sha256=I6DkcltPe08OBxpBQ-9rOIWzDlIqPn59et1J4PhHm0c,25098
@@ -133,7 +134,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.7.3.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.8.0.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
@@ -154,7 +155,7 @@ tests/models/__init__.py,sha256=Z2naiUgIjI-nHGsldwze1BqpV5JfDKTaBk-vg6_h6t4,21
 tests/models/test_model_service.py,sha256=lQKFPHlfxhqcnrKHz6l2sGp53lpmC_kCHgQ-sXBbgMw,7556
 tests/predictions/__init__.py,sha256=mFAXy0xOEXVYM7quYfIhwSih3kcEDdD4rLPq5G91Igg,26
 tests/predictions/test_prediction_service.py,sha256=u2MYXNXSXKRAbv-Rz7sV0roitbD2h5U5ZIAaqeFMA1M,12632
-superb_ai_onprem-0.7.3.dist-info/METADATA,sha256=pYY47Zf51PCU697cgAbNk_pLQi6D-efhwOioj9ymFM0,9627
-superb_ai_onprem-0.7.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.7.3.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.7.3.dist-info/RECORD,,
+superb_ai_onprem-0.8.0.dist-info/METADATA,sha256=jY0Ebd_P9C129HWZhPCji3xJW0tcaGWAJxMIUSIteMs,9627
+superb_ai_onprem-0.8.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.8.0.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.8.0.dist-info/RECORD,,