Add beta

Author: jakubno

packages/python-sdk/e2b/api/compatibility/README.md

@@ -1,8 +1,10 @@
 from dataclasses import dataclass
-from typing import Optional, Dict
+from typing import Optional, Dict, Union, Unpack
 from datetime import datetime
 
-from e2b.api.client.models import SandboxState
+from e2b import ConnectionConfig
+from e2b.api.client.models import SandboxState, SandboxDetail, ListedSandbox
+from e2b.connection_config import ApiParams
 
 
 @dataclass
@@ -23,33 +25,64 @@ class SandboxInfo:
     """Sandbox start time."""
     end_at: datetime
     """Sandbox expiration date."""
-    envd_version: Optional[str]
+    state: SandboxState
+    """Sandbox state."""
+    cpu_count: int
+    """Sandbox CPU count."""
+    memory_mb: int
+    """Sandbox Memory size in MiB."""
+    _envd_version: Optional[str]
     """Envd version."""
     _envd_access_token: Optional[str]
     """Envd access token."""
 
+    @classmethod
+    def _from_sandbox_data(
+        cls,
+        sandbox: Union[ListedSandbox, SandboxDetail],
+        envd_version: Optional[str] = None,
+        envd_access_token: Optional[str] = None,
+        sandbox_domain: Optional[str] = None,
+    ):
+        return cls(
+            sandbox_domain=sandbox_domain,
+            sandbox_id=sandbox.sandbox_id,
+            template_id=sandbox.template_id,
+            name=(sandbox.alias if isinstance(sandbox.alias, str) else None),
+            metadata=(sandbox.metadata if isinstance(sandbox.metadata, dict) else {}),
+            started_at=sandbox.started_at,
+            end_at=sandbox.end_at,
+            state=sandbox.state,
+            cpu_count=sandbox.cpu_count,
+            memory_mb=sandbox.memory_mb,
+            _envd_version=envd_version,
+            _envd_access_token=envd_access_token,
+        )
 
-@dataclass
-class ListedSandbox:
-    """Information about a sandbox."""
+    @classmethod
+    def _from_listed_sandbox(cls, listed_sandbox: ListedSandbox):
+        return cls._from_sandbox_data(listed_sandbox)
 
-    sandbox_id: str
-    """Sandbox ID."""
-    template_id: str
-    """Template ID."""
-    name: Optional[str]
-    """Template Alias."""
-    state: SandboxState
-    """Sandbox state."""
-    cpu_count: int
-    """Sandbox CPU count."""
-    memory_mb: int
-    """Sandbox Memory size in MB."""
-    metadata: Dict[str, str]
-    """Saved sandbox metadata."""
-    started_at: datetime
-    """Sandbox start time."""
-    end_at: datetime
+    @classmethod
+    def _from_sandbox_detail(cls, sandbox_detail: SandboxDetail):
+        return cls._from_sandbox_data(
+            sandbox_detail,
+            (
+                sandbox_detail.envd_version
+                if isinstance(sandbox_detail.envd_version, str)
+                else None
+            ),
+            (
+                sandbox_detail.envd_access_token
+                if isinstance(sandbox_detail.envd_access_token, str)
+                else None
+            ),
+            sandbox_domain=(
+                sandbox_detail.domain
+                if isinstance(sandbox_detail.domain, str)
+                else None
+            ),
+        )
 
 
 @dataclass
@@ -60,6 +93,14 @@ class SandboxQuery:
     """Filter sandboxes by metadata."""
 
 
+@dataclass
+class SandboxQueryBeta(SandboxQuery):
+    """Query parameters for listing sandboxes."""
+
+    state: Optional[list[SandboxState]] = None
+    """Filter sandboxes by state."""
+
+
 @dataclass
 class SandboxMetrics:
     """Sandbox metrics."""
@@ -78,3 +119,34 @@ class SandboxMetrics:
     """Memory used in bytes."""
     timestamp: datetime
     """Timestamp of the metric entry."""
+
+
+class SandboxPaginatorBase:
+    def __init__(
+        self,
+        query: Optional[SandboxQueryBeta] = None,
+        limit: Optional[int] = None,
+        next_token: Optional[str] = None,
+        **opts: Unpack[ApiParams],
+    ):
+        self._config = ConnectionConfig(**opts)
+
+        self.query = query
+        self.limit = limit
+
+        self._has_next = True
+        self._next_token = next_token
+
+    @property
+    def has_next(self) -> bool:
+        """
+        Returns True if there are more items to fetch.
+        """
+        return self._has_next
+
+    @property
+    def next_token(self) -> Optional[str]:
+        """
+        Returns the next token to use for pagination.
+        """
+        return self._next_token

packages/python-sdk/e2b/api/compatibility/modified_get_v2_sandboxes.py

@@ -265,7 +265,7 @@ async def connect(
             sandbox_id=sandbox_id,
             sandbox_domain=response.sandbox_domain,
             connection_config=connection_config,
-            envd_version=response.envd_version,
+            envd_version=response._envd_version,
             envd_access_token=envd_access_token,
         )