superb-ai-onprem 0.8.3__tar.gz → 0.9.1__tar.gz

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.8.3
+Version: 0.9.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.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/__init__.py

@@ -10,7 +10,6 @@ from .slices.service import SliceService
 from .activities.service import ActivityService
 from .exports.service import ExportService
 from .contents.service import ContentService
-from .predictions.service import PredictionService
 from .models.service import ModelService
 from .inferences.service import InferService
 
@@ -21,7 +20,6 @@ from .entities import (
     Scene,
     Annotation,
     AnnotationVersion,
-    Prediction,
     DataMeta,
     Dataset,
     Slice,
@@ -90,7 +88,6 @@ __all__ = (
     "ActivityService",
     "ExportService",
     "ContentService",
-    "PredictionService",
     "ModelService",
     "InferService",
 
@@ -99,7 +96,6 @@ __all__ = (
     "Scene",
     "Annotation",
     "AnnotationVersion",
-    "Prediction",
     "DataMeta",
     "Dataset",
     "Slice",
@@ -109,7 +105,6 @@ __all__ = (
     "Export",
     "Content",
     "Frame",
-    "PredictionSet",
     "Model",
     "Comment",
     "Reply",

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.8.3'
-__version_tuple__ = version_tuple = (0, 8, 3)
+__version__ = version = '0.9.1'
+__version_tuple__ = version_tuple = (0, 9, 1)
 
-__commit_id__ = commit_id = 'gb54e702a6'
+__commit_id__ = commit_id = 'gbc82964b9'

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/data/README.md

@@ -199,7 +199,6 @@ For detailed entity documentation with comprehensive field descriptions, see the
 - **[🎬 Frame](entities/frame.py)** - Video frame entities with timestamp/geo data
 - **[🎭 Scene](entities/scene.py)** - File content representation (images, videos, text)
 - **[🏷️ DataMeta](entities/data_meta.py)** - Structured custom metadata entities
-- **[🔮 Prediction](entities/prediction.py)** - ML model prediction results with confidence scores
 
 Each entity file contains:
 - **Comprehensive class documentation**
@@ -219,7 +218,6 @@ from spb_onprem.data.entities import (
     Frame,          # Video frame data with timestamps and geo-location
     Scene,          # File content representation (images, videos, documents)  
     DataMeta,       # Structured custom metadata with type validation
-    Prediction      # ML model results with confidence and performance metrics
 )
 
 # Entity relationship example with comment system
@@ -240,7 +238,6 @@ data = Data(
             )
         ]
     ),
-    predictions=[Prediction(set_id="model_v1.0", meta={"confidence": 0.95})],
     meta=[DataMeta(key="camera_model", value="Canon EOS", type=DataMetaTypes.STRING)]
 )
 ```

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/data/entities/__init__.py

@@ -2,7 +2,6 @@ from .annotation import Annotation, AnnotationVersion
 from .comment import Comment, Reply
 from .data_meta import DataMeta
 from .data import Data
-from .prediction import Prediction
 from .scene import Scene
 from .data_slice import DataSlice
 from .frame import Frame
@@ -16,7 +15,6 @@ __all__ = (
     "AnnotationVersion",
     "Comment",
     "Reply",
-    "Prediction",
     "DataMeta",
     "DataSlice",
 )

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/data/entities/data.py

@@ -3,7 +3,6 @@ from spb_onprem.base_model import CustomBaseModel, Field
 from spb_onprem.data.enums import DataType
 from .scene import Scene
 from .annotation import Annotation
-from .prediction import Prediction
 from .data_meta import DataMeta
 from .data_slice import DataSlice
 from .frame import Frame
@@ -19,7 +18,6 @@ class Data(CustomBaseModel):
     # 식별자
     id: Optional[str] = Field(None, description="데이터 고유 식별자")
     dataset_id: Optional[str] = Field(None, alias="datasetId", description="상위 데이터셋 ID")
-    slice_ids: Optional[List[str]] = Field(None, alias="sliceIds", description="소속된 슬라이스 ID 목록")
     key: Optional[str] = Field(None, description="사용자 정의 고유 키")
     
     # 데이터 타입 및 내용
@@ -30,8 +28,7 @@ class Data(CustomBaseModel):
     
     # 어노테이션 및 예측
     annotation: Optional[Annotation] = Field(None, description="어노테이션 데이터")
-    predictions: Optional[List[Prediction]] = Field(None, description="ML 모델 예측 결과")
-    
+
     # 메타데이터
     meta: Optional[List[DataMeta]] = Field(None, description="커스텀 메타데이터 목록")
     

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/data/params/__init__.py

@@ -37,9 +37,6 @@ from .data_list import (
 from .get_data_detail import (
     get_data_detail_params,
 )
-from .get_evaluation_value_list import (
-    get_evaluation_value_list_params,
-)
 from .remove_data_from_slice import (
     remove_data_from_slice_params
 )
@@ -49,12 +46,6 @@ from .insert_data_to_slice import (
 from .delete_data import (
     delete_data_params
 )
-from .insert_prediction import (
-    insert_prediction_params
-)
-from .delete_prediction import (
-    delete_prediction_params
-)
 from .update_annotation import (
     update_annotation_params
 )
@@ -99,7 +90,6 @@ __all__ = [
     "get_data_id_list_params",
     "get_data_list_params",
     "get_data_detail_params",
-    "get_evaluation_value_list_params",
     "DateTimeRangeFilterOption",
     "UserFilterOption",
     "NumericRangeFilter",
@@ -126,8 +116,6 @@ __all__ = [
     "remove_data_from_slice_params",
     "insert_data_to_slice_params",
     "delete_data_params",
-    "insert_prediction_params",
-    "delete_prediction_params",
     "update_annotation_params",
     "insert_annotation_version_params",
     "delete_annotation_version_params",

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/data/params/create_data.py

@@ -13,7 +13,7 @@ def create_params(
         "datasetId": data.dataset_id,
         "key": data.key,
         "type": data.type.value,
-        "slices": data.slice_ids if data.slice_ids is not None else [],
+        "slices": [slice.id for slice in data.slices] if data.slices is not None else [],
         "scene": [
             {
                 "type": scene.type.value,
@@ -39,16 +39,6 @@ def create_params(
             ] if data.annotation.versions is not None else [],
             "meta": data.annotation.meta
         } if data.annotation is not None else None,
-        "predictions": [
-            {
-                "setId": prediction.set_id,
-                "content": {
-                    "id": prediction.content.id,
-                },
-                "meta": prediction.meta,
-            }
-            for prediction in data.predictions
-        ] if data.predictions is not None else [],
         "meta": [
             {
                 "key": meta.key,

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/data/queries.py

@@ -5,12 +5,9 @@ from .params import (
     get_data_id_list_params,
     get_data_list_params,
     get_data_detail_params,
-    get_evaluation_value_list_params,
     remove_data_from_slice_params,
     insert_data_to_slice_params,
     delete_data_params,
-    insert_prediction_params,
-    delete_prediction_params,
     update_annotation_params,
     insert_annotation_version_params,
     delete_annotation_version_params,
@@ -42,7 +39,6 @@ class Schemas:
     DATA = '''
         id
         datasetId
-        sliceIds
         key
         type
         scene {
@@ -120,13 +116,6 @@ class Schemas:
             }
             meta
         }
-        predictions {
-            setId
-            content {
-                id
-            }
-            meta
-        }
         thumbnail {
             id
         }
@@ -155,7 +144,6 @@ class Queries():
                 $scene: [SceneInput!],
                 $thumbnail: ContentBaseInput,
                 $annotation: AnnotationInput,
-                $predictions: [PredictionInput!],
                 $meta: [DataMetaInput!]
             ) {{
             createData(
@@ -166,7 +154,6 @@ class Queries():
                 scene: $scene,
                 thumbnail: $thumbnail,
                 annotation: $annotation,
-                predictions: $predictions,
                 meta: $meta,
             ) 
                 {{
@@ -325,46 +312,6 @@ class Queries():
         ''',
         "variables": delete_data_params,
     }
-
-    INSERT_PREDICTION = {
-        "name": "insertPrediction",
-        "query": f'''
-            mutation (
-                $dataset_id: ID!,
-                $data_id: ID!,
-                $prediction: PredictionInput!,
-            ) {{
-                insertPrediction(
-                    datasetId: $dataset_id,
-                    dataId: $data_id,
-                    prediction: $prediction,
-                ) {{
-                    {Schemas.DATA}
-                }}
-            }}
-        ''',
-        "variables": insert_prediction_params,
-    }
-
-    DELETE_PREDICTION = {
-        "name": "deletePrediction",
-        "query": f'''
-            mutation (
-                $dataset_id: ID!,
-                $data_id: ID!,
-                $set_id: ID!,
-            ) {{
-                deletePrediction(
-                    datasetId: $dataset_id,
-                    dataId: $data_id,
-                    setId: $set_id,
-                ) {{
-                    {Schemas.DATA}
-                }}
-            }}
-        ''',
-        "variables": delete_prediction_params,
-    }
     
     UPDATE_ANNOTATION = {
         "name": "updateAnnotation",
@@ -627,31 +574,3 @@ class Queries():
         ''',
         "variables": update_frames_params,
     }
-    
-    GET_EVALUATION_VALUE_LIST = {
-        "name": "evaluationValueList",
-        "query": '''
-            query GetEvaluationValueList(
-                $datasetId: String!,
-                $predictionSetId: String!,
-                $filter: DiagnosisFilter,
-                $length: Int,
-                $cursor: String
-            ) {
-                evaluationValueList(
-                    datasetId: $datasetId,
-                    predictionSetId: $predictionSetId,
-                    filter: $filter,
-                    length: $length,
-                    cursor: $cursor
-                ) {
-                    totalCount
-                    next
-                    data {
-                        dataId
-                    }
-                }
-            }
-        ''',
-        "variables": get_evaluation_value_list_params
-    }

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/data/service.py

@@ -17,7 +17,6 @@ from .queries import Queries
 from .entities import (
     Data,
     AnnotationVersion,
-    Prediction,
     Frame,
     DataMeta,
 )
@@ -297,60 +296,6 @@ class DataService(BaseService):
         )
         return response
 
-    def insert_prediction(
-        self,
-        dataset_id: str,
-        data_id: str,
-        prediction: Prediction,
-    ):
-        """Insert a prediction.
-
-        Args:
-            dataset_id (str): The dataset id.
-            data_id (str): The data id.
-            prediction (Prediction): The prediction.
-
-        Returns:
-            Data: The updated data.
-        """
-        response = self.request_gql(
-            Queries.INSERT_PREDICTION,
-            Queries.INSERT_PREDICTION["variables"](
-                dataset_id=dataset_id,
-                data_id=data_id,
-                prediction=prediction,
-            )
-        )
-        data = Data.model_validate(response)
-        return data
-    
-    def delete_prediction(
-        self,
-        dataset_id: str,
-        data_id: str,
-        set_id: str,
-    ):
-        """Delete a prediction.
-
-        Args:
-            dataset_id (str): The dataset id.
-            data_id (str): The data id.
-            set_id (str): The set id.
-
-        Returns:
-            Data: The updated data.
-        """
-        response = self.request_gql(
-            Queries.DELETE_PREDICTION,
-            Queries.DELETE_PREDICTION["variables"](
-                dataset_id=dataset_id,
-                data_id=data_id,
-                set_id=set_id,
-            )
-        )
-        data = Data.model_validate(response)
-        return data
-    
     def update_annotation(
         self,
         dataset_id: str,
@@ -781,66 +726,3 @@ class DataService(BaseService):
         )
         data = Data.model_validate(response)
         return data
-    
-    def get_evaluation_value_list(
-        self,
-        dataset_id: str,
-        prediction_set_id: str,
-        filter: Union[UndefinedType, dict] = Undefined,
-        length: int = 50,
-        cursor: Union[UndefinedType, str] = Undefined
-    ) -> dict:
-        """Get evaluation values list for diagnosis filtering.
-        
-        Retrieves evaluation values for diagnosis filtering with pagination support.
-        
-        Args:
-            dataset_id (str): The dataset ID.
-            prediction_set_id (str): The prediction set ID.
-            filter (Union[UndefinedType, dict]): Diagnosis filter for evaluation values.
-            length (int): Number of items to retrieve per page.
-            cursor (Union[UndefinedType, str]): Cursor for pagination.
-        
-        Returns:
-            dict: Response containing totalCount, next cursor, and data list with dataId fields.
-        """
-        if dataset_id is None:
-            raise BadParameterError("dataset_id is required.")
-        if prediction_set_id is None:
-            raise BadParameterError("prediction_set_id is required.")
-
-        response = self.request_gql(
-            Queries.GET_EVALUATION_VALUE_LIST,
-            Queries.GET_EVALUATION_VALUE_LIST["variables"](
-                dataset_id=dataset_id,
-                prediction_set_id=prediction_set_id,
-                filter=filter,
-                length=length,
-                cursor=cursor
-            )
-        )
-        return response.get("evaluationValueList", {})
-    
-    def get_total_data_id_count_in_evaluation_value(
-        self,
-        dataset_id: str,
-        prediction_set_id: str,
-        filter: Union[UndefinedType, dict] = Undefined
-    ) -> int:
-        """Get total count of data IDs in evaluation values for diagnosis filtering.
-        
-        Args:
-            dataset_id (str): The dataset ID.
-            prediction_set_id (str): The prediction set ID.
-            filter (Union[UndefinedType, dict]): Diagnosis filter for evaluation values.
-        
-        Returns:
-            int: Total count of evaluation values.
-        """
-        result = self.get_evaluation_value_list(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id,
-            filter=filter,
-            length=1
-        )
-        return result.get("totalCount", 0)

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/spb_onprem/entities.py

@@ -3,7 +3,6 @@ from .data.entities import (
     Scene,
     Annotation,
     AnnotationVersion,
-    Prediction,
     DataMeta,
     DataSlice,
     Frame,
@@ -29,7 +28,6 @@ from .activities.entities import (
 )
 from .exports.entities import Export
 from .contents.entities import Content
-from .predictions.entities import PredictionSet
 from .models.entities import Model
 
 __all__ = [
@@ -38,7 +36,6 @@ __all__ = [
     "Scene",
     "Annotation",
     "AnnotationVersion",
-    "Prediction",
     "DataMeta",
     "DataSlice",
     "Dataset",
@@ -48,7 +45,6 @@ __all__ = [
     "Export",
     "Content",
     "Frame",
-    "PredictionSet",
     "Model",
     "Comment",
     "Reply",

{superb_ai_onprem-0.8.3 → superb_ai_onprem-0.9.1}/superb_ai_onprem.egg-info/PKG-INFO

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

@@ -72,14 +72,11 @@ spb_onprem/data/params/data.py
 spb_onprem/data/params/data_list.py
 spb_onprem/data/params/delete_annotation_version.py
 spb_onprem/data/params/delete_data.py
-spb_onprem/data/params/delete_prediction.py
 spb_onprem/data/params/delete_scene.py
 spb_onprem/data/params/delete_slice_annotation_version.py
 spb_onprem/data/params/get_data_detail.py
-spb_onprem/data/params/get_evaluation_value_list.py
 spb_onprem/data/params/insert_annotation_version.py
 spb_onprem/data/params/insert_data_to_slice.py
-spb_onprem/data/params/insert_prediction.py
 spb_onprem/data/params/insert_scene.py
 spb_onprem/data/params/insert_slice_annotation_version.py
 spb_onprem/data/params/remove_data_from_slice.py