@xinghunm/ai-chat 1.0.0 → 1.0.1

package/dist/index.d.mts

@@ -168,6 +168,31 @@ interface ResultSummary {
     headline: string;
     details: string[];
 }
+/**
+ * Merge behavior applied when a keyed custom block is patched repeatedly.
+ */
+type ChatCustomBlockMergePolicy = 'append' | 'replace' | 'ignore-duplicate';
+/**
+ * Extensible custom block rendered by a consumer-provided block renderer.
+ */
+interface ChatCustomBlock {
+    type: 'custom';
+    kind: string;
+    data: unknown;
+    /**
+     * Stable key used to identify the same logical block across streaming patches.
+     * When present, keyed custom blocks are merged by `blockKey` regardless of `kind`.
+     */
+    blockKey?: string;
+    /**
+     * Merge strategy used when another custom block with the same key arrives.
+     *
+     * - `append`: keep appending blocks even if the key matches.
+     * - `replace`: replace the existing keyed block with the latest one.
+     * - `ignore-duplicate`: keep the first keyed block and ignore later duplicates.
+     */
+    mergePolicy?: ChatCustomBlockMergePolicy;
+}
 /**
  * Structured assistant block variants rendered by the chat thread.
  */
@@ -190,11 +215,7 @@ type ChatMessageBlock = {
 } | {
     type: 'questionnaire';
     questionnaire: PlanQuestionnaire;
-} | {
-    type: 'custom';
-    kind: string;
-    data: unknown;
-};
+} | ChatCustomBlock;
 /**
  * Supported message body render orders for mixed text and structured blocks.
  */

package/dist/index.d.ts

@@ -168,6 +168,31 @@ interface ResultSummary {
     headline: string;
     details: string[];
 }
+/**
+ * Merge behavior applied when a keyed custom block is patched repeatedly.
+ */
+type ChatCustomBlockMergePolicy = 'append' | 'replace' | 'ignore-duplicate';
+/**
+ * Extensible custom block rendered by a consumer-provided block renderer.
+ */
+interface ChatCustomBlock {
+    type: 'custom';
+    kind: string;
+    data: unknown;
+    /**
+     * Stable key used to identify the same logical block across streaming patches.
+     * When present, keyed custom blocks are merged by `blockKey` regardless of `kind`.
+     */
+    blockKey?: string;
+    /**
+     * Merge strategy used when another custom block with the same key arrives.
+     *
+     * - `append`: keep appending blocks even if the key matches.
+     * - `replace`: replace the existing keyed block with the latest one.
+     * - `ignore-duplicate`: keep the first keyed block and ignore later duplicates.
+     */
+    mergePolicy?: ChatCustomBlockMergePolicy;
+}
 /**
  * Structured assistant block variants rendered by the chat thread.
  */
@@ -190,11 +215,7 @@ type ChatMessageBlock = {
 } | {
     type: 'questionnaire';
     questionnaire: PlanQuestionnaire;
-} | {
-    type: 'custom';
-    kind: string;
-    data: unknown;
-};
+} | ChatCustomBlock;
 /**
  * Supported message body render orders for mixed text and structured blocks.
  */

package/dist/index.js

@@ -108,6 +108,22 @@ var resolveSessionTitleFromMessage = (message) => {
 var mergeStreamingBlocks = (existingBlocks, incomingBlocks) => {
   const nextBlocks = [...existingBlocks ?? []];
   incomingBlocks.forEach((incomingBlock) => {
+    if (incomingBlock.type === "custom" && incomingBlock.blockKey) {
+      const mergePolicy = incomingBlock.mergePolicy ?? "append";
+      if (mergePolicy !== "append") {
+        const existingIndex2 = nextBlocks.findIndex(
+          (block) => block.type === "custom" && block.blockKey === incomingBlock.blockKey
+        );
+        if (existingIndex2 !== -1) {
+          if (mergePolicy === "replace") {
+            nextBlocks[existingIndex2] = incomingBlock;
+          }
+          return;
+        }
+      }
+      nextBlocks.push(incomingBlock);
+      return;
+    }
     if (incomingBlock.type !== "questionnaire") {
       nextBlocks.push(incomingBlock);
       return;
@@ -1010,7 +1026,7 @@ var getTimelineBlockKey = (block, index) => {
     case "questionnaire":
       return `${index}:questionnaire:${block.questionnaire.questionnaireId}`;
     case "custom":
-      return `${index}:custom:${block.kind}:${stringifyTimelineKeyPart(block.data)}`;
+      return block.blockKey ? `custom:${block.blockKey}` : `${index}:custom:${block.kind}:${stringifyTimelineKeyPart(block.data)}`;
     default:
       return null;
   }

package/dist/index.mjs

@@ -61,6 +61,22 @@ var resolveSessionTitleFromMessage = (message) => {
 var mergeStreamingBlocks = (existingBlocks, incomingBlocks) => {
   const nextBlocks = [...existingBlocks ?? []];
   incomingBlocks.forEach((incomingBlock) => {
+    if (incomingBlock.type === "custom" && incomingBlock.blockKey) {
+      const mergePolicy = incomingBlock.mergePolicy ?? "append";
+      if (mergePolicy !== "append") {
+        const existingIndex2 = nextBlocks.findIndex(
+          (block) => block.type === "custom" && block.blockKey === incomingBlock.blockKey
+        );
+        if (existingIndex2 !== -1) {
+          if (mergePolicy === "replace") {
+            nextBlocks[existingIndex2] = incomingBlock;
+          }
+          return;
+        }
+      }
+      nextBlocks.push(incomingBlock);
+      return;
+    }
     if (incomingBlock.type !== "questionnaire") {
       nextBlocks.push(incomingBlock);
       return;
@@ -963,7 +979,7 @@ var getTimelineBlockKey = (block, index) => {
     case "questionnaire":
       return `${index}:questionnaire:${block.questionnaire.questionnaireId}`;
     case "custom":
-      return `${index}:custom:${block.kind}:${stringifyTimelineKeyPart(block.data)}`;
+      return block.blockKey ? `custom:${block.blockKey}` : `${index}:custom:${block.kind}:${stringifyTimelineKeyPart(block.data)}`;
     default:
       return null;
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xinghunm/ai-chat",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "AI chat React component library",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",