@ai-sdk/harness-pi 1.0.109 → 1.0.111

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/harness-pi",
-  "version": "1.0.109",
+  "version": "1.0.111",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -26,8 +26,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/harness": "1.0.107",
-    "@ai-sdk/provider-utils": "5.0.39",
+    "@ai-sdk/harness": "1.0.109",
+    "@ai-sdk/provider-utils": "5.0.40",
     "@earendil-works/pi-ai": "0.74.2",
     "@earendil-works/pi-coding-agent": "^0.84.3",
     "pi-mcp-adapter": "2.12.1",
@@ -37,7 +37,7 @@
     "zod": "^3.25.76 || ^4.1.8"
   },
   "devDependencies": {
-    "@ai-sdk/sandbox-just-bash": "1.0.107",
+    "@ai-sdk/sandbox-just-bash": "1.0.109",
     "@types/node": "22.19.19",
     "@vercel/ai-tsconfig": "0.0.0",
     "tsup": "^8.5.1",

package/src/pi-harness.ts

@@ -4,7 +4,10 @@ import {
   type HarnessV1BuiltinTool,
 } from '@ai-sdk/harness';
 import { tool } from '@ai-sdk/provider-utils';
-import type { ExtensionFactory } from '@earendil-works/pi-coding-agent';
+import type {
+  ExtensionFactory,
+  ProviderConfig,
+} from '@earendil-works/pi-coding-agent';
 import { z } from 'zod/v4';
 import type { PiAuthenticationMode } from './pi-auth';
 import { piResumeStateSchema } from './pi-resume-state';
@@ -23,6 +26,12 @@ const PI_CLIENT_APP = `ai-sdk/harness-pi/${VERSION}`;
 export type PiHarnessSettings = {
   /** Where Pi sources API keys / gateway credentials from. */
   readonly auth?: PiAuthenticationMode;
+  /**
+   * Explicit Pi provider configurations keyed by provider id. Use this to
+   * register custom models and their API protocol without coupling model
+   * metadata to authentication environment variables.
+   */
+  readonly providers?: Readonly<Record<string, ProviderConfig>>;
   /**
    * Pi's extended-thinking budget level. Maps directly to the SDK's
    * `thinkingLevel` option on `createAgentSession`.
@@ -30,9 +39,9 @@ export type PiHarnessSettings = {
   readonly thinkingLevel?: PiThinkingLevel;
   /**
    * Directory holding Pi's global agent config (auth.json, models.json,
-   * settings.json). When omitted, a per-session temp dir is used. Pass the
-   * user's agent dir (e.g. `~/.pi/agent/`) to reuse their CLI auth and
-   * model settings.
+   * settings.json). When omitted, native subscription auth is discovered from
+   * Pi's default agent directory while model and general settings remain
+   * isolated per session.
    */
   readonly agentDir?: string;
   /**
@@ -149,6 +158,7 @@ export function createPi(
             ? { thinkingLevel: settings.thinkingLevel }
             : {}),
           ...(settings.mcpServers ? { mcpServers: settings.mcpServers } : {}),
+          ...(settings.providers ? { providers: settings.providers } : {}),
           ...(settings.extensionFactories
             ? { extensionFactories: settings.extensionFactories }
             : {}),

package/src/pi-model-resolver.ts

@@ -77,13 +77,15 @@ export function createPiModelResolver({
     if (gatewayMatch) return gatewayMatch;
 
     const scopedMatch = findScopedMatch(effectiveId, models);
-    if (scopedMatch && !modelRegistry.hasConfiguredAuth(scopedMatch)) {
-      const authenticatedFlatMatches = models.filter(
-        m => m.id === effectiveId && modelRegistry.hasConfiguredAuth(m),
-      );
-      if (authenticatedFlatMatches.length === 1) {
-        return authenticatedFlatMatches[0];
-      }
+    if (scopedMatch && modelRegistry.hasConfiguredAuth(scopedMatch)) {
+      return scopedMatch;
+    }
+
+    const authenticatedFlatMatches = models.filter(
+      m => matches(m) && modelRegistry.hasConfiguredAuth(m),
+    );
+    if (authenticatedFlatMatches.length === 1) {
+      return authenticatedFlatMatches[0];
     }
 
     return scopedMatch ?? models.find(matches);

package/src/pi-session.ts

@@ -8,6 +8,7 @@ import {
   type AgentSession,
   type AgentToolResult,
   type ExtensionFactory,
+  type ProviderConfig,
   type Skill,
   type ToolDefinition,
 } from '@earendil-works/pi-coding-agent';
@@ -43,6 +44,7 @@ import {
   resolvePiEnv,
   type PiAuthenticationMode,
 } from './pi-auth';
+import { resolvePiSubscriptionAgentDir } from './pi-subscription';
 import { getPiTerminalError, parseNativeEvent } from './pi-events';
 import { createPiModelResolver } from './pi-model-resolver';
 import { createPiPathMapper } from './pi-paths';
@@ -222,6 +224,7 @@ export interface PiSessionSettings {
   readonly headers?: Readonly<Record<string, string>>;
   readonly thinkingLevel?: PiThinkingLevel;
   readonly mcpServers?: Record<string, unknown>;
+  readonly providers?: Readonly<Record<string, ProviderConfig>>;
   readonly extensionFactories?: ReadonlyArray<ExtensionFactory>;
 }
 
@@ -238,9 +241,9 @@ export interface CreatePiSessionInput {
   readonly abortSignal?: AbortSignal;
   /**
    * Directory holding Pi's global agent config (auth.json, models.json,
-   * settings.json). When omitted, a per-session temp dir is used (the
-   * harness cannot reuse existing CLI logins). Pass the user's agent dir
-   * (e.g. `~/.pi/agent/`) to reuse their CLI auth and model settings.
+   * settings.json). Native auth from this directory is considered after
+   * applicable environment credentials. Model and general settings are only
+   * reused when this option is explicit.
    */
   readonly agentDir?: string;
 }
@@ -428,9 +431,14 @@ export async function createPiSession(
    * outside that record. General Pi settings still use agentDir below.
    */
   const agentDir = input.agentDir ?? hostAgentDir;
+  const nativeAgentDir = resolvePiSubscriptionAgentDir({
+    options: input.settings.auth,
+    env: process.env,
+    agentDir: input.agentDir,
+  });
   const modelRuntime = await createPiModelRuntime({
     auth: input.settings.auth,
-    authPath: path.join(agentDir, 'auth.json'),
+    authPath: path.join(nativeAgentDir ?? hostAgentDir, 'auth.json'),
     modelsPath: path.join(agentDir, 'models.json'),
   });
   const modelRegistry = new ModelRegistry(modelRuntime);
@@ -454,6 +462,14 @@ export async function createPiSession(
     clientApp: input.clientApp,
     headers: input.settings.headers,
   });
+  for (const [provider, config] of Object.entries(
+    input.settings.providers ?? {},
+  )) {
+    modelRegistry.registerProvider(provider, {
+      ...modelRegistry.getRegisteredProviderConfig(provider),
+      ...config,
+    });
+  }
   const resolveModel = createPiModelResolver({
     modelRegistry,
     env: resolverEnv,
@@ -1468,13 +1484,21 @@ export async function createPiSession(
       if (stopped) {
         throw new Error('Pi session has been stopped.');
       }
+      if (piSession == null) {
+        await rebuildPiSession([], true);
+        lastToolsSignature = JSON.stringify([]);
+      }
+      const session = piSession;
+      if (session == null) {
+        throw new Error('Pi session failed to initialize.');
+      }
       /*
        * Pi owns the compaction. We just request it; the resulting
        * `compaction_end` event is observed by the session subscription and
        * translated into a `compaction` stream part. The returned
        * `CompactionResult` is intentionally discarded here.
        */
-      await piSession?.compact(customInstructions);
+      await session.compact(customInstructions);
     },
 
     doDestroy: async () => {

package/src/pi-subscription.ts

@@ -0,0 +1,27 @@
+import { homedir } from 'node:os';
+import { resolve } from 'node:path';
+import { isHarnessAuthenticationEnvironment } from '@ai-sdk/harness/utils';
+import { resolvePiEnv, type PiAuthenticationMode } from './pi-auth';
+
+export function resolvePiSubscriptionAgentDir({
+  options,
+  env,
+  agentDir,
+  homeDirectory = homedir(),
+}: {
+  options: PiAuthenticationMode | undefined;
+  env: NodeJS.ProcessEnv;
+  agentDir?: string;
+  homeDirectory?: string;
+}): string | undefined {
+  if (isHarnessAuthenticationEnvironment(options) || options === 'ai-gateway') {
+    return undefined;
+  }
+  const resolvedEnvironment = resolvePiEnv({ options, env });
+  if (resolvedEnvironment.AI_GATEWAY_API_KEY != null) {
+    return undefined;
+  }
+  return resolve(
+    agentDir ?? env.PI_CODING_AGENT_DIR ?? `${homeDirectory}/.pi/agent`,
+  );
+}

package/src/pi-translate.ts

@@ -1,7 +1,11 @@
 import { randomBytes } from 'node:crypto';
 import type { HarnessV1StreamPart } from '@ai-sdk/harness';
 import { secureJsonParse } from '@ai-sdk/provider-utils';
-import { extractAssistantText, type PiSessionEvent } from './pi-events';
+import {
+  extractAssistantText,
+  getPiTerminalError,
+  type PiSessionEvent,
+} from './pi-events';
 import { serializeToolOutput } from './pi-utils';
 
 /**
@@ -411,7 +415,9 @@ export function translatePiEvent(
       } else {
         state.pendingStepToolCallIds.clear();
         state.stepToolCallCount = undefined;
-        parts.push(...finishStep(state));
+        if (!getPiTerminalError(event)) {
+          parts.push(...finishStep(state));
+        }
       }
       return parts;
     }