@assistant-ui/react-ai-sdk 1.3.32 → 1.3.35

package/src/generativeTools.test.ts

@@ -1,5 +1,5 @@
 import { beforeEach, describe, expect, it, vi } from "vitest";
-import { AISDKToolkit, generativeTools } from "./generativeTools";
+import { AISDKToolkit } from "./generativeTools";
 import { wrapModelContentEnvelope } from "./modelContentEnvelope";
 
 const mocks = vi.hoisted(() => ({
@@ -19,20 +19,9 @@ vi.mock("@ai-sdk/mcp/mcp-stdio", () => ({
   })),
 }));
 
-describe("generativeTools", () => {
-  beforeEach(() => {
-    mocks.close.mockReset();
-    mocks.tools.mockReset();
-    mocks.createMCPClient.mockReset();
-  });
-
-  it("merges frontend tools with toolkit tools", () => {
-    const toolSet = generativeTools({
-      frontendTools: {
-        clientTool: {
-          parameters: { type: "object", properties: {} },
-        },
-      },
+describe("AISDKToolkit.tools()", () => {
+  it("merges frontend tools with toolkit tools", async () => {
+    const toolSet = await new AISDKToolkit({
       toolkit: {
         serverTool: {
           type: "backend",
@@ -41,6 +30,12 @@ describe("generativeTools", () => {
           execute: async () => "ok",
         } as never,
       },
+    }).tools({
+      frontend: {
+        clientTool: {
+          parameters: { type: "object", properties: {} },
+        },
+      },
     });
 
     expect(toolSet.clientTool).toBeDefined();
@@ -48,8 +43,8 @@ describe("generativeTools", () => {
     expect(toolSet.serverTool?.execute).toBeTypeOf("function");
   });
 
-  it("keeps a flat toolkit tool named tools", () => {
-    const toolSet = generativeTools({
+  it("keeps a flat toolkit tool named tools", async () => {
+    const toolSet = await new AISDKToolkit({
       toolkit: {
         tools: {
           type: "backend",
@@ -58,27 +53,14 @@ describe("generativeTools", () => {
           execute: async () => "ok",
         } as never,
       },
-    });
+    }).tools();
 
     expect(toolSet.tools?.description).toBe("Actually a tool, not config");
     expect(toolSet.tools?.execute).toBeTypeOf("function");
   });
 
-  it("rejects MCP entries because they require pooled clients", () => {
-    expect(() =>
-      generativeTools({
-        toolkit: {
-          docs: {
-            type: "mcp",
-            server: { type: "http", url: "http://localhost:3001/mcp" },
-          },
-        },
-      }),
-    ).toThrow(/requires AISDKToolkit/);
-  });
-
-  it("converts provider tools without an execute function", () => {
-    const toolSet = generativeTools({
+  it("converts provider tools without an execute function", async () => {
+    const toolSet = await new AISDKToolkit({
       toolkit: {
         web_search: {
           type: "provider",
@@ -86,7 +68,7 @@ describe("generativeTools", () => {
           args: { searchContextSize: "low" },
         },
       },
-    });
+    }).tools();
 
     expect(toolSet.web_search).toMatchObject({
       type: "provider",
@@ -97,8 +79,8 @@ describe("generativeTools", () => {
     expect(toolSet.web_search).not.toHaveProperty("execute");
   });
 
-  it("forwards provider tool parameters and providerOptions when present", () => {
-    const toolSet = generativeTools({
+  it("forwards provider tool parameters and providerOptions when present", async () => {
+    const toolSet = await new AISDKToolkit({
       toolkit: {
         web_search: {
           type: "provider",
@@ -116,7 +98,7 @@ describe("generativeTools", () => {
           },
         },
       },
-    });
+    }).tools();
 
     expect(toolSet.web_search).toMatchObject({
       type: "provider",
@@ -129,8 +111,8 @@ describe("generativeTools", () => {
     expect(toolSet.web_search).toHaveProperty("inputSchema");
   });
 
-  it("forwards explicit false supportsDeferredResults", () => {
-    const toolSet = generativeTools({
+  it("forwards explicit false supportsDeferredResults", async () => {
+    const toolSet = await new AISDKToolkit({
       toolkit: {
         web_search: {
           type: "provider",
@@ -139,7 +121,7 @@ describe("generativeTools", () => {
           supportsDeferredResults: false,
         },
       },
-    });
+    }).tools();
 
     expect(toolSet.web_search).toMatchObject({
       supportsDeferredResults: false,
@@ -324,18 +306,18 @@ describe("AISDKToolkit", () => {
   });
 });
 
-describe("generativeTools toModelOutput", () => {
+describe("AISDKToolkit toModelOutput", () => {
   const createWeatherTools = (toModelOutput?: any) =>
-    generativeTools({
+    new AISDKToolkit({
       toolkit: {
         get_weather: {
           ...(toModelOutput && { toModelOutput }),
         },
       } as any,
-    });
+    }).tools();
 
   it("adapts assistant-ui model content parts to the AI SDK tool output shape", async () => {
-    const tools = createWeatherTools(({ output }: any) => [
+    const tools = await createWeatherTools(({ output }: any) => [
       { type: "text", text: `Weather card displayed: ${output.location}` },
     ]);
 
@@ -353,7 +335,7 @@ describe("generativeTools toModelOutput", () => {
 
   it("uses stored model content envelopes without re-running the custom projector", async () => {
     let called = false;
-    const tools = createWeatherTools(() => {
+    const tools = await createWeatherTools(() => {
       called = true;
       return [{ type: "text", text: "recomputed" }];
     });
@@ -374,7 +356,7 @@ describe("generativeTools toModelOutput", () => {
   });
 
   it("falls back to default model output when no custom projector is defined", async () => {
-    const tools = createWeatherTools();
+    const tools = await createWeatherTools();
 
     const output = await tools.get_weather!.toModelOutput!({
       toolCallId: "tc-weather",
@@ -389,7 +371,7 @@ describe("generativeTools toModelOutput", () => {
   });
 
   it("uses stored model content envelopes when no custom projector is defined", async () => {
-    const tools = createWeatherTools();
+    const tools = await createWeatherTools();
 
     const output = await tools.get_weather!.toModelOutput!({
       toolCallId: "tc-weather",

package/src/generativeTools.ts

@@ -1,7 +1,7 @@
 import { jsonSchema, type ToolSet } from "ai";
 import type { MCPClient, MCPClientConfig } from "@ai-sdk/mcp";
 import { createMCPClient } from "@ai-sdk/mcp";
-import { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+import { Experimental_StdioMCPTransport } from "#mcp-stdio";
 import {
   toJSONSchema,
   type Tool,
@@ -31,6 +31,11 @@ const neverAbort = new AbortController().signal;
 const parametersToInputSchema = (parameters: Tool["parameters"] | undefined) =>
   jsonSchema(parameters ? toJSONSchema(parameters) : EMPTY_SCHEMA);
 
+/**
+ * @deprecated Options for the deprecated {@link generativeTools}. Use
+ * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /
+ * {@link AISDKToolkitToolsOptions} instead.
+ */
 export interface GenerativeToolsOptions {
   /**
    * The server build of a generative toolkit (schema + server `execute`). Typed
@@ -66,16 +71,25 @@ export type AISDKToolkitToolsOptions = {
  * resolves to the server build — schema + `execute`, with `render` stripped) and
  * pass it here. Tools without an `execute` are still exposed to the model but
  * left for the client to fulfill. `frontendTools` lets the client contribute
- * tools that aren't in the static toolkit. Use {@link AISDKToolkit} when the
- * toolkit contains MCP entries.
+ * tools that aren't in the static toolkit.
+ *
+ * @deprecated Use {@link AISDKToolkit} instead:
+ * `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset
+ * (it also opens MCP server connections), so it replaces `generativeTools`
+ * everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and
+ * `.tools()` is async. `generativeTools` will be removed in a future version.
  *
  * @example
  * ```ts
+ * // Define once at module scope so any MCP connections pool across requests.
+ * const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });
+ *
+ * // In your route handler:
  * const { tools } = await req.json();
  * streamText({
  *   model,
  *   messages,
- *   tools: generativeTools({ toolkit: docsToolkit, frontendTools: tools }),
+ *   tools: await aiToolkit.tools({ frontend: tools }),
  * });
  * ```
  */

package/src/index.native.ts

@@ -0,0 +1,3 @@
+/// <reference types="@assistant-ui/core/react" />
+
+export * from "./client";

package/src/index.ts

@@ -1,18 +1,6 @@
 /// <reference types="@assistant-ui/core/react" />
 
-export { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime";
-export { useChatRuntime } from "./ui/use-chat/useChatRuntime";
-export type { UseChatRuntimeOptions } from "./ui/use-chat/useChatRuntime";
-export { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport";
-export {
-  RESUMABLE_STREAM_ID_HEADER,
-  createResumableSessionStorage,
-} from "./ui/resumable";
-export type {
-  AssistantChatResumableOptions,
-  ResumableClientStorage,
-} from "./ui/resumable";
-export { frontendTools } from "./frontendTools";
+export * from "./client";
 export {
   AISDKToolkit,
   generativeTools,
@@ -20,6 +8,3 @@ export {
   type AISDKToolkitToolsOptions,
   type GenerativeToolsOptions,
 } from "./generativeTools";
-export { injectQuoteContext } from "./injectQuoteContext";
-export type { ThreadTokenUsage, TokenUsageExtractableMessage } from "./usage";
-export { getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage";

package/src/mcp-stdio.node.ts

@@ -0,0 +1 @@
+export { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";

package/src/mcp-stdio.unsupported.ts

@@ -0,0 +1,12 @@
+import type { Experimental_StdioMCPTransport as NodeStdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+
+class UnsupportedStdioMCPTransport {
+  constructor() {
+    throw new Error(
+      "stdio MCP transport requires a runtime that can spawn a subprocess, such as Node, Bun, or Deno (with --allow-run). Use an HTTP or SSE MCP server config in browser, React Native, edge, or worker runtimes.",
+    );
+  }
+}
+
+export const Experimental_StdioMCPTransport =
+  UnsupportedStdioMCPTransport as unknown as typeof NodeStdioMCPTransport;

package/src/ui/use-chat/useAISDKRuntime.test.ts

@@ -8,10 +8,14 @@ import { validateUIMessages } from "ai";
 // in isolation). Every other dependency — useExternalStoreRuntime,
 // useToolInvocations, the message converter — runs for real.
 vi.mock("./useExternalHistory", () => ({
-  useExternalHistory: vi.fn(() => false),
+  useExternalHistory: vi.fn(() => ({
+    isLoading: false,
+    deleteMessage: vi.fn().mockResolvedValue(undefined),
+  })),
   toExportedMessageRepository: vi.fn(),
 }));
 
+import { useExternalHistory } from "./useExternalHistory";
 import { useAISDKRuntime } from "./useAISDKRuntime";
 
 const createChatHelpers = (messages: any[] = []) => {
@@ -37,9 +41,19 @@ const createChatHelpers = (messages: any[] = []) => {
   return chatHelpers;
 };
 
+const textOf = (message: any): string =>
+  message.content
+    .filter((part: any) => part.type === "text")
+    .map((part: any) => part.text)
+    .join("|");
+
 describe("useAISDKRuntime", () => {
   beforeEach(() => {
     vi.clearAllMocks();
+    vi.mocked(useExternalHistory).mockReturnValue({
+      isLoading: false,
+      deleteMessage: vi.fn().mockResolvedValue(undefined),
+    });
   });
 
   it("sends a new user message through the runtime", async () => {
@@ -271,6 +285,45 @@ describe("useAISDKRuntime", () => {
     );
   });
 
+  it("deletes only the selected message from AI SDK state", async () => {
+    const deleteMessage = vi.fn().mockResolvedValue(undefined);
+    vi.mocked(useExternalHistory).mockReturnValue({
+      isLoading: false,
+      deleteMessage,
+    });
+    const chat = createChatHelpers([
+      { id: "u1", role: "user", parts: [{ type: "text", text: "first" }] },
+      {
+        id: "a1",
+        role: "assistant",
+        parts: [{ type: "text", text: "first-answer" }],
+      },
+      { id: "u2", role: "user", parts: [{ type: "text", text: "second" }] },
+      {
+        id: "a2",
+        role: "assistant",
+        parts: [{ type: "text", text: "second-answer" }],
+      },
+    ]);
+
+    const { result } = renderHook(() => useAISDKRuntime(chat));
+
+    await waitFor(() => {
+      expect(result.current.thread.getState().messages).toHaveLength(4);
+    });
+
+    await act(async () => {
+      result.current.thread.getMessageById("u2").delete();
+    });
+
+    expect(deleteMessage).toHaveBeenCalledWith("u2");
+    expect(chat.messages.map((message: any) => message.id)).toEqual([
+      "u1",
+      "a1",
+      "a2",
+    ]);
+  });
+
   it("edit slices history to parentId and sends the edited message", async () => {
     const chat = createChatHelpers([
       { id: "u1", role: "user", parts: [{ type: "text", text: "first" }] },
@@ -439,4 +492,54 @@ describe("useAISDKRuntime", () => {
     const { result } = renderHook(() => useAISDKRuntime(chat, { suggestions }));
     expect(result.current.thread.getState().suggestions).toEqual(suggestions);
   });
+
+  it("merges consecutive assistant messages into one turn by default", async () => {
+    const chat = createChatHelpers([
+      { id: "u1", role: "user", parts: [{ type: "text", text: "hi" }] },
+      { id: "a1", role: "assistant", parts: [{ type: "text", text: "first" }] },
+      {
+        id: "a2",
+        role: "assistant",
+        parts: [{ type: "text", text: "second" }],
+      },
+    ]);
+
+    const { result } = renderHook(() => useAISDKRuntime(chat));
+
+    await waitFor(() => {
+      expect(result.current.thread.getState().messages.length).toBe(2);
+    });
+
+    const messages = result.current.thread.getState().messages;
+    expect(messages.map((m: any) => m.role)).toEqual(["user", "assistant"]);
+    expect(textOf(messages[1])).toBe("first|second");
+  });
+
+  it('keeps consecutive assistant messages separate when joinStrategy is "none"', async () => {
+    const chat = createChatHelpers([
+      { id: "u1", role: "user", parts: [{ type: "text", text: "hi" }] },
+      { id: "a1", role: "assistant", parts: [{ type: "text", text: "first" }] },
+      {
+        id: "a2",
+        role: "assistant",
+        parts: [{ type: "text", text: "second" }],
+      },
+    ]);
+
+    const { result } = renderHook(() =>
+      useAISDKRuntime(chat, { joinStrategy: "none" }),
+    );
+
+    await waitFor(() => {
+      expect(result.current.thread.getState().messages.length).toBe(3);
+    });
+
+    const messages = result.current.thread.getState().messages;
+    expect(messages.map((m: any) => m.role)).toEqual([
+      "user",
+      "assistant",
+      "assistant",
+    ]);
+    expect(messages.slice(1).map(textOf)).toEqual(["first", "second"]);
+  });
 });

package/src/ui/use-chat/useAISDKRuntime.ts

@@ -6,6 +6,7 @@ import { isToolUIPart, generateId } from "ai";
 import {
   useExternalStoreRuntime,
   useRuntimeAdapters,
+  type JoinStrategy,
 } from "@assistant-ui/core/react";
 import type { ToolExecutionStatus } from "@assistant-ui/core";
 import type {
@@ -82,6 +83,15 @@ export type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {
    * (for example, an SSE reconnect endpoint keyed by turn id).
    */
   onResume?: ExternalStoreAdapter["onResume"];
+  /**
+   * How consecutive assistant messages are rendered.
+   *
+   * `"concat-content"` (the default) merges them into a single thread message.
+   * `"none"` keeps each assistant message as its own thread message, which is
+   * useful when a backend persists proactive or consecutive assistant messages
+   * as separate entries.
+   */
+  joinStrategy?: JoinStrategy | undefined;
 };
 
 export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
@@ -93,6 +103,7 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
     toCreateMessage: customToCreateMessage,
     cancelPendingToolCallsOnSend = true,
     onResume,
+    joinStrategy,
   } = adapter;
   const contextAdapters = useRuntimeAdapters();
   const [toolStatuses, setToolStatuses] = useState<
@@ -126,6 +137,7 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
   const messages = AISDKMessageConverter.useThreadMessages({
     isRunning,
     messages: chatHelpers.messages,
+    joinStrategy,
     metadata: useMemo(
       () => ({
         toolStatuses,
@@ -146,7 +158,7 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
     },
   }));
 
-  const isLoading = useExternalHistory(
+  const { isLoading, deleteMessage: deleteHistoryMessage } = useExternalHistory(
     runtimeRef,
     adapters?.history ?? contextAdapters?.history,
     AISDKMessageConverter.toThreadMessages as (
@@ -303,6 +315,24 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
         metadata: message.runConfig,
       });
     },
+    onDelete: async (messageId) => {
+      const threadMessages = runtimeRef.current.thread.getState().messages;
+      const messageIndex = threadMessages.findIndex(
+        (message) => message.id === messageId,
+      );
+      if (messageIndex === -1) return;
+
+      await deleteHistoryMessage(messageId);
+
+      const deleteIds = new Set(
+        getExternalStoreMessages<UI_MESSAGE>(threadMessages[messageIndex]!).map(
+          (message) => message.id,
+        ),
+      );
+      chatHelpers.setMessages((current) =>
+        current.filter((message) => !deleteIds.has(message.id)),
+      );
+    },
     onReload: async (parentId: string | null, config) => {
       lastRunConfigRef.current = config.runConfig;
       const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);

package/src/ui/use-chat/useChatRuntime.ts

@@ -29,6 +29,7 @@ export type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =
       adapters?: AISDKRuntimeAdapter["adapters"] | undefined;
       toCreateMessage?: CustomToCreateMessageFunction;
       onResume?: AISDKRuntimeAdapter["onResume"];
+      joinStrategy?: AISDKRuntimeAdapter["joinStrategy"];
     };
 
 const useDynamicChatTransport = <UI_MESSAGE extends UIMessage = UIMessage>(
@@ -78,6 +79,7 @@ const useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
     unstable_capabilities: _unstable_capabilities,
     suggestions: _suggestions,
     onResume,
+    joinStrategy,
     ...chatOptions
   } = options ?? {};
   // peel guard: any shared key left in `chatOptions` collapses this to `never`
@@ -103,6 +105,7 @@ const useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
     ...pickExternalStoreSharedOptions(options ?? {}),
     ...(toCreateMessage && { toCreateMessage }),
     ...(onResume && { onResume }),
+    ...(joinStrategy && { joinStrategy }),
   });
 
   if (transport instanceof AssistantChatTransport) {

package/src/ui/use-chat/useExternalHistory.ts

@@ -292,5 +292,34 @@ export const useExternalHistory = <TMessage>(
     };
   }, [formatAdapter, storageFormatAdapter, runtimeRef]);
 
-  return isLoading;
+  const deleteMessage = useCallback(
+    async (messageId: string) => {
+      if (!formatAdapter?.delete) return;
+
+      const messages = runtimeRef.current.thread.getState().messages;
+      const messageIndex = messages.findIndex((m) => m.id === messageId);
+      if (messageIndex === -1) return;
+
+      const previousInnerMessages = messages
+        .slice(0, messageIndex)
+        .flatMap(getExternalStoreMessages<TMessage>);
+      let parentId = previousInnerMessages.at(-1)
+        ? storageFormatAdapter.getId(previousInnerMessages.at(-1)!)
+        : null;
+      const itemsToDelete = getExternalStoreMessages<TMessage>(
+        messages[messageIndex]!,
+      ).map((message) => {
+        const item = { parentId, message };
+        parentId = storageFormatAdapter.getId(message);
+        return item;
+      });
+
+      await formatAdapter.delete(itemsToDelete);
+
+      historyIds.current.delete(messageId);
+    },
+    [formatAdapter, runtimeRef, storageFormatAdapter],
+  );
+
+  return { isLoading, deleteMessage };
 };

package/src/ui/utils/convertMessage.test.ts

@@ -1,5 +1,8 @@
 import { describe, expect, it } from "vitest";
-import type { ReadonlyJSONObject } from "assistant-stream/utils";
+import {
+  getPartialJsonObjectFieldState,
+  type ReadonlyJSONObject,
+} from "assistant-stream/utils";
 import {
   AISDKMessageConverter,
   type AISDKMessageConverterMetadata,
@@ -365,6 +368,34 @@ describe("AISDKMessageConverter", () => {
     expect(toolCall?.argsText).toBe('{"city":"NYC');
   });
 
+  it("attaches partial-JSON meta marking the trailing streaming field", () => {
+    const converted = AISDKMessageConverter.toThreadMessages([
+      {
+        id: "a1",
+        role: "assistant",
+        parts: [
+          {
+            type: "tool-weather",
+            toolCallId: "tc-1",
+            state: "input-streaming",
+            input: { city: "NYC", units: "F" },
+          },
+        ],
+      } as any,
+    ]);
+
+    const toolCall = converted[0]?.content.find(
+      (part): part is any => part.type === "tool-call",
+    );
+    expect(toolCall?.args).toMatchObject({ city: "NYC", units: "F" });
+    expect(getPartialJsonObjectFieldState(toolCall!.args, ["city"])).toBe(
+      "complete",
+    );
+    expect(getPartialJsonObjectFieldState(toolCall!.args, ["units"])).toBe(
+      "partial",
+    );
+  });
+
   it("keeps observed key order from streaming snapshots for final tool args", () => {
     const metadata: AISDKMessageConverterMetadata = {
       toolArgsKeyOrderCache: new Map<string, Map<string, string[]>>(),
@@ -518,19 +549,19 @@ describe("AISDKMessageConverter", () => {
 
     const first = convertWithInput({ city: "NYC" });
     expect(first?.argsText).toBe('{"city":"NYC');
-    expect(first?.args).toEqual({ city: "NYC" });
+    expect(first?.args).toMatchObject({ city: "NYC" });
 
     const dropped = convertWithInput(null);
     expect(dropped?.argsText).toBe('{"city":"NYC');
-    expect(dropped?.args).toEqual({ city: "NYC" });
+    expect(dropped?.args).toMatchObject({ city: "NYC" });
 
     const undef = convertWithInput(undefined);
     expect(undef?.argsText).toBe('{"city":"NYC');
-    expect(undef?.args).toEqual({ city: "NYC" });
+    expect(undef?.args).toMatchObject({ city: "NYC" });
 
     const grown = convertWithInput({ city: "NYC", units: "F" });
     expect(grown?.argsText).toBe('{"city":"NYC","units":"F');
-    expect(grown?.args).toEqual({ city: "NYC", units: "F" });
+    expect(grown?.args).toMatchObject({ city: "NYC", units: "F" });
   });
 
   it("preserves last good input across terminal state transitions", () => {

package/src/ui/utils/convertMessage.ts

@@ -15,7 +15,10 @@ import {
   type ThreadMessageLike,
   type McpAppMetadata,
 } from "@assistant-ui/core";
-import type { ReadonlyJSONObject } from "assistant-stream/utils";
+import {
+  parsePartialJsonObject,
+  type ReadonlyJSONObject,
+} from "assistant-stream/utils";
 import { unwrapModelContentEnvelope } from "../../modelContentEnvelope";
 
 type MessageMetadata = ThreadMessageLike["metadata"];
@@ -259,6 +262,11 @@ function convertParts(
         if (part.state === "input-streaming") {
           // strip closing delimiters added by the AI SDK's fix-json
           argsText = stripClosingDelimiters(argsText);
+          // Re-parse so args carries the partial-JSON meta that marks which
+          // field is still mid-arrival, like every argsText-based runtime.
+          // The key-order cache appends new keys last, so the trailing field
+          // of the stripped text is the streaming frontier.
+          args = parsePartialJsonObject(argsText) ?? args;
         } else {
           metadata.toolArgsKeyOrderCache?.delete(argsKeyOrderCacheKey);
           if (

package/dist/assistant-stream/dist/core/AssistantStream.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/AssistantStreamChunk.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/accumulators/AssistantMessageStream.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/accumulators/assistant-message-accumulator.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/modules/assistant-stream.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/modules/text.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/modules/tool-call.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/serialization/PlainText.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/serialization/assistant-transport/AssistantTransport.d.ts

@@ -1 +0,0 @@
-export { };

package/dist/assistant-stream/dist/core/serialization/data-stream/DataStream.d.ts

@@ -1 +0,0 @@
-export { };