diff --git a/packages/js-sdk/src/api/index.ts b/packages/js-sdk/src/api/index.ts index 4452bd8319..abacdb99ef 100644 --- a/packages/js-sdk/src/api/index.ts +++ b/packages/js-sdk/src/api/index.ts @@ -52,13 +52,13 @@ class ApiClient { this.api = createClient({ baseUrl: config.apiUrl, - // keepalive: true, // TODO: Return keepalive headers: { ...defaultHeaders, ...(config.apiKey && { 'X-API-KEY': config.apiKey }), ...(config.accessToken && { Authorization: `Bearer ${config.accessToken}`, }), + ...{ Connection: 'keep-alive' }, ...config.headers, }, }) diff --git a/packages/js-sdk/src/sandbox/index.ts b/packages/js-sdk/src/sandbox/index.ts index 3bad2634d4..c3ae254833 100644 --- a/packages/js-sdk/src/sandbox/index.ts +++ b/packages/js-sdk/src/sandbox/index.ts @@ -168,6 +168,7 @@ export class Sandbox extends SandboxApi { ...(options ?? {}), headers: headers, redirect: 'follow', + keepalive: true, } return fetch(url, options) @@ -180,8 +181,8 @@ export class Sandbox extends SandboxApi { logger: opts?.logger, accessToken: this.envdAccessToken, headers: this.envdAccessToken - ? { 'X-Access-Token': this.envdAccessToken } - : {}, + ? { 'X-Access-Token': this.envdAccessToken, Connection: 'keep-alive' } + : { Connection: 'keep-alive' }, }, { version: opts?.envdVersion, diff --git a/packages/python-sdk/e2b/api/client/client.py b/packages/python-sdk/e2b/api/client/client.py index eeffd00c86..f7bd11fc49 100644 --- a/packages/python-sdk/e2b/api/client/client.py +++ b/packages/python-sdk/e2b/api/client/client.py @@ -232,7 +232,7 @@ def get_httpx_client(self) -> httpx.Client: self._client = httpx.Client( base_url=self._base_url, cookies=self._cookies, - headers=self._headers, + headers={**self._headers, "Connection": "keep-alive"}, timeout=self._timeout, verify=self._verify_ssl, follow_redirects=self._follow_redirects, diff --git a/packages/python-sdk/e2b_connect/client.py b/packages/python-sdk/e2b_connect/client.py index ab52a3baa7..6a5e57f002 100644 --- a/packages/python-sdk/e2b_connect/client.py +++ b/packages/python-sdk/e2b_connect/client.py @@ -161,7 +161,10 @@ def __init__( self._codec = JSONCodec if json else ProtobufCodec self._response_type = response_type self._compressor = compressor - self._headers = {**{"user-agent": "connect-python"}, **headers} + self._headers = { + **{"user-agent": "connect-python", "Connection": "keep-alive"}, + **headers, + } self._connection_retries = 3 def _prepare_unary_request(