@ai-sdk/harness-pi 1.0.109 → 1.0.111

package/CHANGELOG.md

@@ -1,5 +1,28 @@
 # @ai-sdk/harness-pi
 
+## 1.0.111
+
+### Patch Changes
+
+- @ai-sdk/harness@1.0.109
+
+## 1.0.110
+
+### Patch Changes
+
+- 3df6dc6: fix(harness-pi): initialize cold resumed sessions before compaction
+- 3cd6dd8: fix(harness-pi): surface terminal provider errors before successful step boundaries
+- cdc12a1: feat(harness): add support for using adapter-native subscriptions where available
+- c550630: fix(harness-pi): register explicit custom provider models and prefer authenticated providers during model resolution
+- Updated dependencies [4d1bf28]
+- Updated dependencies [cdc12a1]
+- Updated dependencies [d70a334]
+- Updated dependencies [81ba84c]
+- Updated dependencies [813bb36]
+- Updated dependencies [c43e4b7]
+  - @ai-sdk/harness@1.0.108
+  - @ai-sdk/provider-utils@5.0.40
+
 ## 1.0.109
 
 ### Patch Changes

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import * as _ai_sdk_harness from '@ai-sdk/harness';
 import { HarnessV1Authentication, HarnessV1, HarnessV1BuiltinTool } from '@ai-sdk/harness';
-import { ExtensionFactory } from '@earendil-works/pi-coding-agent';
+import { ProviderConfig, ExtensionFactory } from '@earendil-works/pi-coding-agent';
 
 /**
  * Pi auth options. Choose an explicit mode or rely on 'auto' (precedence:
@@ -17,6 +17,12 @@ type PiThinkingLevel = 'off' | 'minimal' | 'low' | 'medium' | 'high' | 'xhigh' |
 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`.
@@ -24,9 +30,9 @@ 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;
     /**

package/dist/index.js

@@ -136,7 +136,7 @@ import {
 import { access } from "fs/promises";
 
 // src/version.ts
-var VERSION = true ? "1.0.109" : "0.0.0-test";
+var VERSION = true ? "1.0.111" : "0.0.0-test";
 
 // src/pi-auth.ts
 var DEFAULT_GATEWAY_BASE_URL = "https://ai-gateway.vercel.sh";
@@ -534,6 +534,28 @@ async function registerCustomProviders({
   }
 }
 
+// src/pi-subscription.ts
+import { homedir } from "os";
+import { resolve } from "path";
+import { isHarnessAuthenticationEnvironment as isHarnessAuthenticationEnvironment2 } from "@ai-sdk/harness/utils";
+function resolvePiSubscriptionAgentDir({
+  options,
+  env,
+  agentDir,
+  homeDirectory = homedir()
+}) {
+  if (isHarnessAuthenticationEnvironment2(options) || options === "ai-gateway") {
+    return void 0;
+  }
+  const resolvedEnvironment = resolvePiEnv({ options, env });
+  if (resolvedEnvironment.AI_GATEWAY_API_KEY != null) {
+    return void 0;
+  }
+  return resolve(
+    agentDir ?? env.PI_CODING_AGENT_DIR ?? `${homeDirectory}/.pi/agent`
+  );
+}
+
 // src/pi-events.ts
 import { z as z2 } from "zod/v4";
 var piSessionEventSchema = z2.looseObject({
@@ -662,13 +684,14 @@ function createPiModelResolver({
     const gatewayMatch = useGateway ? models.find((m) => m.provider === "vercel-ai-gateway" && matches(m)) : void 0;
     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);
   };
@@ -1286,7 +1309,9 @@ function translatePiEvent(event, state) {
       } else {
         state.pendingStepToolCallIds.clear();
         state.stepToolCallCount = void 0;
-        parts.push(...finishStep(state));
+        if (!getPiTerminalError(event)) {
+          parts.push(...finishStep(state));
+        }
       }
       return parts;
     }
@@ -2168,9 +2193,14 @@ async function createPiSession(input) {
     readableRoots: [{ sandboxDir: sandboxSkillRootDir }]
   });
   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: path6.join(agentDir, "auth.json"),
+    authPath: path6.join(nativeAgentDir ?? hostAgentDir, "auth.json"),
     modelsPath: path6.join(agentDir, "models.json")
   });
   const modelRegistry = new ModelRegistry(modelRuntime);
@@ -2189,6 +2219,14 @@ async function createPiSession(input) {
     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
@@ -2405,8 +2443,8 @@ async function createPiSession(input) {
     deferredRerun?.cancel();
     let resolveDone;
     let rejectDone;
-    const done = new Promise((resolve, reject) => {
-      resolveDone = resolve;
+    const done = new Promise((resolve2, reject) => {
+      resolveDone = resolve2;
       rejectDone = reject;
     });
     let settled = false;
@@ -2538,8 +2576,8 @@ async function createPiSession(input) {
         currentEmit?.(part);
       }
     }
-    return new Promise((resolve) => {
-      pendingToolApprovals.set(args.toolCallId, { resolve });
+    return new Promise((resolve2) => {
+      pendingToolApprovals.set(args.toolCallId, { resolve: resolve2 });
     });
   }
   function buildToolDefinitions(userTools) {
@@ -2579,7 +2617,7 @@ async function createPiSession(input) {
     let resourcesReloaded = false;
     if (piSession) {
       await disposePiSession();
-      await new Promise((resolve) => setTimeout(resolve, 25));
+      await new Promise((resolve2) => setTimeout(resolve2, 25));
       if (hasExtensionFactories) {
         await resourceLoader.reload();
         resourcesReloaded = true;
@@ -2889,7 +2927,15 @@ async function createPiSession(input) {
       if (stopped) {
         throw new Error("Pi session has been stopped.");
       }
-      await piSession?.compact(customInstructions);
+      if (piSession == null) {
+        await rebuildPiSession([], true);
+        lastToolsSignature = JSON.stringify([]);
+      }
+      const session = piSession;
+      if (session == null) {
+        throw new Error("Pi session failed to initialize.");
+      }
+      await session.compact(customInstructions);
     },
     doDestroy: async () => {
       if (stopped) return;
@@ -3207,8 +3253,8 @@ function buildUserToolDefinition(spec, pending) {
     description: spec.description ?? `User-registered tool ${spec.name}`,
     parameters: toolSpecToTypeBoxParameters(schema),
     async execute(toolCallId) {
-      return new Promise((resolve) => {
-        pending.set(toolCallId, { resolve });
+      return new Promise((resolve2) => {
+        pending.set(toolCallId, { resolve: resolve2 });
       }).then((output) => asPiToolResult(serializeToolOutput(output)));
     }
   });
@@ -3309,6 +3355,7 @@ function createPi(settings = {}) {
           ...settings.auth ? { auth: settings.auth } : {},
           ...settings.thinkingLevel ? { thinkingLevel: settings.thinkingLevel } : {},
           ...settings.mcpServers ? { mcpServers: settings.mcpServers } : {},
+          ...settings.providers ? { providers: settings.providers } : {},
           ...settings.extensionFactories ? { extensionFactories: settings.extensionFactories } : {},
           ...startOpts.headers ? { headers: startOpts.headers } : {}
         },