Return NotFoundError/Exception for get info and set timeout, instead of generic error

Author: ValentaTomas

packages/js-sdk/src/sandbox/sandboxApi.ts

@@ -357,6 +357,10 @@ export class SandboxApi {
       signal: config.getSignal(opts?.requestTimeoutMs),
     })
 
+    if (res.error?.code === 404) {
+      throw new NotFoundError(`Sandbox ${sandboxId} not found`)
+    }
+
     const err = handleApiError(res)
     if (err) {
       throw err
@@ -376,6 +380,10 @@ export class SandboxApi {
       signal: config.getSignal(opts?.requestTimeoutMs),
     })
 
+    if (res.error?.code === 404) {
+      throw new NotFoundError(`Sandbox ${sandboxId} not found`)
+    }
+
     const err = handleApiError(res)
     if (err) {
       throw err

packages/python-sdk/e2b/sandbox_async/sandbox_api.py

@@ -76,6 +76,9 @@ async def _cls_get_info(
                 client=api_client,
             )
 
+            if res.status_code == 404:
+                raise NotFoundException(f"Sandbox {sandbox_id} not found")
+
             if res.status_code >= 300:
                 raise handle_api_exception(res)
 
@@ -139,6 +142,9 @@ async def _cls_set_timeout(
                 body=PostSandboxesSandboxIDTimeoutBody(timeout=timeout),
             )
 
+            if res.status_code == 404:
+                raise NotFoundException(f"Paused sandbox {sandbox_id} not found")
+
             if res.status_code >= 300:
                 raise handle_api_exception(res)
 

packages/python-sdk/e2b/sandbox_sync/sandbox_api.py

@@ -75,6 +75,9 @@ def _cls_get_info(
                 client=api_client,
             )
 
+            if res.status_code == 404:
+                raise NotFoundException(f"Sandbox {sandbox_id} not found")
+
             if res.status_code >= 300:
                 raise handle_api_exception(res)
 
@@ -138,6 +141,9 @@ def _cls_set_timeout(
                 body=PostSandboxesSandboxIDTimeoutBody(timeout=timeout),
             )
 
+            if res.status_code == 404:
+                raise NotFoundException(f"Sandbox {sandbox_id} not found")
+
             if res.status_code >= 300:
                 raise handle_api_exception(res)