blaxel 0.2.32__py3-none-any.whl → 0.2.34__py3-none-any.whl

blaxel/core/volume/volume.py

@@ -1,15 +1,55 @@
 import uuid
-from typing import Dict, Union
+from typing import Callable, Dict, List, Union
 
 from ..client.api.volumes.create_volume import asyncio as create_volume
+from ..client.api.volumes.create_volume import sync as create_volume_sync
 from ..client.api.volumes.delete_volume import asyncio as delete_volume
+from ..client.api.volumes.delete_volume import sync as delete_volume_sync
 from ..client.api.volumes.get_volume import asyncio as get_volume
+from ..client.api.volumes.get_volume import sync as get_volume_sync
 from ..client.api.volumes.list_volumes import asyncio as list_volumes
+from ..client.api.volumes.list_volumes import sync as list_volumes_sync
 from ..client.client import client
 from ..client.models import Metadata, Volume, VolumeSpec
 from ..client.types import UNSET
 
 
+class _AsyncDeleteDescriptor:
+    """Descriptor that provides both class-level and instance-level delete functionality."""
+
+    def __init__(self, delete_func: Callable):
+        self._delete_func = delete_func
+
+    def __get__(self, instance, owner):
+        if instance is None:
+            # Called on the class: VolumeInstance.delete("name")
+            return self._delete_func
+        else:
+            # Called on an instance: instance.delete()
+            async def instance_delete() -> Volume:
+                return await self._delete_func(instance.metadata.name or "")
+
+            return instance_delete
+
+
+class _SyncDeleteDescriptor:
+    """Descriptor that provides both class-level and instance-level delete functionality (sync)."""
+
+    def __init__(self, delete_func: Callable):
+        self._delete_func = delete_func
+
+    def __get__(self, instance, owner):
+        if instance is None:
+            # Called on the class: SyncVolumeInstance.delete("name")
+            return self._delete_func
+        else:
+            # Called on an instance: instance.delete()
+            def instance_delete() -> Volume:
+                return self._delete_func(instance.metadata.name or "")
+
+            return instance_delete
+
+
 class VolumeCreateConfiguration:
     """Simplified configuration for creating volumes with default values."""
 
@@ -17,12 +57,14 @@ class VolumeCreateConfiguration:
         self,
         name: str | None = None,
         display_name: str | None = None,
+        labels: Dict[str, str] | None = None,
         size: int | None = None,  # Size in MB
         region: str | None = None,  # AWS region
         template: str | None = None,  # Template
     ):
         self.name = name
         self.display_name = display_name
+        self.labels = labels
         self.size = size
         self.region = region
         self.template = template
@@ -32,6 +74,7 @@ class VolumeCreateConfiguration:
         return cls(
             name=data.get("name"),
             display_name=data.get("display_name"),
+            labels=data.get("labels"),
             size=data.get("size"),
             region=data.get("region"),
             template=data.get("template"),
@@ -90,6 +133,7 @@ class VolumeInstance:
                 metadata=Metadata(
                     name=config.name or default_name,
                     display_name=config.display_name or config.name or default_name,
+                    labels=config.labels,
                 ),
                 spec=VolumeSpec(
                     size=config.size or default_size,
@@ -103,6 +147,7 @@ class VolumeInstance:
                 metadata=Metadata(
                     name=volume_config.name or default_name,
                     display_name=volume_config.display_name or volume_config.name or default_name,
+                    labels=volume_config.labels,
                 ),
                 spec=VolumeSpec(
                     size=volume_config.size or default_size,
@@ -138,11 +183,6 @@ class VolumeInstance:
         response = await list_volumes(client=client)
         return [cls(volume) for volume in response or []]
 
-    @classmethod
-    async def delete(cls, volume_name: str) -> Volume:
-        response = await delete_volume(volume_name=volume_name, client=client)
-        return response
-
     @classmethod
     async def create_if_not_exists(
         cls, config: Union[VolumeCreateConfiguration, Volume, Dict[str, any]]
@@ -171,3 +211,157 @@ class VolumeInstance:
                 volume_instance = await cls.get(name)
                 return volume_instance
             raise e
+
+
+class SyncVolumeInstance:
+    """Synchronous volume instance for managing persistent storage."""
+
+    def __init__(self, volume: Volume):
+        self.volume = volume
+
+    @property
+    def metadata(self):
+        return self.volume.metadata
+
+    @property
+    def spec(self):
+        return self.volume.spec
+
+    @property
+    def status(self):
+        return self.volume.status
+
+    @property
+    def name(self):
+        return self.volume.metadata.name if self.volume.metadata else None
+
+    @property
+    def display_name(self):
+        return self.volume.metadata.display_name if self.volume.metadata else None
+
+    @property
+    def size(self):
+        return self.volume.spec.size if self.volume.spec else None
+
+    @property
+    def region(self):
+        return self.volume.spec.region if self.volume.spec else None
+
+    @property
+    def template(self):
+        return self.volume.spec.template if self.volume.spec else None
+
+    @classmethod
+    def create(
+        cls, config: Union[VolumeCreateConfiguration, Volume, Dict[str, any]]
+    ) -> "SyncVolumeInstance":
+        """Create a new volume synchronously."""
+        # Generate default values
+        default_name = f"volume-{uuid.uuid4().hex[:8]}"
+        default_size = 1024  # 1GB in MB
+
+        # Handle different configuration types
+        if isinstance(config, Volume):
+            volume = config
+        elif isinstance(config, VolumeCreateConfiguration):
+            volume = Volume(
+                metadata=Metadata(
+                    name=config.name or default_name,
+                    display_name=config.display_name or config.name or default_name,
+                    labels=config.labels,
+                ),
+                spec=VolumeSpec(
+                    size=config.size or default_size,
+                    region=config.region or UNSET,
+                    template=config.template or UNSET,
+                ),
+            )
+        elif isinstance(config, dict):
+            volume_config = VolumeCreateConfiguration.from_dict(config)
+            volume = Volume(
+                metadata=Metadata(
+                    name=volume_config.name or default_name,
+                    display_name=volume_config.display_name or volume_config.name or default_name,
+                    labels=volume_config.labels,
+                ),
+                spec=VolumeSpec(
+                    size=volume_config.size or default_size,
+                    region=volume_config.region or UNSET,
+                    template=volume_config.template or UNSET,
+                ),
+            )
+        else:
+            raise ValueError(
+                f"Invalid config type: {type(config)}. Expected VolumeCreateConfiguration, Volume, or dict."
+            )
+
+        # Ensure required fields have defaults
+        if not volume.metadata:
+            volume.metadata = Metadata(name=default_name)
+        if not volume.metadata.name:
+            volume.metadata.name = default_name
+        if not volume.spec:
+            volume.spec = VolumeSpec(size=default_size)
+        if not volume.spec.size:
+            volume.spec.size = default_size
+
+        response = create_volume_sync(client=client, body=volume)
+        return cls(response)
+
+    @classmethod
+    def get(cls, volume_name: str) -> "SyncVolumeInstance":
+        """Get a volume by name synchronously."""
+        response = get_volume_sync(volume_name=volume_name, client=client)
+        return cls(response)
+
+    @classmethod
+    def list(cls) -> List["SyncVolumeInstance"]:
+        """List all volumes synchronously."""
+        response = list_volumes_sync(client=client)
+        return [cls(volume) for volume in response or []]
+
+    @classmethod
+    def create_if_not_exists(
+        cls, config: Union[VolumeCreateConfiguration, Volume, Dict[str, any]]
+    ) -> "SyncVolumeInstance":
+        """Create a volume if it doesn't exist, otherwise return existing."""
+        try:
+            return cls.create(config)
+        except Exception as e:
+            # Check if it's a 409 conflict error (volume already exists)
+            if (hasattr(e, "status_code") and e.status_code == 409) or (
+                hasattr(e, "code") and e.code in [409, "VOLUME_ALREADY_EXISTS"]
+            ):
+                # Extract name from different configuration types
+                if isinstance(config, VolumeCreateConfiguration):
+                    name = config.name
+                elif isinstance(config, dict):
+                    name = config.get("name")
+                elif isinstance(config, Volume):
+                    name = config.metadata.name if config.metadata else None
+                else:
+                    name = None
+
+                if not name:
+                    raise ValueError("Volume name is required")
+
+                volume_instance = cls.get(name)
+                return volume_instance
+            raise e
+
+
+async def _delete_volume_by_name(volume_name: str) -> Volume:
+    """Delete a volume by name (async)."""
+    response = await delete_volume(volume_name=volume_name, client=client)
+    return response
+
+
+def _delete_volume_by_name_sync(volume_name: str) -> Volume:
+    """Delete a volume by name (sync)."""
+    response = delete_volume_sync(volume_name=volume_name, client=client)
+    return response
+
+
+# Assign the delete descriptors to support both class-level and instance-level calls
+VolumeInstance.delete = _AsyncDeleteDescriptor(_delete_volume_by_name)
+SyncVolumeInstance.delete = _SyncDeleteDescriptor(_delete_volume_by_name_sync)

{blaxel-0.2.32.dist-info → blaxel-0.2.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.32
+Version: 0.2.34
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai
@@ -17,7 +17,6 @@ Requires-Dist: pyjwt>=2.0.0
 Requires-Dist: python-dateutil>=2.8.0
 Requires-Dist: pyyaml>=6.0.0
 Requires-Dist: requests>=2.32.3
-Requires-Dist: sentry-sdk>=2.46.0
 Requires-Dist: tomli>=2.2.1
 Requires-Dist: websockets<16.0.0
 Provides-Extra: all

{blaxel-0.2.32.dist-info → blaxel-0.2.34.dist-info}/RECORD

@@ -1,5 +1,5 @@
-blaxel/__init__.py,sha256=DodOCaAPXa8pstbUXBPzRjID-7e_I2akWFDmtctRxkc,413
-blaxel/core/__init__.py,sha256=CKMC7TaCYdOdnwqcJCN9VjBbNg366coZUGTxI1mgFQQ,1710
+blaxel/__init__.py,sha256=JFHDlTbbIPz2qG5ETm0mcQ1NX7AXph_tw6L5PVKLXzg,413
+blaxel/core/__init__.py,sha256=CU0gXpVRbuQZNWoCJuuhZS0ZhXPEu0cg-3XzoYMrBm4,1756
 blaxel/core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 blaxel/core/agents/__init__.py,sha256=MJZga99lU8JWUUPHd4rmUfdo7ALwWgF7CQq95SfT2OI,4456
 blaxel/core/authentication/__init__.py,sha256=5uHv73xZ2aZRXzGHc1sP-XWqZUryfXxfC9dMX7-gd6o,3241
@@ -330,11 +330,12 @@ blaxel/core/client/models/workspace.py,sha256=3Z-bpTYukkbqW46B39Nljr4XZtAV41AqqK
 blaxel/core/client/models/workspace_labels.py,sha256=M_ZbpQeVNdgqWv2RmTYO7znBczUJ6zi5_XS1EDin3Gk,1253
 blaxel/core/client/models/workspace_runtime.py,sha256=Pdr_Q-ugs_5famziqGSyBzUkKQ6e-pAFRGza6GhppSA,1672
 blaxel/core/client/models/workspace_user.py,sha256=f28XZuP3V46y2YYbfiIiNbnIMpy3JINLwpy_mX9N1iU,3430
-blaxel/core/common/__init__.py,sha256=K5uIZe98NIjYkjMPdKvP5RuwzIfv1Obm6g3aWh2qjZ0,599
-blaxel/core/common/autoload.py,sha256=1v6-MwjifmPLnfFA11deNYxydSPTKIawc3nQGOb4l50,6375
+blaxel/core/common/__init__.py,sha256=A69U94NslPOYjKvja0bN5z98eguJPC6UaGaVN2-FdRY,736
+blaxel/core/common/autoload.py,sha256=Tq6cmmNcOwUy77ACWM-4Ap8DgRY6GWkAOJmIwugRAPU,1481
 blaxel/core/common/env.py,sha256=05Jm2mw0-KIgR7QaNVyvZPP91B9OlxlJ6mcx6Mqfji0,1234
 blaxel/core/common/internal.py,sha256=NDTFh9Duj84os8GkMXjGzn-UVS9zBDyLAcfPxIpoQGA,3218
 blaxel/core/common/logger.py,sha256=Jt0MCJgYDPq36rl7UyKRDJH76a-AwYdfggNeNYJt6N0,4779
+blaxel/core/common/sentry.py,sha256=P_v1vWivlh-usXV1_JeJ603r1OHoWaR5jaYV2JJ6vDM,9759
 blaxel/core/common/settings.py,sha256=f7ZP8VpB8gKESptyb6S82gTHQsbmx6RcaAkDkxy2JpE,4599
 blaxel/core/common/webhook.py,sha256=N1f2bamP7wRyPyCfmAZKMdjeB3aQ6d6pcafHyVZKtPk,5330
 blaxel/core/jobs/__init__.py,sha256=LZTtkOwqUyMjRTdeLv4EXLbVhgebwvY95jd56IGO4MQ,17082
@@ -343,7 +344,7 @@ blaxel/core/mcp/client.py,sha256=EZ7l5w3bTXaD41nalHzM-byxfQK-JdcmQqxg3zGpVO4,550
 blaxel/core/mcp/server.py,sha256=edAztWBlukERw9-dzS2Sk96TP8R3-CSofY1CZDu19ZA,5967
 blaxel/core/models/__init__.py,sha256=ydz1txqIVyOhehItut-AOnLMnGp7AtCD2zku9gkvAsE,1722
 blaxel/core/sandbox/__init__.py,sha256=SQ9YH0I9ruAYscY_W75jqIhNbejbY_92xL7VqM083Mc,1345
-blaxel/core/sandbox/types.py,sha256=h_e0Zk_BNRjix1RRtomqVwdG64ViA34KwM4ZQIM5Bp0,12601
+blaxel/core/sandbox/types.py,sha256=mi2fbD5vvPzgyDbmN1YA-LkFS6Iaq0dLtI4OxQDb16s,12817
 blaxel/core/sandbox/client/__init__.py,sha256=N26bD5o1jsTb48oExow6Rgivd8ylaU9jaWZfZsVilP8,128
 blaxel/core/sandbox/client/client.py,sha256=EGCYliUHCk4RyIBlydEZyVpc_VUiIIGPuu2E-xYeKFY,7074
 blaxel/core/sandbox/client/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
@@ -438,8 +439,8 @@ blaxel/core/sandbox/default/filesystem.py,sha256=Gn1G3DpMmDXplUEpXVENXkjUDAEkWAK
 blaxel/core/sandbox/default/interpreter.py,sha256=TJSryQvq2rWedyhMU69tOlXVOF1iIpgIbV2z3_mF72E,11316
 blaxel/core/sandbox/default/network.py,sha256=3ZvrJB_9JdZrclNkwifZOIciz2OqzV0LQfbebjZXLIY,358
 blaxel/core/sandbox/default/preview.py,sha256=dV_xuu9Efop5TnzuFJPeLUZ7CEepuYkJedx01fDVMX4,6132
-blaxel/core/sandbox/default/process.py,sha256=hUdFxFD5w8Md827gVwRijhY55D4BQEbmxuLrqFrCC6I,11965
-blaxel/core/sandbox/default/sandbox.py,sha256=b641_MHILDdsz9ZkbSIoRAmbB_0grbd3C_SG5JOwe18,12576
+blaxel/core/sandbox/default/process.py,sha256=7nI1wJXeZWoveBesC13wur-ghIjP5POZ38G8wqCdJTw,16983
+blaxel/core/sandbox/default/sandbox.py,sha256=B7MGnVnelV4iOh1PSBWbMwO6Vq6KNNxpQQ42mMNgShU,13485
 blaxel/core/sandbox/default/session.py,sha256=XzVpPOH_az6T38Opp4Hmj3RIg7QCzA1l5wh1YDh7czc,5313
 blaxel/core/sandbox/sync/__init__.py,sha256=iqTRxQYbJyHTXoA4MHaigeXFxi9wtJ3o9XygZuFe3bM,372
 blaxel/core/sandbox/sync/action.py,sha256=9t8nCS9KX6SBflJmzK8dsen_bDim3zElBTebY6N6JQc,2136
@@ -448,14 +449,14 @@ blaxel/core/sandbox/sync/filesystem.py,sha256=FoxM9EJ5sXGysf-x22tbt9yrcbbpaunTD3
 blaxel/core/sandbox/sync/interpreter.py,sha256=5cAzwnt5BgnByGimagMBotjGW2vMAz4vutBBrrFV9-A,11062
 blaxel/core/sandbox/sync/network.py,sha256=QkCFKfFayvwL1J4JYwOuXPGlYQuX4J9Jj55Kf_kD-ig,283
 blaxel/core/sandbox/sync/preview.py,sha256=w3bC8iA3QecHiLkRvITmQ6LTT9Co_93G24QpZFgEQSE,6379
-blaxel/core/sandbox/sync/process.py,sha256=U4aYv-uRXDOKScz6Ets-QkwZZ1kdgR9ve04CkUaUkEU,9624
-blaxel/core/sandbox/sync/sandbox.py,sha256=jNqwMJXIpjC8Fs9nn-ujfSpFK2PgCvEePISKQKaMhH0,10515
+blaxel/core/sandbox/sync/process.py,sha256=W-ZUM6VyFDxTmexHTQn9PI6iRc0QiB9JMOEq__r2bBA,14913
+blaxel/core/sandbox/sync/sandbox.py,sha256=mz1_AJkoIBIgrAtX5pGn5h3lhmKI1jtKB75fMmoJ3UM,11418
 blaxel/core/sandbox/sync/session.py,sha256=e0CVbW2LBRYTwm4RL52S0UdNvhNfuFLo6AYE5hk9DH0,4931
 blaxel/core/tools/__init__.py,sha256=OK2TFqeXAIi6CC7xtL8fFl-4DvCB7jjihkhx6RTld_c,13147
 blaxel/core/tools/common.py,sha256=fjapOcJwFCS1Do9FCwqmdDTXpo1pZ-9vbfSjadJ9l8g,1321
 blaxel/core/tools/types.py,sha256=EXa-10iOOXvd8zB2IsTS0gWrgoC2hqbIv6iK5m6E5t8,693
-blaxel/core/volume/__init__.py,sha256=W7MLNeA5AvpvE1NwyTzWrkwypNaj6xTIbRmILtoaItE,177
-blaxel/core/volume/volume.py,sha256=10j4fnhsX9IDBSEI-KG5RrAZc2Q5bLnedy9-E_ToUyU,6068
+blaxel/core/volume/__init__.py,sha256=YgEquO_GVxtcymSZoIJTEPvL8KeXtHIobNjGecEKtwM,219
+blaxel/core/volume/volume.py,sha256=24ihurOCGfvadeWNvJCEI3OAK1UpLUWpYacALS8u0MI,13153
 blaxel/crewai/__init__.py,sha256=HdAZxRDgC2zW-O5xYzb_Xuysb66_fMykNJVPxlTmFnY,123
 blaxel/crewai/model.py,sha256=iR-40o8pGBFrktYNKcTaMdqbF_F1SFVavNOhhi2fW3w,1957
 blaxel/crewai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -500,7 +501,7 @@ blaxel/telemetry/instrumentation/map.py,sha256=PCzZJj39yiYVYJrxLBNP-NW-tjjYyTijw
 blaxel/telemetry/instrumentation/utils.py,sha256=FGyMY5ZE4f-0JdZpm_R_BCoKLJ18hftz8vsh7ftDwMk,1889
 blaxel/telemetry/log/log.py,sha256=vtzUIFIIj4MTTKUigILDYXN8NHHPOo44OaKukpyIjQg,2407
 blaxel/telemetry/log/logger.py,sha256=IcFWCd1yyWWGAjAd2i0pDYqpZHQ61pmcaQ7Kf4bC8lg,4150
-blaxel-0.2.32.dist-info/METADATA,sha256=yQQO_JdQpH1Ka7_HxoTU2he4DXnWLWh8--eD8LWzdZc,10108
-blaxel-0.2.32.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-blaxel-0.2.32.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.32.dist-info/RECORD,,
+blaxel-0.2.34.dist-info/METADATA,sha256=L6AhAF7TXWMZCy4FRqjnkHYUEOl-WBxbDO09pVzZdv0,10074
+blaxel-0.2.34.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+blaxel-0.2.34.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.34.dist-info/RECORD,,