@ai-sdk/harness-pi 1.0.120 → 1.0.121

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/harness-pi",
-  "version": "1.0.120",
+  "version": "1.0.121",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -26,10 +26,10 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/harness": "1.0.118",
+    "@ai-sdk/harness": "1.0.119",
     "@ai-sdk/provider-utils": "5.0.45",
     "@earendil-works/pi-ai": "0.74.2",
-    "@earendil-works/pi-coding-agent": "^0.84.3",
+    "@earendil-works/pi-coding-agent": "^0.85.1",
     "pi-mcp-adapter": "2.12.1",
     "typebox": "^1.1.38"
   },
@@ -37,7 +37,7 @@
     "zod": "^3.25.76 || ^4.1.8"
   },
   "devDependencies": {
-    "@ai-sdk/sandbox-just-bash": "1.0.118",
+    "@ai-sdk/sandbox-just-bash": "1.0.119",
     "@types/node": "22.19.19",
     "@vercel/ai-tsconfig": "0.0.0",
     "tsup": "^8.5.1",

package/src/index.ts

@@ -9,4 +9,4 @@ export const pi = createPi();
 export { createPi } from './pi-harness';
 export { VERSION } from './version';
 export type { PiHarnessSettings } from './pi-harness';
-export type { PiAuthenticationMode } from './pi-auth';
+export type { PiAuthenticationMode, PiCredentialStore } from './pi-auth';

package/src/pi-auth.ts

@@ -12,7 +12,10 @@ import { access } from 'node:fs/promises';
 import { VERSION } from './version';
 
 type ProviderConfigInput = Parameters<ModelRegistry['registerProvider']>[1];
-type PiCredentialStore = NonNullable<CreateModelRuntimeOptions['credentials']>;
+/** Persistent credential storage consumed by Pi's model runtime. */
+export type PiCredentialStore = NonNullable<
+  CreateModelRuntimeOptions['credentials']
+>;
 type PiCredential = Exclude<
   Awaited<ReturnType<PiCredentialStore['read']>>,
   undefined
@@ -116,24 +119,27 @@ function scopePiProviderEnvironment({
 
 export async function createPiModelRuntime({
   auth,
+  credentials,
   authPath,
   modelsPath,
 }: {
   auth: PiAuthenticationMode | undefined;
+  credentials?: PiCredentialStore;
   authPath: string;
   modelsPath: string;
 }): Promise<ModelRuntime> {
   if (!isHarnessAuthenticationEnvironment(auth)) {
     return ModelRuntime.create({
-      authPath,
+      ...(credentials ? { credentials } : { authPath }),
       modelsPath,
       allowModelNetwork: false,
     });
   }
 
-  const isolatedCredentials = createIsolatedPiCredentialStore();
+  const isolatedCredentials =
+    credentials == null ? createIsolatedPiCredentialStore() : undefined;
   const modelRuntime = await ModelRuntime.create({
-    credentials: isolatedCredentials.credentials,
+    credentials: credentials ?? isolatedCredentials!.credentials,
     modelsPath: null,
     allowModelNetwork: false,
   });
@@ -164,7 +170,7 @@ export async function createPiModelRuntime({
     modelRuntime,
     authenticationEnvironment: auth,
   });
-  isolatedCredentials.finishInitialization();
+  isolatedCredentials?.finishInitialization();
   await modelRuntime.refresh({ allowNetwork: false });
 
   return modelRuntime;

package/src/pi-harness.ts

@@ -9,7 +9,7 @@ import type {
   ProviderConfig,
 } from '@earendil-works/pi-coding-agent';
 import { z } from 'zod/v4';
-import type { PiAuthenticationMode } from './pi-auth';
+import type { PiAuthenticationMode, PiCredentialStore } from './pi-auth';
 import { piResumeStateSchema } from './pi-resume-state';
 import { createPiSession, type PiThinkingLevel } from './pi-session';
 import { VERSION } from './version';
@@ -26,6 +26,19 @@ const PI_CLIENT_APP = `ai-sdk/harness-pi/${VERSION}`;
 export 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
@@ -154,6 +167,12 @@ export function createPi(
         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 }
             : {}),
@@ -166,6 +185,7 @@ export function createPi(
         },
         clientApp: PI_CLIENT_APP,
         isResume: lifecycleState != null,
+        ...(lifecycleState ? { resumeStateType: lifecycleState.type } : {}),
         permissionMode: startOpts.permissionMode,
         builtinToolFiltering: startOpts.builtinToolFiltering,
         ...(resumeData?.sessionFileName

package/src/pi-session.ts

@@ -43,6 +43,7 @@ import {
   registerPiProviders,
   resolvePiEnv,
   type PiAuthenticationMode,
+  type PiCredentialStore,
 } from './pi-auth';
 import { resolvePiSubscriptionAgentDir } from './pi-subscription';
 import { getPiTerminalError, parseNativeEvent } from './pi-events';
@@ -100,7 +101,14 @@ type PiMcpAdapterModule = {
  * same-process resume instead of stopping it and resolving the promise as an
  * error. Cross-process resume still falls back to the persisted session file.
  */
-const parkedPiSessions = new Map<string, HarnessV1Session>();
+const parkedPiSessions = new Map<
+  string,
+  {
+    session: HarnessV1Session;
+    input: CreatePiSessionInput;
+    stateType: 'continue-turn' | 'resume-session';
+  }
+>();
 
 /**
  * Whether a discovered resource path belongs to a specific directory.
@@ -221,6 +229,8 @@ export type PiThinkingLevel =
 
 export interface PiSessionSettings {
   readonly auth?: PiAuthenticationMode;
+  readonly credentials?: PiCredentialStore;
+  readonly reattachInProcess?: boolean;
   readonly headers?: Readonly<Record<string, string>>;
   readonly thinkingLevel?: PiThinkingLevel;
   readonly mcpServers?: Record<string, unknown>;
@@ -235,6 +245,9 @@ export interface CreatePiSessionInput {
   readonly settings: PiSessionSettings;
   readonly clientApp: string;
   readonly isResume: boolean;
+  readonly resumeStateType?:
+    | HarnessV1ContinueTurnState['type']
+    | HarnessV1ResumeSessionState['type'];
   readonly permissionMode?: HarnessV1PermissionMode;
   readonly builtinToolFiltering?: HarnessV1BuiltinToolFiltering;
   readonly resumeSessionFileName?: string;
@@ -248,6 +261,28 @@ export interface CreatePiSessionInput {
   readonly agentDir?: string;
 }
 
+function hasCompatibleReattachSettings(
+  parked: CreatePiSessionInput,
+  current: CreatePiSessionInput,
+): boolean {
+  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
+  );
+}
+
 interface PendingToolResult {
   resolve: (value: unknown) => void;
 }
@@ -319,13 +354,27 @@ export async function createPiSession(
   input: CreatePiSessionInput,
 ): Promise<HarnessV1Session> {
   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,
+        };
+      }
+
+      // The caller explicitly requested cold restoration, supplied settings
+      // that cannot safely reuse the live runtime, or supplied a resume-session
+      // state that represents progress beyond the parked continue-turn. Retire
+      // the stale live session before rebuilding from the persisted journal
+      // with this request's runtime settings.
+      await parked.session.doDestroy();
     }
   }
 
@@ -438,6 +487,7 @@ export async function createPiSession(
   });
   const modelRuntime = await createPiModelRuntime({
     auth: input.settings.auth,
+    credentials: input.settings.credentials,
     authPath: path.join(nativeAgentDir ?? hostAgentDir, 'auth.json'),
     modelsPath: path.join(agentDir, 'models.json'),
   });
@@ -1521,7 +1571,11 @@ export async function createPiSession(
 
     doDetach: async (): Promise<HarnessV1ResumeSessionState> => {
       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();
@@ -1551,7 +1605,11 @@ export async function createPiSession(
         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();