superb-ai-onprem 0.7.3__tar.gz → 0.8.1__tar.gz

{superb_ai_onprem-0.7.3 → superb_ai_onprem-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.7.3
+Version: 0.8.1
 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 → superb_ai_onprem-0.8.1}/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.1'
+__version_tuple__ = version_tuple = (0, 8, 1)
 
-__commit_id__ = commit_id = 'g17dcb4312'
+__commit_id__ = commit_id = 'gc24fcd383'

{superb_ai_onprem-0.7.3 → superb_ai_onprem-0.8.1}/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"
 )

superb_ai_onprem-0.8.1/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
+    }

{superb_ai_onprem-0.7.3 → superb_ai_onprem-0.8.1}/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",

{superb_ai_onprem-0.7.3 → superb_ai_onprem-0.8.1}/spb_onprem/contents/service.py

@@ -158,6 +158,25 @@ 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 → superb_ai_onprem-0.8.1}/superb_ai_onprem.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.7.3
+Version: 0.8.1
 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 → superb_ai_onprem-0.8.1}/superb_ai_onprem.egg-info/SOURCES.txt

@@ -44,6 +44,7 @@ spb_onprem/contents/params/__init__.py
 spb_onprem/contents/params/create.py
 spb_onprem/contents/params/delete_content.py
 spb_onprem/contents/params/get_download_url.py
+spb_onprem/contents/params/get_upload_url.py
 spb_onprem/data/README.md
 spb_onprem/data/__init__.py
 spb_onprem/data/queries.py