@ai-sdk/harness-pi 1.0.104 → 1.0.105

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @ai-sdk/harness-pi
 
+## 1.0.105
+
+### Patch Changes
+
+- a514695: fix(harness-pi): let extensions read host-backed session workspaces
+- c359fc0: fix(harness-pi): expose tools registered by inline extensions to models
+- Updated dependencies [9e1d1b2]
+- Updated dependencies [a495511]
+  - @ai-sdk/provider-utils@5.0.37
+  - @ai-sdk/harness@1.0.103
+
 ## 1.0.104
 
 ### Patch Changes

package/dist/index.js

@@ -111,7 +111,8 @@ import {
   SessionManager,
   SettingsManager
 } from "@earendil-works/pi-coding-agent";
-import { mkdir as mkdir3, rm as rm2 } from "fs/promises";
+import { randomUUID } from "crypto";
+import { mkdir as mkdir3, rm as rm2, writeFile as writeFile3 } from "fs/promises";
 import { tmpdir } from "os";
 import path6 from "path";
 import { Type as Type2 } from "typebox";
@@ -135,7 +136,7 @@ import {
 import { access } from "fs/promises";
 
 // src/version.ts
-var VERSION = true ? "1.0.104" : "0.0.0-test";
+var VERSION = true ? "1.0.105" : "0.0.0-test";
 
 // src/pi-auth.ts
 var DEFAULT_GATEWAY_BASE_URL = "https://ai-gateway.vercel.sh";
@@ -856,7 +857,7 @@ function createPiRemoteOps(options) {
     }
     return Buffer.from(bytes);
   };
-  const writeFile3 = async (inputPath, content) => {
+  const writeFile4 = async (inputPath, content) => {
     const remotePath = options.paths.toSandboxPath(inputPath);
     const resolvedPath = await resolveWritableSandboxPath(
       remotePath,
@@ -882,7 +883,7 @@ function createPiRemoteOps(options) {
     const updated = `${current.slice(0, index)}${newText}${current.slice(
       index + oldText.length
     )}`;
-    await writeFile3(inputPath, updated);
+    await writeFile4(inputPath, updated);
     return updated;
   };
   const listDirectory = async (inputPath = ".", limit = 500) => {
@@ -972,7 +973,7 @@ function createPiRemoteOps(options) {
   return {
     paths: options.paths,
     readBuffer,
-    writeFile: writeFile3,
+    writeFile: writeFile4,
     editFile,
     listDirectory,
     findFiles,
@@ -2074,6 +2075,26 @@ function resolveActivePiBuiltinNames(toolFiltering) {
     (native) => !toolFiltering.toolNames.includes(NATIVE_TO_COMMON[native] ?? native)
   );
 }
+async function isWorkspaceAvailableOnHost({
+  sandbox,
+  sessionWorkDir
+}) {
+  const probePath = path6.join(
+    sessionWorkDir,
+    `.ai-sdk-harness-pi-${randomUUID()}`
+  );
+  const probeContent = randomUUID();
+  try {
+    await writeFile3(probePath, probeContent, { flag: "wx" });
+    const sandboxContent = await sandbox.readBinaryFile({ path: probePath });
+    return sandboxContent != null && Buffer.from(sandboxContent).equals(Buffer.from(probeContent));
+  } catch {
+    return false;
+  } finally {
+    await rm2(probePath, { force: true }).catch(() => {
+    });
+  }
+}
 async function createPiSession(input) {
   if (input.isResume) {
     const parkedSession = parkedPiSessions.get(input.sessionId);
@@ -2087,16 +2108,20 @@ async function createPiSession(input) {
   }
   const safeSessionId = input.sessionId.replace(/[\\/: ]/g, "-");
   const hostRoot = path6.join(tmpdir(), "ai-sdk-harness", "pi", safeSessionId);
-  const hostWorkDir = path6.join(hostRoot, "workspace");
   const hostAgentDir = path6.join(hostRoot, "agent");
   const hostSessionDir = path6.join(hostRoot, "sessions");
+  const toolSafeSandboxSession = getRestrictedSandboxSession(
+    input.sandboxSession
+  );
   const sessionWorkDir = input.sessionWorkDir;
+  const workspaceAvailableOnHost = await isWorkspaceAvailableOnHost({
+    sandbox: toolSafeSandboxSession,
+    sessionWorkDir
+  });
+  const hostWorkDir = workspaceAvailableOnHost ? path6.resolve(sessionWorkDir) : path6.join(hostRoot, "workspace");
   await mkdir3(hostWorkDir, { recursive: true });
   await mkdir3(hostAgentDir, { recursive: true });
   await mkdir3(hostSessionDir, { recursive: true });
-  const toolSafeSandboxSession = getRestrictedSandboxSession(
-    input.sandboxSession
-  );
   const sandboxHomeDir = await resolveSandboxHomeDir({
     sandbox: toolSafeSandboxSession,
     ...input.abortSignal ? { abortSignal: input.abortSignal } : {}
@@ -2126,13 +2151,17 @@ async function createPiSession(input) {
       ...input.abortSignal ? { abortSignal: input.abortSignal } : {}
     });
   }
-  await syncHostWorkspaceFromSandbox({
-    sandbox: toolSafeSandboxSession,
-    sandboxWorkDir: input.sessionWorkDir,
-    hostWorkDir
-  });
+  if (!workspaceAvailableOnHost) {
+    await syncHostWorkspaceFromSandbox({
+      sandbox: toolSafeSandboxSession,
+      sandboxWorkDir: input.sessionWorkDir,
+      hostWorkDir
+    });
+  }
   const workspaceVfs = new PiWorkspaceVfs();
-  workspaceVfs.mount(hostWorkDir, sessionWorkDir);
+  if (!workspaceAvailableOnHost) {
+    workspaceVfs.mount(hostWorkDir, sessionWorkDir);
+  }
   const paths = createPiPathMapper({
     hostWorkDir,
     sandboxWorkDir: sessionWorkDir,
@@ -2567,7 +2596,7 @@ async function createPiSession(input) {
       settingsManager,
       resourceLoader,
       customTools,
-      ...hasMcpServers ? { noTools: "builtin" } : { tools: toolNames },
+      ...hasExtensionFactories ? { noTools: "builtin" } : { tools: toolNames },
       ...input.settings.thinkingLevel ? { thinkingLevel: input.settings.thinkingLevel } : {},
       ...activeResolvedModel ? { model: activeResolvedModel } : {}
     });
@@ -2656,11 +2685,13 @@ async function createPiSession(input) {
           await reloadResourcesOnly();
           turnAbortController.signal.throwIfAborted();
         }
-        await syncHostWorkspaceFromSandbox({
-          sandbox: toolSafeSandboxSession,
-          sandboxWorkDir: input.sessionWorkDir,
-          hostWorkDir
-        });
+        if (!workspaceAvailableOnHost) {
+          await syncHostWorkspaceFromSandbox({
+            sandbox: toolSafeSandboxSession,
+            sandboxWorkDir: input.sessionWorkDir,
+            hostWorkDir
+          });
+        }
         turnAbortController.signal.throwIfAborted();
         translatorState = createPiTranslatorState({
           builtinToolNames: [...PI_NATIVE_BUILTIN_NAMES],