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

scalebox/exceptions.py

@@ -1,83 +1,83 @@
-def sandbox_timeout_exception(message: str) -> "TimeoutException":
-    return TimeoutException(
-        f"{message}: This error is likely due to sandbox timeout. You can modify the sandbox timeout by passing 'timeout' when starting the sandbox or calling '.set_timeout' on the sandbox with the desired timeout."
-    )
-
-
-def request_timeout_error() -> Exception:
-    return TimeoutException(
-        "Request timed out — the 'request_timeout' option can be used to increase this timeout",
-    )
-
-
-def execution_timeout_error() -> Exception:
-    return TimeoutException(
-        "Execution timed out — the 'timeout' option can be used to increase this timeout",
-    )
-
-
-class SandboxException(Exception):
-    """
-    Base class for all sandbox errors.
-
-    Raised when a general sandbox exception occurs.
-    """
-
-    pass
-
-
-class TimeoutException(SandboxException):
-    """
-    Raised when a timeout occurs.
-
-    The `unavailable` exception type is caused by sandbox timeout.\n
-    The `canceled` exception type is caused by exceeding request timeout.\n
-    The `deadline_exceeded` exception type is caused by exceeding the timeout for process, watch, etc.\n
-    The `unknown` exception type is sometimes caused by the sandbox timeout when the request is not processed correctly.\n
-    """
-
-    pass
-
-
-class InvalidArgumentException(SandboxException):
-    """
-    Raised when an invalid argument is provided.
-    """
-
-    pass
-
-
-class NotEnoughSpaceException(SandboxException):
-    """
-    Raised when there is not enough disk space.
-    """
-
-    pass
-
-
-class NotFoundException(SandboxException):
-    """
-    Raised when a resource is not found.
-    """
-
-    pass
-
-
-class AuthenticationException(SandboxException):
-    """
-    Raised when authentication fails.
-    """
-
-    pass
-
-
-class TemplateException(SandboxException):
-    """
-    Exception raised when the template uses old envd version. It isn't compatible with the new SDK.
-    """
-
-
-class RateLimitException(SandboxException):
-    """
-    Raised when the API rate limit is exceeded.
-    """
+def sandbox_timeout_exception(message: str) -> "TimeoutException":
+    return TimeoutException(
+        f"{message}: This error is likely due to sandbox timeout. You can modify the sandbox timeout by passing 'timeout' when starting the sandbox or calling '.set_timeout' on the sandbox with the desired timeout."
+    )
+
+
+def request_timeout_error() -> Exception:
+    return TimeoutException(
+        "Request timed out — the 'request_timeout' option can be used to increase this timeout",
+    )
+
+
+def execution_timeout_error() -> Exception:
+    return TimeoutException(
+        "Execution timed out — the 'timeout' option can be used to increase this timeout",
+    )
+
+
+class SandboxException(Exception):
+    """
+    Base class for all sandbox errors.
+
+    Raised when a general sandbox exception occurs.
+    """
+
+    pass
+
+
+class TimeoutException(SandboxException):
+    """
+    Raised when a timeout occurs.
+
+    The `unavailable` exception type is caused by sandbox timeout.\n
+    The `canceled` exception type is caused by exceeding request timeout.\n
+    The `deadline_exceeded` exception type is caused by exceeding the timeout for process, watch, etc.\n
+    The `unknown` exception type is sometimes caused by the sandbox timeout when the request is not processed correctly.\n
+    """
+
+    pass
+
+
+class InvalidArgumentException(SandboxException):
+    """
+    Raised when an invalid argument is provided.
+    """
+
+    pass
+
+
+class NotEnoughSpaceException(SandboxException):
+    """
+    Raised when there is not enough disk space.
+    """
+
+    pass
+
+
+class NotFoundException(SandboxException):
+    """
+    Raised when a resource is not found.
+    """
+
+    pass
+
+
+class AuthenticationException(SandboxException):
+    """
+    Raised when authentication fails.
+    """
+
+    pass
+
+
+class TemplateException(SandboxException):
+    """
+    Exception raised when the template uses old envd version. It isn't compatible with the new SDK.
+    """
+
+
+class RateLimitException(SandboxException):
+    """
+    Raised when the API rate limit is exceeded.
+    """

scalebox/generated/api.py

@@ -1,61 +1,61 @@
-import json
-
-import httpx
-
-from ..exceptions import (
-    AuthenticationException,
-    InvalidArgumentException,
-    NotEnoughSpaceException,
-    NotFoundException,
-    SandboxException,
-    sandbox_timeout_exception,
-)
-
-ENVD_API_UPLOAD_FILES_ROUTE = "/upload"
-ENVD_API_DOWNLOAD_FILES_ROUTE = "/download"
-ENVD_API_FILES_ROUTE = "/files"
-ENVD_API_HEALTH_ROUTE = "/health"
-
-
-def get_message(e: httpx.Response) -> str:
-    try:
-        message = e.json().get("message", e.text)
-    except json.JSONDecodeError:
-        message = e.text
-
-    return message
-
-
-def handle_envd_api_exception(res: httpx.Response):
-    if res.is_success:
-        return
-
-    res.read()
-
-    return format_envd_api_exception(res.status_code, get_message(res))
-
-
-async def ahandle_envd_api_exception(res: httpx.Response):
-    if res.is_success:
-        return
-
-    await res.aread()
-
-    return format_envd_api_exception(res.status_code, get_message(res))
-
-
-def format_envd_api_exception(status_code: int, message: str):
-    if status_code == 400:
-        return InvalidArgumentException(message)
-    elif status_code == 401:
-        return AuthenticationException(message)
-    elif status_code == 404:
-        return NotFoundException(message)
-    elif status_code == 429:
-        return SandboxException(f"{message}: The requests are being rate limited.")
-    elif status_code == 502:
-        return sandbox_timeout_exception(message)
-    elif status_code == 507:
-        return NotEnoughSpaceException(message)
-    else:
-        return SandboxException(f"{status_code}: {message}")
+import json
+
+import httpx
+
+from ..exceptions import (
+    AuthenticationException,
+    InvalidArgumentException,
+    NotEnoughSpaceException,
+    NotFoundException,
+    SandboxException,
+    sandbox_timeout_exception,
+)
+
+ENVD_API_UPLOAD_FILES_ROUTE = "/upload"
+ENVD_API_DOWNLOAD_FILES_ROUTE = "/download"
+ENVD_API_FILES_ROUTE = "/files"
+ENVD_API_HEALTH_ROUTE = "/health"
+
+
+def get_message(e: httpx.Response) -> str:
+    try:
+        message = e.json().get("message", e.text)
+    except json.JSONDecodeError:
+        message = e.text
+
+    return message
+
+
+def handle_envd_api_exception(res: httpx.Response):
+    if res.is_success:
+        return
+
+    res.read()
+
+    return format_envd_api_exception(res.status_code, get_message(res))
+
+
+async def ahandle_envd_api_exception(res: httpx.Response):
+    if res.is_success:
+        return
+
+    await res.aread()
+
+    return format_envd_api_exception(res.status_code, get_message(res))
+
+
+def format_envd_api_exception(status_code: int, message: str):
+    if status_code == 400:
+        return InvalidArgumentException(message)
+    elif status_code == 401:
+        return AuthenticationException(message)
+    elif status_code == 404:
+        return NotFoundException(message)
+    elif status_code == 429:
+        return SandboxException(f"{message}: The requests are being rate limited.")
+    elif status_code == 502:
+        return sandbox_timeout_exception(message)
+    elif status_code == 507:
+        return NotEnoughSpaceException(message)
+    else:
+        return SandboxException(f"{status_code}: {message}")