@assistant-ui/core 0.2.9 → 0.2.11

package/dist/react/primitive-hooks/useMessageBranching.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui, useAuiState } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useMessageBranching.ts
 const useMessageBranching = () => {

package/dist/react/primitive-hooks/useMessageBranching.js.map

@@ -1 +1 @@
-{"version":3,"file":"useMessageBranching.js","names":[],"sources":["../../../src/react/primitive-hooks/useMessageBranching.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport const useMessageBranching = () => {\n  const aui = useAui();\n  const branchNumber = useAuiState((s) => s.message.branchNumber);\n  const branchCount = useAuiState((s) => s.message.branchCount);\n\n  const goToPrev = useCallback(() => {\n    aui.message().switchToBranch({ position: \"previous\" });\n  }, [aui]);\n\n  const goToNext = useCallback(() => {\n    aui.message().switchToBranch({ position: \"next\" });\n  }, [aui]);\n\n  return { branchNumber, branchCount, goToPrev, goToNext };\n};\n"],"mappings":";;;AAGA,MAAa,4BAA4B;CACvC,MAAM,MAAM,OAAO;CAYnB,OAAO;EAAE,cAXY,aAAa,MAAM,EAAE,QAAQ,YAW9B;EAAG,aAVH,aAAa,MAAM,EAAE,QAAQ,WAUhB;EAAG,UARnB,kBAAkB;GACjC,IAAI,QAAQ,EAAE,eAAe,EAAE,UAAU,WAAW,CAAC;EACvD,GAAG,CAAC,GAAG,CAMoC;EAAG,UAJ7B,kBAAkB;GACjC,IAAI,QAAQ,EAAE,eAAe,EAAE,UAAU,OAAO,CAAC;EACnD,GAAG,CAAC,GAAG,CAE8C;CAAE;AACzD"}
+{"version":3,"file":"useMessageBranching.js","names":[],"sources":["../../../src/react/primitive-hooks/useMessageBranching.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport const useMessageBranching = () => {\n  const aui = useAui();\n  const branchNumber = useAuiState((s) => s.message.branchNumber);\n  const branchCount = useAuiState((s) => s.message.branchCount);\n\n  const goToPrev = useCallback(() => {\n    aui.message().switchToBranch({ position: \"previous\" });\n  }, [aui]);\n\n  const goToNext = useCallback(() => {\n    aui.message().switchToBranch({ position: \"next\" });\n  }, [aui]);\n\n  return { branchNumber, branchCount, goToPrev, goToNext };\n};\n"],"mappings":";;;AAGA,MAAa,4BAA4B;CACvC,MAAM,MAAM,OAAO;CAYnB,OAAO;EAAE,cAXY,aAAa,MAAM,EAAE,QAAQ,YAW9B;EAAG,aAVH,aAAa,MAAM,EAAE,QAAQ,WAUhB;EAAG,UARnB,kBAAkB;GACjC,IAAI,QAAQ,CAAC,CAAC,eAAe,EAAE,UAAU,WAAW,CAAC;EACvD,GAAG,CAAC,GAAG,CAMoC;EAAG,UAJ7B,kBAAkB;GACjC,IAAI,QAAQ,CAAC,CAAC,eAAe,EAAE,UAAU,OAAO,CAAC;EACnD,GAAG,CAAC,GAAG,CAE8C;CAAE;AACzD"}

package/dist/react/primitive-hooks/useMessageReload.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui, useAuiState } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useMessageReload.ts
 const useMessageReload = () => {

package/dist/react/primitive-hooks/useMessageReload.js.map

@@ -1 +1 @@
-{"version":3,"file":"useMessageReload.js","names":[],"sources":["../../../src/react/primitive-hooks/useMessageReload.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport const useMessageReload = () => {\n  const aui = useAui();\n  const canReload = useAuiState((s) => s.message.role === \"assistant\");\n\n  const reload = useCallback(() => {\n    aui.message().reload();\n  }, [aui]);\n\n  return { reload, canReload };\n};\n"],"mappings":";;;AAGA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,MAAM,YAAY,aAAa,MAAM,EAAE,QAAQ,SAAS,WAAW;CAMnE,OAAO;EAAE,QAJM,kBAAkB;GAC/B,IAAI,QAAQ,EAAE,OAAO;EACvB,GAAG,CAAC,GAAG,CAEO;EAAG;CAAU;AAC7B"}
+{"version":3,"file":"useMessageReload.js","names":[],"sources":["../../../src/react/primitive-hooks/useMessageReload.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport const useMessageReload = () => {\n  const aui = useAui();\n  const canReload = useAuiState((s) => s.message.role === \"assistant\");\n\n  const reload = useCallback(() => {\n    aui.message().reload();\n  }, [aui]);\n\n  return { reload, canReload };\n};\n"],"mappings":";;;AAGA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,MAAM,YAAY,aAAa,MAAM,EAAE,QAAQ,SAAS,WAAW;CAMnE,OAAO;EAAE,QAJM,kBAAkB;GAC/B,IAAI,QAAQ,CAAC,CAAC,OAAO;EACvB,GAAG,CAAC,GAAG,CAEO;EAAG;CAAU;AAC7B"}

package/dist/react/primitive-hooks/useSuggestionTrigger.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui, useAuiState } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useSuggestionTrigger.ts
 const useSuggestionTrigger = ({ prompt, send, clearComposer = true }) => {

package/dist/react/primitive-hooks/useSuggestionTrigger.js.map

@@ -1 +1 @@
-{"version":3,"file":"useSuggestionTrigger.js","names":[],"sources":["../../../src/react/primitive-hooks/useSuggestionTrigger.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport type UseSuggestionTriggerOptions = {\n  prompt: string;\n  send?: boolean | undefined;\n  clearComposer?: boolean | undefined;\n};\n\nexport const useSuggestionTrigger = ({\n  prompt,\n  send,\n  clearComposer = true,\n}: UseSuggestionTriggerOptions) => {\n  const aui = useAui();\n  const disabled = useAuiState((s) => s.thread.isDisabled);\n  const resolvedSend = send ?? false;\n\n  const trigger = useCallback(() => {\n    const isRunning = aui.thread().getState().isRunning;\n\n    if (resolvedSend && !isRunning) {\n      aui.thread().append({\n        content: [{ type: \"text\", text: prompt }],\n        runConfig: aui.composer().getState().runConfig,\n      });\n      if (clearComposer) {\n        aui.composer().setText(\"\");\n      }\n    } else {\n      if (clearComposer) {\n        aui.composer().setText(prompt);\n      } else {\n        const currentText = aui.composer().getState().text;\n        aui\n          .composer()\n          .setText(currentText.trim() ? `${currentText} ${prompt}` : prompt);\n      }\n    }\n  }, [aui, resolvedSend, clearComposer, prompt]);\n\n  return { trigger, disabled };\n};\n"],"mappings":";;;AASA,MAAa,wBAAwB,EACnC,QACA,MACA,gBAAgB,WACiB;CACjC,MAAM,MAAM,OAAO;CACnB,MAAM,WAAW,aAAa,MAAM,EAAE,OAAO,UAAU;CACvD,MAAM,eAAe,QAAQ;CAyB7B,OAAO;EAAE,SAvBO,kBAAkB;GAChC,MAAM,YAAY,IAAI,OAAO,EAAE,SAAS,EAAE;GAE1C,IAAI,gBAAgB,CAAC,WAAW;IAC9B,IAAI,OAAO,EAAE,OAAO;KAClB,SAAS,CAAC;MAAE,MAAM;MAAQ,MAAM;KAAO,CAAC;KACxC,WAAW,IAAI,SAAS,EAAE,SAAS,EAAE;IACvC,CAAC;IACD,IAAI,eACF,IAAI,SAAS,EAAE,QAAQ,EAAE;GAE7B,OACE,IAAI,eACF,IAAI,SAAS,EAAE,QAAQ,MAAM;QACxB;IACL,MAAM,cAAc,IAAI,SAAS,EAAE,SAAS,EAAE;IAC9C,IACG,SAAS,EACT,QAAQ,YAAY,KAAK,IAAI,GAAG,YAAY,GAAG,WAAW,MAAM;GACrE;EAEJ,GAAG;GAAC;GAAK;GAAc;GAAe;EAAM,CAE7B;EAAG;CAAS;AAC7B"}
+{"version":3,"file":"useSuggestionTrigger.js","names":[],"sources":["../../../src/react/primitive-hooks/useSuggestionTrigger.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport type UseSuggestionTriggerOptions = {\n  prompt: string;\n  send?: boolean | undefined;\n  clearComposer?: boolean | undefined;\n};\n\nexport const useSuggestionTrigger = ({\n  prompt,\n  send,\n  clearComposer = true,\n}: UseSuggestionTriggerOptions) => {\n  const aui = useAui();\n  const disabled = useAuiState((s) => s.thread.isDisabled);\n  const resolvedSend = send ?? false;\n\n  const trigger = useCallback(() => {\n    const isRunning = aui.thread().getState().isRunning;\n\n    if (resolvedSend && !isRunning) {\n      aui.thread().append({\n        content: [{ type: \"text\", text: prompt }],\n        runConfig: aui.composer().getState().runConfig,\n      });\n      if (clearComposer) {\n        aui.composer().setText(\"\");\n      }\n    } else {\n      if (clearComposer) {\n        aui.composer().setText(prompt);\n      } else {\n        const currentText = aui.composer().getState().text;\n        aui\n          .composer()\n          .setText(currentText.trim() ? `${currentText} ${prompt}` : prompt);\n      }\n    }\n  }, [aui, resolvedSend, clearComposer, prompt]);\n\n  return { trigger, disabled };\n};\n"],"mappings":";;;AASA,MAAa,wBAAwB,EACnC,QACA,MACA,gBAAgB,WACiB;CACjC,MAAM,MAAM,OAAO;CACnB,MAAM,WAAW,aAAa,MAAM,EAAE,OAAO,UAAU;CACvD,MAAM,eAAe,QAAQ;CAyB7B,OAAO;EAAE,SAvBO,kBAAkB;GAChC,MAAM,YAAY,IAAI,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC;GAE1C,IAAI,gBAAgB,CAAC,WAAW;IAC9B,IAAI,OAAO,CAAC,CAAC,OAAO;KAClB,SAAS,CAAC;MAAE,MAAM;MAAQ,MAAM;KAAO,CAAC;KACxC,WAAW,IAAI,SAAS,CAAC,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IACD,IAAI,eACF,IAAI,SAAS,CAAC,CAAC,QAAQ,EAAE;GAE7B,OACE,IAAI,eACF,IAAI,SAAS,CAAC,CAAC,QAAQ,MAAM;QACxB;IACL,MAAM,cAAc,IAAI,SAAS,CAAC,CAAC,SAAS,CAAC,CAAC;IAC9C,IACG,SAAS,CAAC,CACV,QAAQ,YAAY,KAAK,IAAI,GAAG,YAAY,GAAG,WAAW,MAAM;GACrE;EAEJ,GAAG;GAAC;GAAK;GAAc;GAAe;EAAM,CAE7B;EAAG;CAAS;AAC7B"}

package/dist/react/primitive-hooks/useThreadListItemArchive.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useThreadListItemArchive.ts
 const useThreadListItemArchive = () => {

package/dist/react/primitive-hooks/useThreadListItemArchive.js.map

@@ -1 +1 @@
-{"version":3,"file":"useThreadListItemArchive.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemArchive.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemArchive = () => {\n  const aui = useAui();\n\n  const archive = useCallback(() => {\n    aui.threadListItem().archive();\n  }, [aui]);\n\n  return { archive };\n};\n"],"mappings":";;;AAGA,MAAa,iCAAiC;CAC5C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,SAJO,kBAAkB;EAChC,IAAI,eAAe,EAAE,QAAQ;CAC/B,GAAG,CAAC,GAAG,CAEQ,EAAE;AACnB"}
+{"version":3,"file":"useThreadListItemArchive.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemArchive.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemArchive = () => {\n  const aui = useAui();\n\n  const archive = useCallback(() => {\n    aui.threadListItem().archive();\n  }, [aui]);\n\n  return { archive };\n};\n"],"mappings":";;;AAGA,MAAa,iCAAiC;CAC5C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,SAJO,kBAAkB;EAChC,IAAI,eAAe,CAAC,CAAC,QAAQ;CAC/B,GAAG,CAAC,GAAG,CAEQ,EAAE;AACnB"}

package/dist/react/primitive-hooks/useThreadListItemDelete.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useThreadListItemDelete.ts
 const useThreadListItemDelete = () => {

package/dist/react/primitive-hooks/useThreadListItemDelete.js.map

@@ -1 +1 @@
-{"version":3,"file":"useThreadListItemDelete.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemDelete.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemDelete = () => {\n  const aui = useAui();\n\n  const deleteThread = useCallback(() => {\n    aui.threadListItem().delete();\n  }, [aui]);\n\n  return { delete: deleteThread };\n};\n"],"mappings":";;;AAGA,MAAa,gCAAgC;CAC3C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,QAJY,kBAAkB;EACrC,IAAI,eAAe,EAAE,OAAO;CAC9B,GAAG,CAAC,GAAG,CAEqB,EAAE;AAChC"}
+{"version":3,"file":"useThreadListItemDelete.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemDelete.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemDelete = () => {\n  const aui = useAui();\n\n  const deleteThread = useCallback(() => {\n    aui.threadListItem().delete();\n  }, [aui]);\n\n  return { delete: deleteThread };\n};\n"],"mappings":";;;AAGA,MAAa,gCAAgC;CAC3C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,QAJY,kBAAkB;EACrC,IAAI,eAAe,CAAC,CAAC,OAAO;CAC9B,GAAG,CAAC,GAAG,CAEqB,EAAE;AAChC"}

package/dist/react/primitive-hooks/useThreadListItemTrigger.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useThreadListItemTrigger.ts
 const useThreadListItemTrigger = () => {

package/dist/react/primitive-hooks/useThreadListItemTrigger.js.map

@@ -1 +1 @@
-{"version":3,"file":"useThreadListItemTrigger.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemTrigger.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemTrigger = () => {\n  const aui = useAui();\n\n  const switchTo = useCallback(() => {\n    aui.threadListItem().switchTo();\n  }, [aui]);\n\n  return { switchTo };\n};\n"],"mappings":";;;AAGA,MAAa,iCAAiC;CAC5C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,UAJQ,kBAAkB;EACjC,IAAI,eAAe,EAAE,SAAS;CAChC,GAAG,CAAC,GAAG,CAES,EAAE;AACpB"}
+{"version":3,"file":"useThreadListItemTrigger.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemTrigger.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemTrigger = () => {\n  const aui = useAui();\n\n  const switchTo = useCallback(() => {\n    aui.threadListItem().switchTo();\n  }, [aui]);\n\n  return { switchTo };\n};\n"],"mappings":";;;AAGA,MAAa,iCAAiC;CAC5C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,UAJQ,kBAAkB;EACjC,IAAI,eAAe,CAAC,CAAC,SAAS;CAChC,GAAG,CAAC,GAAG,CAES,EAAE;AACpB"}

package/dist/react/primitive-hooks/useThreadListItemUnarchive.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useThreadListItemUnarchive.ts
 const useThreadListItemUnarchive = () => {

package/dist/react/primitive-hooks/useThreadListItemUnarchive.js.map

@@ -1 +1 @@
-{"version":3,"file":"useThreadListItemUnarchive.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemUnarchive.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemUnarchive = () => {\n  const aui = useAui();\n\n  const unarchive = useCallback(() => {\n    aui.threadListItem().unarchive();\n  }, [aui]);\n\n  return { unarchive };\n};\n"],"mappings":";;;AAGA,MAAa,mCAAmC;CAC9C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,WAJS,kBAAkB;EAClC,IAAI,eAAe,EAAE,UAAU;CACjC,GAAG,CAAC,GAAG,CAEU,EAAE;AACrB"}
+{"version":3,"file":"useThreadListItemUnarchive.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListItemUnarchive.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListItemUnarchive = () => {\n  const aui = useAui();\n\n  const unarchive = useCallback(() => {\n    aui.threadListItem().unarchive();\n  }, [aui]);\n\n  return { unarchive };\n};\n"],"mappings":";;;AAGA,MAAa,mCAAmC;CAC9C,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,WAJS,kBAAkB;EAClC,IAAI,eAAe,CAAC,CAAC,UAAU;CACjC,GAAG,CAAC,GAAG,CAEU,EAAE;AACrB"}

package/dist/react/primitive-hooks/useThreadListLoadMore.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui, useAuiState } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useThreadListLoadMore.ts
 const useThreadListLoadMore = () => {

package/dist/react/primitive-hooks/useThreadListLoadMore.js.map

@@ -1 +1 @@
-{"version":3,"file":"useThreadListLoadMore.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListLoadMore.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport const useThreadListLoadMore = () => {\n  const aui = useAui();\n  const disabled = useAuiState(\n    (s) => !s.threads.hasMore || s.threads.isLoading || s.threads.isLoadingMore,\n  );\n\n  const loadMore = useCallback(() => {\n    aui.threads().loadMore();\n  }, [aui]);\n\n  return { loadMore, disabled };\n};\n"],"mappings":";;;AAGA,MAAa,8BAA8B;CACzC,MAAM,MAAM,OAAO;CACnB,MAAM,WAAW,aACd,MAAM,CAAC,EAAE,QAAQ,WAAW,EAAE,QAAQ,aAAa,EAAE,QAAQ,aAChE;CAMA,OAAO;EAAE,UAJQ,kBAAkB;GACjC,IAAI,QAAQ,EAAE,SAAS;EACzB,GAAG,CAAC,GAAG,CAES;EAAG;CAAS;AAC9B"}
+{"version":3,"file":"useThreadListLoadMore.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListLoadMore.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\n\nexport const useThreadListLoadMore = () => {\n  const aui = useAui();\n  const disabled = useAuiState(\n    (s) => !s.threads.hasMore || s.threads.isLoading || s.threads.isLoadingMore,\n  );\n\n  const loadMore = useCallback(() => {\n    aui.threads().loadMore();\n  }, [aui]);\n\n  return { loadMore, disabled };\n};\n"],"mappings":";;;AAGA,MAAa,8BAA8B;CACzC,MAAM,MAAM,OAAO;CACnB,MAAM,WAAW,aACd,MAAM,CAAC,EAAE,QAAQ,WAAW,EAAE,QAAQ,aAAa,EAAE,QAAQ,aAChE;CAMA,OAAO;EAAE,UAJQ,kBAAkB;GACjC,IAAI,QAAQ,CAAC,CAAC,SAAS;EACzB,GAAG,CAAC,GAAG,CAES;EAAG;CAAS;AAC9B"}

package/dist/react/primitive-hooks/useThreadListNew.js

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useThreadListNew.ts
 const useThreadListNew = () => {

package/dist/react/primitive-hooks/useThreadListNew.js.map

@@ -1 +1 @@
-{"version":3,"file":"useThreadListNew.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListNew.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListNew = () => {\n  const aui = useAui();\n\n  const switchToNewThread = useCallback(() => {\n    aui.threads().switchToNewThread();\n  }, [aui]);\n\n  return { switchToNewThread };\n};\n"],"mappings":";;;AAGA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,mBAJiB,kBAAkB;EAC1C,IAAI,QAAQ,EAAE,kBAAkB;CAClC,GAAG,CAAC,GAAG,CAEkB,EAAE;AAC7B"}
+{"version":3,"file":"useThreadListNew.js","names":[],"sources":["../../../src/react/primitive-hooks/useThreadListNew.ts"],"sourcesContent":["import { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nexport const useThreadListNew = () => {\n  const aui = useAui();\n\n  const switchToNewThread = useCallback(() => {\n    aui.threads().switchToNewThread();\n  }, [aui]);\n\n  return { switchToNewThread };\n};\n"],"mappings":";;;AAGA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CAMnB,OAAO,EAAE,mBAJiB,kBAAkB;EAC1C,IAAI,QAAQ,CAAC,CAAC,kBAAkB;CAClC,GAAG,CAAC,GAAG,CAEkB,EAAE;AAC7B"}

package/dist/react/primitive-hooks/useVoice.js

@@ -1,4 +1,4 @@
-import { useCallback, useSyncExternalStore } from "react";
+import { useCallback, useSyncExternalStore } from "@assistant-ui/tap/react-shim";
 import { useAui, useAuiState } from "@assistant-ui/store";
 //#region src/react/primitive-hooks/useVoice.ts
 const useVoiceState = () => {

package/dist/react/primitive-hooks/useVoice.js.map

@@ -1 +1 @@
-{"version":3,"file":"useVoice.js","names":[],"sources":["../../../src/react/primitive-hooks/useVoice.ts"],"sourcesContent":["import { useCallback, useSyncExternalStore } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { VoiceSessionState } from \"../../runtime/interfaces/thread-runtime-core\";\n\nexport const useVoiceState = (): VoiceSessionState | undefined => {\n  return useAuiState((s) => s.thread.voice);\n};\n\nconst getServerVolume = () => 0;\n\nexport const useVoiceVolume = (): number => {\n  const aui = useAui();\n  const thread = aui.thread();\n  return useSyncExternalStore(\n    thread.subscribeVoiceVolume,\n    thread.getVoiceVolume,\n    getServerVolume,\n  );\n};\n\nexport const useVoiceControls = () => {\n  const aui = useAui();\n\n  const connect = useCallback(() => {\n    aui.thread().connectVoice();\n  }, [aui]);\n\n  const disconnect = useCallback(() => {\n    aui.thread().disconnectVoice();\n  }, [aui]);\n\n  const mute = useCallback(() => {\n    aui.thread().muteVoice();\n  }, [aui]);\n\n  const unmute = useCallback(() => {\n    aui.thread().unmuteVoice();\n  }, [aui]);\n\n  return { connect, disconnect, mute, unmute };\n};\n"],"mappings":";;;AAIA,MAAa,sBAAqD;CAChE,OAAO,aAAa,MAAM,EAAE,OAAO,KAAK;AAC1C;AAEA,MAAM,wBAAwB;AAE9B,MAAa,uBAA+B;CAE1C,MAAM,SADM,OACK,EAAE,OAAO;CAC1B,OAAO,qBACL,OAAO,sBACP,OAAO,gBACP,eACF;AACF;AAEA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CAkBnB,OAAO;EAAE,SAhBO,kBAAkB;GAChC,IAAI,OAAO,EAAE,aAAa;EAC5B,GAAG,CAAC,GAAG,CAcQ;EAAG,YAZC,kBAAkB;GACnC,IAAI,OAAO,EAAE,gBAAgB;EAC/B,GAAG,CAAC,GAAG,CAUoB;EAAG,MARjB,kBAAkB;GAC7B,IAAI,OAAO,EAAE,UAAU;EACzB,GAAG,CAAC,GAAG,CAM0B;EAAG,QAJrB,kBAAkB;GAC/B,IAAI,OAAO,EAAE,YAAY;EAC3B,GAAG,CAAC,GAAG,CAEkC;CAAE;AAC7C"}
+{"version":3,"file":"useVoice.js","names":[],"sources":["../../../src/react/primitive-hooks/useVoice.ts"],"sourcesContent":["import { useCallback, useSyncExternalStore } from \"react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { VoiceSessionState } from \"../../runtime/interfaces/thread-runtime-core\";\n\nexport const useVoiceState = (): VoiceSessionState | undefined => {\n  return useAuiState((s) => s.thread.voice);\n};\n\nconst getServerVolume = () => 0;\n\nexport const useVoiceVolume = (): number => {\n  const aui = useAui();\n  const thread = aui.thread();\n  return useSyncExternalStore(\n    thread.subscribeVoiceVolume,\n    thread.getVoiceVolume,\n    getServerVolume,\n  );\n};\n\nexport const useVoiceControls = () => {\n  const aui = useAui();\n\n  const connect = useCallback(() => {\n    aui.thread().connectVoice();\n  }, [aui]);\n\n  const disconnect = useCallback(() => {\n    aui.thread().disconnectVoice();\n  }, [aui]);\n\n  const mute = useCallback(() => {\n    aui.thread().muteVoice();\n  }, [aui]);\n\n  const unmute = useCallback(() => {\n    aui.thread().unmuteVoice();\n  }, [aui]);\n\n  return { connect, disconnect, mute, unmute };\n};\n"],"mappings":";;;AAIA,MAAa,sBAAqD;CAChE,OAAO,aAAa,MAAM,EAAE,OAAO,KAAK;AAC1C;AAEA,MAAM,wBAAwB;AAE9B,MAAa,uBAA+B;CAE1C,MAAM,SADM,OACK,CAAC,CAAC,OAAO;CAC1B,OAAO,qBACL,OAAO,sBACP,OAAO,gBACP,eACF;AACF;AAEA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CAkBnB,OAAO;EAAE,SAhBO,kBAAkB;GAChC,IAAI,OAAO,CAAC,CAAC,aAAa;EAC5B,GAAG,CAAC,GAAG,CAcQ;EAAG,YAZC,kBAAkB;GACnC,IAAI,OAAO,CAAC,CAAC,gBAAgB;EAC/B,GAAG,CAAC,GAAG,CAUoB;EAAG,MARjB,kBAAkB;GAC7B,IAAI,OAAO,CAAC,CAAC,UAAU;EACzB,GAAG,CAAC,GAAG,CAM0B;EAAG,QAJrB,kBAAkB;GAC/B,IAAI,OAAO,CAAC,CAAC,YAAY;EAC3B,GAAG,CAAC,GAAG,CAEkC;CAAE;AAC7C"}

package/dist/react/primitives/chainOfThought/ChainOfThoughtParts.d.ts

@@ -1,5 +1,5 @@
-import { ReasoningMessagePartComponent, ToolCallMessagePartComponent } from "../../types/MessagePartComponentTypes.js";
 import { PartState } from "../../../store/scopes/part.js";
+import { ReasoningMessagePartComponent, ToolCallMessagePartComponent } from "../../types/MessagePartComponentTypes.js";
 import { ComponentType, FC, PropsWithChildren, ReactNode } from "react";
 
 //#region src/react/primitives/chainOfThought/ChainOfThoughtParts.d.ts

package/dist/react/primitives/chainOfThought/ChainOfThoughtParts.js

@@ -1,6 +1,6 @@
 import { ChainOfThoughtPartByIndexProvider } from "../../providers/ChainOfThoughtPartByIndexProvider.js";
 import { MessagePartComponent } from "../message/MessageParts.js";
-import { useMemo } from "react";
+import { useMemo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import { jsx } from "react/jsx-runtime";
 //#region src/react/primitives/chainOfThought/ChainOfThoughtParts.tsx

package/dist/react/primitives/chainOfThought/ChainOfThoughtParts.js.map

@@ -1 +1 @@
-{"version":3,"file":"ChainOfThoughtParts.js","names":[],"sources":["../../../../src/react/primitives/chainOfThought/ChainOfThoughtParts.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type PropsWithChildren,\n  type ReactNode,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport { ChainOfThoughtPartByIndexProvider } from \"../../providers/ChainOfThoughtPartByIndexProvider\";\nimport { MessagePartComponent } from \"../message/MessageParts\";\nimport type {\n  ReasoningMessagePartComponent,\n  ToolCallMessagePartComponent,\n} from \"../../types/MessagePartComponentTypes\";\n\ntype ChainOfThoughtPartsComponentConfig = {\n  /** Component for rendering reasoning parts */\n  Reasoning?: ReasoningMessagePartComponent | undefined;\n  /** Fallback component for tool-call parts */\n  tools?: {\n    Fallback?: ToolCallMessagePartComponent | undefined;\n  };\n  /** Layout component to wrap the rendered parts when expanded */\n  Layout?: ComponentType<PropsWithChildren> | undefined;\n};\n\nexport namespace ChainOfThoughtPrimitiveParts {\n  export type Props =\n    | {\n        /**\n         * @deprecated Use the children render function instead.\n         */\n        components?: ChainOfThoughtPartsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each part. Receives the part. */\n        children: (value: { part: PartState }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst ChainOfThoughtPrimitivePartsInner: FC<{\n  children: (value: { part: PartState }) => ReactNode;\n}> = ({ children }) => {\n  const partsLength = useAuiState((s) => s.chainOfThought.parts.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: partsLength }, (_, index) => (\n        <ChainOfThoughtPartByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.chainOfThought().part({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get part() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </ChainOfThoughtPartByIndexProvider>\n      )),\n    [partsLength, children],\n  );\n};\n\n/**\n * Renders the parts within a chain of thought, with support for collapsed/expanded states.\n *\n * When collapsed, no parts are shown. When expanded, all parts are rendered\n * using the provided component configuration through the part scope mechanism.\n */\nexport const ChainOfThoughtPrimitiveParts: FC<\n  ChainOfThoughtPrimitiveParts.Props\n> = ({ components, children }) => {\n  if (children) {\n    return (\n      <ChainOfThoughtPrimitivePartsInner>\n        {children}\n      </ChainOfThoughtPrimitivePartsInner>\n    );\n  }\n\n  // oxlint-disable-next-line react-hooks/rules-of-hooks -- intentional conditional hook below the early return above\n  const messageComponents = useMemo(\n    () => ({\n      Reasoning: components?.Reasoning,\n      tools: {\n        Fallback: components?.tools?.Fallback,\n      },\n    }),\n    [components?.Reasoning, components?.tools?.Fallback],\n  );\n\n  const Layout = components?.Layout;\n\n  return (\n    <ChainOfThoughtPrimitivePartsInner>\n      {() =>\n        Layout ? (\n          <Layout>\n            <MessagePartComponent components={messageComponents} />\n          </Layout>\n        ) : (\n          <MessagePartComponent components={messageComponents} />\n        )\n      }\n    </ChainOfThoughtPrimitivePartsInner>\n  );\n};\n\nChainOfThoughtPrimitiveParts.displayName = \"ChainOfThoughtPrimitive.Parts\";\n"],"mappings":";;;;;;AA2CA,MAAM,qCAEA,EAAE,eAAe;CACrB,MAAM,cAAc,aAAa,MAAM,EAAE,eAAe,MAAM,MAAM;CAEpE,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,YAAY,IAAI,GAAG,UACtC,oBAAC,mCAAD;EAAsD;YACpD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,eAAe,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE,SAAS;cAG9C,YACA,SAAS,EACP,IAAI,OAAO;IACT,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACK,GAdK,KAcL,CACpC,GACH,CAAC,aAAa,QAAQ,CACxB;AACF;;;;;;;AAQA,MAAa,gCAER,EAAE,YAAY,eAAe;CAChC,IAAI,UACF,OACE,oBAAC,mCAAD,EACG,SACgC,CAAA;CAKvC,MAAM,oBAAoB,eACjB;EACL,WAAW,YAAY;EACvB,OAAO,EACL,UAAU,YAAY,OAAO,SAC/B;CACF,IACA,CAAC,YAAY,WAAW,YAAY,OAAO,QAAQ,CACrD;CAEA,MAAM,SAAS,YAAY;CAE3B,OACE,oBAAC,mCAAD,EAAA,gBAEI,SACE,oBAAC,QAAD,EAAA,UACE,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAChD,CAAA,IAER,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAGzB,CAAA;AAEvC;AAEA,6BAA6B,cAAc"}
+{"version":3,"file":"ChainOfThoughtParts.js","names":[],"sources":["../../../../src/react/primitives/chainOfThought/ChainOfThoughtParts.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type PropsWithChildren,\n  type ReactNode,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport { ChainOfThoughtPartByIndexProvider } from \"../../providers/ChainOfThoughtPartByIndexProvider\";\nimport { MessagePartComponent } from \"../message/MessageParts\";\nimport type {\n  ReasoningMessagePartComponent,\n  ToolCallMessagePartComponent,\n} from \"../../types/MessagePartComponentTypes\";\n\ntype ChainOfThoughtPartsComponentConfig = {\n  /** Component for rendering reasoning parts */\n  Reasoning?: ReasoningMessagePartComponent | undefined;\n  /** Fallback component for tool-call parts */\n  tools?: {\n    Fallback?: ToolCallMessagePartComponent | undefined;\n  };\n  /** Layout component to wrap the rendered parts when expanded */\n  Layout?: ComponentType<PropsWithChildren> | undefined;\n};\n\nexport namespace ChainOfThoughtPrimitiveParts {\n  export type Props =\n    | {\n        /**\n         * @deprecated Use the children render function instead.\n         */\n        components?: ChainOfThoughtPartsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each part. Receives the part. */\n        children: (value: { part: PartState }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst ChainOfThoughtPrimitivePartsInner: FC<{\n  children: (value: { part: PartState }) => ReactNode;\n}> = ({ children }) => {\n  const partsLength = useAuiState((s) => s.chainOfThought.parts.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: partsLength }, (_, index) => (\n        <ChainOfThoughtPartByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.chainOfThought().part({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get part() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </ChainOfThoughtPartByIndexProvider>\n      )),\n    [partsLength, children],\n  );\n};\n\n/**\n * Renders the parts within a chain of thought, with support for collapsed/expanded states.\n *\n * When collapsed, no parts are shown. When expanded, all parts are rendered\n * using the provided component configuration through the part scope mechanism.\n */\nexport const ChainOfThoughtPrimitiveParts: FC<\n  ChainOfThoughtPrimitiveParts.Props\n> = ({ components, children }) => {\n  if (children) {\n    return (\n      <ChainOfThoughtPrimitivePartsInner>\n        {children}\n      </ChainOfThoughtPrimitivePartsInner>\n    );\n  }\n\n  // oxlint-disable-next-line react-hooks/rules-of-hooks -- intentional conditional hook below the early return above\n  const messageComponents = useMemo(\n    () => ({\n      Reasoning: components?.Reasoning,\n      tools: {\n        Fallback: components?.tools?.Fallback,\n      },\n    }),\n    [components?.Reasoning, components?.tools?.Fallback],\n  );\n\n  const Layout = components?.Layout;\n\n  return (\n    <ChainOfThoughtPrimitivePartsInner>\n      {() =>\n        Layout ? (\n          <Layout>\n            <MessagePartComponent components={messageComponents} />\n          </Layout>\n        ) : (\n          <MessagePartComponent components={messageComponents} />\n        )\n      }\n    </ChainOfThoughtPrimitivePartsInner>\n  );\n};\n\nChainOfThoughtPrimitiveParts.displayName = \"ChainOfThoughtPrimitive.Parts\";\n"],"mappings":";;;;;;AA2CA,MAAM,qCAEA,EAAE,eAAe;CACrB,MAAM,cAAc,aAAa,MAAM,EAAE,eAAe,MAAM,MAAM;CAEpE,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,YAAY,IAAI,GAAG,UACtC,oBAAC,mCAAD;EAAsD;YACpD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,eAAe,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS;cAG9C,YACA,SAAS,EACP,IAAI,OAAO;IACT,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACK,GAdK,KAcL,CACpC,GACH,CAAC,aAAa,QAAQ,CACxB;AACF;;;;;;;AAQA,MAAa,gCAER,EAAE,YAAY,eAAe;CAChC,IAAI,UACF,OACE,oBAAC,mCAAD,EACG,SACgC,CAAA;CAKvC,MAAM,oBAAoB,eACjB;EACL,WAAW,YAAY;EACvB,OAAO,EACL,UAAU,YAAY,OAAO,SAC/B;CACF,IACA,CAAC,YAAY,WAAW,YAAY,OAAO,QAAQ,CACrD;CAEA,MAAM,SAAS,YAAY;CAE3B,OACE,oBAAC,mCAAD,EAAA,gBAEI,SACE,oBAAC,QAAD,EAAA,UACE,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAChD,CAAA,IAER,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAGzB,CAAA;AAEvC;AAEA,6BAA6B,cAAc"}

package/dist/react/primitives/composer/ComposerAttachments.js

@@ -1,5 +1,5 @@
 import { ComposerAttachmentByIndexProvider } from "../../providers/AttachmentByIndexProvider.js";
-import { memo, useMemo } from "react";
+import { memo, useMemo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import { jsx } from "react/jsx-runtime";
 //#region src/react/primitives/composer/ComposerAttachments.tsx

package/dist/react/primitives/composer/ComposerAttachments.js.map

@@ -1 +1 @@
-{"version":3,"file":"ComposerAttachments.js","names":[],"sources":["../../../../src/react/primitives/composer/ComposerAttachments.tsx"],"sourcesContent":["import type { Attachment } from \"../../../types/attachment\";\nimport {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  memo,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport { ComposerAttachmentByIndexProvider } from \"../../providers/AttachmentByIndexProvider\";\n\ntype ComposerAttachmentsComponentConfig = {\n  Image?: ComponentType | undefined;\n  Document?: ComponentType | undefined;\n  File?: ComponentType | undefined;\n  Attachment?: ComponentType | undefined;\n};\n\nexport namespace ComposerPrimitiveAttachments {\n  export type Props =\n    | {\n        /** @deprecated Use the children render function instead. */\n        components: ComposerAttachmentsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each attachment. Receives the attachment. */\n        children: (value: { attachment: Attachment }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst getComponent = (\n  components: ComposerAttachmentsComponentConfig | undefined,\n  attachment: Attachment,\n) => {\n  const type = attachment.type;\n  switch (type) {\n    case \"image\":\n      return components?.Image ?? components?.Attachment;\n    case \"document\":\n      return components?.Document ?? components?.Attachment;\n    case \"file\":\n      return components?.File ?? components?.Attachment;\n    default:\n      return components?.Attachment;\n  }\n};\n\nconst AttachmentComponent: FC<{\n  components: ComposerAttachmentsComponentConfig | undefined;\n}> = ({ components }) => {\n  const attachment = useAuiState((s) => s.attachment);\n  if (!attachment) return null;\n\n  const Component = getComponent(components, attachment);\n  if (!Component) return null;\n  return <Component />;\n};\n\nexport namespace ComposerPrimitiveAttachmentByIndex {\n  export type Props = {\n    index: number;\n    components?: ComposerAttachmentsComponentConfig;\n  };\n}\n\n/**\n * Renders a single attachment at the specified index within the composer.\n */\nexport const ComposerPrimitiveAttachmentByIndex: FC<ComposerPrimitiveAttachmentByIndex.Props> =\n  memo(\n    ({ index, components }) => {\n      return (\n        <ComposerAttachmentByIndexProvider index={index}>\n          <AttachmentComponent components={components} />\n        </ComposerAttachmentByIndexProvider>\n      );\n    },\n    (prev, next) =>\n      prev.index === next.index &&\n      prev.components?.Image === next.components?.Image &&\n      prev.components?.Document === next.components?.Document &&\n      prev.components?.File === next.components?.File &&\n      prev.components?.Attachment === next.components?.Attachment,\n  );\n\nComposerPrimitiveAttachmentByIndex.displayName =\n  \"ComposerPrimitive.AttachmentByIndex\";\n\nconst ComposerPrimitiveAttachmentsInner: FC<{\n  children: (value: { attachment: Attachment }) => ReactNode;\n}> = ({ children }) => {\n  const attachmentsCount = useAuiState((s) => s.composer.attachments.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: attachmentsCount }, (_, index) => (\n        <ComposerAttachmentByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.composer().attachment({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get attachment() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </ComposerAttachmentByIndexProvider>\n      )),\n    [attachmentsCount, children],\n  );\n};\n\nexport const ComposerPrimitiveAttachments: FC<\n  ComposerPrimitiveAttachments.Props\n> = ({ components, children }) => {\n  if (components) {\n    return (\n      <ComposerPrimitiveAttachmentsInner>\n        {({ attachment }) => {\n          const Component = getComponent(components, attachment);\n          if (!Component) return null;\n          return <Component />;\n        }}\n      </ComposerPrimitiveAttachmentsInner>\n    );\n  }\n  return (\n    <ComposerPrimitiveAttachmentsInner>\n      {children}\n    </ComposerPrimitiveAttachmentsInner>\n  );\n};\n\nComposerPrimitiveAttachments.displayName = \"ComposerPrimitive.Attachments\";\n"],"mappings":";;;;;AAgCA,MAAM,gBACJ,YACA,eACG;CAEH,QADa,WAAW,MACxB;EACE,KAAK,SACH,OAAO,YAAY,SAAS,YAAY;EAC1C,KAAK,YACH,OAAO,YAAY,YAAY,YAAY;EAC7C,KAAK,QACH,OAAO,YAAY,QAAQ,YAAY;EACzC,SACE,OAAO,YAAY;CACvB;AACF;AAEA,MAAM,uBAEA,EAAE,iBAAiB;CACvB,MAAM,aAAa,aAAa,MAAM,EAAE,UAAU;CAClD,IAAI,CAAC,YAAY,OAAO;CAExB,MAAM,YAAY,aAAa,YAAY,UAAU;CACrD,IAAI,CAAC,WAAW,OAAO;CACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;AACrB;;;;AAYA,MAAa,qCACX,MACG,EAAE,OAAO,iBAAiB;CACzB,OACE,oBAAC,mCAAD;EAA0C;YACxC,oBAAC,qBAAD,EAAiC,WAAa,CAAA;CACb,CAAA;AAEvC,IACC,MAAM,SACL,KAAK,UAAU,KAAK,SACpB,KAAK,YAAY,UAAU,KAAK,YAAY,SAC5C,KAAK,YAAY,aAAa,KAAK,YAAY,YAC/C,KAAK,YAAY,SAAS,KAAK,YAAY,QAC3C,KAAK,YAAY,eAAe,KAAK,YAAY,UACrD;AAEF,mCAAmC,cACjC;AAEF,MAAM,qCAEA,EAAE,eAAe;CACrB,MAAM,mBAAmB,aAAa,MAAM,EAAE,SAAS,YAAY,MAAM;CAEzE,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,iBAAiB,IAAI,GAAG,UAC3C,oBAAC,mCAAD;EAAsD;YACpD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,SAAS,EAAE,WAAW,EAAE,MAAM,CAAC,EAAE,SAAS;cAG9C,YACA,SAAS,EACP,IAAI,aAAa;IACf,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACK,GAdK,KAcL,CACpC,GACH,CAAC,kBAAkB,QAAQ,CAC7B;AACF;AAEA,MAAa,gCAER,EAAE,YAAY,eAAe;CAChC,IAAI,YACF,OACE,oBAAC,mCAAD,EAAA,WACI,EAAE,iBAAiB;EACnB,MAAM,YAAY,aAAa,YAAY,UAAU;EACrD,IAAI,CAAC,WAAW,OAAO;EACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;CACrB,EACiC,CAAA;CAGvC,OACE,oBAAC,mCAAD,EACG,SACgC,CAAA;AAEvC;AAEA,6BAA6B,cAAc"}
+{"version":3,"file":"ComposerAttachments.js","names":[],"sources":["../../../../src/react/primitives/composer/ComposerAttachments.tsx"],"sourcesContent":["import type { Attachment } from \"../../../types/attachment\";\nimport {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  memo,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport { ComposerAttachmentByIndexProvider } from \"../../providers/AttachmentByIndexProvider\";\n\ntype ComposerAttachmentsComponentConfig = {\n  Image?: ComponentType | undefined;\n  Document?: ComponentType | undefined;\n  File?: ComponentType | undefined;\n  Attachment?: ComponentType | undefined;\n};\n\nexport namespace ComposerPrimitiveAttachments {\n  export type Props =\n    | {\n        /** @deprecated Use the children render function instead. */\n        components: ComposerAttachmentsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each attachment. Receives the attachment. */\n        children: (value: { attachment: Attachment }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst getComponent = (\n  components: ComposerAttachmentsComponentConfig | undefined,\n  attachment: Attachment,\n) => {\n  const type = attachment.type;\n  switch (type) {\n    case \"image\":\n      return components?.Image ?? components?.Attachment;\n    case \"document\":\n      return components?.Document ?? components?.Attachment;\n    case \"file\":\n      return components?.File ?? components?.Attachment;\n    default:\n      return components?.Attachment;\n  }\n};\n\nconst AttachmentComponent: FC<{\n  components: ComposerAttachmentsComponentConfig | undefined;\n}> = ({ components }) => {\n  const attachment = useAuiState((s) => s.attachment);\n  if (!attachment) return null;\n\n  const Component = getComponent(components, attachment);\n  if (!Component) return null;\n  return <Component />;\n};\n\nexport namespace ComposerPrimitiveAttachmentByIndex {\n  export type Props = {\n    index: number;\n    components?: ComposerAttachmentsComponentConfig;\n  };\n}\n\n/**\n * Renders a single attachment at the specified index within the composer.\n */\nexport const ComposerPrimitiveAttachmentByIndex: FC<ComposerPrimitiveAttachmentByIndex.Props> =\n  memo(\n    ({ index, components }) => {\n      return (\n        <ComposerAttachmentByIndexProvider index={index}>\n          <AttachmentComponent components={components} />\n        </ComposerAttachmentByIndexProvider>\n      );\n    },\n    (prev, next) =>\n      prev.index === next.index &&\n      prev.components?.Image === next.components?.Image &&\n      prev.components?.Document === next.components?.Document &&\n      prev.components?.File === next.components?.File &&\n      prev.components?.Attachment === next.components?.Attachment,\n  );\n\nComposerPrimitiveAttachmentByIndex.displayName =\n  \"ComposerPrimitive.AttachmentByIndex\";\n\nconst ComposerPrimitiveAttachmentsInner: FC<{\n  children: (value: { attachment: Attachment }) => ReactNode;\n}> = ({ children }) => {\n  const attachmentsCount = useAuiState((s) => s.composer.attachments.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: attachmentsCount }, (_, index) => (\n        <ComposerAttachmentByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.composer().attachment({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get attachment() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </ComposerAttachmentByIndexProvider>\n      )),\n    [attachmentsCount, children],\n  );\n};\n\nexport const ComposerPrimitiveAttachments: FC<\n  ComposerPrimitiveAttachments.Props\n> = ({ components, children }) => {\n  if (components) {\n    return (\n      <ComposerPrimitiveAttachmentsInner>\n        {({ attachment }) => {\n          const Component = getComponent(components, attachment);\n          if (!Component) return null;\n          return <Component />;\n        }}\n      </ComposerPrimitiveAttachmentsInner>\n    );\n  }\n  return (\n    <ComposerPrimitiveAttachmentsInner>\n      {children}\n    </ComposerPrimitiveAttachmentsInner>\n  );\n};\n\nComposerPrimitiveAttachments.displayName = \"ComposerPrimitive.Attachments\";\n"],"mappings":";;;;;AAgCA,MAAM,gBACJ,YACA,eACG;CAEH,QADa,WAAW,MACxB;EACE,KAAK,SACH,OAAO,YAAY,SAAS,YAAY;EAC1C,KAAK,YACH,OAAO,YAAY,YAAY,YAAY;EAC7C,KAAK,QACH,OAAO,YAAY,QAAQ,YAAY;EACzC,SACE,OAAO,YAAY;CACvB;AACF;AAEA,MAAM,uBAEA,EAAE,iBAAiB;CACvB,MAAM,aAAa,aAAa,MAAM,EAAE,UAAU;CAClD,IAAI,CAAC,YAAY,OAAO;CAExB,MAAM,YAAY,aAAa,YAAY,UAAU;CACrD,IAAI,CAAC,WAAW,OAAO;CACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;AACrB;;;;AAYA,MAAa,qCACX,MACG,EAAE,OAAO,iBAAiB;CACzB,OACE,oBAAC,mCAAD;EAA0C;YACxC,oBAAC,qBAAD,EAAiC,WAAa,CAAA;CACb,CAAA;AAEvC,IACC,MAAM,SACL,KAAK,UAAU,KAAK,SACpB,KAAK,YAAY,UAAU,KAAK,YAAY,SAC5C,KAAK,YAAY,aAAa,KAAK,YAAY,YAC/C,KAAK,YAAY,SAAS,KAAK,YAAY,QAC3C,KAAK,YAAY,eAAe,KAAK,YAAY,UACrD;AAEF,mCAAmC,cACjC;AAEF,MAAM,qCAEA,EAAE,eAAe;CACrB,MAAM,mBAAmB,aAAa,MAAM,EAAE,SAAS,YAAY,MAAM;CAEzE,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,iBAAiB,IAAI,GAAG,UAC3C,oBAAC,mCAAD;EAAsD;YACpD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,SAAS,CAAC,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS;cAG9C,YACA,SAAS,EACP,IAAI,aAAa;IACf,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACK,GAdK,KAcL,CACpC,GACH,CAAC,kBAAkB,QAAQ,CAC7B;AACF;AAEA,MAAa,gCAER,EAAE,YAAY,eAAe;CAChC,IAAI,YACF,OACE,oBAAC,mCAAD,EAAA,WACI,EAAE,iBAAiB;EACnB,MAAM,YAAY,aAAa,YAAY,UAAU;EACrD,IAAI,CAAC,WAAW,OAAO;EACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;CACrB,EACiC,CAAA;CAGvC,OACE,oBAAC,mCAAD,EACG,SACgC,CAAA;AAEvC;AAEA,6BAA6B,cAAc"}

package/dist/react/primitives/composer/ComposerQueue.js

@@ -1,5 +1,5 @@
 import { QueueItemByIndexProvider } from "../../providers/QueueItemByIndexProvider.js";
-import { memo, useMemo } from "react";
+import { memo, useMemo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import { jsx } from "react/jsx-runtime";
 //#region src/react/primitives/composer/ComposerQueue.tsx

package/dist/react/primitives/composer/ComposerQueue.js.map

@@ -1 +1 @@
-{"version":3,"file":"ComposerQueue.js","names":[],"sources":["../../../../src/react/primitives/composer/ComposerQueue.tsx"],"sourcesContent":["import { type FC, type ReactNode, memo, useMemo } from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport type { QueueItemState } from \"../../../store/scopes/queue-item\";\nimport { QueueItemByIndexProvider } from \"../../providers/QueueItemByIndexProvider\";\n\nexport namespace ComposerPrimitiveQueue {\n  export type Props = {\n    /** Render function called for each queue item. Receives the queue item state. */\n    children: (value: { queueItem: QueueItemState }) => ReactNode;\n  };\n}\n\nconst ComposerPrimitiveQueueInner: FC<{\n  children: (value: { queueItem: QueueItemState }) => ReactNode;\n}> = ({ children }) => {\n  const queue = useAuiState((s) => s.composer.queue.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: queue }, (_, index) => (\n        <QueueItemByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.composer().queueItem({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get queueItem() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </QueueItemByIndexProvider>\n      )),\n    [queue, children],\n  );\n};\n\n/**\n * Renders all queue items in the composer.\n *\n * @example\n * ```tsx\n * <ComposerPrimitive.Queue>\n *   {({ queueItem }) => (\n *     <div>\n *       <QueueItemPrimitive.Text />\n *       <QueueItemPrimitive.Steer>Run Now</QueueItemPrimitive.Steer>\n *     </div>\n *   )}\n * </ComposerPrimitive.Queue>\n * ```\n */\nexport const ComposerPrimitiveQueue = memo(ComposerPrimitiveQueueInner);\n\nComposerPrimitiveQueue.displayName = \"ComposerPrimitive.Queue\";\n"],"mappings":";;;;;AAYA,MAAM,+BAEA,EAAE,eAAe;CACrB,MAAM,QAAQ,aAAa,MAAM,EAAE,SAAS,MAAM,MAAM;CAExD,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,MAAM,IAAI,GAAG,UAChC,oBAAC,0BAAD;EAA6C;YAC3C,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,SAAS;cAG7C,YACA,SAAS,EACP,IAAI,YAAY;IACd,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACJ,GAdK,KAcL,CAC3B,GACH,CAAC,OAAO,QAAQ,CAClB;AACF;;;;;;;;;;;;;;;;AAiBA,MAAa,yBAAyB,KAAK,2BAA2B;AAEtE,uBAAuB,cAAc"}
+{"version":3,"file":"ComposerQueue.js","names":[],"sources":["../../../../src/react/primitives/composer/ComposerQueue.tsx"],"sourcesContent":["import { type FC, type ReactNode, memo, useMemo } from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport type { QueueItemState } from \"../../../store/scopes/queue-item\";\nimport { QueueItemByIndexProvider } from \"../../providers/QueueItemByIndexProvider\";\n\nexport namespace ComposerPrimitiveQueue {\n  export type Props = {\n    /** Render function called for each queue item. Receives the queue item state. */\n    children: (value: { queueItem: QueueItemState }) => ReactNode;\n  };\n}\n\nconst ComposerPrimitiveQueueInner: FC<{\n  children: (value: { queueItem: QueueItemState }) => ReactNode;\n}> = ({ children }) => {\n  const queue = useAuiState((s) => s.composer.queue.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: queue }, (_, index) => (\n        <QueueItemByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.composer().queueItem({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get queueItem() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </QueueItemByIndexProvider>\n      )),\n    [queue, children],\n  );\n};\n\n/**\n * Renders all queue items in the composer.\n *\n * @example\n * ```tsx\n * <ComposerPrimitive.Queue>\n *   {({ queueItem }) => (\n *     <div>\n *       <QueueItemPrimitive.Text />\n *       <QueueItemPrimitive.Steer>Run Now</QueueItemPrimitive.Steer>\n *     </div>\n *   )}\n * </ComposerPrimitive.Queue>\n * ```\n */\nexport const ComposerPrimitiveQueue = memo(ComposerPrimitiveQueueInner);\n\nComposerPrimitiveQueue.displayName = \"ComposerPrimitive.Queue\";\n"],"mappings":";;;;;AAYA,MAAM,+BAEA,EAAE,eAAe;CACrB,MAAM,QAAQ,aAAa,MAAM,EAAE,SAAS,MAAM,MAAM;CAExD,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,MAAM,IAAI,GAAG,UAChC,oBAAC,0BAAD;EAA6C;YAC3C,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,SAAS,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS;cAG7C,YACA,SAAS,EACP,IAAI,YAAY;IACd,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACJ,GAdK,KAcL,CAC3B,GACH,CAAC,OAAO,QAAQ,CAClB;AACF;;;;;;;;;;;;;;;;AAiBA,MAAa,yBAAyB,KAAK,2BAA2B;AAEtE,uBAAuB,cAAc"}

package/dist/react/primitives/generativeUI/GenerativeUI.js

@@ -1,5 +1,5 @@
 "use client";
-import { createElement, useMemo } from "react";
+import { createElement, useMemo } from "@assistant-ui/tap/react-shim";
 import { useAuiState } from "@assistant-ui/store";
 import { Fragment as Fragment$1, jsx } from "react/jsx-runtime";
 //#region src/react/primitives/generativeUI/GenerativeUI.tsx

package/dist/react/primitives/generativeUI/GenerativeUI.js.map

@@ -1 +1 @@
-{"version":3,"file":"GenerativeUI.js","names":[],"sources":["../../../../src/react/primitives/generativeUI/GenerativeUI.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  createElement,\n  useMemo,\n} from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport type {\n  GenerativeUINode,\n  GenerativeUISpec,\n} from \"../../../types/message\";\nimport type {\n  GenerativeUIComponentRegistry,\n  GenerativeUIRenderProps,\n} from \"../../types/MessagePartComponentTypes\";\n\n/**\n * Thrown when a generative-ui spec references a component name that is not\n * present in the consumer-provided allowlist. The allowlist is the security\n * boundary in the same-realm rendering path — there is no fallback by\n * default. Pass `Fallback` to opt into a soft-fail UX.\n */\nexport class GenerativeUIRenderError extends Error {\n  public readonly componentName: string;\n\n  constructor(\n    componentName: string,\n    message = `Component \"${componentName}\" is not in the generative-ui allowlist.`,\n  ) {\n    super(message);\n    this.name = \"GenerativeUIRenderError\";\n    this.componentName = componentName;\n  }\n}\n\nconst isObjectNode = (\n  node: GenerativeUINode,\n): node is Exclude<GenerativeUINode, string> =>\n  typeof node === \"object\" && node !== null;\n\nconst renderNode = (\n  node: GenerativeUINode | undefined,\n  components: GenerativeUIComponentRegistry,\n  Fallback: GenerativeUIRenderProps[\"Fallback\"],\n  path: string,\n): ReactNode => {\n  if (node === undefined || node === null) return null;\n\n  if (typeof node === \"string\") return node;\n\n  if (!isObjectNode(node) || !(\"component\" in node)) {\n    if (\n      typeof process !== \"undefined\" &&\n      process.env?.NODE_ENV !== \"production\"\n    ) {\n      console.warn(`[generative-ui] Skipping malformed node at ${path}:`, node);\n    }\n    return null;\n  }\n\n  const { component, props, children, key } = node;\n\n  const Resolved = components[component];\n  if (!Resolved) {\n    if (Fallback) {\n      return <Fallback key={key ?? path} component={component} props={props} />;\n    }\n    throw new GenerativeUIRenderError(component);\n  }\n\n  const renderedChildren = children?.length\n    ? children.map((child, i) =>\n        renderNode(child, components, Fallback, `${path}/${i}`),\n      )\n    : undefined;\n\n  return createElement(\n    Resolved,\n    { ...(props ?? {}), key: key ?? path },\n    ...(renderedChildren ?? []),\n  );\n};\n\nconst normalizeRoot = (\n  spec: GenerativeUISpec | undefined,\n): readonly GenerativeUINode[] => {\n  if (!spec || spec.root === undefined || spec.root === null) return [];\n  const root = spec.root;\n  return Array.isArray(root)\n    ? (root as readonly GenerativeUINode[])\n    : [root as GenerativeUINode];\n};\n\n/**\n * Internal renderer. Resolves a {@link GenerativeUISpec} against the consumer\n * allowlist. Used by `MessagePrimitive.GenerativeUI` and by\n * `MessagePrimitive.Parts` when handling a `generative-ui` part.\n */\nexport const GenerativeUIRender: FC<GenerativeUIRenderProps> = ({\n  spec,\n  components,\n  Fallback,\n}) => {\n  const nodes = useMemo(() => normalizeRoot(spec), [spec]);\n\n  return (\n    <>\n      {nodes.map((node, i) => renderNode(node, components, Fallback, `${i}`))}\n    </>\n  );\n};\n\nGenerativeUIRender.displayName = \"GenerativeUIRender\";\n\nexport namespace MessagePrimitiveGenerativeUI {\n  export type Props = {\n    /**\n     * The component allowlist. Keys are the names referenced in the spec\n     * (e.g. `\"Card\"`, `\"Button\"`), values are the React components.\n     *\n     * This is the security boundary — any name not in the allowlist is\n     * rejected with {@link GenerativeUIRenderError}.\n     */\n    components: GenerativeUIComponentRegistry;\n    /**\n     * Optional override spec. If omitted, the primitive reads the\n     * `generative-ui` part from the surrounding `MessagePartProvider` /\n     * `PartByIndexProvider` context.\n     */\n    spec?: GenerativeUISpec | undefined;\n    /** Optional fallback for unknown component names. */\n    Fallback?:\n      | ComponentType<{ component: string; props?: unknown }>\n      | undefined;\n  };\n}\n\n/**\n * Renders a generative-ui message part using a consumer-provided allowlist.\n *\n * The agent emits a `generative-ui` message part containing a JSON spec\n * (see {@link GenerativeUISpec}). This primitive walks the spec and resolves\n * each `component` name against the allowlist. Names not in the allowlist\n * throw {@link GenerativeUIRenderError} unless a `Fallback` is provided.\n *\n * Stream-friendly: a partial spec renders progressively as it is filled in.\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GenerativeUI\n *   components={{ Card: MyCard, Button: MyButton }}\n * />\n * ```\n */\nexport const MessagePrimitiveGenerativeUI: FC<\n  MessagePrimitiveGenerativeUI.Props\n> = ({ components, spec, Fallback }) => {\n  // Selector reads store state only — combining with the `spec` prop inside\n  // the selector closes over a value that may change identity per render and\n  // would trigger spurious tearing-detection re-renders in\n  // `useSyncExternalStore`.\n  const storeSpec = useAuiState((s) => {\n    const part = s.part as { type?: string; spec?: GenerativeUISpec };\n    return part?.type === \"generative-ui\" ? part.spec : undefined;\n  });\n  const partSpec = spec ?? storeSpec;\n\n  if (!partSpec) return null;\n\n  return (\n    <GenerativeUIRender\n      spec={partSpec}\n      components={components}\n      Fallback={Fallback}\n    />\n  );\n};\n\nMessagePrimitiveGenerativeUI.displayName = \"MessagePrimitive.GenerativeUI\";\n"],"mappings":";;;;;;;;;;;AAyBA,IAAa,0BAAb,cAA6C,MAAM;CACjD;CAEA,YACE,eACA,UAAU,cAAc,cAAc,2CACtC;EACA,MAAM,OAAO;EACb,KAAK,OAAO;EACZ,KAAK,gBAAgB;CACvB;AACF;AAEA,MAAM,gBACJ,SAEA,OAAO,SAAS,YAAY,SAAS;AAEvC,MAAM,cACJ,MACA,YACA,UACA,SACc;CACd,IAAI,SAAS,KAAA,KAAa,SAAS,MAAM,OAAO;CAEhD,IAAI,OAAO,SAAS,UAAU,OAAO;CAErC,IAAI,CAAC,aAAa,IAAI,KAAK,EAAE,eAAe,OAAO;EACjD,IACE,OAAO,YAAY,eACnB,QAAQ,KAAK,aAAa,cAE1B,QAAQ,KAAK,8CAA8C,KAAK,IAAI,IAAI;EAE1E,OAAO;CACT;CAEA,MAAM,EAAE,WAAW,OAAO,UAAU,QAAQ;CAE5C,MAAM,WAAW,WAAW;CAC5B,IAAI,CAAC,UAAU;EACb,IAAI,UACF,OAAO,oBAAC,UAAD;GAAuC;GAAkB;EAAQ,GAAlD,OAAO,IAA2C;EAE1E,MAAM,IAAI,wBAAwB,SAAS;CAC7C;CAEA,MAAM,mBAAmB,UAAU,SAC/B,SAAS,KAAK,OAAO,MACnB,WAAW,OAAO,YAAY,UAAU,GAAG,KAAK,GAAG,GAAG,CACxD,IACA,KAAA;CAEJ,OAAO,cACL,UACA;EAAE,GAAI,SAAS,CAAC;EAAI,KAAK,OAAO;CAAK,GACrC,GAAI,oBAAoB,CAAC,CAC3B;AACF;AAEA,MAAM,iBACJ,SACgC;CAChC,IAAI,CAAC,QAAQ,KAAK,SAAS,KAAA,KAAa,KAAK,SAAS,MAAM,OAAO,CAAC;CACpE,MAAM,OAAO,KAAK;CAClB,OAAO,MAAM,QAAQ,IAAI,IACpB,OACD,CAAC,IAAwB;AAC/B;;;;;;AAOA,MAAa,sBAAmD,EAC9D,MACA,YACA,eACI;CAGJ,OACE,oBAAA,YAAA,EAAA,UAHY,cAAc,cAAc,IAAI,GAAG,CAAC,IAAI,CAI7C,EAAE,KAAK,MAAM,MAAM,WAAW,MAAM,YAAY,UAAU,GAAG,GAAG,CAAC,EACtE,CAAA;AAEN;AAEA,mBAAmB,cAAc;;;;;;;;;;;;;;;;;;AA0CjC,MAAa,gCAER,EAAE,YAAY,MAAM,eAAe;CAKtC,MAAM,YAAY,aAAa,MAAM;EACnC,MAAM,OAAO,EAAE;EACf,OAAO,MAAM,SAAS,kBAAkB,KAAK,OAAO,KAAA;CACtD,CAAC;CACD,MAAM,WAAW,QAAQ;CAEzB,IAAI,CAAC,UAAU,OAAO;CAEtB,OACE,oBAAC,oBAAD;EACE,MAAM;EACM;EACF;CACX,CAAA;AAEL;AAEA,6BAA6B,cAAc"}
+{"version":3,"file":"GenerativeUI.js","names":[],"sources":["../../../../src/react/primitives/generativeUI/GenerativeUI.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  createElement,\n  useMemo,\n} from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport type {\n  GenerativeUINode,\n  GenerativeUISpec,\n} from \"../../../types/message\";\nimport type {\n  GenerativeUIComponentRegistry,\n  GenerativeUIRenderProps,\n} from \"../../types/MessagePartComponentTypes\";\n\n/**\n * Thrown when a generative-ui spec references a component name that is not\n * present in the consumer-provided allowlist. The allowlist is the security\n * boundary in the same-realm rendering path — there is no fallback by\n * default. Pass `Fallback` to opt into a soft-fail UX.\n */\nexport class GenerativeUIRenderError extends Error {\n  public readonly componentName: string;\n\n  constructor(\n    componentName: string,\n    message = `Component \"${componentName}\" is not in the generative-ui allowlist.`,\n  ) {\n    super(message);\n    this.name = \"GenerativeUIRenderError\";\n    this.componentName = componentName;\n  }\n}\n\nconst isObjectNode = (\n  node: GenerativeUINode,\n): node is Exclude<GenerativeUINode, string> =>\n  typeof node === \"object\" && node !== null;\n\nconst renderNode = (\n  node: GenerativeUINode | undefined,\n  components: GenerativeUIComponentRegistry,\n  Fallback: GenerativeUIRenderProps[\"Fallback\"],\n  path: string,\n): ReactNode => {\n  if (node === undefined || node === null) return null;\n\n  if (typeof node === \"string\") return node;\n\n  if (!isObjectNode(node) || !(\"component\" in node)) {\n    if (\n      typeof process !== \"undefined\" &&\n      process.env?.NODE_ENV !== \"production\"\n    ) {\n      console.warn(`[generative-ui] Skipping malformed node at ${path}:`, node);\n    }\n    return null;\n  }\n\n  const { component, props, children, key } = node;\n\n  const Resolved = components[component];\n  if (!Resolved) {\n    if (Fallback) {\n      return <Fallback key={key ?? path} component={component} props={props} />;\n    }\n    throw new GenerativeUIRenderError(component);\n  }\n\n  const renderedChildren = children?.length\n    ? children.map((child, i) =>\n        renderNode(child, components, Fallback, `${path}/${i}`),\n      )\n    : undefined;\n\n  return createElement(\n    Resolved,\n    { ...(props ?? {}), key: key ?? path },\n    ...(renderedChildren ?? []),\n  );\n};\n\nconst normalizeRoot = (\n  spec: GenerativeUISpec | undefined,\n): readonly GenerativeUINode[] => {\n  if (!spec || spec.root === undefined || spec.root === null) return [];\n  const root = spec.root;\n  return Array.isArray(root)\n    ? (root as readonly GenerativeUINode[])\n    : [root as GenerativeUINode];\n};\n\n/**\n * Internal renderer. Resolves a {@link GenerativeUISpec} against the consumer\n * allowlist. Used by `MessagePrimitive.GenerativeUI` and by\n * `MessagePrimitive.Parts` when handling a `generative-ui` part.\n */\nexport const GenerativeUIRender: FC<GenerativeUIRenderProps> = ({\n  spec,\n  components,\n  Fallback,\n}) => {\n  const nodes = useMemo(() => normalizeRoot(spec), [spec]);\n\n  return (\n    <>\n      {nodes.map((node, i) => renderNode(node, components, Fallback, `${i}`))}\n    </>\n  );\n};\n\nGenerativeUIRender.displayName = \"GenerativeUIRender\";\n\nexport namespace MessagePrimitiveGenerativeUI {\n  export type Props = {\n    /**\n     * The component allowlist. Keys are the names referenced in the spec\n     * (e.g. `\"Card\"`, `\"Button\"`), values are the React components.\n     *\n     * This is the security boundary — any name not in the allowlist is\n     * rejected with {@link GenerativeUIRenderError}.\n     */\n    components: GenerativeUIComponentRegistry;\n    /**\n     * Optional override spec. If omitted, the primitive reads the\n     * `generative-ui` part from the surrounding `MessagePartProvider` /\n     * `PartByIndexProvider` context.\n     */\n    spec?: GenerativeUISpec | undefined;\n    /** Optional fallback for unknown component names. */\n    Fallback?:\n      | ComponentType<{ component: string; props?: unknown }>\n      | undefined;\n  };\n}\n\n/**\n * Renders a generative-ui message part using a consumer-provided allowlist.\n *\n * The agent emits a `generative-ui` message part containing a JSON spec\n * (see {@link GenerativeUISpec}). This primitive walks the spec and resolves\n * each `component` name against the allowlist. Names not in the allowlist\n * throw {@link GenerativeUIRenderError} unless a `Fallback` is provided.\n *\n * Stream-friendly: a partial spec renders progressively as it is filled in.\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GenerativeUI\n *   components={{ Card: MyCard, Button: MyButton }}\n * />\n * ```\n */\nexport const MessagePrimitiveGenerativeUI: FC<\n  MessagePrimitiveGenerativeUI.Props\n> = ({ components, spec, Fallback }) => {\n  // Selector reads store state only — combining with the `spec` prop inside\n  // the selector closes over a value that may change identity per render and\n  // would trigger spurious tearing-detection re-renders in\n  // `useSyncExternalStore`.\n  const storeSpec = useAuiState((s) => {\n    const part = s.part as { type?: string; spec?: GenerativeUISpec };\n    return part?.type === \"generative-ui\" ? part.spec : undefined;\n  });\n  const partSpec = spec ?? storeSpec;\n\n  if (!partSpec) return null;\n\n  return (\n    <GenerativeUIRender\n      spec={partSpec}\n      components={components}\n      Fallback={Fallback}\n    />\n  );\n};\n\nMessagePrimitiveGenerativeUI.displayName = \"MessagePrimitive.GenerativeUI\";\n"],"mappings":";;;;;;;;;;;AAyBA,IAAa,0BAAb,cAA6C,MAAM;CACjD;CAEA,YACE,eACA,UAAU,cAAc,cAAc,2CACtC;EACA,MAAM,OAAO;EACb,KAAK,OAAO;EACZ,KAAK,gBAAgB;CACvB;AACF;AAEA,MAAM,gBACJ,SAEA,OAAO,SAAS,YAAY,SAAS;AAEvC,MAAM,cACJ,MACA,YACA,UACA,SACc;CACd,IAAI,SAAS,KAAA,KAAa,SAAS,MAAM,OAAO;CAEhD,IAAI,OAAO,SAAS,UAAU,OAAO;CAErC,IAAI,CAAC,aAAa,IAAI,KAAK,EAAE,eAAe,OAAO;EACjD,IACE,OAAO,YAAY,eACnB,QAAQ,KAAK,aAAa,cAE1B,QAAQ,KAAK,8CAA8C,KAAK,IAAI,IAAI;EAE1E,OAAO;CACT;CAEA,MAAM,EAAE,WAAW,OAAO,UAAU,QAAQ;CAE5C,MAAM,WAAW,WAAW;CAC5B,IAAI,CAAC,UAAU;EACb,IAAI,UACF,OAAO,oBAAC,UAAD;GAAuC;GAAkB;EAAQ,GAAlD,OAAO,IAA2C;EAE1E,MAAM,IAAI,wBAAwB,SAAS;CAC7C;CAEA,MAAM,mBAAmB,UAAU,SAC/B,SAAS,KAAK,OAAO,MACnB,WAAW,OAAO,YAAY,UAAU,GAAG,KAAK,GAAG,GAAG,CACxD,IACA,KAAA;CAEJ,OAAO,cACL,UACA;EAAE,GAAI,SAAS,CAAC;EAAI,KAAK,OAAO;CAAK,GACrC,GAAI,oBAAoB,CAAC,CAC3B;AACF;AAEA,MAAM,iBACJ,SACgC;CAChC,IAAI,CAAC,QAAQ,KAAK,SAAS,KAAA,KAAa,KAAK,SAAS,MAAM,OAAO,CAAC;CACpE,MAAM,OAAO,KAAK;CAClB,OAAO,MAAM,QAAQ,IAAI,IACpB,OACD,CAAC,IAAwB;AAC/B;;;;;;AAOA,MAAa,sBAAmD,EAC9D,MACA,YACA,eACI;CAGJ,OACE,oBAAA,YAAA,EAAA,UAHY,cAAc,cAAc,IAAI,GAAG,CAAC,IAAI,CAI7C,CAAC,CAAC,KAAK,MAAM,MAAM,WAAW,MAAM,YAAY,UAAU,GAAG,GAAG,CAAC,EACtE,CAAA;AAEN;AAEA,mBAAmB,cAAc;;;;;;;;;;;;;;;;;;AA0CjC,MAAa,gCAER,EAAE,YAAY,MAAM,eAAe;CAKtC,MAAM,YAAY,aAAa,MAAM;EACnC,MAAM,OAAO,EAAE;EACf,OAAO,MAAM,SAAS,kBAAkB,KAAK,OAAO,KAAA;CACtD,CAAC;CACD,MAAM,WAAW,QAAQ;CAEzB,IAAI,CAAC,UAAU,OAAO;CAEtB,OACE,oBAAC,oBAAD;EACE,MAAM;EACM;EACF;CACX,CAAA;AAEL;AAEA,6BAA6B,cAAc"}

package/dist/react/primitives/message/MessageAttachments.js

@@ -1,5 +1,5 @@
 import { MessageAttachmentByIndexProvider } from "../../providers/AttachmentByIndexProvider.js";
-import { memo, useMemo } from "react";
+import { memo, useMemo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import { jsx } from "react/jsx-runtime";
 //#region src/react/primitives/message/MessageAttachments.tsx

package/dist/react/primitives/message/MessageAttachments.js.map

@@ -1 +1 @@
-{"version":3,"file":"MessageAttachments.js","names":[],"sources":["../../../../src/react/primitives/message/MessageAttachments.tsx"],"sourcesContent":["import type { CompleteAttachment } from \"../../../types/attachment\";\nimport {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  memo,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport { MessageAttachmentByIndexProvider } from \"../../providers/AttachmentByIndexProvider\";\n\ntype MessageAttachmentsComponentConfig = {\n  Image?: ComponentType | undefined;\n  Document?: ComponentType | undefined;\n  File?: ComponentType | undefined;\n  Attachment?: ComponentType | undefined;\n};\n\nexport namespace MessagePrimitiveAttachments {\n  export type Props =\n    | {\n        /** @deprecated Use the children render function instead. */\n        components: MessageAttachmentsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each attachment. Receives the attachment. */\n        children: (value: { attachment: CompleteAttachment }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst getComponent = (\n  components: MessageAttachmentsComponentConfig | undefined,\n  attachment: CompleteAttachment,\n) => {\n  const type = attachment.type;\n  switch (type) {\n    case \"image\":\n      return components?.Image ?? components?.Attachment;\n    case \"document\":\n      return components?.Document ?? components?.Attachment;\n    case \"file\":\n      return components?.File ?? components?.Attachment;\n    default:\n      return components?.Attachment;\n  }\n};\n\nconst AttachmentComponent: FC<{\n  components: MessageAttachmentsComponentConfig | undefined;\n}> = ({ components }) => {\n  const attachment = useAuiState((s) => s.attachment);\n  if (!attachment) return null;\n\n  const Component = getComponent(components, attachment as CompleteAttachment);\n  if (!Component) return null;\n  return <Component />;\n};\n\nexport namespace MessagePrimitiveAttachmentByIndex {\n  export type Props = {\n    index: number;\n    components?: MessageAttachmentsComponentConfig;\n  };\n}\n\n/**\n * Renders a single attachment at the specified index within the current message.\n */\nexport const MessagePrimitiveAttachmentByIndex: FC<MessagePrimitiveAttachmentByIndex.Props> =\n  memo(\n    ({ index, components }) => {\n      return (\n        <MessageAttachmentByIndexProvider index={index}>\n          <AttachmentComponent components={components} />\n        </MessageAttachmentByIndexProvider>\n      );\n    },\n    (prev, next) =>\n      prev.index === next.index &&\n      prev.components?.Image === next.components?.Image &&\n      prev.components?.Document === next.components?.Document &&\n      prev.components?.File === next.components?.File &&\n      prev.components?.Attachment === next.components?.Attachment,\n  );\n\nMessagePrimitiveAttachmentByIndex.displayName =\n  \"MessagePrimitive.AttachmentByIndex\";\n\nconst MessagePrimitiveAttachmentsInner: FC<{\n  children: (value: { attachment: CompleteAttachment }) => ReactNode;\n}> = ({ children }) => {\n  const attachmentsCount = useAuiState((s) => {\n    if (s.message.role !== \"user\") return 0;\n    return (s.message.attachments ?? []).length;\n  });\n\n  return useMemo(\n    () =>\n      Array.from({ length: attachmentsCount }, (_, index) => (\n        <MessageAttachmentByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.message().attachment({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get attachment() {\n                  return getItem() as CompleteAttachment;\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </MessageAttachmentByIndexProvider>\n      )),\n    [attachmentsCount, children],\n  );\n};\n\nexport const MessagePrimitiveAttachments: FC<\n  MessagePrimitiveAttachments.Props\n> = ({ components, children }) => {\n  if (components) {\n    return (\n      <MessagePrimitiveAttachmentsInner>\n        {({ attachment }) => {\n          const Component = getComponent(components, attachment);\n          if (!Component) return null;\n          return <Component />;\n        }}\n      </MessagePrimitiveAttachmentsInner>\n    );\n  }\n  return (\n    <MessagePrimitiveAttachmentsInner>\n      {children}\n    </MessagePrimitiveAttachmentsInner>\n  );\n};\n\nMessagePrimitiveAttachments.displayName = \"MessagePrimitive.Attachments\";\n"],"mappings":";;;;;AAgCA,MAAM,gBACJ,YACA,eACG;CAEH,QADa,WAAW,MACxB;EACE,KAAK,SACH,OAAO,YAAY,SAAS,YAAY;EAC1C,KAAK,YACH,OAAO,YAAY,YAAY,YAAY;EAC7C,KAAK,QACH,OAAO,YAAY,QAAQ,YAAY;EACzC,SACE,OAAO,YAAY;CACvB;AACF;AAEA,MAAM,uBAEA,EAAE,iBAAiB;CACvB,MAAM,aAAa,aAAa,MAAM,EAAE,UAAU;CAClD,IAAI,CAAC,YAAY,OAAO;CAExB,MAAM,YAAY,aAAa,YAAY,UAAgC;CAC3E,IAAI,CAAC,WAAW,OAAO;CACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;AACrB;;;;AAYA,MAAa,oCACX,MACG,EAAE,OAAO,iBAAiB;CACzB,OACE,oBAAC,kCAAD;EAAyC;YACvC,oBAAC,qBAAD,EAAiC,WAAa,CAAA;CACd,CAAA;AAEtC,IACC,MAAM,SACL,KAAK,UAAU,KAAK,SACpB,KAAK,YAAY,UAAU,KAAK,YAAY,SAC5C,KAAK,YAAY,aAAa,KAAK,YAAY,YAC/C,KAAK,YAAY,SAAS,KAAK,YAAY,QAC3C,KAAK,YAAY,eAAe,KAAK,YAAY,UACrD;AAEF,kCAAkC,cAChC;AAEF,MAAM,oCAEA,EAAE,eAAe;CACrB,MAAM,mBAAmB,aAAa,MAAM;EAC1C,IAAI,EAAE,QAAQ,SAAS,QAAQ,OAAO;EACtC,QAAQ,EAAE,QAAQ,eAAe,CAAC,GAAG;CACvC,CAAC;CAED,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,iBAAiB,IAAI,GAAG,UAC3C,oBAAC,kCAAD;EAAqD;YACnD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC,EAAE,SAAS;cAG7C,YACA,SAAS,EACP,IAAI,aAAa;IACf,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACI,GAdK,KAcL,CACnC,GACH,CAAC,kBAAkB,QAAQ,CAC7B;AACF;AAEA,MAAa,+BAER,EAAE,YAAY,eAAe;CAChC,IAAI,YACF,OACE,oBAAC,kCAAD,EAAA,WACI,EAAE,iBAAiB;EACnB,MAAM,YAAY,aAAa,YAAY,UAAU;EACrD,IAAI,CAAC,WAAW,OAAO;EACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;CACrB,EACgC,CAAA;CAGtC,OACE,oBAAC,kCAAD,EACG,SAC+B,CAAA;AAEtC;AAEA,4BAA4B,cAAc"}
+{"version":3,"file":"MessageAttachments.js","names":[],"sources":["../../../../src/react/primitives/message/MessageAttachments.tsx"],"sourcesContent":["import type { CompleteAttachment } from \"../../../types/attachment\";\nimport {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  memo,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport { MessageAttachmentByIndexProvider } from \"../../providers/AttachmentByIndexProvider\";\n\ntype MessageAttachmentsComponentConfig = {\n  Image?: ComponentType | undefined;\n  Document?: ComponentType | undefined;\n  File?: ComponentType | undefined;\n  Attachment?: ComponentType | undefined;\n};\n\nexport namespace MessagePrimitiveAttachments {\n  export type Props =\n    | {\n        /** @deprecated Use the children render function instead. */\n        components: MessageAttachmentsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each attachment. Receives the attachment. */\n        children: (value: { attachment: CompleteAttachment }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst getComponent = (\n  components: MessageAttachmentsComponentConfig | undefined,\n  attachment: CompleteAttachment,\n) => {\n  const type = attachment.type;\n  switch (type) {\n    case \"image\":\n      return components?.Image ?? components?.Attachment;\n    case \"document\":\n      return components?.Document ?? components?.Attachment;\n    case \"file\":\n      return components?.File ?? components?.Attachment;\n    default:\n      return components?.Attachment;\n  }\n};\n\nconst AttachmentComponent: FC<{\n  components: MessageAttachmentsComponentConfig | undefined;\n}> = ({ components }) => {\n  const attachment = useAuiState((s) => s.attachment);\n  if (!attachment) return null;\n\n  const Component = getComponent(components, attachment as CompleteAttachment);\n  if (!Component) return null;\n  return <Component />;\n};\n\nexport namespace MessagePrimitiveAttachmentByIndex {\n  export type Props = {\n    index: number;\n    components?: MessageAttachmentsComponentConfig;\n  };\n}\n\n/**\n * Renders a single attachment at the specified index within the current message.\n */\nexport const MessagePrimitiveAttachmentByIndex: FC<MessagePrimitiveAttachmentByIndex.Props> =\n  memo(\n    ({ index, components }) => {\n      return (\n        <MessageAttachmentByIndexProvider index={index}>\n          <AttachmentComponent components={components} />\n        </MessageAttachmentByIndexProvider>\n      );\n    },\n    (prev, next) =>\n      prev.index === next.index &&\n      prev.components?.Image === next.components?.Image &&\n      prev.components?.Document === next.components?.Document &&\n      prev.components?.File === next.components?.File &&\n      prev.components?.Attachment === next.components?.Attachment,\n  );\n\nMessagePrimitiveAttachmentByIndex.displayName =\n  \"MessagePrimitive.AttachmentByIndex\";\n\nconst MessagePrimitiveAttachmentsInner: FC<{\n  children: (value: { attachment: CompleteAttachment }) => ReactNode;\n}> = ({ children }) => {\n  const attachmentsCount = useAuiState((s) => {\n    if (s.message.role !== \"user\") return 0;\n    return (s.message.attachments ?? []).length;\n  });\n\n  return useMemo(\n    () =>\n      Array.from({ length: attachmentsCount }, (_, index) => (\n        <MessageAttachmentByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.message().attachment({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get attachment() {\n                  return getItem() as CompleteAttachment;\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </MessageAttachmentByIndexProvider>\n      )),\n    [attachmentsCount, children],\n  );\n};\n\nexport const MessagePrimitiveAttachments: FC<\n  MessagePrimitiveAttachments.Props\n> = ({ components, children }) => {\n  if (components) {\n    return (\n      <MessagePrimitiveAttachmentsInner>\n        {({ attachment }) => {\n          const Component = getComponent(components, attachment);\n          if (!Component) return null;\n          return <Component />;\n        }}\n      </MessagePrimitiveAttachmentsInner>\n    );\n  }\n  return (\n    <MessagePrimitiveAttachmentsInner>\n      {children}\n    </MessagePrimitiveAttachmentsInner>\n  );\n};\n\nMessagePrimitiveAttachments.displayName = \"MessagePrimitive.Attachments\";\n"],"mappings":";;;;;AAgCA,MAAM,gBACJ,YACA,eACG;CAEH,QADa,WAAW,MACxB;EACE,KAAK,SACH,OAAO,YAAY,SAAS,YAAY;EAC1C,KAAK,YACH,OAAO,YAAY,YAAY,YAAY;EAC7C,KAAK,QACH,OAAO,YAAY,QAAQ,YAAY;EACzC,SACE,OAAO,YAAY;CACvB;AACF;AAEA,MAAM,uBAEA,EAAE,iBAAiB;CACvB,MAAM,aAAa,aAAa,MAAM,EAAE,UAAU;CAClD,IAAI,CAAC,YAAY,OAAO;CAExB,MAAM,YAAY,aAAa,YAAY,UAAgC;CAC3E,IAAI,CAAC,WAAW,OAAO;CACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;AACrB;;;;AAYA,MAAa,oCACX,MACG,EAAE,OAAO,iBAAiB;CACzB,OACE,oBAAC,kCAAD;EAAyC;YACvC,oBAAC,qBAAD,EAAiC,WAAa,CAAA;CACd,CAAA;AAEtC,IACC,MAAM,SACL,KAAK,UAAU,KAAK,SACpB,KAAK,YAAY,UAAU,KAAK,YAAY,SAC5C,KAAK,YAAY,aAAa,KAAK,YAAY,YAC/C,KAAK,YAAY,SAAS,KAAK,YAAY,QAC3C,KAAK,YAAY,eAAe,KAAK,YAAY,UACrD;AAEF,kCAAkC,cAChC;AAEF,MAAM,oCAEA,EAAE,eAAe;CACrB,MAAM,mBAAmB,aAAa,MAAM;EAC1C,IAAI,EAAE,QAAQ,SAAS,QAAQ,OAAO;EACtC,QAAQ,EAAE,QAAQ,eAAe,CAAC,EAAA,CAAG;CACvC,CAAC;CAED,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,iBAAiB,IAAI,GAAG,UAC3C,oBAAC,kCAAD;EAAqD;YACnD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,QAAQ,CAAC,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS;cAG7C,YACA,SAAS,EACP,IAAI,aAAa;IACf,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACI,GAdK,KAcL,CACnC,GACH,CAAC,kBAAkB,QAAQ,CAC7B;AACF;AAEA,MAAa,+BAER,EAAE,YAAY,eAAe;CAChC,IAAI,YACF,OACE,oBAAC,kCAAD,EAAA,WACI,EAAE,iBAAiB;EACnB,MAAM,YAAY,aAAa,YAAY,UAAU;EACrD,IAAI,CAAC,WAAW,OAAO;EACvB,OAAO,oBAAC,WAAD,CAAY,CAAA;CACrB,EACgC,CAAA;CAGtC,OACE,oBAAC,kCAAD,EACG,SAC+B,CAAA;AAEtC;AAEA,4BAA4B,cAAc"}

package/dist/react/primitives/message/MessageGroupedParts.js

@@ -1,7 +1,7 @@
 "use client";
 import { MessagePartChildren } from "./MessageParts.js";
 import { GROUPBY_MEMO_KEY, buildGroupTree } from "../../utils/groupParts.js";
-import { Fragment, useMemo } from "react";
+import { Fragment, useMemo } from "@assistant-ui/tap/react-shim";
 import { useAuiState } from "@assistant-ui/store";
 import { Fragment as Fragment$1, jsx, jsxs } from "react/jsx-runtime";
 import { useShallow } from "zustand/shallow";

package/dist/react/primitives/message/MessageGroupedParts.js.map

@@ -1 +1 @@
-{"version":3,"file":"MessageGroupedParts.js","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallow } from \"zustand/shallow\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport {\n  buildGroupTree,\n  GROUPBY_MEMO_KEY,\n  type GroupByContext,\n  type GroupNode,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    readonly indices: readonly number[];\n  };\n\n  /**\n   * Synthetic trailing slot for a streaming/loading affordance (a\n   * \"thinking…\" dot, etc.). Surfaced through the same `{ part }` channel\n   * as groups and leaf parts so a single `switch (part.type)` renders it\n   * via `case \"indicator\"`.\n   *\n   * It is only ever emitted while the message is running, so its presence\n   * alone means \"render your loading UI here\" — there's no `status` to\n   * branch on.\n   */\n  export type IndicatorPart = {\n    readonly type: \"indicator\";\n  };\n\n  /**\n   * When to emit the synthetic {@link IndicatorPart}. It is **only** emitted\n   * while the message is running (streaming); the mode further restricts\n   * which running states qualify:\n   * - `\"never\"` — never.\n   * - `\"empty\"` — only when the message has no parts yet.\n   * - `\"no-text\"` (default) — when the last part isn't `text`/`reasoning`\n   *   (e.g. it ended on a tool call, so the assistant likely isn't done).\n   * - `\"always\"` — whenever the message is running, regardless of parts.\n   */\n  export type IndicatorMode = \"never\" | \"empty\" | \"no-text\" | \"always\";\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`), a single enriched leaf part, or the synthetic\n     * {@link IndicatorPart} (`type: \"indicator\"`). Use one switch over\n     * `part.type` to handle all three.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState | IndicatorPart;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to a group-key path. Adjacent parts that share a\n     * prefix coalesce into the same group. Return `[]` (or `null`) to\n     * leave a part ungrouped.\n     *\n     * Group keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can tell groups apart from real part types.\n     *\n     * **Prefer {@link groupPartByType}** for the common case of mapping by\n     * `part.type` — it ships a stable memo fingerprint so the tree\n     * survives unrelated re-renders. Use an inline function only when\n     * the helper isn't expressive enough (e.g. branching on\n     * `part.toolName` or part metadata).\n     *\n     * The second argument is a {@link GroupByContext} carrying the tool-UI\n     * registry, for grouping that depends on it (e.g. standalone tool calls).\n     *\n     * @example\n     * ```tsx\n     * import { groupPartByType } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.GroupedParts\n     *   groupBy={groupPartByType({\n     *     reasoning: [\"group-thought\", \"group-reasoning\"],\n     *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n     *   })}\n     * >\n     * ```\n     */\n    readonly groupBy: (\n      part: PartState,\n      context: GroupByContext,\n    ) => readonly TKey[] | null;\n\n    /**\n     * Controls emission of the synthetic {@link IndicatorPart} — a\n     * trailing `{ part: { type: \"indicator\", status } }` render call you\n     * handle with `case \"indicator\"` to show loading/status UI.\n     *\n     * @default \"no-text\"\n     * @see IndicatorMode\n     */\n    readonly indicator?: IndicatorMode;\n\n    /**\n     * Render function called once per group node, once per leaf part, and\n     * (when the `indicator` condition is met) once for the trailing\n     * {@link IndicatorPart}. Switch on `part.type`: `\"group-…\"` cases wrap\n     * `children`; real part types (`\"text\"`, `\"tool-call\"`, …) render the\n     * part directly; `\"indicator\"` renders status/loading UI.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `respondToApproval`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst COMPLETE_STATUS: MessagePartStatus = Object.freeze({ type: \"complete\" });\n\nconst shouldShowIndicator = (\n  mode: MessagePrimitiveGroupedParts.IndicatorMode,\n  parts: readonly PartState[],\n  isRunning: boolean,\n): boolean => {\n  // The indicator is a streaming affordance — never show it on a settled\n  // message, whatever the mode.\n  if (!isRunning) return false;\n\n  switch (mode) {\n    case \"never\":\n      return false;\n    case \"always\":\n      return true;\n    case \"empty\":\n      return parts.length === 0;\n    case \"no-text\": {\n      const last = parts[parts.length - 1];\n      return (\n        last !== undefined && last.type !== \"text\" && last.type !== \"reasoning\"\n      );\n    }\n  }\n};\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by absolute part index, not structural nodeKey — prevents zombie fiber subscriptions when parts reshape (#4051).\n    return (\n      <MessagePartChildren key={`part-${node.index}`} index={node.index}>\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const status = parts[node.indices.at(-1)!]?.status ?? COMPLETE_STATUS;\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={groupPartByType({\n *     reasoning: [\"group-thought\", \"group-reasoning\"],\n *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n *   })}\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       case \"indicator\":       return <LoadingDots />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  indicator = \"no-text\",\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallow((s) => s.message.parts));\n  // Handed to `groupBy` as its `context` argument (see GroupByContext).\n  const toolUIs = useAuiState((s) => s.tools.toolUIs);\n  // Subscribe to a boolean, not the status object: the tree only needs to\n  // re-render when running-ness flips, and `\"never\"` opts out entirely.\n  const isRunning = useAuiState((s) =>\n    indicator === \"never\" ? false : s.message.status?.type === \"running\",\n  );\n\n  // Helpers like `groupPartByType` tag the function with `GROUPBY_MEMO_KEY`\n  // (a stable string fingerprint of the helper config). When present,\n  // memo on `[parts, memoKey]` so the tree survives unrelated renders.\n  // For inline `groupBy`, fall back to recomputing each render — O(n)\n  // and cheap.\n  const memoKey = (groupBy as { [GROUPBY_MEMO_KEY]?: string })[\n    GROUPBY_MEMO_KEY\n  ];\n  const memoDep = memoKey ?? groupBy;\n  const tree = useMemo(() => {\n    const context: GroupByContext = { toolUIs };\n    return buildGroupTree(parts.map((part) => groupBy(part, context) ?? []));\n    // oxlint-disable-next-line tap-hooks/exhaustive-deps -- groupBy is captured via memoDep (either its identity or the helper's memoKey fingerprint); listing it directly would defeat the helper-tagged memo path\n  }, [parts, memoDep, toolUIs]);\n\n  return (\n    <>\n      {tree.map((node) => renderNode(node, parts, children))}\n      {shouldShowIndicator(indicator, parts, isRunning) &&\n        children({\n          part: { type: \"indicator\" },\n          children: <PartChildrenSentinel />,\n        })}\n    </>\n  );\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;AAsIA,MAAM,kBAAqC,OAAO,OAAO,EAAE,MAAM,WAAW,CAAC;AAE7E,MAAM,uBACJ,MACA,OACA,cACY;CAGZ,IAAI,CAAC,WAAW,OAAO;CAEvB,QAAQ,MAAR;EACE,KAAK,SACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,SACH,OAAO,MAAM,WAAW;EAC1B,KAAK,WAAW;GACd,MAAM,OAAO,MAAM,MAAM,SAAS;GAClC,OACE,SAAS,KAAA,KAAa,KAAK,SAAS,UAAU,KAAK,SAAS;EAEhE;CACF;AACF;;;;;;;AAQA,MAAM,6BAAiC;CACrC,MAAM,IAAI,MACR,8LAGF;AACF;AAEA,MAAM,cACJ,MACA,OACA,WACc;CACd,IAAI,KAAK,SAAS,QAEhB,OACE,oBAAC,qBAAD;EAAgD,OAAO,KAAK;aACxD,EAAE,WAAW,OAAO;GAAE;GAAM,UAAU,oBAAC,sBAAD,CAAuB,CAAA;EAAE,CAAC;CAC/C,GAFK,QAAQ,KAAK,OAElB;CAIzB,MAAM,SAAS,MAAM,KAAK,QAAQ,GAAG,EAAE,IAAK,UAAU;CAOtD,OACE,oBAAC,UAAD,EAAA,UACG,OAAO;EACN,MAAM;GARV,MAAM,KAAK;GACX;GACA,SAAS,KAAK;EAMI;EACd,UACE,oBAAA,YAAA,EAAA,UAAG,KAAK,SAAS,KAAK,UAAU,WAAW,OAAO,OAAO,MAAM,CAAC,EAAI,CAAA;CAExE,CAAC,EACO,GAPK,KAAK,OAOV;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAa,gCAAgE,EAC3E,SACA,YAAY,WACZ,eACyD;CACzD,MAAM,QAAQ,YAAY,YAAY,MAAM,EAAE,QAAQ,KAAK,CAAC;CAE5D,MAAM,UAAU,aAAa,MAAM,EAAE,MAAM,OAAO;CAGlD,MAAM,YAAY,aAAa,MAC7B,cAAc,UAAU,QAAQ,EAAE,QAAQ,QAAQ,SAAS,SAC7D;CAiBA,OACE,qBAAA,YAAA,EAAA,UAAA,CAPW,cAAc;EACzB,MAAM,UAA0B,EAAE,QAAQ;EAC1C,OAAO,eAAe,MAAM,KAAK,SAAS,QAAQ,MAAM,OAAO,KAAK,CAAC,CAAC,CAAC;CAEzE,GAAG;EAAC;EARa,QACf,qBAEyB;EAKP;CAAO,CAInB,EAAE,KAAK,SAAS,WAAW,MAAM,OAAO,QAAQ,CAAC,GACpD,oBAAoB,WAAW,OAAO,SAAS,KAC9C,SAAS;EACP,MAAM,EAAE,MAAM,YAAY;EAC1B,UAAU,oBAAC,sBAAD,CAAuB,CAAA;CACnC,CAAC,CACH,EAAA,CAAA;AAEN;AAEA,6BAA6B,cAAc"}
+{"version":3,"file":"MessageGroupedParts.js","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallow } from \"zustand/shallow\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport {\n  buildGroupTree,\n  GROUPBY_MEMO_KEY,\n  type GroupByContext,\n  type GroupNode,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    readonly indices: readonly number[];\n  };\n\n  /**\n   * Synthetic trailing slot for a streaming/loading affordance (a\n   * \"thinking…\" dot, etc.). Surfaced through the same `{ part }` channel\n   * as groups and leaf parts so a single `switch (part.type)` renders it\n   * via `case \"indicator\"`.\n   *\n   * It is only ever emitted while the message is running, so its presence\n   * alone means \"render your loading UI here\" — there's no `status` to\n   * branch on.\n   */\n  export type IndicatorPart = {\n    readonly type: \"indicator\";\n  };\n\n  /**\n   * When to emit the synthetic {@link IndicatorPart}. It is **only** emitted\n   * while the message is running (streaming); the mode further restricts\n   * which running states qualify:\n   * - `\"never\"` — never.\n   * - `\"empty\"` — only when the message has no parts yet.\n   * - `\"no-text\"` (default) — when the last part isn't `text`/`reasoning`\n   *   (e.g. it ended on a tool call, so the assistant likely isn't done).\n   * - `\"always\"` — whenever the message is running, regardless of parts.\n   */\n  export type IndicatorMode = \"never\" | \"empty\" | \"no-text\" | \"always\";\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`), a single enriched leaf part, or the synthetic\n     * {@link IndicatorPart} (`type: \"indicator\"`). Use one switch over\n     * `part.type` to handle all three.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState | IndicatorPart;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to a group-key path. Adjacent parts that share a\n     * prefix coalesce into the same group. Return `[]` (or `null`) to\n     * leave a part ungrouped.\n     *\n     * Group keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can tell groups apart from real part types.\n     *\n     * **Prefer {@link groupPartByType}** for the common case of mapping by\n     * `part.type` — it ships a stable memo fingerprint so the tree\n     * survives unrelated re-renders. Use an inline function only when\n     * the helper isn't expressive enough (e.g. branching on\n     * `part.toolName` or part metadata).\n     *\n     * The second argument is a {@link GroupByContext} carrying the tool-UI\n     * registry, for grouping that depends on it (e.g. standalone tool calls).\n     *\n     * @example\n     * ```tsx\n     * import { groupPartByType } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.GroupedParts\n     *   groupBy={groupPartByType({\n     *     reasoning: [\"group-thought\", \"group-reasoning\"],\n     *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n     *   })}\n     * >\n     * ```\n     */\n    readonly groupBy: (\n      part: PartState,\n      context: GroupByContext,\n    ) => readonly TKey[] | null;\n\n    /**\n     * Controls emission of the synthetic {@link IndicatorPart} — a\n     * trailing `{ part: { type: \"indicator\", status } }` render call you\n     * handle with `case \"indicator\"` to show loading/status UI.\n     *\n     * @default \"no-text\"\n     * @see IndicatorMode\n     */\n    readonly indicator?: IndicatorMode;\n\n    /**\n     * Render function called once per group node, once per leaf part, and\n     * (when the `indicator` condition is met) once for the trailing\n     * {@link IndicatorPart}. Switch on `part.type`: `\"group-…\"` cases wrap\n     * `children`; real part types (`\"text\"`, `\"tool-call\"`, …) render the\n     * part directly; `\"indicator\"` renders status/loading UI.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `respondToApproval`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst COMPLETE_STATUS: MessagePartStatus = Object.freeze({ type: \"complete\" });\n\nconst shouldShowIndicator = (\n  mode: MessagePrimitiveGroupedParts.IndicatorMode,\n  parts: readonly PartState[],\n  isRunning: boolean,\n): boolean => {\n  // The indicator is a streaming affordance — never show it on a settled\n  // message, whatever the mode.\n  if (!isRunning) return false;\n\n  switch (mode) {\n    case \"never\":\n      return false;\n    case \"always\":\n      return true;\n    case \"empty\":\n      return parts.length === 0;\n    case \"no-text\": {\n      const last = parts[parts.length - 1];\n      return (\n        last !== undefined && last.type !== \"text\" && last.type !== \"reasoning\"\n      );\n    }\n  }\n};\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by absolute part index, not structural nodeKey — prevents zombie fiber subscriptions when parts reshape (#4051).\n    return (\n      <MessagePartChildren key={`part-${node.index}`} index={node.index}>\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const status = parts[node.indices.at(-1)!]?.status ?? COMPLETE_STATUS;\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={groupPartByType({\n *     reasoning: [\"group-thought\", \"group-reasoning\"],\n *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n *   })}\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       case \"indicator\":       return <LoadingDots />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  indicator = \"no-text\",\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallow((s) => s.message.parts));\n  // Handed to `groupBy` as its `context` argument (see GroupByContext).\n  const toolUIs = useAuiState((s) => s.tools.toolUIs);\n  // Subscribe to a boolean, not the status object: the tree only needs to\n  // re-render when running-ness flips, and `\"never\"` opts out entirely.\n  const isRunning = useAuiState((s) =>\n    indicator === \"never\" ? false : s.message.status?.type === \"running\",\n  );\n\n  // Helpers like `groupPartByType` tag the function with `GROUPBY_MEMO_KEY`\n  // (a stable string fingerprint of the helper config). When present,\n  // memo on `[parts, memoKey]` so the tree survives unrelated renders.\n  // For inline `groupBy`, fall back to recomputing each render — O(n)\n  // and cheap.\n  const memoKey = (groupBy as { [GROUPBY_MEMO_KEY]?: string })[\n    GROUPBY_MEMO_KEY\n  ];\n  const memoDep = memoKey ?? groupBy;\n  const tree = useMemo(() => {\n    const context: GroupByContext = { toolUIs };\n    return buildGroupTree(parts.map((part) => groupBy(part, context) ?? []));\n    // oxlint-disable-next-line react/exhaustive-deps -- groupBy is captured via memoDep (either its identity or the helper's memoKey fingerprint); listing it directly would defeat the helper-tagged memo path\n  }, [parts, memoDep, toolUIs]);\n\n  return (\n    <>\n      {tree.map((node) => renderNode(node, parts, children))}\n      {shouldShowIndicator(indicator, parts, isRunning) &&\n        children({\n          part: { type: \"indicator\" },\n          children: <PartChildrenSentinel />,\n        })}\n    </>\n  );\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;AAsIA,MAAM,kBAAqC,OAAO,OAAO,EAAE,MAAM,WAAW,CAAC;AAE7E,MAAM,uBACJ,MACA,OACA,cACY;CAGZ,IAAI,CAAC,WAAW,OAAO;CAEvB,QAAQ,MAAR;EACE,KAAK,SACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,SACH,OAAO,MAAM,WAAW;EAC1B,KAAK,WAAW;GACd,MAAM,OAAO,MAAM,MAAM,SAAS;GAClC,OACE,SAAS,KAAA,KAAa,KAAK,SAAS,UAAU,KAAK,SAAS;EAEhE;CACF;AACF;;;;;;;AAQA,MAAM,6BAAiC;CACrC,MAAM,IAAI,MACR,8LAGF;AACF;AAEA,MAAM,cACJ,MACA,OACA,WACc;CACd,IAAI,KAAK,SAAS,QAEhB,OACE,oBAAC,qBAAD;EAAgD,OAAO,KAAK;aACxD,EAAE,WAAW,OAAO;GAAE;GAAM,UAAU,oBAAC,sBAAD,CAAuB,CAAA;EAAE,CAAC;CAC/C,GAFK,QAAQ,KAAK,OAElB;CAIzB,MAAM,SAAS,MAAM,KAAK,QAAQ,GAAG,EAAE,EAAG,EAAE,UAAU;CAOtD,OACE,oBAAC,UAAD,EAAA,UACG,OAAO;EACN,MAAM;GARV,MAAM,KAAK;GACX;GACA,SAAS,KAAK;EAMI;EACd,UACE,oBAAA,YAAA,EAAA,UAAG,KAAK,SAAS,KAAK,UAAU,WAAW,OAAO,OAAO,MAAM,CAAC,EAAI,CAAA;CAExE,CAAC,EACO,GAPK,KAAK,OAOV;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAa,gCAAgE,EAC3E,SACA,YAAY,WACZ,eACyD;CACzD,MAAM,QAAQ,YAAY,YAAY,MAAM,EAAE,QAAQ,KAAK,CAAC;CAE5D,MAAM,UAAU,aAAa,MAAM,EAAE,MAAM,OAAO;CAGlD,MAAM,YAAY,aAAa,MAC7B,cAAc,UAAU,QAAQ,EAAE,QAAQ,QAAQ,SAAS,SAC7D;CAiBA,OACE,qBAAA,YAAA,EAAA,UAAA,CAPW,cAAc;EACzB,MAAM,UAA0B,EAAE,QAAQ;EAC1C,OAAO,eAAe,MAAM,KAAK,SAAS,QAAQ,MAAM,OAAO,KAAK,CAAC,CAAC,CAAC;CAEzE,GAAG;EAAC;EARa,QACf,qBAEyB;EAKP;CAAO,CAInB,CAAC,CAAC,KAAK,SAAS,WAAW,MAAM,OAAO,QAAQ,CAAC,GACpD,oBAAoB,WAAW,OAAO,SAAS,KAC9C,SAAS;EACP,MAAM,EAAE,MAAM,YAAY;EAC1B,UAAU,oBAAC,sBAAD,CAAuB,CAAA;CACnC,CAAC,CACH,EAAA,CAAA;AAEN;AAEA,6BAA6B,cAAc"}

package/dist/react/primitives/message/MessageParts.d.ts

@@ -1,5 +1,5 @@
-import { DataMessagePartComponent, EmptyMessagePartComponent, FileMessagePartComponent, GenerativeUIComponentRegistry, ImageMessagePartComponent, QuoteMessagePartComponent, ReasoningGroupComponent, ReasoningMessagePartComponent, SourceMessagePartComponent, TextMessagePartComponent, ToolCallMessagePartComponent, ToolCallMessagePartProps, Unstable_AudioMessagePartComponent } from "../../types/MessagePartComponentTypes.js";
 import { PartState } from "../../../store/scopes/part.js";
+import { DataMessagePartComponent, EmptyMessagePartComponent, FileMessagePartComponent, GenerativeUIComponentRegistry, ImageMessagePartComponent, QuoteMessagePartComponent, ReasoningGroupComponent, ReasoningMessagePartComponent, SourceMessagePartComponent, TextMessagePartComponent, ToolCallMessagePartComponent, ToolCallMessagePartProps, Unstable_AudioMessagePartComponent } from "../../types/MessagePartComponentTypes.js";
 import { ComponentType, FC, PropsWithChildren, ReactNode } from "react";
 
 //#region src/react/primitives/message/MessageParts.d.ts

package/dist/react/primitives/message/MessageParts.js

@@ -4,7 +4,7 @@ import { TextMessagePartProvider } from "../../providers/TextMessagePartProvider
 import { ChainOfThoughtByIndicesProvider } from "../../providers/ChainOfThoughtByIndicesProvider.js";
 import { getMessageQuote } from "../../utils/getMessageQuote.js";
 import { GenerativeUIRender } from "../generativeUI/GenerativeUI.js";
-import { memo, useMemo } from "react";
+import { memo, useMemo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAui, useAuiState } from "@assistant-ui/store";
 import { Fragment as Fragment$1, jsx, jsxs } from "react/jsx-runtime";
 import { useShallow } from "zustand/shallow";