@ai-sdk/harness-pi 1.0.98 → 1.0.99

package/CHANGELOG.md

@@ -1,5 +1,12 @@
 # @ai-sdk/harness-pi
 
+## 1.0.99
+
+### Patch Changes
+
+- e115d16: fix(harness-pi): stream Pi tool input while it is generated
+  - @ai-sdk/harness@1.0.97
+
 ## 1.0.98
 
 ### Patch Changes

package/dist/index.js

@@ -134,7 +134,7 @@ import {
 import { access } from "fs/promises";
 
 // src/version.ts
-var VERSION = true ? "1.0.98" : "0.0.0-test";
+var VERSION = true ? "1.0.99" : "0.0.0-test";
 
 // src/pi-auth.ts
 var DEFAULT_GATEWAY_BASE_URL = "https://ai-gateway.vercel.sh";
@@ -522,7 +522,13 @@ var piSessionEventSchema = z2.looseObject({
   type: z2.string(),
   assistantMessageEvent: z2.looseObject({
     type: z2.string().optional(),
-    delta: z2.string().optional()
+    delta: z2.string().optional(),
+    // `toolcall_start` / `toolcall_delta` / `toolcall_end` address a content
+    // block by index rather than by tool call id. The id and name live in the
+    // partial assistant message at that index, which Pi fills in before the
+    // first delta arrives.
+    contentIndex: z2.number().optional(),
+    partial: z2.looseObject({ content: z2.array(z2.unknown()).optional() }).optional()
   }).optional(),
   toolCallId: z2.string().optional(),
   toolName: z2.string().optional(),
@@ -1047,6 +1053,7 @@ function createPiTranslatorState(options = {}) {
     currentReasoningId: void 0,
     reasoningStarted: false,
     observedToolNames: /* @__PURE__ */ new Map(),
+    streamingToolInputIds: /* @__PURE__ */ new Map(),
     pendingStepToolCallIds: /* @__PURE__ */ new Set(),
     stepToolCallCount: void 0,
     stepOpen: false,
@@ -1091,6 +1098,26 @@ function resolveToolName(state, nativeName) {
   const common = state.nativeToCommonNameMap.get(nativeName);
   return { wire: common ?? nativeName, native: nativeName };
 }
+function resolveToolDispatch(state, nativeName) {
+  const isMcpTool = !state.hostToolNames.has(nativeName) && (nativeName === "mcp" || nativeName.startsWith("mcp__"));
+  return {
+    isMcpTool,
+    providerExecuted: state.builtinToolNames.has(nativeName) || isMcpTool
+  };
+}
+function readStreamingToolCall(event) {
+  const update = event.assistantMessageEvent;
+  const contentIndex = update?.contentIndex;
+  if (typeof contentIndex !== "number") return void 0;
+  const block = update?.partial?.content?.[contentIndex];
+  if (!block || typeof block !== "object") return void 0;
+  const record = block;
+  if (record.type !== "toolCall") return void 0;
+  const { id, name } = record;
+  if (typeof id !== "string" || id.length === 0) return void 0;
+  if (typeof name !== "string" || name.length === 0) return void 0;
+  return { contentIndex, id, name };
+}
 function finishStep(state) {
   if (!state.stepOpen || state.pendingStepToolCallIds.size > 0) return [];
   state.stepOpen = false;
@@ -1141,6 +1168,7 @@ function translatePiEvent(event, state) {
       state.currentTextId = void 0;
       state.currentReasoningId = void 0;
       state.reasoningStarted = false;
+      state.streamingToolInputIds.clear();
       return [];
     }
     case "message_update": {
@@ -1189,6 +1217,37 @@ function translatePiEvent(event, state) {
         });
         return parts;
       }
+      if (update.type === "toolcall_start") {
+        const call = readStreamingToolCall(event);
+        if (!call) return [];
+        const { wire, native } = resolveToolName(state, call.name);
+        const { isMcpTool, providerExecuted } = resolveToolDispatch(
+          state,
+          native
+        );
+        state.streamingToolInputIds.set(call.contentIndex, call.id);
+        return [
+          {
+            type: "tool-input-start",
+            id: call.id,
+            toolName: wire,
+            ...providerExecuted ? { providerExecuted: true } : {},
+            ...isMcpTool ? { dynamic: true } : {}
+          }
+        ];
+      }
+      if (update.type === "toolcall_delta" || update.type === "toolcall_end") {
+        const contentIndex = update.contentIndex;
+        if (typeof contentIndex !== "number") return [];
+        const id = state.streamingToolInputIds.get(contentIndex);
+        if (id === void 0) return [];
+        if (update.type === "toolcall_end") {
+          state.streamingToolInputIds.delete(contentIndex);
+          return [{ type: "tool-input-end", id }];
+        }
+        if (typeof update.delta !== "string") return [];
+        return [{ type: "tool-input-delta", id, delta: update.delta }];
+      }
       return [];
     }
     case "message_end":
@@ -1231,8 +1290,10 @@ function translatePiEvent(event, state) {
       if (!event.toolCallId || !event.toolName) return [];
       const { wire, native } = resolveToolName(state, event.toolName);
       state.observedToolNames.set(event.toolCallId, wire);
-      const isMcpTool = !state.hostToolNames.has(native) && (native === "mcp" || native.startsWith("mcp__"));
-      const providerExecuted = state.builtinToolNames.has(native) || isMcpTool;
+      const { isMcpTool, providerExecuted } = resolveToolDispatch(
+        state,
+        native
+      );
       if (isMcpTool) state.dynamicToolCallIds.add(event.toolCallId);
       const input = serializeToolOutput(event.args ?? event.input ?? {});
       return [