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

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

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

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

{superb_ai_onprem-0.5.8 → superb_ai_onprem-0.5.10}/spb_onprem/data/entities/data.py

@@ -24,7 +24,6 @@ class Data(CustomBaseModel):
     annotation: Optional[Annotation] = None
     predictions: Optional[List[Prediction]] = None
     meta: Optional[List[DataMeta]] = None
-    system_meta: Optional[List[DataMeta]] = Field(None, alias="systemMeta")
     created_at: Optional[str] = Field(None, alias="createdAt")
     created_by: Optional[str] = Field(None, alias="createdBy")
     updated_at: Optional[str] = Field(None, alias="updatedAt")

{superb_ai_onprem-0.5.8 → superb_ai_onprem-0.5.10}/spb_onprem/data/params/create_data.py

@@ -36,7 +36,7 @@ def create_params(
                     "meta": version.meta,
                 }
                 for version in data.annotation.versions
-            ],
+            ] if data.annotation.versions is not None else [],
             "meta": data.annotation.meta
         } if data.annotation is not None else None,
         "predictions": [
@@ -56,13 +56,5 @@ def create_params(
                 "value": meta.value,
             }
             for meta in data.meta
-        ] if data.meta is not None else None,
-        "systemMeta": [
-            {
-                "key": meta.key,
-                "type": meta.type.value,
-                "value": meta.value,
-            }
-            for meta in data.system_meta
-        ] if data.system_meta is not None else None,
+        ] if data.meta is not None else [],
     }

{superb_ai_onprem-0.5.8 → superb_ai_onprem-0.5.10}/spb_onprem/data/params/update_data.py

@@ -23,10 +23,6 @@ def update_params(
         Optional[List[DataMeta]],
         UndefinedType
     ] = Undefined,
-    system_meta: Union[
-        Optional[List[DataMeta]],
-        UndefinedType
-    ] = Undefined,
 ):
     """Make the variables for the updateData query.
 
@@ -56,17 +52,5 @@ def update_params(
             }
             for meta in meta
         ] if meta is not None else None
-    
-    if system_meta is not Undefined:
-        if system_meta is not None and not isinstance(system_meta, list):
-            raise ValueError("meta must be a list of DataMeta or None.")
-        variables["systemMeta"] = [
-            {
-                "key": meta.key,
-                "type": meta.type.value,
-                "value": meta.value,
-            }
-            for meta in system_meta
-        ] if system_meta is not None else None
 
     return variables

{superb_ai_onprem-0.5.8 → superb_ai_onprem-0.5.10}/spb_onprem/data/params/upsert_data_meta.py

@@ -11,10 +11,6 @@ def upsert_data_meta_params(
         Optional[List[DataMeta]],
         UndefinedType
     ] = Undefined,
-    system_meta: Union[
-        Optional[List[DataMeta]],
-        UndefinedType
-    ] = Undefined,
 ):
     """Make the variables for the upsertDataMeta query.
 
@@ -37,12 +33,4 @@ def upsert_data_meta_params(
             for item in meta
         ] if meta is not None else None
 
-    if system_meta is not Undefined:
-        if system_meta is not None and not isinstance(system_meta, list):
-            raise ValueError("system_meta must be a list of DataMeta or None.")
-        variables["system_meta"] = [
-            item.model_dump(by_alias=True, exclude_unset=True)
-            for item in system_meta
-        ] if system_meta is not None else None
-
     return variables

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

@@ -122,7 +122,6 @@ class Queries():
                 $annotation: AnnotationInput,
                 $predictions: [PredictionInput!],
                 $meta: [DataMetaInput!]
-                $systemMeta: [DataMetaInput!]
             ) {{
             createData(
                 datasetId: $datasetId,
@@ -134,7 +133,6 @@ class Queries():
                 annotation: $annotation,
                 predictions: $predictions,
                 meta: $meta,
-                systemMeta: $systemMeta
             ) 
                 {{
                     {Schemas.DATA}
@@ -151,15 +149,13 @@ class Queries():
                 $dataset_id: ID!,
                 $data_id: ID!,
                 $key: String,
-                $meta: [DataMetaInput!],
-                $systemMeta: [DataMetaInput!]
+                $meta: [DataMetaInput!]
             ) {{
             updateData(
                 datasetId: $dataset_id,
                 id: $data_id,
                 key: $key,
-                meta: $meta,
-                systemMeta: $systemMeta
+                meta: $meta
             ) 
                 {{
                     {Schemas.DATA}

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

@@ -1,19 +1,13 @@
 """
-base_service.py
-
-This module defines the BaseService class, which serves as an abstract base class for services that handle data operations.
+This module defines the DataService class for handling data-related operations.
 
 Classes:
-    BaseService: An abstract base class that requires the implementation of the create_data method.
+    DataService: A service class that provides methods for data management operations.
 """
-from io import BytesIO
 from typing import (
     Optional, List, Union,
 )
 
-from spb_onprem.contents.service import (
-    ContentService
-)
 from spb_onprem.base_service import BaseService
 from spb_onprem.base_types import (
     Undefined,
@@ -22,17 +16,11 @@ from spb_onprem.base_types import (
 from .queries import Queries
 from .entities import (
     Data,
-    Scene,
     AnnotationVersion,
-    Annotation,
-    DataMeta,
     Prediction,
     Frame,
 )
 from .enums import (
-    DataType,
-    SceneType,
-    DataMetaValue,
     DataStatus,
 )
 from .params import (
@@ -177,102 +165,23 @@ 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(
+    def create_data(
         self,
-        dataset_id: str,
-        key: str,
-        image_content: Union[
-            BytesIO,
-            str,
-        ],
-        slices: Optional[List[str]] = None,
-        annotation: Optional[dict] = None,
-        predictions: Optional[List[dict]] = None,
-        meta: Optional[List[dict]] = None,
-        system_meta: Optional[List[dict]] = None,
+        data: Data,
     ):
-        """Create an image data.
+        """Create data in the dataset.
 
         Args:
-            dataset_id (str): The dataset id.
-            key (str): The key of the data.
-            image_content (Union[BytesIO, str]): The image content. If str, it is considered as a file path.
-            slices (Optional[List[str]]): The slices to add the data to.
-            annotation (Optional[dict]): The annotation data.
-            predictions (Optional[List[dict]]): The predictions data.
-            meta (Optional[List[dict]]): The meta data.
-            system_meta (Optional[List[dict]]): The system meta data.
+            data (Data): The data object to create.
 
         Returns:
             Data: The created data.
         """
-        content_service = ContentService()
-        if isinstance(image_content, str):
-            content = content_service.upload_content(
-                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,
-            )
-
         response = self.request_gql(
             Queries.CREATE,
-            Queries.CREATE["variables"](
-                dataset_id=dataset_id,
-                key=key,
-                type=DataType.SUPERB_IMAGE,
-                slices=slices,
-                scene=[{
-                    "id": f"{key}_scene_0",
-                    "type": SceneType.IMAGE,
-                    "content": content.model_dump(by_alias=True),
-                    "meta": {}
-                }],
-                thumbnail=content.model_dump(by_alias=True),
-                annotation=annotation,
-                predictions=predictions,
-                meta=meta,
-                system_meta=system_meta,
-            )
+            Queries.CREATE["variables"](data)
         )
-        data = Data.model_validate(response)
-        return data
+        return Data.model_validate(response)
 
     def update_data(
         self,
@@ -286,10 +195,6 @@ class DataService(BaseService):
             List[dict],
             UndefinedType,
         ] = Undefined,
-        system_meta: Union[
-            List[dict],
-            UndefinedType,
-        ] = Undefined,
     ):
         """Update a data.
 
@@ -298,7 +203,6 @@ class DataService(BaseService):
             data_id (str): The data id.
             key (Union[str, UndefinedType], optional): The key of the data. Defaults to Undefined.
             meta (Union[List[dict], UndefinedType], optional): The meta data. Defaults to Undefined.
-            system_meta (Union[List[dict], UndefinedType], optional): The system meta data. Defaults to Undefined.
 
         Returns:
             Data: The updated data.
@@ -310,7 +214,6 @@ class DataService(BaseService):
                 data_id=data_id,
                 key=key,
                 meta=meta,
-                system_meta=system_meta,
             )
         )
         data = Data.model_validate(response)

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

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