feat(local): cua on the local workstation

pyproject.toml

@@ -44,7 +44,7 @@ browser = [
   "hai-drivers[web]>=0.1.3",
 ]
 workstation = [
-  "hai-drivers[browser-cli,desktop-cli]>=0.1.6",
+  "hai-drivers[browser-cli,cua-cli,desktop-cli]>=0.1.9",
 ]
 all = [
   "hai-agents[browser,cli,desktop,workstation]",

src/hai_agents_local/workstation.py

@@ -1,4 +1,4 @@
-"""Local workstation bridge: the agent's shell runs on this machine and drives it through the desk and web CLIs."""
+"""Local workstation bridge: the agent's shell runs on this machine and drives it through the desk, cua and web CLIs."""
 
 from __future__ import annotations
 
@@ -58,6 +58,7 @@ def create_driver(self) -> ManagedCodeSandboxInterface:
             str(self.workspace),
             environment_variables={
                 "COORDINATE_SYSTEM": "0-1000",
+                "CUA_SESSION": self.session_id,
                 "PATH": os.pathsep.join([os.path.dirname(desk), path]),
             },
         )

tests/test_local.py

@@ -689,12 +689,12 @@ def test_workstation_serves_a_shell_in_its_session_folder_with_the_cli_commands(
         bridge._driver = bridge.create_driver()
         try:
             result, error = bridge._dispatch(
-                "execute", {"command": "pwd -P; echo $COORDINATE_SYSTEM; command -v desk web click"}
+                "execute", {"command": "pwd -P; echo $COORDINATE_SYSTEM $CUA_SESSION; command -v desk web cua click"}
             )
             assert error is None and result["exit_code"] == 0
             out = result["stdout"].split()
             assert out[0] == str((tmp_path / "hai" / bridge.session_id).resolve())
-            assert out[1] == "0-1000" and len(out) == 5
+            assert out[1:3] == ["0-1000", bridge.session_id] and len(out) == 7
         finally:
             bridge._driver.close()