Align keepalive on client with proxies and envd

e2b-dev · Sep 11, 2024 · a4991a2 · a4991a2
1 parent bc129ca
commit a4991a2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/packages/python-sdk/e2b/sandbox/main.py b/packages/python-sdk/e2b/sandbox/main.py
@@ -12,7 +12,7 @@ class SandboxSetup(ABC):
     _limits = Limits(
         max_keepalive_connections=10,
         max_connections=20,
-        keepalive_expiry=600,
+        keepalive_expiry=540,
     )
 
     envd_port = 49983

diff --git a/packages/python-sdk/e2b/sandbox_sync/sandbox_api.py b/packages/python-sdk/e2b/sandbox_sync/sandbox_api.py
@@ -17,7 +17,6 @@
 
 
 class SandboxApi(SandboxApiBase):
-
     @classmethod
     def list(
         cls,
@@ -33,7 +32,9 @@ def list(
             request_timeout=request_timeout,
         )
 
-        with ApiClient(config, transport=HTTPTransport(limits=SandboxApiBase._limits)) as api_client:
+        with ApiClient(
+            config, transport=HTTPTransport(limits=SandboxApiBase._limits)
+        ) as api_client:
             res = get_sandboxes.sync_detailed(
                 client=api_client,
             )
@@ -79,7 +80,9 @@ def _cls_kill(
         if config.debug:
             return True
 
-        with ApiClient(config, transport=HTTPTransport(limits=SandboxApiBase._limits)) as api_client:
+        with ApiClient(
+            config, transport=HTTPTransport(limits=SandboxApiBase._limits)
+        ) as api_client:
             res = delete_sandboxes_sandbox_id.sync_detailed(
                 sandbox_id,
                 client=api_client,
@@ -110,7 +113,9 @@ def _cls_set_timeout(
             request_timeout=request_timeout,
         )
 
-        with ApiClient(config, transport=HTTPTransport(limits=SandboxApiBase._limits)) as api_client:
+        with ApiClient(
+            config, transport=HTTPTransport(limits=SandboxApiBase._limits)
+        ) as api_client:
             res = post_sandboxes_sandbox_id_timeout.sync_detailed(
                 sandbox_id,
                 client=api_client,
@@ -139,7 +144,9 @@ def _create_sandbox(
             request_timeout=request_timeout,
         )
 
-        with ApiClient(config, transport=HTTPTransport(limits=SandboxApiBase._limits)) as api_client:
+        with ApiClient(
+            config, transport=HTTPTransport(limits=SandboxApiBase._limits)
+        ) as api_client:
             res = post_sandboxes.sync_detailed(
                 body=NewSandbox(
                     template_id=template,