superb-ai-onprem 0.5.6__tar.gz → 0.5.8__tar.gz

{superb_ai_onprem-0.5.6/superb_ai_onprem.egg-info → superb_ai_onprem-0.5.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.6
+Version: 0.5.8
 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.5.6 → superb_ai_onprem-0.5.8}/spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.6'
-__version_tuple__ = version_tuple = (0, 5, 6)
+__version__ = version = '0.5.8'
+__version_tuple__ = version_tuple = (0, 5, 8)
 
-__commit_id__ = commit_id = 'g4c010a872'
+__commit_id__ = commit_id = 'g07632cdb4'

{superb_ai_onprem-0.5.6 → superb_ai_onprem-0.5.8}/spb_onprem/data/queries.py

@@ -581,7 +581,7 @@ class Queries():
             mutation (
                 $dataset_id: ID!,
                 $data_id: ID!,
-                $frames: [DataFrameInput!],
+                $frames: [DataFrameInput!]!,
             ) {{
                 updateFrames(
                     datasetId: $dataset_id,

{superb_ai_onprem-0.5.6 → superb_ai_onprem-0.5.8}/spb_onprem/data/service.py

@@ -177,6 +177,36 @@ class DataService(BaseService):
             response.get("totalCount", 0)
         )
 
+    def _detect_image_type(self, file_data: BytesIO) -> str:
+        """Detect image type from BytesIO data using magic numbers.
+        
+        Args:
+            file_data (BytesIO): The image data.
+            
+        Returns:
+            str: The MIME type of the image.
+            
+        Raises:
+            BadParameterError: If the file format is not a supported image type.
+        """
+        current_pos = file_data.tell()
+        file_data.seek(0)
+        header = file_data.read(12)
+        file_data.seek(current_pos)
+        
+        if header.startswith(b'\xff\xd8\xff'):
+            return 'image/jpeg'
+        elif header.startswith(b'\x89PNG\r\n\x1a\n'):
+            return 'image/png'
+        elif header.startswith(b'GIF8'):
+            return 'image/gif'
+        elif header.startswith(b'RIFF') and header[8:12] == b'WEBP':
+            return 'image/webp'
+        else:
+            raise BadParameterError(
+                "Unsupported image format. Only JPEG, PNG, GIF, and WebP formats are supported."
+            )
+
     def create_image_data(
         self,
         dataset_id: str,
@@ -208,17 +238,17 @@ class DataService(BaseService):
         """
         content_service = ContentService()
         if isinstance(image_content, str):
-            with open(image_content, "rb") as f:
-                content = content_service.upload_content(
-                    f.read(),
-                    key,
-                    f.name.split(".")[-1]
-                )
-        else:
             content = content_service.upload_content(
-                image_content.read(),
+                image_content,
+                key,
+            )    
+        else:
+            # Detect the image type from BytesIO data
+            content_type = self._detect_image_type(image_content)
+            content = content_service.upload_content_with_data(
+                image_content,
+                content_type,
                 key,
-                "jpg"
             )
 
         response = self.request_gql(

{superb_ai_onprem-0.5.6 → superb_ai_onprem-0.5.8/superb_ai_onprem.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.6
+Version: 0.5.8
 Summary: Python SDK for Superb AI On-premise
 Home-page: https://github.com/Superb-AI-Suite/superb-ai-onprem-python
 Author: Superb AI