superb-ai-onprem 0.5.7__py3-none-any.whl → 0.5.8__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.5.7'
-__version_tuple__ = version_tuple = (0, 5, 7)
+__version__ = version = '0.5.8'
+__version_tuple__ = version_tuple = (0, 5, 8)
 
 __commit_id__ = commit_id = None

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.7.dist-info → superb_ai_onprem-0.5.8.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 spb_onprem/__init__.py,sha256=lGZRdgpRdBFM97SawPJIC-a5uRzrd5jdb9i5yHDeveQ,2649
-spb_onprem/_version.py,sha256=NvV7p6eu_Rli4DWHJnEcpyTUiImNPPDyoDonzzIsNwA,704
+spb_onprem/_version.py,sha256=94GfHP_pnw8tNHpsnH9lv-lFOG6x-yD9RalL7Ee8XUA,704
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=dPfr3mGXYlqadOXycu6RBFX1HcZ1qzEsskLoOxERLOU,5737
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
@@ -32,7 +32,7 @@ spb_onprem/contents/params/create.py,sha256=JLMbNiIecOsVvVSYY6DfUpWjZxZKuO9-NSpu
 spb_onprem/contents/params/get_download_url.py,sha256=3oaiqmjJRxw0U59SoTxoih1pcU-cuQvcU-g89lCcqtQ,98
 spb_onprem/data/__init__.py,sha256=5XKxNm2BlKufrX5uRQreUEzJ-nerTrozKpG1RJL5wt8,68
 spb_onprem/data/queries.py,sha256=B5vd_8Q8ziwtz44GozbZ-yEN5zKAgMY8wKex-YJJJSs,15607
-spb_onprem/data/service.py,sha256=ItNnGoUrsXcuwNMYJpdEUD9fHnBBb4aCQnHbT5b9_7s,24799
+spb_onprem/data/service.py,sha256=Nt6kO64pZVy_pVVmoGgCPl_1cf6-VHb9fsiD7iigiiU,25876
 spb_onprem/data/entities/__init__.py,sha256=ZQ2YWAHIan2ZLa1z5ozPSJwW3pqdlsJCxmAT3NuG89c,376
 spb_onprem/data/entities/annotation.py,sha256=pS97Nx3TWMBrb-iiBfRct4RXl7hPMy2IOzTIpXkmBJU,751
 spb_onprem/data/entities/data.py,sha256=9wD6fk6K-GDaCesWhxA4vW8IyrlG2Tn7xqR1ZB7VN6k,1293
@@ -108,7 +108,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.5.7.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.5.8.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
@@ -121,7 +121,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.5.7.dist-info/METADATA,sha256=CbJKmxefDAost1yu_Ak16tNRUQBoJUlpaEtbBdwk0Lw,5817
-superb_ai_onprem-0.5.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.5.7.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.5.7.dist-info/RECORD,,
+superb_ai_onprem-0.5.8.dist-info/METADATA,sha256=uuwPhCI6hNstRsrePMgkC2O3caJVLUeChVb8r6do8sI,5817
+superb_ai_onprem-0.5.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.5.8.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.5.8.dist-info/RECORD,,