scalebox-sdk 0.1.4__py3-none-any.whl → 0.1.25__py3-none-any.whl

scalebox/__init__.py

@@ -9,7 +9,7 @@ A multi-language code execution sandbox with support for:
 - Real-time callbacks and monitoring
 """
 
-__version__ = "0.1.4"
+__version__ = "0.1.25"
 __author__ = "ScaleBox Team"
 __email__ = "dev@scalebox.dev"
 

scalebox/api/__init__.py

@@ -1,128 +1,128 @@
-import json
-import logging
-from dataclasses import dataclass
-from typing import Optional
-
-from httpx import Limits
-
-from ..api.client.client import AuthenticatedClient
-from ..api.client.types import Response
-from ..api.metadata import default_headers
-from ..connection_config import ConnectionConfig
-from ..exceptions import AuthenticationException, RateLimitException, SandboxException
-
-logger = logging.getLogger(__name__)
-
-
-@dataclass
-class SandboxCreateResponse:
-    sandbox_id: str
-    sandbox_domain: Optional[str]
-    envd_version: str
-    envd_access_token: str
-
-
-def handle_api_exception(e: Response):
-    try:
-        body = json.loads(e.content) if e.content else {}
-    except json.JSONDecodeError:
-        body = {}
-
-    if e.status_code == 429:
-        return RateLimitException(
-            f"{e.status_code}: Rate limit exceeded, please try again later."
-        )
-
-    if "message" in body:
-        return SandboxException(f"{e.status_code}: {body['message']}")
-    return SandboxException(f"{e.status_code}: {e.content}")
-
-
-class ApiClient(AuthenticatedClient):
-    """
-    The client for interacting with the CSX API.
-    """
-
-    def __init__(
-        self,
-        config: ConnectionConfig,
-        require_api_key: bool = True,
-        require_access_token: bool = False,
-        limits: Optional[Limits] = None,
-        *args,
-        **kwargs,
-    ):
-        if require_api_key and require_access_token:
-            raise AuthenticationException(
-                "Only one of api_key or access_token can be required, not both",
-            )
-
-        if not require_api_key and not require_access_token:
-            raise AuthenticationException(
-                "Either api_key or access_token is required",
-            )
-
-        token = None
-        if require_api_key:
-            if config.api_key is None:
-                raise AuthenticationException(
-                    "API key is required, please visit the Team tab at https://dev/dashboard to get your API key. "
-                    "You can either set the environment variable `E2B_API_KEY` "
-                    'or you can pass it directly to the sandbox like Sandbox(api_key="e2b_...")',
-                )
-            token = config.api_key
-
-        if require_access_token:
-            if config.access_token is None:
-                raise AuthenticationException(
-                    "Access token is required, please visit the Personal tab at https://dev/dashboard to get your access token. "
-                    "You can set the environment variable `E2B_ACCESS_TOKEN` or pass the `access_token` in options.",
-                )
-            token = config.access_token
-
-        auth_header_name = "X-API-KEY" if require_api_key else "Authorization"
-        prefix = "" if require_api_key else "Bearer"
-
-        headers = {
-            **default_headers,
-            **(config.headers or {}),
-        }
-
-        super().__init__(
-            base_url=config.api_url,
-            httpx_args={
-                "event_hooks": {
-                    "request": [self._log_request],
-                    "response": [self._log_response],
-                },
-                "proxy": config.proxy,
-                "limits": limits,
-            },
-            headers=headers,
-            token=token,
-            auth_header_name=auth_header_name,
-            prefix=prefix,
-            *args,
-            **kwargs,
-        )
-
-    def _log_request(self, request):
-        logger.info(f"Request {request.method} {request.url}")
-
-    def _log_response(self, response: Response):
-        if response.status_code >= 400:
-            logger.error(f"Response {response.status_code}")
-        else:
-            logger.info(f"Response {response.status_code}")
-
-
-# We need to override the logging hooks for the async usage
-class AsyncApiClient(ApiClient):
-    async def _log_request(self, request):
-        logger.info(f"Request {request.method} {request.url}")
-
-    async def _log_response(self, response: Response):
-        if response.status_code >= 400:
-            logger.error(f"Response {response.status_code}")
-        else:
-            logger.info(f"Response {response.status_code}")
+import json
+import logging
+from dataclasses import dataclass
+from typing import Optional
+
+from httpx import Limits
+
+from ..api.client.client import AuthenticatedClient
+from ..api.client.types import Response
+from ..api.metadata import default_headers
+from ..connection_config import ConnectionConfig
+from ..exceptions import AuthenticationException, RateLimitException, SandboxException
+
+logger = logging.getLogger(__name__)
+
+
+@dataclass
+class SandboxCreateResponse:
+    sandbox_id: str
+    sandbox_domain: Optional[str]
+    envd_version: str
+    envd_access_token: str
+
+
+def handle_api_exception(e: Response):
+    try:
+        body = json.loads(e.content) if e.content else {}
+    except json.JSONDecodeError:
+        body = {}
+
+    if e.status_code == 429:
+        return RateLimitException(
+            f"{e.status_code}: Rate limit exceeded, please try again later."
+        )
+
+    if "message" in body:
+        return SandboxException(f"{e.status_code}: {body['message']}")
+    return SandboxException(f"{e.status_code}: {e.content}")
+
+
+class ApiClient(AuthenticatedClient):
+    """
+    The client for interacting with the CSX API.
+    """
+
+    def __init__(
+        self,
+        config: ConnectionConfig,
+        require_api_key: bool = True,
+        require_access_token: bool = False,
+        limits: Optional[Limits] = None,
+        *args,
+        **kwargs,
+    ):
+        if require_api_key and require_access_token:
+            raise AuthenticationException(
+                "Only one of api_key or access_token can be required, not both",
+            )
+
+        if not require_api_key and not require_access_token:
+            raise AuthenticationException(
+                "Either api_key or access_token is required",
+            )
+
+        token = None
+        if require_api_key:
+            if config.api_key is None:
+                raise AuthenticationException(
+                    "API key is required, please visit the Team tab at https://dev/dashboard to get your API key. "
+                    "You can either set the environment variable `SBX_API_KEY` "
+                    'or you can pass it directly to the sandbox like Sandbox(api_key="sbx_...")',
+                )
+            token = config.api_key
+
+        if require_access_token:
+            if config.access_token is None:
+                raise AuthenticationException(
+                    "Access token is required, please visit the Personal tab at https://dev/dashboard to get your access token. "
+                    "You can set the environment variable `SBX_ACCESS_TOKEN` or pass the `access_token` in options.",
+                )
+            token = config.access_token
+
+        auth_header_name = "X-API-KEY" if require_api_key else "Authorization"
+        prefix = "" if require_api_key else "Bearer"
+
+        headers = {
+            **default_headers,
+            **(config.headers or {}),
+        }
+
+        super().__init__(
+            base_url=config.api_url,
+            httpx_args={
+                "event_hooks": {
+                    "request": [self._log_request],
+                    "response": [self._log_response],
+                },
+                "proxy": config.proxy,
+                "limits": limits,
+            },
+            headers=headers,
+            token=token,
+            auth_header_name=auth_header_name,
+            prefix=prefix,
+            *args,
+            **kwargs,
+        )
+
+    def _log_request(self, request):
+        logger.info(f"Request {request.method} {request.url}")
+
+    def _log_response(self, response: Response):
+        if response.status_code >= 400:
+            logger.error(f"Response {response.status_code}")
+        else:
+            logger.info(f"Response {response.status_code}")
+
+
+# We need to override the logging hooks for the async usage
+class AsyncApiClient(ApiClient):
+    async def _log_request(self, request):
+        logger.info(f"Request {request.method} {request.url}")
+
+    async def _log_response(self, response: Response):
+        if response.status_code >= 400:
+            logger.error(f"Response {response.status_code}")
+        else:
+            logger.info(f"Response {response.status_code}")

scalebox/api/client/__init__.py

@@ -1,8 +1,8 @@
-"""A client library for accessing E2B API"""
-
-from .client import AuthenticatedClient, Client
-
-__all__ = (
-    "AuthenticatedClient",
-    "Client",
-)
+"""A client library for accessing Scalebox API"""
+
+from .client import AuthenticatedClient, Client
+
+__all__ = (
+    "AuthenticatedClient",
+    "Client",
+)

scalebox/api/client/api/sandboxes/get_sandboxes.py

@@ -17,6 +17,7 @@ def _get_kwargs(
     params: dict[str, Any] = {}
 
     params["metadata"] = metadata
+    params["status"] = "running"
 
     params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
 
@@ -35,9 +36,11 @@ def _parse_response(
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
-        for response_200_item_data in _response_200:
+        data=_response_200["data"]
+        if data["sandboxes"] is None:
+            return None
+        for response_200_item_data in data["sandboxes"]:
             response_200_item = ListedSandbox.from_dict(response_200_item_data)
-
             response_200.append(response_200_item)
 
         return response_200
@@ -95,7 +98,6 @@ def sync_detailed(
     response = client.get_httpx_client().request(
         **kwargs,
     )
-
     return _build_response(client=client, response=response)
 
 

scalebox/api/client/api/sandboxes/get_sandboxes_sandbox_id_metrics.py

@@ -39,9 +39,9 @@ def _parse_response(
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
-        print(_response_200)
+        # print(_response_200)
         for response_200_item_data in _response_200["data"]["metrics"]:
-            print(response_200_item_data)
+            # print(response_200_item_data)
             response_200_item = SandboxMetric.from_dict(response_200_item_data)
 
             response_200.append(response_200_item)

scalebox/api/client/api/sandboxes/post_sandboxes.py

@@ -35,7 +35,7 @@ def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
 ) -> Optional[Union[Error, Sandbox]]:
     if response.status_code == 201:
-        print(response.json())
+        # print(response.json())
         response_201 = Sandbox.from_dict(response.json())
 
         return response_201
@@ -89,7 +89,7 @@ def sync_detailed(
     kwargs = _get_kwargs(
         body=body,
     )
-    print(kwargs)
+    # print(kwargs)
     response = client.get_httpx_client().request(
         **kwargs,
     )