@tanstack/ai 0.0.1

package/dist/esm/message-converters.js

@@ -0,0 +1,162 @@
+function getTextContent(content) {
+  if (content === null) {
+    return "";
+  }
+  if (typeof content === "string") {
+    return content;
+  }
+  return content.filter((part) => part.type === "text").map((part) => part.content).join("");
+}
+function convertMessagesToModelMessages(messages) {
+  const modelMessages = [];
+  for (const msg of messages) {
+    if ("parts" in msg) {
+      modelMessages.push(...uiMessageToModelMessages(msg));
+    } else {
+      modelMessages.push(msg);
+    }
+  }
+  return modelMessages;
+}
+function uiMessageToModelMessages(uiMessage) {
+  const messages = [];
+  if (uiMessage.role === "system") {
+    return messages;
+  }
+  const textParts = [];
+  const toolCallParts = [];
+  const toolResultParts = [];
+  for (const part of uiMessage.parts) {
+    if (part.type === "text") {
+      textParts.push(part);
+    } else if (part.type === "tool-call") {
+      toolCallParts.push(part);
+    } else if (part.type === "tool-result") {
+      toolResultParts.push(part);
+    }
+  }
+  const content = textParts.map((p) => p.content).join("") || null;
+  const toolCalls = toolCallParts.length > 0 ? toolCallParts.filter(
+    (p) => p.state === "input-complete" || p.state === "approval-responded" || p.output !== void 0
+    // Include if has output (client tool result)
+  ).map((p) => ({
+    id: p.id,
+    type: "function",
+    function: {
+      name: p.name,
+      arguments: p.arguments
+    }
+  })) : void 0;
+  if (uiMessage.role !== "assistant" || content || !toolCalls) {
+    messages.push({
+      role: uiMessage.role,
+      content,
+      ...toolCalls && toolCalls.length > 0 && { toolCalls }
+    });
+  } else if (toolCalls.length > 0) {
+    messages.push({
+      role: "assistant",
+      content,
+      toolCalls
+    });
+  }
+  for (const toolResultPart of toolResultParts) {
+    if (toolResultPart.state === "complete" || toolResultPart.state === "error") {
+      messages.push({
+        role: "tool",
+        content: toolResultPart.content,
+        toolCallId: toolResultPart.toolCallId
+      });
+    }
+  }
+  return messages;
+}
+function modelMessageToUIMessage(modelMessage, id) {
+  const parts = [];
+  const textContent = getTextContent(modelMessage.content);
+  if (textContent) {
+    parts.push({
+      type: "text",
+      content: textContent
+    });
+  }
+  if (modelMessage.toolCalls && modelMessage.toolCalls.length > 0) {
+    for (const toolCall of modelMessage.toolCalls) {
+      parts.push({
+        type: "tool-call",
+        id: toolCall.id,
+        name: toolCall.function.name,
+        arguments: toolCall.function.arguments,
+        state: "input-complete"
+        // Model messages have complete arguments
+      });
+    }
+  }
+  if (modelMessage.role === "tool" && modelMessage.toolCallId) {
+    parts.push({
+      type: "tool-result",
+      toolCallId: modelMessage.toolCallId,
+      content: getTextContent(modelMessage.content),
+      state: "complete"
+    });
+  }
+  return {
+    id: id || generateMessageId(),
+    role: modelMessage.role === "tool" ? "assistant" : modelMessage.role,
+    parts
+  };
+}
+function modelMessagesToUIMessages(modelMessages) {
+  const uiMessages = [];
+  let currentAssistantMessage = null;
+  for (const msg of modelMessages) {
+    if (msg.role === "tool") {
+      if (currentAssistantMessage && currentAssistantMessage.role === "assistant") {
+        currentAssistantMessage.parts.push({
+          type: "tool-result",
+          toolCallId: msg.toolCallId,
+          content: getTextContent(msg.content),
+          state: "complete"
+        });
+      } else {
+        const toolResultUIMessage = modelMessageToUIMessage(msg);
+        uiMessages.push(toolResultUIMessage);
+      }
+    } else {
+      const uiMessage = modelMessageToUIMessage(msg);
+      uiMessages.push(uiMessage);
+      if (msg.role === "assistant") {
+        currentAssistantMessage = uiMessage;
+      } else {
+        currentAssistantMessage = null;
+      }
+    }
+  }
+  return uiMessages;
+}
+function normalizeToUIMessage(message, generateId) {
+  if ("parts" in message) {
+    return {
+      ...message,
+      id: message.id || generateId(),
+      createdAt: message.createdAt || /* @__PURE__ */ new Date()
+    };
+  } else {
+    return {
+      ...modelMessageToUIMessage(message, generateId()),
+      createdAt: /* @__PURE__ */ new Date()
+    };
+  }
+}
+function generateMessageId() {
+  return `msg-${Date.now()}-${Math.random().toString(36).substring(7)}`;
+}
+export {
+  convertMessagesToModelMessages,
+  generateMessageId,
+  modelMessageToUIMessage,
+  modelMessagesToUIMessages,
+  normalizeToUIMessage,
+  uiMessageToModelMessages
+};
+//# sourceMappingURL=message-converters.js.map

package/dist/esm/message-converters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-converters.js","sources":["../../src/message-converters.ts"],"sourcesContent":["/**\n * Message Converters\n *\n * Functions for converting between UIMessage and ModelMessage formats.\n */\n\nimport type {\n  ContentPart,\n  MessagePart,\n  ModelMessage,\n  TextPart,\n  ToolCallPart,\n  ToolResultPart,\n  UIMessage,\n} from './types'\n\n/**\n * Helper to extract text content from string or ContentPart array\n * For multimodal content, this extracts only the text parts\n */\nfunction getTextContent(content: string | null | Array<ContentPart>): string {\n  if (content === null) {\n    return ''\n  }\n  if (typeof content === 'string') {\n    return content\n  }\n  // Extract text from ContentPart array\n  return content\n    .filter((part) => part.type === 'text')\n    .map((part) => part.content)\n    .join('')\n}\n\n/**\n * Convert UIMessages or ModelMessages to ModelMessages\n */\nexport function convertMessagesToModelMessages(\n  messages: Array<UIMessage | ModelMessage>,\n): Array<ModelMessage> {\n  const modelMessages: Array<ModelMessage> = []\n  for (const msg of messages) {\n    if ('parts' in msg) {\n      // UIMessage - convert to ModelMessages\n      modelMessages.push(...uiMessageToModelMessages(msg))\n    } else {\n      // Already ModelMessage\n      modelMessages.push(msg)\n    }\n  }\n  return modelMessages\n}\n\n/**\n * Convert a UIMessage to ModelMessage(s)\n *\n * This conversion handles the parts-based structure:\n * - Text parts → content field\n * - ToolCall parts → toolCalls array\n * - ToolResult parts → separate role=\"tool\" messages\n *\n * @param uiMessage - The UIMessage to convert\n * @returns An array of ModelMessages (may be multiple if tool results are present)\n */\nexport function uiMessageToModelMessages(\n  uiMessage: UIMessage,\n): Array<ModelMessage> {\n  const messages: Array<ModelMessage> = []\n\n  // Skip system messages - they're handled via systemPrompts, not ModelMessages\n  if (uiMessage.role === 'system') {\n    return messages\n  }\n\n  // Separate parts by type\n  // Note: thinking parts are UI-only and not included in ModelMessages\n  const textParts: Array<TextPart> = []\n  const toolCallParts: Array<ToolCallPart> = []\n  const toolResultParts: Array<ToolResultPart> = []\n\n  for (const part of uiMessage.parts) {\n    if (part.type === 'text') {\n      textParts.push(part)\n    } else if (part.type === 'tool-call') {\n      toolCallParts.push(part)\n    } else if (part.type === 'tool-result') {\n      toolResultParts.push(part)\n    }\n    // thinking parts are skipped - they're UI-only\n  }\n\n  // Build the main message (user or assistant)\n  const content = textParts.map((p) => p.content).join('') || null\n  const toolCalls =\n    toolCallParts.length > 0\n      ? toolCallParts\n          .filter(\n            (p) =>\n              p.state === 'input-complete' ||\n              p.state === 'approval-responded' ||\n              p.output !== undefined, // Include if has output (client tool result)\n          )\n          .map((p) => ({\n            id: p.id,\n            type: 'function' as const,\n            function: {\n              name: p.name,\n              arguments: p.arguments,\n            },\n          }))\n      : undefined\n\n  // Create the main message\n  if (uiMessage.role !== 'assistant' || content || !toolCalls) {\n    messages.push({\n      role: uiMessage.role,\n      content,\n      ...(toolCalls && toolCalls.length > 0 && { toolCalls }),\n    })\n  } else if (toolCalls.length > 0) {\n    // Assistant message with only tool calls\n    messages.push({\n      role: 'assistant',\n      content,\n      toolCalls,\n    })\n  }\n\n  // Add tool result messages (only completed ones)\n  for (const toolResultPart of toolResultParts) {\n    if (\n      toolResultPart.state === 'complete' ||\n      toolResultPart.state === 'error'\n    ) {\n      messages.push({\n        role: 'tool',\n        content: toolResultPart.content,\n        toolCallId: toolResultPart.toolCallId,\n      })\n    }\n  }\n\n  return messages\n}\n\n/**\n * Convert a ModelMessage to UIMessage\n *\n * This conversion creates a parts-based structure:\n * - content field → TextPart\n * - toolCalls array → ToolCallPart[]\n * - role=\"tool\" messages should be converted separately and merged\n *\n * @param modelMessage - The ModelMessage to convert\n * @param id - Optional ID for the UIMessage (generated if not provided)\n * @returns A UIMessage with parts\n */\nexport function modelMessageToUIMessage(\n  modelMessage: ModelMessage,\n  id?: string,\n): UIMessage {\n  const parts: Array<MessagePart> = []\n\n  // Handle content (convert multimodal content to text for UI)\n  const textContent = getTextContent(modelMessage.content)\n  if (textContent) {\n    parts.push({\n      type: 'text',\n      content: textContent,\n    })\n  }\n\n  // Handle tool calls\n  if (modelMessage.toolCalls && modelMessage.toolCalls.length > 0) {\n    for (const toolCall of modelMessage.toolCalls) {\n      parts.push({\n        type: 'tool-call',\n        id: toolCall.id,\n        name: toolCall.function.name,\n        arguments: toolCall.function.arguments,\n        state: 'input-complete', // Model messages have complete arguments\n      })\n    }\n  }\n\n  // Handle tool results (when role is \"tool\")\n  if (modelMessage.role === 'tool' && modelMessage.toolCallId) {\n    parts.push({\n      type: 'tool-result',\n      toolCallId: modelMessage.toolCallId,\n      content: getTextContent(modelMessage.content),\n      state: 'complete',\n    })\n  }\n\n  return {\n    id: id || generateMessageId(),\n    role: modelMessage.role === 'tool' ? 'assistant' : modelMessage.role,\n    parts,\n  }\n}\n\n/**\n * Convert an array of ModelMessages to UIMessages\n *\n * This handles merging tool result messages with their corresponding assistant messages\n *\n * @param modelMessages - Array of ModelMessages to convert\n * @returns Array of UIMessages\n */\nexport function modelMessagesToUIMessages(\n  modelMessages: Array<ModelMessage>,\n): Array<UIMessage> {\n  const uiMessages: Array<UIMessage> = []\n  let currentAssistantMessage: UIMessage | null = null\n\n  for (const msg of modelMessages) {\n    if (msg.role === 'tool') {\n      // Tool result - merge into the last assistant message if possible\n      if (\n        currentAssistantMessage &&\n        currentAssistantMessage.role === 'assistant'\n      ) {\n        currentAssistantMessage.parts.push({\n          type: 'tool-result',\n          toolCallId: msg.toolCallId!,\n          content: getTextContent(msg.content),\n          state: 'complete',\n        })\n      } else {\n        // No assistant message to merge into, create a standalone one\n        const toolResultUIMessage = modelMessageToUIMessage(msg)\n        uiMessages.push(toolResultUIMessage)\n      }\n    } else {\n      // Regular message\n      const uiMessage = modelMessageToUIMessage(msg)\n      uiMessages.push(uiMessage)\n\n      // Track assistant messages for potential tool result merging\n      if (msg.role === 'assistant') {\n        currentAssistantMessage = uiMessage\n      } else {\n        currentAssistantMessage = null\n      }\n    }\n  }\n\n  return uiMessages\n}\n\n/**\n * Normalize a message (UIMessage or ModelMessage) to a UIMessage\n * Ensures the message has an ID and createdAt timestamp\n *\n * @param message - Either a UIMessage or ModelMessage\n * @param generateId - Function to generate a message ID if needed\n * @returns A UIMessage with guaranteed id and createdAt\n */\nexport function normalizeToUIMessage(\n  message: UIMessage | ModelMessage,\n  generateId: () => string,\n): UIMessage {\n  if ('parts' in message) {\n    // Already a UIMessage\n    return {\n      ...message,\n      id: message.id || generateId(),\n      createdAt: message.createdAt || new Date(),\n    }\n  } else {\n    // ModelMessage - convert to UIMessage\n    return {\n      ...modelMessageToUIMessage(message, generateId()),\n      createdAt: new Date(),\n    }\n  }\n}\n\n/**\n * Generate a unique message ID\n */\nexport function generateMessageId(): string {\n  return `msg-${Date.now()}-${Math.random().toString(36).substring(7)}`\n}\n"],"names":[],"mappings":"AAoBA,SAAS,eAAe,SAAqD;AAC3E,MAAI,YAAY,MAAM;AACpB,WAAO;AAAA,EACT;AACA,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO;AAAA,EACT;AAEA,SAAO,QACJ,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,SAAS,KAAK,OAAO,EAC1B,KAAK,EAAE;AACZ;AAKO,SAAS,+BACd,UACqB;AACrB,QAAM,gBAAqC,CAAA;AAC3C,aAAW,OAAO,UAAU;AAC1B,QAAI,WAAW,KAAK;AAElB,oBAAc,KAAK,GAAG,yBAAyB,GAAG,CAAC;AAAA,IACrD,OAAO;AAEL,oBAAc,KAAK,GAAG;AAAA,IACxB;AAAA,EACF;AACA,SAAO;AACT;AAaO,SAAS,yBACd,WACqB;AACrB,QAAM,WAAgC,CAAA;AAGtC,MAAI,UAAU,SAAS,UAAU;AAC/B,WAAO;AAAA,EACT;AAIA,QAAM,YAA6B,CAAA;AACnC,QAAM,gBAAqC,CAAA;AAC3C,QAAM,kBAAyC,CAAA;AAE/C,aAAW,QAAQ,UAAU,OAAO;AAClC,QAAI,KAAK,SAAS,QAAQ;AACxB,gBAAU,KAAK,IAAI;AAAA,IACrB,WAAW,KAAK,SAAS,aAAa;AACpC,oBAAc,KAAK,IAAI;AAAA,IACzB,WAAW,KAAK,SAAS,eAAe;AACtC,sBAAgB,KAAK,IAAI;AAAA,IAC3B;AAAA,EAEF;AAGA,QAAM,UAAU,UAAU,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;AAC5D,QAAM,YACJ,cAAc,SAAS,IACnB,cACG;AAAA,IACC,CAAC,MACC,EAAE,UAAU,oBACZ,EAAE,UAAU,wBACZ,EAAE,WAAW;AAAA;AAAA,EAAA,EAEhB,IAAI,CAAC,OAAO;AAAA,IACX,IAAI,EAAE;AAAA,IACN,MAAM;AAAA,IACN,UAAU;AAAA,MACR,MAAM,EAAE;AAAA,MACR,WAAW,EAAE;AAAA,IAAA;AAAA,EACf,EACA,IACJ;AAGN,MAAI,UAAU,SAAS,eAAe,WAAW,CAAC,WAAW;AAC3D,aAAS,KAAK;AAAA,MACZ,MAAM,UAAU;AAAA,MAChB;AAAA,MACA,GAAI,aAAa,UAAU,SAAS,KAAK,EAAE,UAAA;AAAA,IAAU,CACtD;AAAA,EACH,WAAW,UAAU,SAAS,GAAG;AAE/B,aAAS,KAAK;AAAA,MACZ,MAAM;AAAA,MACN;AAAA,MACA;AAAA,IAAA,CACD;AAAA,EACH;AAGA,aAAW,kBAAkB,iBAAiB;AAC5C,QACE,eAAe,UAAU,cACzB,eAAe,UAAU,SACzB;AACA,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS,eAAe;AAAA,QACxB,YAAY,eAAe;AAAA,MAAA,CAC5B;AAAA,IACH;AAAA,EACF;AAEA,SAAO;AACT;AAcO,SAAS,wBACd,cACA,IACW;AACX,QAAM,QAA4B,CAAA;AAGlC,QAAM,cAAc,eAAe,aAAa,OAAO;AACvD,MAAI,aAAa;AACf,UAAM,KAAK;AAAA,MACT,MAAM;AAAA,MACN,SAAS;AAAA,IAAA,CACV;AAAA,EACH;AAGA,MAAI,aAAa,aAAa,aAAa,UAAU,SAAS,GAAG;AAC/D,eAAW,YAAY,aAAa,WAAW;AAC7C,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,IAAI,SAAS;AAAA,QACb,MAAM,SAAS,SAAS;AAAA,QACxB,WAAW,SAAS,SAAS;AAAA,QAC7B,OAAO;AAAA;AAAA,MAAA,CACR;AAAA,IACH;AAAA,EACF;AAGA,MAAI,aAAa,SAAS,UAAU,aAAa,YAAY;AAC3D,UAAM,KAAK;AAAA,MACT,MAAM;AAAA,MACN,YAAY,aAAa;AAAA,MACzB,SAAS,eAAe,aAAa,OAAO;AAAA,MAC5C,OAAO;AAAA,IAAA,CACR;AAAA,EACH;AAEA,SAAO;AAAA,IACL,IAAI,MAAM,kBAAA;AAAA,IACV,MAAM,aAAa,SAAS,SAAS,cAAc,aAAa;AAAA,IAChE;AAAA,EAAA;AAEJ;AAUO,SAAS,0BACd,eACkB;AAClB,QAAM,aAA+B,CAAA;AACrC,MAAI,0BAA4C;AAEhD,aAAW,OAAO,eAAe;AAC/B,QAAI,IAAI,SAAS,QAAQ;AAEvB,UACE,2BACA,wBAAwB,SAAS,aACjC;AACA,gCAAwB,MAAM,KAAK;AAAA,UACjC,MAAM;AAAA,UACN,YAAY,IAAI;AAAA,UAChB,SAAS,eAAe,IAAI,OAAO;AAAA,UACnC,OAAO;AAAA,QAAA,CACR;AAAA,MACH,OAAO;AAEL,cAAM,sBAAsB,wBAAwB,GAAG;AACvD,mBAAW,KAAK,mBAAmB;AAAA,MACrC;AAAA,IACF,OAAO;AAEL,YAAM,YAAY,wBAAwB,GAAG;AAC7C,iBAAW,KAAK,SAAS;AAGzB,UAAI,IAAI,SAAS,aAAa;AAC5B,kCAA0B;AAAA,MAC5B,OAAO;AACL,kCAA0B;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAUO,SAAS,qBACd,SACA,YACW;AACX,MAAI,WAAW,SAAS;AAEtB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,IAAI,QAAQ,MAAM,WAAA;AAAA,MAClB,WAAW,QAAQ,aAAa,oBAAI,KAAA;AAAA,IAAK;AAAA,EAE7C,OAAO;AAEL,WAAO;AAAA,MACL,GAAG,wBAAwB,SAAS,YAAY;AAAA,MAChD,+BAAe,KAAA;AAAA,IAAK;AAAA,EAExB;AACF;AAKO,SAAS,oBAA4B;AAC1C,SAAO,OAAO,KAAK,IAAA,CAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,UAAU,CAAC,CAAC;AACrE;"}

package/dist/esm/stream/index.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Stream Processing Module
+ *
+ * Unified stream processing for both server and client.
+ */
+export { StreamProcessor, createReplayStream } from './processor.js';
+export type { StreamProcessorEvents, StreamProcessorHandlers, StreamProcessorOptions, } from './processor.js';
+export { ImmediateStrategy, PunctuationStrategy, BatchStrategy, WordBoundaryStrategy, CompositeStrategy, } from './strategies.js';
+export { PartialJSONParser, defaultJSONParser, parsePartialJSON, } from './json-parser.js';
+export type { JSONParser } from './json-parser.js';
+export type { ChunkStrategy, ChunkRecording, InternalToolCallState, ProcessorResult, ProcessorState, ToolCallState, ToolResultState, } from './types.js';

package/dist/esm/stream/json-parser.d.ts

@@ -0,0 +1,38 @@
+/**
+ * JSON Parser for partial/incomplete JSON strings
+ *
+ * Used during streaming to parse tool call arguments that may be incomplete.
+ */
+/**
+ * JSON Parser interface - allows for custom parser implementations
+ */
+export interface JSONParser {
+    /**
+     * Parse a JSON string (may be incomplete/partial)
+     * @param jsonString - The JSON string to parse
+     * @returns The parsed object, or undefined if parsing fails
+     */
+    parse: (jsonString: string) => any;
+}
+/**
+ * Partial JSON Parser implementation using the partial-json library
+ * This parser can handle incomplete JSON strings during streaming
+ */
+export declare class PartialJSONParser implements JSONParser {
+    /**
+     * Parse a potentially incomplete JSON string
+     * @param jsonString - The JSON string to parse (may be incomplete)
+     * @returns The parsed object, or undefined if parsing fails
+     */
+    parse(jsonString: string): any;
+}
+/**
+ * Default parser instance
+ */
+export declare const defaultJSONParser: PartialJSONParser;
+/**
+ * Parse partial JSON string (convenience function)
+ * @param jsonString - The JSON string to parse (may be incomplete)
+ * @returns The parsed object, or undefined if parsing fails
+ */
+export declare function parsePartialJSON(jsonString: string): any;

package/dist/esm/stream/json-parser.js

@@ -0,0 +1,28 @@
+import { parse } from "partial-json";
+class PartialJSONParser {
+  /**
+   * Parse a potentially incomplete JSON string
+   * @param jsonString - The JSON string to parse (may be incomplete)
+   * @returns The parsed object, or undefined if parsing fails
+   */
+  parse(jsonString) {
+    if (!jsonString || jsonString.trim() === "") {
+      return void 0;
+    }
+    try {
+      return parse(jsonString);
+    } catch {
+      return void 0;
+    }
+  }
+}
+const defaultJSONParser = new PartialJSONParser();
+function parsePartialJSON(jsonString) {
+  return defaultJSONParser.parse(jsonString);
+}
+export {
+  PartialJSONParser,
+  defaultJSONParser,
+  parsePartialJSON
+};
+//# sourceMappingURL=json-parser.js.map

package/dist/esm/stream/json-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-parser.js","sources":["../../../src/stream/json-parser.ts"],"sourcesContent":["/**\n * JSON Parser for partial/incomplete JSON strings\n *\n * Used during streaming to parse tool call arguments that may be incomplete.\n */\n\nimport { parse as parsePartialJSONLib } from 'partial-json'\n\n/**\n * JSON Parser interface - allows for custom parser implementations\n */\nexport interface JSONParser {\n  /**\n   * Parse a JSON string (may be incomplete/partial)\n   * @param jsonString - The JSON string to parse\n   * @returns The parsed object, or undefined if parsing fails\n   */\n  parse: (jsonString: string) => any\n}\n\n/**\n * Partial JSON Parser implementation using the partial-json library\n * This parser can handle incomplete JSON strings during streaming\n */\nexport class PartialJSONParser implements JSONParser {\n  /**\n   * Parse a potentially incomplete JSON string\n   * @param jsonString - The JSON string to parse (may be incomplete)\n   * @returns The parsed object, or undefined if parsing fails\n   */\n  parse(jsonString: string): any {\n    if (!jsonString || jsonString.trim() === '') {\n      return undefined\n    }\n\n    try {\n      return parsePartialJSONLib(jsonString)\n    } catch {\n      // If partial parsing fails, return undefined\n      // This is expected during early streaming when we have very little data\n      return undefined\n    }\n  }\n}\n\n/**\n * Default parser instance\n */\nexport const defaultJSONParser = new PartialJSONParser()\n\n/**\n * Parse partial JSON string (convenience function)\n * @param jsonString - The JSON string to parse (may be incomplete)\n * @returns The parsed object, or undefined if parsing fails\n */\nexport function parsePartialJSON(jsonString: string): any {\n  return defaultJSONParser.parse(jsonString)\n}\n"],"names":["parsePartialJSONLib"],"mappings":";AAwBO,MAAM,kBAAwC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMnD,MAAM,YAAyB;AAC7B,QAAI,CAAC,cAAc,WAAW,KAAA,MAAW,IAAI;AAC3C,aAAO;AAAA,IACT;AAEA,QAAI;AACF,aAAOA,MAAoB,UAAU;AAAA,IACvC,QAAQ;AAGN,aAAO;AAAA,IACT;AAAA,EACF;AACF;AAKO,MAAM,oBAAoB,IAAI,kBAAA;AAO9B,SAAS,iBAAiB,YAAyB;AACxD,SAAO,kBAAkB,MAAM,UAAU;AAC3C;"}

package/dist/esm/stream/message-updaters.d.ts

@@ -0,0 +1,44 @@
+import { UIMessage } from '../types.js';
+import { ToolCallState, ToolResultState } from './types.js';
+/**
+ * Update or add a text part to a message.
+ *
+ * If the last part is a text part, update it (continuing the same text segment).
+ * Otherwise, create a new text part (starting a new text segment after tool calls).
+ */
+export declare function updateTextPart(messages: Array<UIMessage>, messageId: string, content: string): Array<UIMessage>;
+/**
+ * Update or add a tool call part to a message.
+ */
+export declare function updateToolCallPart(messages: Array<UIMessage>, messageId: string, toolCall: {
+    id: string;
+    name: string;
+    arguments: string;
+    state: ToolCallState;
+}): Array<UIMessage>;
+/**
+ * Update or add a tool result part to a message.
+ */
+export declare function updateToolResultPart(messages: Array<UIMessage>, messageId: string, toolCallId: string, content: string, state: ToolResultState, error?: string): Array<UIMessage>;
+/**
+ * Update a tool call part with approval request metadata.
+ */
+export declare function updateToolCallApproval(messages: Array<UIMessage>, messageId: string, toolCallId: string, approvalId: string): Array<UIMessage>;
+/**
+ * Update a tool call part's state (e.g., to "input-complete").
+ */
+export declare function updateToolCallState(messages: Array<UIMessage>, messageId: string, toolCallId: string, state: ToolCallState): Array<UIMessage>;
+/**
+ * Update a tool call part with output.
+ * Searches all messages to find the tool call by ID.
+ */
+export declare function updateToolCallWithOutput(messages: Array<UIMessage>, toolCallId: string, output: any, state?: ToolCallState, errorText?: string): Array<UIMessage>;
+/**
+ * Update a tool call part with approval response.
+ * Searches all messages to find the tool call by approval ID.
+ */
+export declare function updateToolCallApprovalResponse(messages: Array<UIMessage>, approvalId: string, approved: boolean): Array<UIMessage>;
+/**
+ * Update or add a thinking part to a message.
+ */
+export declare function updateThinkingPart(messages: Array<UIMessage>, messageId: string, content: string): Array<UIMessage>;

package/dist/esm/stream/message-updaters.js

@@ -0,0 +1,141 @@
+function updateTextPart(messages, messageId, content) {
+  return messages.map((msg) => {
+    if (msg.id !== messageId) {
+      return msg;
+    }
+    const parts = [...msg.parts];
+    const lastPart = parts.length > 0 ? parts[parts.length - 1] : null;
+    if (lastPart && lastPart.type === "text") {
+      parts[parts.length - 1] = { type: "text", content };
+    } else {
+      parts.push({ type: "text", content });
+    }
+    return { ...msg, parts };
+  });
+}
+function updateToolCallPart(messages, messageId, toolCall) {
+  return messages.map((msg) => {
+    if (msg.id !== messageId) {
+      return msg;
+    }
+    const parts = [...msg.parts];
+    const existingPartIndex = parts.findIndex(
+      (p) => p.type === "tool-call" && p.id === toolCall.id
+    );
+    const toolCallPart = {
+      type: "tool-call",
+      id: toolCall.id,
+      name: toolCall.name,
+      arguments: toolCall.arguments,
+      state: toolCall.state
+    };
+    if (existingPartIndex >= 0) {
+      parts[existingPartIndex] = toolCallPart;
+    } else {
+      parts.push(toolCallPart);
+    }
+    return { ...msg, parts };
+  });
+}
+function updateToolResultPart(messages, messageId, toolCallId, content, state, error) {
+  return messages.map((msg) => {
+    if (msg.id !== messageId) {
+      return msg;
+    }
+    const parts = [...msg.parts];
+    const resultPartIndex = parts.findIndex(
+      (p) => p.type === "tool-result" && p.toolCallId === toolCallId
+    );
+    const toolResultPart = {
+      type: "tool-result",
+      toolCallId,
+      content,
+      state,
+      ...error && { error }
+    };
+    if (resultPartIndex >= 0) {
+      parts[resultPartIndex] = toolResultPart;
+    } else {
+      parts.push(toolResultPart);
+    }
+    return { ...msg, parts };
+  });
+}
+function updateToolCallApproval(messages, messageId, toolCallId, approvalId) {
+  return messages.map((msg) => {
+    if (msg.id !== messageId) {
+      return msg;
+    }
+    const parts = [...msg.parts];
+    const toolCallPart = parts.find(
+      (p) => p.type === "tool-call" && p.id === toolCallId
+    );
+    if (toolCallPart) {
+      toolCallPart.state = "approval-requested";
+      toolCallPart.approval = {
+        id: approvalId,
+        needsApproval: true
+      };
+    }
+    return { ...msg, parts };
+  });
+}
+function updateToolCallWithOutput(messages, toolCallId, output, state, errorText) {
+  return messages.map((msg) => {
+    const parts = [...msg.parts];
+    const toolCallPart = parts.find(
+      (p) => p.type === "tool-call" && p.id === toolCallId
+    );
+    if (toolCallPart) {
+      toolCallPart.output = errorText ? { error: errorText } : output;
+      if (state) {
+        toolCallPart.state = state;
+      } else {
+        toolCallPart.state = "input-complete";
+      }
+    }
+    return { ...msg, parts };
+  });
+}
+function updateToolCallApprovalResponse(messages, approvalId, approved) {
+  return messages.map((msg) => {
+    const parts = [...msg.parts];
+    const toolCallPart = parts.find(
+      (p) => p.type === "tool-call" && p.approval?.id === approvalId
+    );
+    if (toolCallPart && toolCallPart.approval) {
+      toolCallPart.approval.approved = approved;
+      toolCallPart.state = "approval-responded";
+    }
+    return { ...msg, parts };
+  });
+}
+function updateThinkingPart(messages, messageId, content) {
+  return messages.map((msg) => {
+    if (msg.id !== messageId) {
+      return msg;
+    }
+    const parts = [...msg.parts];
+    const thinkingPartIndex = parts.findIndex((p) => p.type === "thinking");
+    const thinkingPart = {
+      type: "thinking",
+      content
+    };
+    if (thinkingPartIndex >= 0) {
+      parts[thinkingPartIndex] = thinkingPart;
+    } else {
+      parts.push(thinkingPart);
+    }
+    return { ...msg, parts };
+  });
+}
+export {
+  updateTextPart,
+  updateThinkingPart,
+  updateToolCallApproval,
+  updateToolCallApprovalResponse,
+  updateToolCallPart,
+  updateToolCallWithOutput,
+  updateToolResultPart
+};
+//# sourceMappingURL=message-updaters.js.map

package/dist/esm/stream/message-updaters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-updaters.js","sources":["../../../src/stream/message-updaters.ts"],"sourcesContent":["/**\n * Message Updaters (Internal)\n *\n * Internal helper functions for updating UIMessage parts.\n * These are used by StreamProcessor to manage the message array.\n */\n\nimport type {\n  ThinkingPart,\n  ToolCallPart,\n  ToolResultPart,\n  UIMessage,\n} from '../types'\nimport type { ToolCallState, ToolResultState } from './types'\n\n/**\n * Update or add a text part to a message.\n *\n * If the last part is a text part, update it (continuing the same text segment).\n * Otherwise, create a new text part (starting a new text segment after tool calls).\n */\nexport function updateTextPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  content: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const lastPart = parts.length > 0 ? parts[parts.length - 1] : null\n\n    if (lastPart && lastPart.type === 'text') {\n      // Update the last text part (continuing same text segment)\n      parts[parts.length - 1] = { type: 'text', content }\n    } else {\n      // Create new text part (starting new text segment after tool calls/results)\n      parts.push({ type: 'text', content })\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a tool call part to a message.\n */\nexport function updateToolCallPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCall: {\n    id: string\n    name: string\n    arguments: string\n    state: ToolCallState\n  },\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    // Find by ID, not index!\n    const existingPartIndex = parts.findIndex(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCall.id,\n    )\n\n    const toolCallPart: ToolCallPart = {\n      type: 'tool-call',\n      id: toolCall.id,\n      name: toolCall.name,\n      arguments: toolCall.arguments,\n      state: toolCall.state,\n    }\n\n    if (existingPartIndex >= 0) {\n      // Update existing tool call\n      parts[existingPartIndex] = toolCallPart\n    } else {\n      // Add new tool call at the end (preserve natural streaming order)\n      parts.push(toolCallPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a tool result part to a message.\n */\nexport function updateToolResultPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  content: string,\n  state: ToolResultState,\n  error?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const resultPartIndex = parts.findIndex(\n      (p): p is ToolResultPart =>\n        p.type === 'tool-result' && p.toolCallId === toolCallId,\n    )\n\n    const toolResultPart: ToolResultPart = {\n      type: 'tool-result',\n      toolCallId,\n      content,\n      state,\n      ...(error && { error }),\n    }\n\n    if (resultPartIndex >= 0) {\n      parts[resultPartIndex] = toolResultPart\n    } else {\n      parts.push(toolResultPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with approval request metadata.\n */\nexport function updateToolCallApproval(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  approvalId: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      toolCallPart.state = 'approval-requested'\n      toolCallPart.approval = {\n        id: approvalId,\n        needsApproval: true,\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part's state (e.g., to \"input-complete\").\n */\nexport function updateToolCallState(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  state: ToolCallState,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      toolCallPart.state = state\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with output.\n * Searches all messages to find the tool call by ID.\n */\nexport function updateToolCallWithOutput(\n  messages: Array<UIMessage>,\n  toolCallId: string,\n  output: any,\n  state?: ToolCallState,\n  errorText?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      toolCallPart.output = errorText ? { error: errorText } : output\n      if (state) {\n        toolCallPart.state = state\n      } else {\n        toolCallPart.state = 'input-complete'\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with approval response.\n * Searches all messages to find the tool call by approval ID.\n */\nexport function updateToolCallApprovalResponse(\n  messages: Array<UIMessage>,\n  approvalId: string,\n  approved: boolean,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart =>\n        p.type === 'tool-call' && p.approval?.id === approvalId,\n    )\n\n    if (toolCallPart && toolCallPart.approval) {\n      toolCallPart.approval.approved = approved\n      toolCallPart.state = 'approval-responded'\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a thinking part to a message.\n */\nexport function updateThinkingPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  content: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const thinkingPartIndex = parts.findIndex((p) => p.type === 'thinking')\n\n    const thinkingPart: ThinkingPart = {\n      type: 'thinking',\n      content,\n    }\n\n    if (thinkingPartIndex >= 0) {\n      // Update existing thinking part\n      parts[thinkingPartIndex] = thinkingPart\n    } else {\n      // Add new thinking part at the end (preserve natural streaming order)\n      parts.push(thinkingPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n"],"names":[],"mappings":"AAqBO,SAAS,eACd,UACA,WACA,SACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,WAAW,MAAM,SAAS,IAAI,MAAM,MAAM,SAAS,CAAC,IAAI;AAE9D,QAAI,YAAY,SAAS,SAAS,QAAQ;AAExC,YAAM,MAAM,SAAS,CAAC,IAAI,EAAE,MAAM,QAAQ,QAAA;AAAA,IAC5C,OAAO;AAEL,YAAM,KAAK,EAAE,MAAM,QAAQ,SAAS;AAAA,IACtC;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,mBACd,UACA,WACA,UAMkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAE3B,UAAM,oBAAoB,MAAM;AAAA,MAC9B,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO,SAAS;AAAA,IAAA;AAGxE,UAAM,eAA6B;AAAA,MACjC,MAAM;AAAA,MACN,IAAI,SAAS;AAAA,MACb,MAAM,SAAS;AAAA,MACf,WAAW,SAAS;AAAA,MACpB,OAAO,SAAS;AAAA,IAAA;AAGlB,QAAI,qBAAqB,GAAG;AAE1B,YAAM,iBAAiB,IAAI;AAAA,IAC7B,OAAO;AAEL,YAAM,KAAK,YAAY;AAAA,IACzB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,qBACd,UACA,WACA,YACA,SACA,OACA,OACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,kBAAkB,MAAM;AAAA,MAC5B,CAAC,MACC,EAAE,SAAS,iBAAiB,EAAE,eAAe;AAAA,IAAA;AAGjD,UAAM,iBAAiC;AAAA,MACrC,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA;AAAA,MACA,GAAI,SAAS,EAAE,MAAA;AAAA,IAAM;AAGvB,QAAI,mBAAmB,GAAG;AACxB,YAAM,eAAe,IAAI;AAAA,IAC3B,OAAO;AACL,YAAM,KAAK,cAAc;AAAA,IAC3B;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,uBACd,UACA,WACA,YACA,YACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,IAAA;AAG/D,QAAI,cAAc;AAChB,mBAAa,QAAQ;AACrB,mBAAa,WAAW;AAAA,QACtB,IAAI;AAAA,QACJ,eAAe;AAAA,MAAA;AAAA,IAEnB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAiCO,SAAS,yBACd,UACA,YACA,QACA,OACA,WACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,IAAA;AAG/D,QAAI,cAAc;AAChB,mBAAa,SAAS,YAAY,EAAE,OAAO,cAAc;AACzD,UAAI,OAAO;AACT,qBAAa,QAAQ;AAAA,MACvB,OAAO;AACL,qBAAa,QAAQ;AAAA,MACvB;AAAA,IACF;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAMO,SAAS,+BACd,UACA,YACA,UACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MACC,EAAE,SAAS,eAAe,EAAE,UAAU,OAAO;AAAA,IAAA;AAGjD,QAAI,gBAAgB,aAAa,UAAU;AACzC,mBAAa,SAAS,WAAW;AACjC,mBAAa,QAAQ;AAAA,IACvB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,mBACd,UACA,WACA,SACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,oBAAoB,MAAM,UAAU,CAAC,MAAM,EAAE,SAAS,UAAU;AAEtE,UAAM,eAA6B;AAAA,MACjC,MAAM;AAAA,MACN;AAAA,IAAA;AAGF,QAAI,qBAAqB,GAAG;AAE1B,YAAM,iBAAiB,IAAI;AAAA,IAC7B,OAAO;AAEL,YAAM,KAAK,YAAY;AAAA,IACzB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;"}