@ai-sdk/harness-pi 1.0.120 → 1.0.121

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @ai-sdk/harness-pi
 
+## 1.0.121
+
+### Patch Changes
+
+- 9e9f18f: fix(harness-pi): support stateless session restoration and injected credentials
+- 2adbb77: feat(harness): update underlying harness SDKs to their latest versions
+- Updated dependencies [125f493]
+  - @ai-sdk/harness@1.0.119
+
 ## 1.0.120
 
 ### Patch Changes

package/README.md

@@ -51,6 +51,25 @@ try {
 
 The adapter requires a `HarnessV1SandboxProvider`. Pi has no in-sandbox bridge, so the sandbox doesn't need to expose any ports — `@ai-sdk/sandbox-vercel` or `@ai-sdk/sandbox-just-bash` both work.
 
+## Stateless session configuration
+
+By default, a suspended turn can reuse its live Pi session in the same process.
+For stateless or multi-replica applications, set `reattachInProcess: false` so
+continuations restore persisted state using the current request's settings.
+Application-managed credentials can be supplied through a `PiCredentialStore`;
+this replaces Pi's file-backed `auth.json` storage.
+
+```ts
+import { createPi, type PiCredentialStore } from '@ai-sdk/harness-pi';
+
+declare const credentials: PiCredentialStore;
+
+const harness = createPi({
+  credentials,
+  reattachInProcess: false,
+});
+```
+
 ## Inline extensions
 
 Use `extensionFactories` to load trusted inline Pi extensions for each harness session:

package/dist/index.d.ts

@@ -1,7 +1,9 @@
 import * as _ai_sdk_harness from '@ai-sdk/harness';
 import { HarnessV1Authentication, HarnessV1, HarnessV1BuiltinTool } from '@ai-sdk/harness';
-import { ProviderConfig, ExtensionFactory } from '@earendil-works/pi-coding-agent';
+import { CreateModelRuntimeOptions, ProviderConfig, ExtensionFactory } from '@earendil-works/pi-coding-agent';
 
+/** Persistent credential storage consumed by Pi's model runtime. */
+type PiCredentialStore = NonNullable<CreateModelRuntimeOptions['credentials']>;
 /**
  * Pi auth options. Choose an explicit mode or rely on 'auto' (precedence:
  * explicit gateway, then OpenAI / Anthropic / custom environment variables).
@@ -17,6 +19,19 @@ type PiThinkingLevel = 'off' | 'minimal' | 'low' | 'medium' | 'high' | 'xhigh' |
 type PiHarnessSettings = {
     /** Where Pi sources API keys / gateway credentials from. */
     readonly auth?: PiAuthenticationMode;
+    /**
+     * Application-owned credential storage for Pi's model runtime. When set,
+     * this replaces Pi's file-backed auth.json credential storage.
+     */
+    readonly credentials?: PiCredentialStore;
+    /**
+     * Whether a suspended turn may reuse its live Pi session in this process.
+     * Disable this in stateless or multi-replica applications so every request
+     * restores from persisted lifecycle state with the current settings.
+     *
+     * @default true
+     */
+    readonly reattachInProcess?: boolean;
     /**
      * Explicit Pi provider configurations keyed by provider id. Use this to
      * register custom models and their API protocol without coupling model
@@ -122,4 +137,4 @@ declare const pi: _ai_sdk_harness.HarnessV1<{
     readonly ls: _ai_sdk_harness.HarnessV1BuiltinTool;
 }>;
 
-export { type PiAuthenticationMode, type PiHarnessSettings, VERSION, createPi, pi };
+export { type PiAuthenticationMode, type PiCredentialStore, type PiHarnessSettings, VERSION, createPi, pi };

package/dist/index.js

@@ -136,7 +136,7 @@ import {
 import { access } from "fs/promises";
 
 // src/version.ts
-var VERSION = true ? "1.0.120" : "0.0.0-test";
+var VERSION = true ? "1.0.121" : "0.0.0-test";
 
 // src/pi-auth.ts
 var DEFAULT_GATEWAY_BASE_URL = "https://ai-gateway.vercel.sh";
@@ -203,19 +203,20 @@ function scopePiProviderEnvironment({
 }
 async function createPiModelRuntime({
   auth,
+  credentials,
   authPath,
   modelsPath
 }) {
   if (!isHarnessAuthenticationEnvironment(auth)) {
     return ModelRuntime.create({
-      authPath,
+      ...credentials ? { credentials } : { authPath },
       modelsPath,
       allowModelNetwork: false
     });
   }
-  const isolatedCredentials = createIsolatedPiCredentialStore();
+  const isolatedCredentials = credentials == null ? createIsolatedPiCredentialStore() : void 0;
   const modelRuntime = await ModelRuntime.create({
-    credentials: isolatedCredentials.credentials,
+    credentials: credentials ?? isolatedCredentials.credentials,
     modelsPath: null,
     allowModelNetwork: false
   });
@@ -237,7 +238,7 @@ async function createPiModelRuntime({
     modelRuntime,
     authenticationEnvironment: auth
   });
-  isolatedCredentials.finishInitialization();
+  isolatedCredentials?.finishInitialization();
   await modelRuntime.refresh({ allowNetwork: false });
   return modelRuntime;
 }
@@ -2228,6 +2229,9 @@ function resolveActivePiBuiltinNames(toolFiltering) {
     (native) => !toolFiltering.toolNames.includes(NATIVE_TO_COMMON[native] ?? native)
   );
 }
+function hasCompatibleReattachSettings(parked, current) {
+  return parked.sandboxSession === current.sandboxSession && parked.sessionWorkDir === current.sessionWorkDir && parked.clientApp === current.clientApp && parked.permissionMode === current.permissionMode && parked.builtinToolFiltering === current.builtinToolFiltering && parked.abortSignal === current.abortSignal && parked.agentDir === current.agentDir && parked.settings.auth === current.settings.auth && parked.settings.credentials === current.settings.credentials && parked.settings.headers === current.settings.headers && parked.settings.thinkingLevel === current.settings.thinkingLevel && parked.settings.mcpServers === current.settings.mcpServers && parked.settings.providers === current.settings.providers && parked.settings.extensionFactories === current.settings.extensionFactories;
+}
 async function isWorkspaceAvailableOnHost({
   sandbox,
   sessionWorkDir
@@ -2250,13 +2254,16 @@ async function isWorkspaceAvailableOnHost({
 }
 async function createPiSession(input) {
   if (input.isResume) {
-    const parkedSession = parkedPiSessions.get(input.sessionId);
-    if (parkedSession) {
+    const parked = parkedPiSessions.get(input.sessionId);
+    if (parked) {
       parkedPiSessions.delete(input.sessionId);
-      return {
-        ...parkedSession,
-        isResume: true
-      };
+      if (input.settings.reattachInProcess !== false && (input.resumeStateType == null || input.resumeStateType === parked.stateType) && hasCompatibleReattachSettings(parked.input, input)) {
+        return {
+          ...parked.session,
+          isResume: true
+        };
+      }
+      await parked.session.doDestroy();
     }
   }
   const safeSessionId = input.sessionId.replace(/[\\/: ]/g, "-");
@@ -2328,6 +2335,7 @@ async function createPiSession(input) {
   });
   const modelRuntime = await createPiModelRuntime({
     auth: input.settings.auth,
+    credentials: input.settings.credentials,
     authPath: path6.join(nativeAgentDir ?? hostAgentDir, "auth.json"),
     modelsPath: path6.join(agentDir, "models.json")
   });
@@ -3084,7 +3092,11 @@ async function createPiSession(input) {
     doStop,
     doDetach: async () => {
       if (activeTurn != null || pendingToolResults.size > 0) {
-        parkedPiSessions.set(input.sessionId, sessionImpl);
+        parkedPiSessions.set(input.sessionId, {
+          session: sessionImpl,
+          input,
+          stateType: "resume-session"
+        });
         if (sessionFileName) {
           try {
             await persistSessionFile();
@@ -3105,7 +3117,11 @@ async function createPiSession(input) {
         throw new Error("Pi session has been stopped.");
       }
       if (activeTurn != null && (pendingToolResults.size > 0 || pendingToolApprovals.size > 0)) {
-        parkedPiSessions.set(input.sessionId, sessionImpl);
+        parkedPiSessions.set(input.sessionId, {
+          session: sessionImpl,
+          input,
+          stateType: "continue-turn"
+        });
         if (sessionFileName) {
           try {
             await persistSessionFile();
@@ -3481,6 +3497,8 @@ function createPi(settings = {}) {
         sessionWorkDir: startOpts.sessionWorkDir,
         settings: {
           ...settings.auth ? { auth: settings.auth } : {},
+          ...settings.credentials ? { credentials: settings.credentials } : {},
+          ...settings.reattachInProcess != null ? { reattachInProcess: settings.reattachInProcess } : {},
           ...settings.thinkingLevel ? { thinkingLevel: settings.thinkingLevel } : {},
           ...settings.mcpServers ? { mcpServers: settings.mcpServers } : {},
           ...settings.providers ? { providers: settings.providers } : {},
@@ -3489,6 +3507,7 @@ function createPi(settings = {}) {
         },
         clientApp: PI_CLIENT_APP,
         isResume: lifecycleState != null,
+        ...lifecycleState ? { resumeStateType: lifecycleState.type } : {},
         permissionMode: startOpts.permissionMode,
         builtinToolFiltering: startOpts.builtinToolFiltering,
         ...resumeData?.sessionFileName ? { resumeSessionFileName: resumeData.sessionFileName } : {},