blaxel 0.2.1rc76__py3-none-any.whl → 0.2.1rc78__py3-none-any.whl

blaxel/core/sandbox/__init__.py

@@ -8,6 +8,7 @@ from .sandbox import (
 from .types import (
     CopyResponse,
     SandboxConfiguration,
+    SandboxCreateConfiguration,
     SandboxFilesystemFile,
     SessionCreateOptions,
     SessionWithToken,
@@ -19,6 +20,7 @@ __all__ = [
     "SessionCreateOptions",
     "SessionWithToken",
     "SandboxConfiguration",
+    "SandboxCreateConfiguration",
     "WatchEvent",
     "SandboxFilesystemFile",
     "CopyResponse",

blaxel/core/sandbox/sandbox.py

@@ -15,7 +15,7 @@ from .network import SandboxNetwork
 from .preview import SandboxPreviews
 from .process import SandboxProcess
 from .session import SandboxSessions
-from .types import SandboxConfiguration, SessionWithToken
+from .types import SandboxConfiguration, SandboxCreateConfiguration, SessionWithToken
 
 logger = logging.getLogger(__name__)
 
@@ -69,22 +69,53 @@ class SandboxInstance:
 
     @classmethod
     async def create(
-        cls, sandbox: Union[Sandbox, Dict[str, Any], None] = None
+        cls, sandbox: Union[Sandbox, SandboxCreateConfiguration, Dict[str, Any], None] = None
     ) -> "SandboxInstance":
-        if sandbox is None:
-            sandbox = Sandbox()
-        if isinstance(sandbox, dict):
-            sandbox = Sandbox.from_dict(sandbox)
-        if not sandbox.metadata:
-            sandbox.metadata = Metadata(name=uuid.uuid4().hex)
-        if not sandbox.spec:
-            sandbox.spec = SandboxSpec(runtime=Runtime(image="blaxel/prod-base:latest"))
-        if not sandbox.spec.runtime:
-            sandbox.spec.runtime = Runtime(image="blaxel/prod-base:latest")
-
-        sandbox.spec.runtime.image = sandbox.spec.runtime.image or "blaxel/prod-base:latest"
-        sandbox.spec.runtime.memory = sandbox.spec.runtime.memory or 4096
-        sandbox.spec.runtime.generation = sandbox.spec.runtime.generation or "mk3"
+        # Generate default values
+        default_name = f"sandbox-{uuid.uuid4().hex[:8]}"
+        default_image = "blaxel/prod-base:latest"
+        default_memory = 4096
+
+        # Handle SandboxCreateConfiguration or simple dict with name/image/memory keys
+        if sandbox is None or isinstance(sandbox, (SandboxCreateConfiguration, dict)) and (
+            not isinstance(sandbox, Sandbox) and (
+                sandbox is None or
+                'name' in (sandbox if isinstance(sandbox, dict) else sandbox.__dict__) or
+                'image' in (sandbox if isinstance(sandbox, dict) else sandbox.__dict__) or
+                'memory' in (sandbox if isinstance(sandbox, dict) else sandbox.__dict__)
+            )
+        ):
+            if sandbox is None:
+                sandbox = SandboxCreateConfiguration()
+            elif isinstance(sandbox, dict) and not isinstance(sandbox, Sandbox):
+                sandbox = SandboxCreateConfiguration.from_dict(sandbox)
+
+            # Set defaults if not provided
+            name = sandbox.name or default_name
+            image = sandbox.image or default_image
+            memory = sandbox.memory or default_memory
+
+            # Create full Sandbox object
+            sandbox = Sandbox(
+                metadata=Metadata(name=name),
+                spec=SandboxSpec(runtime=Runtime(image=image, memory=memory, generation="mk3"))
+            )
+        else:
+            # Handle existing Sandbox object or dict conversion
+            if isinstance(sandbox, dict):
+                sandbox = Sandbox.from_dict(sandbox)
+
+            # Set defaults for missing fields
+            if not sandbox.metadata:
+                sandbox.metadata = Metadata(name=uuid.uuid4().hex.replace('-', ''))
+            if not sandbox.spec:
+                sandbox.spec = SandboxSpec(runtime=Runtime(image=default_image))
+            if not sandbox.spec.runtime:
+                sandbox.spec.runtime = Runtime(image=default_image, memory=default_memory)
+
+            sandbox.spec.runtime.image = sandbox.spec.runtime.image or default_image
+            sandbox.spec.runtime.memory = sandbox.spec.runtime.memory or default_memory
+            sandbox.spec.runtime.generation = sandbox.spec.runtime.generation or "mk3"
 
         response = await create_sandbox(
             client=client,
@@ -115,14 +146,30 @@ class SandboxInstance:
 
     @classmethod
     async def create_if_not_exists(
-        cls, sandbox: Union[Sandbox, Dict[str, Any]]
+        cls, sandbox: Union[Sandbox, SandboxCreateConfiguration, Dict[str, Any]]
     ) -> "SandboxInstance":
         """Create a sandbox if it doesn't exist, otherwise return existing."""
-        if isinstance(sandbox, dict):
-            sandbox = Sandbox.from_dict(sandbox)
-
         try:
-            sandbox_instance = await cls.get(sandbox.metadata.name)
+            # Extract name from different configuration types
+            if isinstance(sandbox, SandboxCreateConfiguration):
+                name = sandbox.name
+            elif isinstance(sandbox, dict):
+                if 'name' in sandbox:
+                    name = sandbox['name']
+                elif 'metadata' in sandbox and isinstance(sandbox['metadata'], dict):
+                    name = sandbox['metadata'].get('name')
+                else:
+                    # If no name provided, we can't check if it exists, so create new
+                    return await cls.create(sandbox)
+            elif isinstance(sandbox, Sandbox):
+                name = sandbox.metadata.name if sandbox.metadata else None
+            else:
+                name = None
+
+            if not name:
+                raise ValueError("Sandbox name is required")
+
+            sandbox_instance = await cls.get(name)
             return sandbox_instance
         except Exception as e:
             # Check if it's a 404 error (sandbox not found)

blaxel/core/sandbox/types.py

@@ -101,3 +101,24 @@ class CopyResponse:
         self.message = message
         self.source = source
         self.destination = destination
+
+
+class SandboxCreateConfiguration:
+    """Simplified configuration for creating sandboxes with default values."""
+    def __init__(
+        self,
+        name: Optional[str] = None,
+        image: Optional[str] = None,
+        memory: Optional[int] = None,
+    ):
+        self.name = name
+        self.image = image
+        self.memory = memory
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "SandboxCreateConfiguration":
+        return cls(
+            name=data.get("name"),
+            image=data.get("image"),
+            memory=data.get("memory"),
+        )

blaxel/googleadk/tools.py

@@ -69,4 +69,3 @@ async def bl_tools(tools_names: list[str], **kwargs: Any) -> list[BaseTool]:
     tools = bl_tools_core(tools_names, **kwargs)
     await tools.initialize()
     return [GoogleADKTool(tool) for tool in tools.get_tools()]
-    return [GoogleADKTool(tool) for tool in tools.get_tools()]

{blaxel-0.2.1rc76.dist-info → blaxel-0.2.1rc78.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.1rc76
+Version: 0.2.1rc78
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai

{blaxel-0.2.1rc76.dist-info → blaxel-0.2.1rc78.dist-info}/RECORD

@@ -304,15 +304,15 @@ blaxel/core/mcp/__init__.py,sha256=5VjkiQFb1QWW5QKRgwPHARlxZJ9Xqaz0diJTpM8LLF0,1
 blaxel/core/mcp/client.py,sha256=aK3wSnsO8DmT1BZqw4eiCMF71Jwvni6Qga0DhPP806Y,5437
 blaxel/core/mcp/server.py,sha256=tXySGZKgK3IllYOzYOecp58BixKBkmAIvQp_4nSM_Ww,5919
 blaxel/core/models/__init__.py,sha256=HbRDsMnUFHkPC-MMkzPXh4mUqkVjqO6p3j7m00N_XSo,1722
-blaxel/core/sandbox/__init__.py,sha256=eWTH7PWNQZe9efqParwgXzltc7L4OyPtT_REqCdQybU,551
+blaxel/core/sandbox/__init__.py,sha256=oF3sX5MbwSqfwhOtF5ODYWwapHffbkp2UI78jPBn78U,617
 blaxel/core/sandbox/action.py,sha256=9Zjkco7YkLzBThD3N2Hr5SpeEiqU_-Ktk8HlKpkpiAg,2802
 blaxel/core/sandbox/filesystem.py,sha256=dyIvDdlPZO0ijD6mXXX8Yl0t75VijQ6_uMz_9rJd-_4,11317
 blaxel/core/sandbox/network.py,sha256=P5jLd4AAg1zgyIK4qGWvZaDZ5BzIcxRx2ffz_JLsLMI,357
 blaxel/core/sandbox/preview.py,sha256=M6FulOxPghUBpb5fLxu1Rd3ekLeCbZ_dgt4s1X2Cneo,5354
 blaxel/core/sandbox/process.py,sha256=7zEngDTs2XiNsMm9TZ4lEDUiRpS3af8dw60pEvRuHDY,8357
-blaxel/core/sandbox/sandbox.py,sha256=tsRGvkb2dnGM5pmOiIxWvOmT42P9BDXWtW3h6_CIffI,5802
+blaxel/core/sandbox/sandbox.py,sha256=4sG_i_VcuGx_cKEGMY-2eUUuGmtQBYMNIToL7YDuluM,8151
 blaxel/core/sandbox/session.py,sha256=4SH1tyXcQ9UqJx4lMwxAlp7x9Te_anDrdSEG6AlNkvU,4496
-blaxel/core/sandbox/types.py,sha256=gmNAt8x7PrJZHpw2_2aWWDE5uG6H-uEgdIKTYIw4G0g,2981
+blaxel/core/sandbox/types.py,sha256=yaKkp5Sn0b4Ri2_ZbdEjXuA-rjo2_DfDIHYz-cueZCA,3564
 blaxel/core/sandbox/client/__init__.py,sha256=N26bD5o1jsTb48oExow6Rgivd8ylaU9jaWZfZsVilP8,128
 blaxel/core/sandbox/client/client.py,sha256=tcP8cJ4Q3dV9aB3yQ01dDXO-ekfsa3WGGFz4DQAEf8I,7079
 blaxel/core/sandbox/client/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
@@ -365,7 +365,7 @@ blaxel/crewai/tools.py,sha256=wMMZvSrmhuoFRFux2j31PeSVMbElhs5e49jtvRNxZxY,848
 blaxel/googleadk/__init__.py,sha256=WfYZvtkh5IF042qWLaiWjbDarJZCX9ghHGiTrat8eps,127
 blaxel/googleadk/model.py,sha256=ZSBGapgcrhRogc1kBFKfuLS36UUkcQL6nZD22yVUEJM,2236
 blaxel/googleadk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-blaxel/googleadk/tools.py,sha256=mgHYc25A82VKYx4VUhV2Nn3XnjSdMgwlBpHJvvq5oEY,2328
+blaxel/googleadk/tools.py,sha256=cyV_vRSYYBNw-5zdXQYTeb62ElyCrdGPTj3yUu6pMwU,2265
 blaxel/langgraph/__init__.py,sha256=lw9d7bl5TsYbemToCtus5P6XnhzR4SAcBWM-1Pffc_U,126
 blaxel/langgraph/model.py,sha256=B0onMzIRZEc5P6oa3SxLcI42M5_736NLez64lng61DU,2286
 blaxel/langgraph/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -402,7 +402,7 @@ blaxel/telemetry/instrumentation/map.py,sha256=PCzZJj39yiYVYJrxLBNP-NW-tjjYyTijw
 blaxel/telemetry/instrumentation/utils.py,sha256=KInMYZH-mu9_wvetmf0EmgrfN3Sw8IWk2Y95v2u90_U,1901
 blaxel/telemetry/log/log.py,sha256=RvQByRjZMoP_dRaAZu8oK6DTegsHs-xV4W-UIqis6CA,2461
 blaxel/telemetry/log/logger.py,sha256=NPAS3g82ryROjvc_DEZaTIfrcehoLEZoP-JkLxADxc0,4113
-blaxel-0.2.1rc76.dist-info/METADATA,sha256=9VGgAjrkZM7Xsn5l7Im1K7x2QVwZa8lBcoSiXpmtI2Q,9879
-blaxel-0.2.1rc76.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.1rc76.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.1rc76.dist-info/RECORD,,
+blaxel-0.2.1rc78.dist-info/METADATA,sha256=r9IaM_vYSHVywuvDayI47ZwwxFDKCtLXlqGlogLKHps,9879
+blaxel-0.2.1rc78.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.1rc78.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.1rc78.dist-info/RECORD,,