Fix headers for unsecure sandbox

Author: jakubno

python/e2b_code_interpreter/code_interpreter_async.py

@@ -191,6 +191,10 @@ async def run_code(
         request_timeout = request_timeout or self.connection_config.request_timeout
         context_id = context.id if context else None
 
+        headers = {}
+        if self._envd_access_token:
+            headers = {"X-Access-Token": self._envd_access_token}
+
         try:
             async with self._client.stream(
                 "POST",
@@ -201,7 +205,7 @@ async def run_code(
                     "language": language,
                     "env_vars": envs,
                 },
-                headers={"X-Access-Token": self._envd_access_token},
+                headers=headers,
                 timeout=(request_timeout, timeout, request_timeout, request_timeout),
             ) as response:
                 err = await aextract_exception(response)
@@ -249,10 +253,14 @@ async def create_code_context(
         if cwd:
             data["cwd"] = cwd
 
+        headers = {}
+        if self._envd_access_token:
+            headers = {"X-Access-Token": self._envd_access_token}
+
         try:
             response = await self._client.post(
                 f"{self._jupyter_url}/contexts",
-                headers={"X-Access-Token": self._envd_access_token},
+                headers=headers,
                 json=data,
                 timeout=request_timeout or self.connection_config.request_timeout,
             )

python/e2b_code_interpreter/code_interpreter_sync.py

@@ -188,6 +188,10 @@ def run_code(
         request_timeout = request_timeout or self.connection_config.request_timeout
         context_id = context.id if context else None
 
+        headers = {}
+        if self._envd_access_token:
+            headers = {"X-Access-Token": self._envd_access_token or ""}
+
         try:
             with self._client.stream(
                 "POST",
@@ -198,7 +202,7 @@ def run_code(
                     "language": language,
                     "env_vars": envs,
                 },
-                headers={"X-Access-Token": self._envd_access_token},
+                headers=headers,
                 timeout=(request_timeout, timeout, request_timeout, request_timeout),
             ) as response:
                 err = extract_exception(response)
@@ -246,11 +250,15 @@ def create_code_context(
         if cwd:
             data["cwd"] = cwd
 
+        headers = {}
+        if self._envd_access_token:
+            headers = {"X-Access-Token": self._envd_access_token or ""}
+
         try:
             response = self._client.post(
                 f"{self._jupyter_url}/contexts",
                 json=data,
-                headers={"X-Access-Token": self._envd_access_token},
+                headers=headers,
                 timeout=request_timeout or self.connection_config.request_timeout,
             )