superb-ai-onprem 0.3.2__py3-none-any.whl → 0.3.4__py3-none-any.whl

spb_onprem/__init__.py

@@ -9,6 +9,7 @@ from .data.service import DataService
 from .slices.service import SliceService
 from .activities.service import ActivityService
 from .exports.service import ExportService
+from .contents.service import ContentService
 
 # Core Entities and Enums
 from .entities import (
@@ -24,6 +25,7 @@ from .entities import (
     Activity,
     ActivityHistory,
     Export,
+    Content,
 
     # Enums
     DataType,
@@ -57,6 +59,7 @@ __all__ = (
     "SliceService",
     "ActivityService",
     "ExportService",
+    "ContentService",
 
     # Core Entities
     "Data",
@@ -70,6 +73,8 @@ __all__ = (
     "Activity",
     "ActivityHistory",
     "Export",
+    "Content",
+    
     # Enums
     "DataType",
     "SceneType",

spb_onprem/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.2'
-__version_tuple__ = version_tuple = (0, 3, 2)
+__version__ = version = '0.3.4'
+__version_tuple__ = version_tuple = (0, 3, 4)

spb_onprem/contents/queries.py

@@ -37,3 +37,21 @@ class Queries:
         ''',
         "variables": create_variables
     }
+    
+    @staticmethod
+    def get_download_url_params(
+        id: str,
+    ):
+        return {
+            "id": id
+        }
+    
+    GET_DOWNLOAD_URL = {
+        "name": "generateContentDownloadURL",
+        "query": '''
+            mutation GenerateContentDownloadURL($id: ID!) {
+                generateContentDownloadURL(id: $id) 
+            }
+        ''',
+        "variables": get_download_url_params
+    }

spb_onprem/contents/service.py

@@ -130,3 +130,21 @@ class ContentService(BaseService):
         # Retrieve the uploaded content details
         content = response['content']
         return BaseContent.model_validate(content)
+
+    def get_download_url(
+        self,
+        id: str,
+    ) -> str:
+        '''
+        Gets the content from the server.
+        Args:
+            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)
+        )
+        return response

spb_onprem/data/enums/data_type.py

@@ -8,3 +8,4 @@ class DataType(str, Enum):
     """
     SUPERB_IMAGE = "SUPERB_IMAGE"
     MCAP = "MCAP"
+    SUPERB_3D = "SUPERB_3D"

spb_onprem/data/enums/scene_type.py

@@ -7,4 +7,5 @@ class SceneType(str, Enum):
     This is used to determine the type of the file.
     """
     IMAGE = "IMAGE"
+    MCAP = "MCAP"
     ETC = "ETC"

spb_onprem/data/params/data.py

@@ -17,8 +17,13 @@ def get_params(
         dict: The variables for the data query.
     """
     
-    return {
+    params = {
         "dataset_id": dataset_id,
-        "key": data_key,
-        "id": data_id,
     }
+    
+    if data_key is not Undefined:
+        params["key"] = data_key
+    if data_id is not Undefined:
+        params["id"] = data_id
+        
+    return params

spb_onprem/entities.py

@@ -22,6 +22,7 @@ from .activities.entities import (
     SchemaType,
 )
 from .exports.entities import Export
+from .contents.entities import Content
 
 __all__ = [
     # Core Entities
@@ -36,6 +37,7 @@ __all__ = [
     "Activity",
     "ActivityHistory",
     "Export",
+    "Content",
 
     # Enums
     "DataType",

{superb_ai_onprem-0.3.2.dist-info → superb_ai_onprem-0.3.4.dist-info}/METADATA

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

@@ -1,9 +1,9 @@
-spb_onprem/__init__.py,sha256=Iucr68oU5pQ3lhHlyI36HUaReA7gsanGuLCSAVZh7GI,1761
-spb_onprem/_version.py,sha256=5NopxuphNnyGZECYEkvIUFi0KZxwtDHmTpW5R266eSo,511
+spb_onprem/__init__.py,sha256=luroRpOs-soHW_qKNeRzkpRR64KXxKPfcpC-wQO4h60,1861
+spb_onprem/_version.py,sha256=Mxups7YfGBY2vvCok_hLJKCtU6O1WHQJfsMY-bAJ0Yg,511
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=dPfr3mGXYlqadOXycu6RBFX1HcZ1qzEsskLoOxERLOU,5737
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
-spb_onprem/entities.py,sha256=VFwaSdHIVB6NhxX5-pFRNPzCqVLNEZS-O2JETIYMopk,823
+spb_onprem/entities.py,sha256=VOFosKVcj85xk8N8SLFYGxZv4EtDAnzzmE3NBj144Tk,877
 spb_onprem/exceptions.py,sha256=jx5rTGsVZ5shOdbgQzk8GcSyMWFtb_3xhPq6Sylwc5o,478
 spb_onprem/searches.py,sha256=VRj9vpvKU8iOdBJUA6BwfvOjUbWrKBo9fPJvNlRmKr0,750
 spb_onprem/activities/__init__.py,sha256=iTHUza8mtiBd_QEPH-0tzrOsd_VSwSQNMbzVWfebQ6c,175
@@ -22,8 +22,8 @@ spb_onprem/activities/params/start_activity.py,sha256=uBf78FYkjGkM5g8shIoMgz8qoC
 spb_onprem/activities/params/update_activity.py,sha256=QU3h2YkPFkqHEHXv6078V2PWEa272h60ST6FYPPZLWU,2615
 spb_onprem/activities/params/update_activity_history.py,sha256=ScWxOG7pZ-7WcTzvJleZIm-TsZVTRkwNrTCyNDyyVSc,1620
 spb_onprem/contents/__init__.py,sha256=9EfIMQxbJuUZVUqsTa3Ji-yDidFPQQB5gnJI4R01YWI,74
-spb_onprem/contents/queries.py,sha256=tGMVH8ixv0CW5bJTWICCjWDM_oAN7jkfEQXdAJUVn4Q,851
-spb_onprem/contents/service.py,sha256=KoPUffr_DEGOMIOwkue0rkidHMPfmAJG0KgJyXN6N_Y,3602
+spb_onprem/contents/queries.py,sha256=idNVcib6Db95n9wkB-tPLX2WKLnvcxMEm08IrOyuxNU,1267
+spb_onprem/contents/service.py,sha256=lYcMfHkEqW4_82ZJQS73YTMvcIgMXv6iCYQi-1V287g,4052
 spb_onprem/contents/entities/__init__.py,sha256=HsQ9J8UDxCx4xYTdMKQto7HCVUQilNozQCkIsceWezk,117
 spb_onprem/contents/entities/base_content.py,sha256=nM7NALpeRjtUKPv7eU0-hlqT1rPtD2mwB6Bvv_2Zc1E,346
 spb_onprem/contents/entities/content.py,sha256=YhTGHE9ykiOgFjvxbnLzSLg2665jPYrDATteB9PbGPE,534
@@ -38,11 +38,11 @@ spb_onprem/data/entities/prediction.py,sha256=Eb2ldNSezeYDnaLQOfC65XWoDGJ0snlvlc
 spb_onprem/data/entities/scene.py,sha256=SJgr5UnGxktyaKjU8FYDaIQmsu7xSJftJOiCgq9uSmo,446
 spb_onprem/data/enums/__init__.py,sha256=IJWaapwesyIiIjuAykZc5fXdMXK2_IiOBa7qNY5cCNk,213
 spb_onprem/data/enums/data_meta_type.py,sha256=9rd12-7C1udbbIGvnuGURKmd5-lndtW7oWQAQwKSf_Q,335
-spb_onprem/data/enums/data_type.py,sha256=BXcmeuWL-y4yIAswI_zsYqH9ozEmgl6uSki7hgbOZ_g,197
-spb_onprem/data/enums/scene_type.py,sha256=ed8fAKfDk9PNG8YgYv3jI59IR9oCr_gkooexAe0448I,187
+spb_onprem/data/enums/data_type.py,sha256=9ErNmOK-RIOrhyt25SE2WVPSx_nTxt3ncTG5UtW92iM,225
+spb_onprem/data/enums/scene_type.py,sha256=pTibjST4nJc3ylUrjQifYBdtLIcS2D0q71bdhyDjbIk,205
 spb_onprem/data/params/__init__.py,sha256=IApFFU6t3lHclvbivLSFdUxhrj1BjO50c3OMG6zP2iY,1311
 spb_onprem/data/params/create_data.py,sha256=8DOfbEchf9GSybiYY1cZoNzNYwILokoAkxXRFybJUAU,2038
-spb_onprem/data/params/data.py,sha256=HXf1a7Peq-WMYGc-R60aJOBGq0nKAK20cZJdgcUim-A,662
+spb_onprem/data/params/data.py,sha256=ZPJRjCbkjOCV73NxlGm5DxbSaNOAztVBRZRE8FFQ9mw,779
 spb_onprem/data/params/data_list.py,sha256=ggh3qzyC5cNufW8xTIMMpg1N2U3FINHMtiem-HMnTqo,3320
 spb_onprem/data/params/delete_annotation_version.py,sha256=R_jBVujor-09VS1Aa8yGP14R8nJ2Aa9OmmKezC4gz4c,457
 spb_onprem/data/params/delete_data.py,sha256=HN4xXW9o4ZFOSjlzRZJ-F0wwtd_RzgI30gVWoULo1p8,319
@@ -93,7 +93,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.3.2.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.3.4.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
@@ -106,7 +106,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.3.2.dist-info/METADATA,sha256=oN3iZHEGaDQ6hr4NX2VmZIkeLk3QzGN5al5gwWTZlKo,5817
-superb_ai_onprem-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.3.2.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.3.2.dist-info/RECORD,,
+superb_ai_onprem-0.3.4.dist-info/METADATA,sha256=JNqfTTOQWsEYEuRd3PAVSm4txEoHjELQNhdMWuDgfQw,5817
+superb_ai_onprem-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.3.4.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.3.4.dist-info/RECORD,,