use-computer 0.0.32__tar.gz → 0.0.34__tar.gz

{use_computer-0.0.32 → use_computer-0.0.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: use-computer
-Version: 0.0.32
+Version: 0.0.34
 Summary: Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu
 Project-URL: Homepage, https://use.computer
 Project-URL: Documentation, https://api.use.computer/docs

{use_computer-0.0.32 → use_computer-0.0.34}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "use-computer"
-version = "0.0.32"
+version = "0.0.34"
 description = "Python SDK for use.computer sandboxes across macOS, iOS simulators, Windows, and Ubuntu"
 readme = "README.md"
 requires-python = ">=3.10"

{use_computer-0.0.32 → use_computer-0.0.34}/use_computer/agents/base/agent.py

@@ -211,9 +211,13 @@ class BaseCUAAgent(RecordingMixin, FinalStateMixin, ArtifactCaptureMixin):
         fire_in_process=None,
     ) -> AgentRunResult:
         """Run against an AsyncSandbox without Harbor."""
+        resolved_logs_dir = Path(logs_dir).resolve() if logs_dir else self.logs_dir
+        resolved_task_dir = (
+            Path(task_dir).resolve() if task_dir else self.task_dir or resolved_logs_dir / "task"
+        )
         options = AgentRunOptions(
-            logs_dir=Path(logs_dir).resolve() if logs_dir else self.logs_dir,
-            task_dir=Path(task_dir).resolve() if task_dir else self.task_dir,
+            logs_dir=resolved_logs_dir,
+            task_dir=resolved_task_dir,
             platform=str(platform),
             max_steps=max_steps,
             fire_in_process=fire_in_process,

{use_computer-0.0.32 → use_computer-0.0.34}/use_computer/harbor/environment/platforms/windows.py

@@ -50,6 +50,31 @@ $profileFile = Join-Path $profileDir 'registrymodifications.xcu'
 '@ | Set-Content -Path $profileFile -Encoding UTF8
 """
 
+_BENCHMARK_USER_ALIAS_COMMAND = r"""
+$profile = [Environment]::GetFolderPath('UserProfile')
+foreach ($name in @('Docker', 'User')) {
+  $alias = Join-Path 'C:\Users' $name
+  if ($alias -ieq $profile) {
+    continue
+  }
+
+  if (Test-Path -LiteralPath $alias) {
+    $item = Get-Item -LiteralPath $alias -Force
+    $targets = @($item.Target)
+    if ($item.LinkType -eq 'Junction' -and $targets -contains $profile) {
+      continue
+    }
+    $backup = "$alias.harbor-backup"
+    if (Test-Path -LiteralPath $backup) {
+      Remove-Item -LiteralPath $backup -Recurse -Force
+    }
+    Move-Item -LiteralPath $alias -Destination $backup -Force
+  }
+
+  New-Item -ItemType Junction -Path $alias -Target $profile -Force | Out-Null
+}
+"""
+
 
 async def exec_on_vm(
     env: Any,
@@ -135,6 +160,7 @@ async def exec_local(
 async def run_windows_setup(env: Any) -> None:
     setup_dir = env._task_dir / "tests" / "setup"
 
+    await _ensure_benchmark_user_aliases(env)
     await _configure_libreoffice_defaults(env)
 
     for setup_file in read_setup_files(setup_dir):
@@ -193,6 +219,18 @@ async def _configure_libreoffice_defaults(env: Any) -> None:
     )
 
 
+async def _ensure_benchmark_user_aliases(env: Any) -> None:
+    await _with_agent_retry(
+        env,
+        "benchmark user aliases",
+        lambda: env.windows_sandbox.shell.run(
+            _BENCHMARK_USER_ALIAS_COMMAND,
+            shell="powershell",
+            timeout=30,
+        ),
+    )
+
+
 async def _with_agent_retry(env: Any, label: str, op):
     attempts = 4
     for attempt in range(1, attempts + 1):

{use_computer-0.0.32 → use_computer-0.0.34}/use_computer/resources/recording.py

@@ -19,7 +19,10 @@ class Recording:
         return RecordingInfo.from_dict(resp.json())
 
     def stop(self, recording_id: str) -> RecordingInfo:
-        resp = self._http.post(f"{self._prefix}/recording/stop", json={"id": recording_id})
+        resp = self._http.post(
+            f"{self._prefix}/recording/stop",
+            json={"id": recording_id, "recording_id": recording_id},
+        )
         resp.raise_for_status()
         return RecordingInfo.from_dict(resp.json())
 
@@ -62,7 +65,10 @@ class AsyncRecording:
         return RecordingInfo.from_dict(resp.json())
 
     async def stop(self, recording_id: str) -> RecordingInfo:
-        resp = await self._http.post(f"{self._prefix}/recording/stop", json={"id": recording_id})
+        resp = await self._http.post(
+            f"{self._prefix}/recording/stop",
+            json={"id": recording_id, "recording_id": recording_id},
+        )
         resp.raise_for_status()
         return RecordingInfo.from_dict(resp.json())