aisolate-client 0.3.7__tar.gz → 0.3.9__tar.gz

{aisolate_client-0.3.7 → aisolate_client-0.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aisolate-client
-Version: 0.3.7
+Version: 0.3.9
 Summary: Client library for sandbox environments
 License: Apache-2.0
 Requires-Python: >=3.10
@@ -16,7 +16,7 @@ Client library for sandbox environments
 ## Local Installation
 
 ```bash
-pip install path/to/aisolate_client-0.3.7-py3-none-any.whl
+pip install path/to/aisolate_client-0.3.9-py3-none-any.whl
 ```
 
 ## Usage

{aisolate_client-0.3.7 → aisolate_client-0.3.9}/README.md

@@ -5,7 +5,7 @@ Client library for sandbox environments
 ## Local Installation
 
 ```bash
-pip install path/to/aisolate_client-0.3.7-py3-none-any.whl
+pip install path/to/aisolate_client-0.3.9-py3-none-any.whl
 ```
 
 ## Usage

{aisolate_client-0.3.7 → aisolate_client-0.3.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aisolate-client"
-version = "0.3.7"
+version = "0.3.9"
 description = "Client library for sandbox environments"
 readme = "README.md"
 requires-python = ">=3.10"

{aisolate_client-0.3.7 → aisolate_client-0.3.9}/src/aisolate/client/sandbox.py

@@ -682,7 +682,11 @@ class Sandbox:
                 sbx_b.run_code("x = 20; print(x)")
             ```
         """
-        ctx = self.create_context(language=language, cwd=cwd, env=env)
+        # The sandbox-wide variables become the context's process environment,
+        # set once at kernel start instead of travelling with every request.
+        ctx = self.create_context(
+            language=language, cwd=cwd, env={**(self._additional_env_vars or {}), **(env or {})}
+        )
 
         # Create a new Sandbox instance that shares the client and has a default
         # context; a relative cwd is a workspace subdir the instance is scoped to.
@@ -692,7 +696,7 @@ class Sandbox:
             context_id=ctx["id"],
             domain=self._domain,
             api_key=self._api_key,
-            additional_env_vars=self._additional_env_vars,
+            additional_env_vars={},
             timeout=self._timeout,
             ping_interval=self._ping_interval,
             ping_timeout=self._ping_timeout,
@@ -732,7 +736,11 @@ class Sandbox:
                 await sbx_b.arun_code("x = 20; print(x)")
             ```
         """
-        ctx = await self.acreate_context(language=language, cwd=cwd, env=env)
+        # The sandbox-wide variables become the context's process environment,
+        # set once at kernel start instead of travelling with every request.
+        ctx = await self.acreate_context(
+            language=language, cwd=cwd, env={**(self._additional_env_vars or {}), **(env or {})}
+        )
 
         # Create a new Sandbox instance that shares the client and has a default
         # context; a relative cwd is a workspace subdir the instance is scoped to.
@@ -743,7 +751,7 @@ class Sandbox:
             context_id=ctx["id"],
             domain=self._domain,
             api_key=self._api_key,
-            additional_env_vars=self._additional_env_vars,
+            additional_env_vars={},
             timeout=self._timeout,
             ping_interval=self._ping_interval,
             ping_timeout=self._ping_timeout,

aisolate_client-0.3.9/src/aisolate/client/tests/test_sandbox_context_env.py

@@ -0,0 +1,52 @@
+"""Sandbox-wide variables travel once, as the bound context's environment.
+
+``additional_env_vars`` carry credentials. A context-bound Sandbox hands them
+to the new context at creation (they become the kernel process environment)
+and stops attaching them to every execute request.
+"""
+
+from aisolate.client.sandbox import Sandbox
+
+
+class _RecordingClient:
+    def __init__(self):
+        self.context_env = None
+        self.execute_env_vars = []
+
+    async def create_context(self, sandbox_id, language, *, cwd=None, env=None):
+        self.context_env = env
+        return {"id": "ctx-1", "language": language, "cwd": cwd}
+
+    async def execute_code(self, sandbox_id, code, *, language, env_vars, context_id):
+        self.execute_env_vars.append(env_vars)
+        return
+        yield  # an empty async event stream
+
+
+def _parent(client):
+    sandbox = Sandbox(
+        sandbox_id="sbx-1",
+        gateway_client=client,
+        additional_env_vars={"API_KEY": "secret", "HOST": "example"},
+    )
+    sandbox._is_active = True
+    return sandbox
+
+
+async def test_bound_sandbox_sets_variables_once_as_context_env():
+    client = _RecordingClient()
+
+    bound = await _parent(client).acreate_context_bound_sandbox("python3", cwd="agent-1", env={"WINDOW": "4000"})
+    await bound.arun_code("1")
+    await bound.arun_code("2")
+
+    assert client.context_env == {"API_KEY": "secret", "HOST": "example", "WINDOW": "4000"}
+    assert client.execute_env_vars == [{}, {}]
+
+
+async def test_explicit_context_env_wins_over_sandbox_wide_value():
+    client = _RecordingClient()
+
+    await _parent(client).acreate_context_bound_sandbox("python3", env={"HOST": "override"})
+
+    assert client.context_env["HOST"] == "override"

{aisolate_client-0.3.7 → aisolate_client-0.3.9}/src/aisolate_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aisolate-client
-Version: 0.3.7
+Version: 0.3.9
 Summary: Client library for sandbox environments
 License: Apache-2.0
 Requires-Python: >=3.10
@@ -16,7 +16,7 @@ Client library for sandbox environments
 ## Local Installation
 
 ```bash
-pip install path/to/aisolate_client-0.3.7-py3-none-any.whl
+pip install path/to/aisolate_client-0.3.9-py3-none-any.whl
 ```
 
 ## Usage

{aisolate_client-0.3.7 → aisolate_client-0.3.9}/src/aisolate_client.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ src/aisolate/client/sandbox.py
 src/aisolate/client/sandbox_manager.py
 src/aisolate/client/tests/__init__.py
 src/aisolate/client/tests/test_gateway_client_demux.py
+src/aisolate/client/tests/test_sandbox_context_env.py
 src/aisolate/client/tests/test_sandbox_workspace_prefix.py
 src/aisolate/client/tests/integration/__init__.py
 src/aisolate/client/tests/integration/conftest.py