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

{superb_ai_onprem-0.3.4/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.4
+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.4 → 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.4'
-__version_tuple__ = version_tuple = (0, 3, 4)
+__version__ = version = '0.3.5'
+__version_tuple__ = version_tuple = (0, 3, 5)

{superb_ai_onprem-0.3.4 → superb_ai_onprem-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
@@ -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 = {

{superb_ai_onprem-0.3.4 → 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']
 
@@ -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 → 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.4
+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