@ai-sdk/harness-pi 1.0.95 → 1.0.97

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @ai-sdk/harness-pi
 
+## 1.0.97
+
+### Patch Changes
+
+- Updated dependencies [371e954]
+- Updated dependencies [87b4858]
+  - @ai-sdk/harness@1.0.95
+
+## 1.0.96
+
+### Patch Changes
+
+- 8961fde: feat(harness): allow changing `model` between turns via call options
+- Updated dependencies [8961fde]
+- Updated dependencies [eb59f2a]
+  - @ai-sdk/harness@1.0.94
+  - @ai-sdk/provider-utils@5.0.34
+
 ## 1.0.95
 
 ### Patch Changes

package/dist/index.js

@@ -134,7 +134,7 @@ import {
 import { access } from "fs/promises";
 
 // src/version.ts
-var VERSION = true ? "1.0.95" : "0.0.0-test";
+var VERSION = true ? "1.0.97" : "0.0.0-test";
 
 // src/pi-auth.ts
 var DEFAULT_GATEWAY_BASE_URL = "https://ai-gateway.vercel.sh";
@@ -2015,7 +2015,7 @@ async function createPiSession(input) {
     modelRegistry,
     env: resolverEnv
   });
-  const resolvedModel = resolveModel(input.settings.model);
+  let activeResolvedModel = resolveModel(input.settings.model);
   const mcpServers = resolvePiMcpServers({
     mcpServers: input.settings.mcpServers
   });
@@ -2240,6 +2240,7 @@ async function createPiSession(input) {
         try {
           const control = await runTurn({
             text: "",
+            ...continueOpts.model ? { model: continueOpts.model } : {},
             skills: continueOpts.skills,
             tools: continueOpts.tools ?? [],
             instructions: continueOpts.instructions,
@@ -2419,7 +2420,7 @@ async function createPiSession(input) {
       customTools,
       ...hasMcpServers ? { noTools: "builtin" } : { tools: toolNames },
       ...input.settings.thinkingLevel ? { thinkingLevel: input.settings.thinkingLevel } : {},
-      ...resolvedModel ? { model: resolvedModel } : {}
+      ...activeResolvedModel ? { model: activeResolvedModel } : {}
     });
     piSession = session;
     if (hasMcpServers) {
@@ -2483,6 +2484,8 @@ async function createPiSession(input) {
         await applySessionInstructions(turnOpts.instructions);
         turnAbortController.signal.throwIfAborted();
         const didAppendDeliveredHostToolResults = appendDeliveredHostToolResults();
+        const nextModel = turnOpts.model == null ? void 0 : resolveModel(turnOpts.model);
+        if (nextModel != null) activeResolvedModel = nextModel;
         const signature = JSON.stringify(userTools.map((t) => t.name).sort());
         const needsRebuild = piSession == null || signature !== lastToolsSignature;
         let resourcesReloaded = false;
@@ -2493,6 +2496,9 @@ async function createPiSession(input) {
           );
           turnAbortController.signal.throwIfAborted();
           lastToolsSignature = signature;
+        } else if (nextModel != null && piSession != null && (piSession.model?.provider !== nextModel.provider || piSession.model.id !== nextModel.id)) {
+          await piSession.setModel(nextModel);
+          turnAbortController.signal.throwIfAborted();
         }
         if (!resourcesReloaded) {
           await reloadResourcesOnly();
@@ -2509,7 +2515,10 @@ async function createPiSession(input) {
           hostToolNames: userTools.map((tool2) => tool2.name),
           nativeToCommon: NATIVE_TO_COMMON
         });
-        currentEmit?.({ type: "stream-start" });
+        currentEmit?.({
+          type: "stream-start",
+          ...piSession?.model?.id ? { modelId: piSession.model.id } : {}
+        });
         if (didAppendDeliveredHostToolResults) {
           currentEmit?.({
             type: "finish-step",
@@ -2636,9 +2645,6 @@ async function createPiSession(input) {
   const sessionImpl = {
     sessionId: input.sessionId,
     isResume: input.isResume,
-    // The model Pi actually resolves to (the configured id, or its default when
-    // unset) — `gen_ai.request.model`.
-    ...resolvedModel?.id ? { modelId: resolvedModel.id } : {},
     // Pi has no bridge to attach to and no on-disk event log to replay; its
     // only resume path is restoring the session file on a fresh/snapshotted
     // sandbox, i.e. `rerun`.
@@ -2651,6 +2657,7 @@ async function createPiSession(input) {
       }
       return runTurn({
         text: extractUserText(promptOpts.prompt),
+        ...promptOpts.model ? { model: promptOpts.model } : {},
         skills: promptOpts.skills,
         tools: promptOpts.tools ?? [],
         instructions: promptOpts.instructions,
@@ -2687,6 +2694,7 @@ async function createPiSession(input) {
       }
       return runTurn({
         text: "",
+        ...continueOpts.model ? { model: continueOpts.model } : {},
         skills: continueOpts.skills,
         tools: continueOpts.tools ?? [],
         instructions: continueOpts.instructions,
@@ -3108,7 +3116,6 @@ function createPi(settings = {}) {
     supportsBuiltinToolFiltering: true,
     lifecycleStateSchema: piResumeStateSchema,
     doStart: async (startOpts) => {
-      const model = startOpts.model ?? settings.model;
       const lifecycleState = startOpts.continueFrom ?? startOpts.resumeFrom;
       const resumeData = lifecycleState?.data;
       return createPiSession({
@@ -3117,7 +3124,7 @@ function createPi(settings = {}) {
         sessionWorkDir: startOpts.sessionWorkDir,
         settings: {
           ...settings.auth ? { auth: settings.auth } : {},
-          ...model == null ? {} : { model },
+          ...settings.model == null ? {} : { model: settings.model },
           ...settings.thinkingLevel ? { thinkingLevel: settings.thinkingLevel } : {},
           ...settings.mcpServers ? { mcpServers: settings.mcpServers } : {},
           ...settings.extensionFactories ? { extensionFactories: settings.extensionFactories } : {}