@assistant-ui/react-native 0.1.40 → 0.1.41

package/dist/primitives/message/MessageContent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageContent.d.ts","names":[],"sources":["../../../src/primitives/message/MessageContent.tsx"],"mappings":";;;KAcK,qBAAqB,wBAAwB;KAEtC;EACV,cAAc;IACZ,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,kBAAkB;IAChB,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,eAAe;IACb,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,mBAAmB;IACjB,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,gBAAgB;IACd,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,cAAc;IACZ,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,cAAc;IACZ,MAAM,QAAQ;MAAsB;;IACpC;QACI;;cAgEK,mBAAkB,YAAA,gBAAA,aAAA,iBAAA,cAAA,YAAA,cAQ5B,wCAAmB,IAAA"}
+{"version":3,"file":"MessageContent.d.ts","names":[],"sources":["../../../src/primitives/message/MessageContent.tsx"],"mappings":";;;KAaK,qBAAqB,wBAAwB;KAGtC;EACV,cAAc;IACZ,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,kBAAkB;IAChB,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,eAAe;IACb,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,mBAAmB;IACjB,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,gBAAgB;IACd,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,cAAc;IACZ,MAAM,QAAQ;MAAsB;;IACpC;QACI;EACN,cAAc;IACZ,MAAM,QAAQ;MAAsB;;IACpC;QACI;;cAsEK,mBAAkB,YAAA,gBAAA,aAAA,iBAAA,cAAA,YAAA,cAQ5B,wCAAmB,IAAA"}

package/dist/primitives/message/MessageContent.js

@@ -191,7 +191,7 @@ const MessageContent = (t0) => {
 	return t2;
 };
 function _temp(s) {
-	return s.message.content;
+	return s.message.parts;
 }
 //#endregion
 export { MessageContent };

package/dist/primitives/message/MessageContent.js.map

@@ -1 +1 @@
-{"version":3,"file":"MessageContent.js","names":["ReactElement","Fragment","useMemo","Text","ThreadUserMessagePart","ThreadAssistantMessagePart","ToolCallMessagePart","DataMessagePart","useAui","useAuiState","ToolCallMessagePartProps","DataMessagePartProps","MessageContentPart","MessageContentProps","renderText","props","part","Extract","type","index","renderToolCall","renderImage","renderReasoning","renderSource","renderFile","renderData","DefaultTextRenderer","t0","$","_c","t1","text","ToolUIDisplay","Fallback","aui","toolName","s","tools","toolUIs","render","Render","t2","message","partMethods","t3","t4","addToolResult","respondToToolApproval","resumeToolCall","DataUIDisplay","name","renders","dataRenderers","renderers","Array","isArray","MessageContent","content","_temp","key","map"],"sources":["../../../src/primitives/message/MessageContent.tsx"],"sourcesContent":["import { type ReactElement, Fragment, useMemo } from \"react\";\nimport { Text } from \"react-native\";\nimport type {\n  ThreadUserMessagePart,\n  ThreadAssistantMessagePart,\n  ToolCallMessagePart,\n  DataMessagePart,\n} from \"@assistant-ui/core\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type {\n  ToolCallMessagePartProps,\n  DataMessagePartProps,\n} from \"@assistant-ui/core/react\";\n\ntype MessageContentPart = ThreadUserMessagePart | ThreadAssistantMessagePart;\n\nexport type MessageContentProps = {\n  renderText?: (props: {\n    part: Extract<MessageContentPart, { type: \"text\" }>;\n    index: number;\n  }) => ReactElement;\n  renderToolCall?: (props: {\n    part: Extract<MessageContentPart, { type: \"tool-call\" }>;\n    index: number;\n  }) => ReactElement;\n  renderImage?: (props: {\n    part: Extract<MessageContentPart, { type: \"image\" }>;\n    index: number;\n  }) => ReactElement;\n  renderReasoning?: (props: {\n    part: Extract<MessageContentPart, { type: \"reasoning\" }>;\n    index: number;\n  }) => ReactElement;\n  renderSource?: (props: {\n    part: Extract<MessageContentPart, { type: \"source\" }>;\n    index: number;\n  }) => ReactElement;\n  renderFile?: (props: {\n    part: Extract<MessageContentPart, { type: \"file\" }>;\n    index: number;\n  }) => ReactElement;\n  renderData?: (props: {\n    part: Extract<MessageContentPart, { type: \"data\" }>;\n    index: number;\n  }) => ReactElement;\n};\n\nconst DefaultTextRenderer = ({\n  part,\n}: {\n  part: Extract<MessageContentPart, { type: \"text\" }>;\n}) => {\n  return <Text>{part.text}</Text>;\n};\n\nconst ToolUIDisplay = ({\n  Fallback,\n  part,\n  index,\n}: {\n  Fallback:\n    | ((props: { part: ToolCallMessagePart; index: number }) => ReactElement)\n    | undefined;\n  part: ToolCallMessagePart;\n  index: number;\n}) => {\n  const aui = useAui();\n  const Render = useAuiState(\n    (s) => s.tools.toolUIs[part.toolName]?.[0]?.render,\n  );\n\n  const partMethods = useMemo(() => aui.message.part({ index }), [aui, index]);\n\n  if (Render) {\n    return (\n      <Render\n        {...(part as ToolCallMessagePartProps)}\n        addResult={partMethods.addToolResult}\n        resume={partMethods.resumeToolCall}\n        respondToApproval={partMethods.respondToToolApproval}\n      />\n    );\n  }\n  if (Fallback) return <Fallback part={part} index={index} />;\n  return null;\n};\n\nconst DataUIDisplay = ({\n  Fallback,\n  part,\n  index,\n}: {\n  Fallback:\n    | ((props: { part: DataMessagePart; index: number }) => ReactElement)\n    | undefined;\n  part: DataMessagePart;\n  index: number;\n}) => {\n  const Render = useAuiState((s) => {\n    const renders = s.dataRenderers.renderers[part.name];\n    if (Array.isArray(renders)) return renders[0];\n    return renders;\n  });\n  if (Render) return <Render {...(part as DataMessagePartProps)} />;\n  if (Fallback) return <Fallback part={part} index={index} />;\n  return null;\n};\n\nexport const MessageContent = ({\n  renderText,\n  renderToolCall,\n  renderImage,\n  renderReasoning,\n  renderSource,\n  renderFile,\n  renderData,\n}: MessageContentProps) => {\n  const content = useAuiState((s) => s.message.content);\n\n  return (\n    <>\n      {content.map((part, index) => {\n        const key = `${part.type}-${index}`;\n        switch (part.type) {\n          case \"text\":\n            return (\n              <Fragment key={key}>\n                {renderText ? (\n                  renderText({ part, index })\n                ) : (\n                  <DefaultTextRenderer part={part} />\n                )}\n              </Fragment>\n            );\n          case \"tool-call\":\n            return (\n              <Fragment key={key}>\n                <ToolUIDisplay\n                  Fallback={renderToolCall}\n                  part={part}\n                  index={index}\n                />\n              </Fragment>\n            );\n          case \"image\":\n            if (!renderImage) return null;\n            return (\n              <Fragment key={key}>{renderImage({ part, index })}</Fragment>\n            );\n          case \"reasoning\":\n            if (!renderReasoning) return null;\n            return (\n              <Fragment key={key}>{renderReasoning({ part, index })}</Fragment>\n            );\n          case \"source\":\n            if (!renderSource) return null;\n            return (\n              <Fragment key={key}>{renderSource({ part, index })}</Fragment>\n            );\n          case \"file\":\n            if (!renderFile) return null;\n            return <Fragment key={key}>{renderFile({ part, index })}</Fragment>;\n          case \"data\":\n            return (\n              <Fragment key={key}>\n                <DataUIDisplay\n                  Fallback={renderData}\n                  part={part}\n                  index={index}\n                />\n              </Fragment>\n            );\n          default:\n            return null;\n        }\n      })}\n    </>\n  );\n};\n"],"mappings":";;;;;;AA+CA,MAAM0B,uBAAsBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAb,SAAAW;CAI5B,IAAAG;CAAA,IAAAF,EAAA,OAAAZ,KAAAe,MAAA;EACQD,KAAA,oBAAC,MAAD,EAAA,UAAOd,KAAIe,KAAN,CAAA;EAAmBH,EAAA,KAAAZ,KAAAe;EAAAH,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,OAAxBE;AAAwB;AAGjC,MAAME,iBAAgBL,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAI,UAAAjB,MAAAG,UAAAQ;CAWrB,MAAAO,MAAY1B,OAAO;CAAE,IAAAsB;CAAA,IAAAF,EAAA,OAAAZ,KAAAmB,UAAA;EAEnBL,MAAAM,MAAOA,EAACC,MAAMC,QAAStB,KAAImB,SAAe,GAAA,EAAQ,EAAAI;EAAAX,EAAA,KAAAZ,KAAAmB;EAAAP,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CADpD,MAAAY,SAAe/B,YACbqB,EACF;CAAE,IAAAW;CAAA,IAAAb,EAAA,OAAAM,IAAAQ,WAAAd,EAAA,OAAAT,OAAA;EAEgCsB,KAAAP,IAAGQ,QAAQ1B,KAAM,EAAAG,MAAQ,CAAC;EAACS,EAAA,KAAAM,IAAAQ;EAAAd,EAAA,KAAAT;EAAAS,EAAA,KAAAa;CAAA,OAAAA,KAAAb,EAAA;CAA7D,MAAAe,cAAkCF;CAElC,IAAID,QAAM;EAGC,MAAAI,KAAA5B;EAAgC,IAAA6B;EAAA,IAAAjB,EAAA,OAAAY,UAAAZ,EAAA,OAAAe,YAAAG,iBAAAlB,EAAA,OAAAe,YAAAI,yBAAAnB,EAAA,OAAAe,YAAAK,kBAAApB,EAAA,OAAAgB,IAAA;GADvCC,KAAA,oBAAC,QAAD;IAAO,GACAD;IACM,WAAAD,YAAWG;IACd,QAAAH,YAAWK;IACA,mBAAAL,YAAWI;GAAsB,CAAA;GACpDnB,EAAA,KAAAY;GAAAZ,EAAA,KAAAe,YAAAG;GAAAlB,EAAA,KAAAe,YAAAI;GAAAnB,EAAA,KAAAe,YAAAK;GAAApB,EAAA,KAAAgB;GAAAhB,EAAA,MAAAiB;EAAA,OAAAA,KAAAjB,EAAA;EAAA,OALFiB;CAKE;CAGN,IAAIZ,UAAQ;EAAA,IAAAW;EAAA,IAAAhB,EAAA,QAAAK,YAAAL,EAAA,QAAAT,SAAAS,EAAA,QAAAZ,MAAA;GAAS4B,KAAA,oBAAC,UAAD;IAAgB5B;IAAaG;GAAK,CAAA;GAAIS,EAAA,MAAAK;GAAAL,EAAA,MAAAT;GAAAS,EAAA,MAAAZ;GAAAY,EAAA,MAAAgB;EAAA,OAAAA,KAAAhB,EAAA;EAAA,OAAtCgB;CAAsC;CAAC,OACrD;AAAI;AAGb,MAAMK,iBAAgBtB,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAI,UAAAjB,MAAAG,UAAAQ;CAUtB,IAAAG;CAAA,IAAAF,EAAA,OAAAZ,KAAAkC,MAAA;EAC4BpB,MAAAM,MAAA;GACzB,MAAAe,UAAgBf,EAACgB,cAAcC,UAAWrC,KAAIkC;GAC9C,IAAII,MAAKC,QAASJ,OAAO,GAAC,OAASA,QAAO;GAAI,OACvCA;EAAO;EACfvB,EAAA,KAAAZ,KAAAkC;EAAAtB,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAJD,MAAAY,SAAe/B,YAAYqB,EAI1B;CACD,IAAIU,QAAM;EAAsB,MAAAC,KAAAzB;EAA4B,IAAA4B;EAAA,IAAAhB,EAAA,OAAAY,UAAAZ,EAAA,OAAAa,IAAA;GAAzCG,KAAA,oBAAC,QAAD,EAAO,GAAMH,GAA4B,CAAA;GAAKb,EAAA,KAAAY;GAAAZ,EAAA,KAAAa;GAAAb,EAAA,KAAAgB;EAAA,OAAAA,KAAAhB,EAAA;EAAA,OAA9CgB;CAA8C;CACjE,IAAIX,UAAQ;EAAA,IAAAQ;EAAA,IAAAb,EAAA,OAAAK,YAAAL,EAAA,OAAAT,SAAAS,EAAA,OAAAZ,MAAA;GAASyB,KAAA,oBAAC,UAAD;IAAgBzB;IAAaG;GAAK,CAAA;GAAIS,EAAA,KAAAK;GAAAL,EAAA,KAAAT;GAAAS,EAAA,KAAAZ;GAAAY,EAAA,KAAAa;EAAA,OAAAA,KAAAb,EAAA;EAAA,OAAtCa;CAAsC;CAAC,OACrD;AAAI;AAGb,MAAae,kBAAiB7B,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAf,YAAAM,gBAAAC,aAAAC,iBAAAC,cAAAC,YAAAC,eAAAE;CAS7B,MAAA8B,UAAgBhD,YAAYiD,KAAwB;CAAE,IAAA5B;CAAA,IAAAF,EAAA,OAAA6B,WAAA7B,EAAA,OAAAH,cAAAG,EAAA,OAAAJ,cAAAI,EAAA,OAAAP,eAAAO,EAAA,OAAAN,mBAAAM,EAAA,OAAAL,gBAAAK,EAAA,OAAAd,cAAAc,EAAA,OAAAR,gBAAA;EAAA,IAAAqB;EAAA,IAAAb,EAAA,OAAAH,cAAAG,EAAA,QAAAJ,cAAAI,EAAA,QAAAP,eAAAO,EAAA,QAAAN,mBAAAM,EAAA,QAAAL,gBAAAK,EAAA,QAAAd,cAAAc,EAAA,QAAAR,gBAAA;GAIrCqB,MAAAzB,MAAAG,UAAA;IACX,MAAAwC,MAAY,GAAG3C,KAAIE,KAAK,GAAIC;IAC5B,QAAQH,KAAIE,MAAZ;KAAiB,KACV,QAAM,OAEP,oBAAC,YAAD,EAAA,UACGJ,aACCA,WAAW;MAAAE;MAAAG;KAAc,CAG3B,IADE,oBAAC,qBAAD,EAA2BH,KAAI,CAAA,EAJ1B,GAAM2C,GAAN;KAME,KAEV,aAAW,OAEZ,oBAAC,YAAD,EAAA,UACE,oBAAC,eAAD;MACYvC,UAAAA;MACJJ;MACCG;KAAK,CAAA,EAJP,GAAMwC,GAAN;KAME,KAEV;MACH,IAAI,CAACtC,aAAW,OAAS;MAAK,OAE5B,oBAAC,YAAD,EAAA,UAAqBA,YAAY;OAAAL;OAAAG;MAAc,CAAC,EAAvC,GAAMwC,GAAN;KAAoD,KAE5D;MACH,IAAI,CAACrC,iBAAe,OAAS;MAAK,OAEhC,oBAAC,YAAD,EAAA,UAAqBA,gBAAgB;OAAAN;OAAAG;MAAc,CAAC,EAA3C,GAAMwC,GAAN;KAAwD,KAEhE;MACH,IAAI,CAACpC,cAAY,OAAS;MAAK,OAE7B,oBAAC,YAAD,EAAA,UAAqBA,aAAa;OAAAP;OAAAG;MAAc,CAAC,EAAxC,GAAMwC,GAAN;KAAqD,KAE7D;MACH,IAAI,CAACnC,YAAU,OAAS;MAAK,OACtB,oBAAC,YAAD,EAAA,UAAqBA,WAAW;OAAAR;OAAAG;MAAc,CAAC,EAAtC,GAAMwC,GAAN;KAAmD,KAChE,QAAM,OAEP,oBAAC,YAAD,EAAA,UACE,oBAAC,eAAD;MACYlC,UAAAA;MACJT;MACCG;KAAK,CAAA,EAJP,GAAMwC,GAAN;KAME,SAAA,OAGN;IACX;GAAC;GACF/B,EAAA,KAAAH;GAAAG,EAAA,MAAAJ;GAAAI,EAAA,MAAAP;GAAAO,EAAA,MAAAN;GAAAM,EAAA,MAAAL;GAAAK,EAAA,MAAAd;GAAAc,EAAA,MAAAR;GAAAQ,EAAA,MAAAa;EAAA,OAAAA,KAAAb,EAAA;EAtDAE,KAAA2B,QAAOG,IAAKnB,EAsDZ;EAACb,EAAA,KAAA6B;EAAA7B,EAAA,KAAAH;EAAAG,EAAA,KAAAJ;EAAAI,EAAA,KAAAP;EAAAO,EAAA,KAAAN;EAAAM,EAAA,KAAAL;EAAAK,EAAA,KAAAd;EAAAc,EAAA,KAAAR;EAAAQ,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,IAAAa;CAAA,IAAAb,EAAA,QAAAE,IAAA;EAvDJW,KAAA,oBAAA,UAAA,EAAA,UACGX,GAsDC,CAAA;EACDF,EAAA,MAAAE;EAAAF,EAAA,MAAAa;CAAA,OAAAA,KAAAb,EAAA;CAAA,OAxDHa;AAwDG;AApEuB,SAAAiB,MAAAtB,GAAA;CAAA,OASOA,EAACM,QAAQe;AAAQ"}
+{"version":3,"file":"MessageContent.js","names":["ReactElement","Fragment","useMemo","Text","ThreadUserMessagePart","ThreadAssistantMessagePart","MessagePartState","useAui","useAuiState","ToolCallMessagePartProps","DataMessagePartProps","MessageContentPart","MessageContentStatePart","MessageContentProps","renderText","props","part","Extract","type","index","renderToolCall","renderImage","renderReasoning","renderSource","renderFile","renderData","DefaultTextRenderer","t0","$","_c","t1","text","ToolUIDisplay","Fallback","aui","toolName","s","tools","toolUIs","render","Render","t2","message","partMethods","t3","t4","addToolResult","respondToToolApproval","resumeToolCall","DataUIDisplay","name","renders","dataRenderers","renderers","Array","isArray","MessageContent","content","_temp","key","map","parts"],"sources":["../../../src/primitives/message/MessageContent.tsx"],"sourcesContent":["import { type ReactElement, Fragment, useMemo } from \"react\";\nimport { Text } from \"react-native\";\nimport type {\n  ThreadUserMessagePart,\n  ThreadAssistantMessagePart,\n  MessagePartState,\n} from \"@assistant-ui/core\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type {\n  ToolCallMessagePartProps,\n  DataMessagePartProps,\n} from \"@assistant-ui/core/react\";\n\ntype MessageContentPart = ThreadUserMessagePart | ThreadAssistantMessagePart;\ntype MessageContentStatePart = MessagePartState;\n\nexport type MessageContentProps = {\n  renderText?: (props: {\n    part: Extract<MessageContentPart, { type: \"text\" }>;\n    index: number;\n  }) => ReactElement;\n  renderToolCall?: (props: {\n    part: Extract<MessageContentPart, { type: \"tool-call\" }>;\n    index: number;\n  }) => ReactElement;\n  renderImage?: (props: {\n    part: Extract<MessageContentPart, { type: \"image\" }>;\n    index: number;\n  }) => ReactElement;\n  renderReasoning?: (props: {\n    part: Extract<MessageContentPart, { type: \"reasoning\" }>;\n    index: number;\n  }) => ReactElement;\n  renderSource?: (props: {\n    part: Extract<MessageContentPart, { type: \"source\" }>;\n    index: number;\n  }) => ReactElement;\n  renderFile?: (props: {\n    part: Extract<MessageContentPart, { type: \"file\" }>;\n    index: number;\n  }) => ReactElement;\n  renderData?: (props: {\n    part: Extract<MessageContentPart, { type: \"data\" }>;\n    index: number;\n  }) => ReactElement;\n};\n\nconst DefaultTextRenderer = ({\n  part,\n}: {\n  part: Extract<MessageContentPart, { type: \"text\" }>;\n}) => {\n  return <Text>{part.text}</Text>;\n};\n\nconst ToolUIDisplay = ({\n  Fallback,\n  part,\n  index,\n}: {\n  Fallback:\n    | ((props: {\n        part: Extract<MessageContentPart, { type: \"tool-call\" }>;\n        index: number;\n      }) => ReactElement)\n    | undefined;\n  part: Extract<MessageContentStatePart, { type: \"tool-call\" }>;\n  index: number;\n}) => {\n  const aui = useAui();\n  const Render = useAuiState(\n    (s) => s.tools.toolUIs[part.toolName]?.[0]?.render,\n  );\n\n  const partMethods = useMemo(() => aui.message.part({ index }), [aui, index]);\n\n  if (Render) {\n    return (\n      <Render\n        {...(part as ToolCallMessagePartProps)}\n        addResult={partMethods.addToolResult}\n        resume={partMethods.resumeToolCall}\n        respondToApproval={partMethods.respondToToolApproval}\n      />\n    );\n  }\n  if (Fallback) return <Fallback part={part} index={index} />;\n  return null;\n};\n\nconst DataUIDisplay = ({\n  Fallback,\n  part,\n  index,\n}: {\n  Fallback:\n    | ((props: {\n        part: Extract<MessageContentPart, { type: \"data\" }>;\n        index: number;\n      }) => ReactElement)\n    | undefined;\n  part: Extract<MessageContentStatePart, { type: \"data\" }>;\n  index: number;\n}) => {\n  const Render = useAuiState((s) => {\n    const renders = s.dataRenderers.renderers[part.name];\n    if (Array.isArray(renders)) return renders[0];\n    return renders;\n  });\n  if (Render) return <Render {...(part as DataMessagePartProps)} />;\n  if (Fallback) return <Fallback part={part} index={index} />;\n  return null;\n};\n\nexport const MessageContent = ({\n  renderText,\n  renderToolCall,\n  renderImage,\n  renderReasoning,\n  renderSource,\n  renderFile,\n  renderData,\n}: MessageContentProps) => {\n  const content = useAuiState((s) => s.message.parts);\n\n  return (\n    <>\n      {content.map((part, index) => {\n        const key = `${part.type}-${index}`;\n        switch (part.type) {\n          case \"text\":\n            return (\n              <Fragment key={key}>\n                {renderText ? (\n                  renderText({ part, index })\n                ) : (\n                  <DefaultTextRenderer part={part} />\n                )}\n              </Fragment>\n            );\n          case \"tool-call\":\n            return (\n              <Fragment key={key}>\n                <ToolUIDisplay\n                  Fallback={renderToolCall}\n                  part={part}\n                  index={index}\n                />\n              </Fragment>\n            );\n          case \"image\":\n            if (!renderImage) return null;\n            return (\n              <Fragment key={key}>{renderImage({ part, index })}</Fragment>\n            );\n          case \"reasoning\":\n            if (!renderReasoning) return null;\n            return (\n              <Fragment key={key}>{renderReasoning({ part, index })}</Fragment>\n            );\n          case \"source\":\n            if (!renderSource) return null;\n            return (\n              <Fragment key={key}>{renderSource({ part, index })}</Fragment>\n            );\n          case \"file\":\n            if (!renderFile) return null;\n            return <Fragment key={key}>{renderFile({ part, index })}</Fragment>;\n          case \"data\":\n            return (\n              <Fragment key={key}>\n                <DataUIDisplay\n                  Fallback={renderData}\n                  part={part}\n                  index={index}\n                />\n              </Fragment>\n            );\n          default:\n            return null;\n        }\n      })}\n    </>\n  );\n};\n"],"mappings":";;;;;;AA+CA,MAAM0B,uBAAsBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAb,SAAAW;CAI5B,IAAAG;CAAA,IAAAF,EAAA,OAAAZ,KAAAe,MAAA;EACQD,KAAA,oBAAC,MAAD,EAAA,UAAOd,KAAIe,KAAN,CAAA;EAAmBH,EAAA,KAAAZ,KAAAe;EAAAH,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,OAAxBE;AAAwB;AAGjC,MAAME,iBAAgBL,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAI,UAAAjB,MAAAG,UAAAQ;CAcrB,MAAAO,MAAY3B,OAAO;CAAE,IAAAuB;CAAA,IAAAF,EAAA,OAAAZ,KAAAmB,UAAA;EAEnBL,MAAAM,MAAOA,EAACC,MAAMC,QAAStB,KAAImB,SAAe,GAAA,EAAQ,EAAAI;EAAAX,EAAA,KAAAZ,KAAAmB;EAAAP,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CADpD,MAAAY,SAAehC,YACbsB,EACF;CAAE,IAAAW;CAAA,IAAAb,EAAA,OAAAM,IAAAQ,WAAAd,EAAA,OAAAT,OAAA;EAEgCsB,KAAAP,IAAGQ,QAAQ1B,KAAM,EAAAG,MAAQ,CAAC;EAACS,EAAA,KAAAM,IAAAQ;EAAAd,EAAA,KAAAT;EAAAS,EAAA,KAAAa;CAAA,OAAAA,KAAAb,EAAA;CAA7D,MAAAe,cAAkCF;CAElC,IAAID,QAAM;EAGC,MAAAI,KAAA5B;EAAgC,IAAA6B;EAAA,IAAAjB,EAAA,OAAAY,UAAAZ,EAAA,OAAAe,YAAAG,iBAAAlB,EAAA,OAAAe,YAAAI,yBAAAnB,EAAA,OAAAe,YAAAK,kBAAApB,EAAA,OAAAgB,IAAA;GADvCC,KAAA,oBAAC,QAAD;IAAO,GACAD;IACM,WAAAD,YAAWG;IACd,QAAAH,YAAWK;IACA,mBAAAL,YAAWI;GAAsB,CAAA;GACpDnB,EAAA,KAAAY;GAAAZ,EAAA,KAAAe,YAAAG;GAAAlB,EAAA,KAAAe,YAAAI;GAAAnB,EAAA,KAAAe,YAAAK;GAAApB,EAAA,KAAAgB;GAAAhB,EAAA,MAAAiB;EAAA,OAAAA,KAAAjB,EAAA;EAAA,OALFiB;CAKE;CAGN,IAAIZ,UAAQ;EAAA,IAAAW;EAAA,IAAAhB,EAAA,QAAAK,YAAAL,EAAA,QAAAT,SAAAS,EAAA,QAAAZ,MAAA;GAAS4B,KAAA,oBAAC,UAAD;IAAgB5B;IAAaG;GAAK,CAAA;GAAIS,EAAA,MAAAK;GAAAL,EAAA,MAAAT;GAAAS,EAAA,MAAAZ;GAAAY,EAAA,MAAAgB;EAAA,OAAAA,KAAAhB,EAAA;EAAA,OAAtCgB;CAAsC;CAAC,OACrD;AAAI;AAGb,MAAMK,iBAAgBtB,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAI,UAAAjB,MAAAG,UAAAQ;CAatB,IAAAG;CAAA,IAAAF,EAAA,OAAAZ,KAAAkC,MAAA;EAC4BpB,MAAAM,MAAA;GACzB,MAAAe,UAAgBf,EAACgB,cAAcC,UAAWrC,KAAIkC;GAC9C,IAAII,MAAKC,QAASJ,OAAO,GAAC,OAASA,QAAO;GAAI,OACvCA;EAAO;EACfvB,EAAA,KAAAZ,KAAAkC;EAAAtB,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAJD,MAAAY,SAAehC,YAAYsB,EAI1B;CACD,IAAIU,QAAM;EAAsB,MAAAC,KAAAzB;EAA4B,IAAA4B;EAAA,IAAAhB,EAAA,OAAAY,UAAAZ,EAAA,OAAAa,IAAA;GAAzCG,KAAA,oBAAC,QAAD,EAAO,GAAMH,GAA4B,CAAA;GAAKb,EAAA,KAAAY;GAAAZ,EAAA,KAAAa;GAAAb,EAAA,KAAAgB;EAAA,OAAAA,KAAAhB,EAAA;EAAA,OAA9CgB;CAA8C;CACjE,IAAIX,UAAQ;EAAA,IAAAQ;EAAA,IAAAb,EAAA,OAAAK,YAAAL,EAAA,OAAAT,SAAAS,EAAA,OAAAZ,MAAA;GAASyB,KAAA,oBAAC,UAAD;IAAgBzB;IAAaG;GAAK,CAAA;GAAIS,EAAA,KAAAK;GAAAL,EAAA,KAAAT;GAAAS,EAAA,KAAAZ;GAAAY,EAAA,KAAAa;EAAA,OAAAA,KAAAb,EAAA;EAAA,OAAtCa;CAAsC;CAAC,OACrD;AAAI;AAGb,MAAae,kBAAiB7B,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAf,YAAAM,gBAAAC,aAAAC,iBAAAC,cAAAC,YAAAC,eAAAE;CAS7B,MAAA8B,UAAgBjD,YAAYkD,KAAsB;CAAE,IAAA5B;CAAA,IAAAF,EAAA,OAAA6B,WAAA7B,EAAA,OAAAH,cAAAG,EAAA,OAAAJ,cAAAI,EAAA,OAAAP,eAAAO,EAAA,OAAAN,mBAAAM,EAAA,OAAAL,gBAAAK,EAAA,OAAAd,cAAAc,EAAA,OAAAR,gBAAA;EAAA,IAAAqB;EAAA,IAAAb,EAAA,OAAAH,cAAAG,EAAA,QAAAJ,cAAAI,EAAA,QAAAP,eAAAO,EAAA,QAAAN,mBAAAM,EAAA,QAAAL,gBAAAK,EAAA,QAAAd,cAAAc,EAAA,QAAAR,gBAAA;GAInCqB,MAAAzB,MAAAG,UAAA;IACX,MAAAwC,MAAY,GAAG3C,KAAIE,KAAK,GAAIC;IAC5B,QAAQH,KAAIE,MAAZ;KAAiB,KACV,QAAM,OAEP,oBAAC,YAAD,EAAA,UACGJ,aACCA,WAAW;MAAAE;MAAAG;KAAc,CAG3B,IADE,oBAAC,qBAAD,EAA2BH,KAAI,CAAA,EAJ1B,GAAM2C,GAAN;KAME,KAEV,aAAW,OAEZ,oBAAC,YAAD,EAAA,UACE,oBAAC,eAAD;MACYvC,UAAAA;MACJJ;MACCG;KAAK,CAAA,EAJP,GAAMwC,GAAN;KAME,KAEV;MACH,IAAI,CAACtC,aAAW,OAAS;MAAK,OAE5B,oBAAC,YAAD,EAAA,UAAqBA,YAAY;OAAAL;OAAAG;MAAc,CAAC,EAAvC,GAAMwC,GAAN;KAAoD,KAE5D;MACH,IAAI,CAACrC,iBAAe,OAAS;MAAK,OAEhC,oBAAC,YAAD,EAAA,UAAqBA,gBAAgB;OAAAN;OAAAG;MAAc,CAAC,EAA3C,GAAMwC,GAAN;KAAwD,KAEhE;MACH,IAAI,CAACpC,cAAY,OAAS;MAAK,OAE7B,oBAAC,YAAD,EAAA,UAAqBA,aAAa;OAAAP;OAAAG;MAAc,CAAC,EAAxC,GAAMwC,GAAN;KAAqD,KAE7D;MACH,IAAI,CAACnC,YAAU,OAAS;MAAK,OACtB,oBAAC,YAAD,EAAA,UAAqBA,WAAW;OAAAR;OAAAG;MAAc,CAAC,EAAtC,GAAMwC,GAAN;KAAmD,KAChE,QAAM,OAEP,oBAAC,YAAD,EAAA,UACE,oBAAC,eAAD;MACYlC,UAAAA;MACJT;MACCG;KAAK,CAAA,EAJP,GAAMwC,GAAN;KAME,SAAA,OAGN;IACX;GAAC;GACF/B,EAAA,KAAAH;GAAAG,EAAA,MAAAJ;GAAAI,EAAA,MAAAP;GAAAO,EAAA,MAAAN;GAAAM,EAAA,MAAAL;GAAAK,EAAA,MAAAd;GAAAc,EAAA,MAAAR;GAAAQ,EAAA,MAAAa;EAAA,OAAAA,KAAAb,EAAA;EAtDAE,KAAA2B,QAAOG,IAAKnB,EAsDZ;EAACb,EAAA,KAAA6B;EAAA7B,EAAA,KAAAH;EAAAG,EAAA,KAAAJ;EAAAI,EAAA,KAAAP;EAAAO,EAAA,KAAAN;EAAAM,EAAA,KAAAL;EAAAK,EAAA,KAAAd;EAAAc,EAAA,KAAAR;EAAAQ,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,IAAAa;CAAA,IAAAb,EAAA,QAAAE,IAAA;EAvDJW,KAAA,oBAAA,UAAA,EAAA,UACGX,GAsDC,CAAA;EACDF,EAAA,MAAAE;EAAAF,EAAA,MAAAa;CAAA,OAAAA,KAAAb,EAAA;CAAA,OAxDHa;AAwDG;AApEuB,SAAAiB,MAAAtB,GAAA;CAAA,OASOA,EAACM,QAAQmB;AAAM"}

package/dist/primitives/message/MessageParts.js

@@ -35,20 +35,8 @@ const MessagePrimitiveParts = (props) => {
 	let t0;
 	if ($[5] !== components) {
 		t0 = components ? {
-			Text: components.Text ?? rnDefaultComponents.Text,
-			Image: components.Image ?? messagePartsDefaultComponents.Image,
-			Reasoning: components.Reasoning ?? messagePartsDefaultComponents.Reasoning,
-			Source: components.Source ?? messagePartsDefaultComponents.Source,
-			File: components.File ?? messagePartsDefaultComponents.File,
-			Unstable_Audio: components.Unstable_Audio ?? messagePartsDefaultComponents.Unstable_Audio,
-			..."ChainOfThought" in components ? { ChainOfThought: components.ChainOfThought } : {
-				tools: components.tools,
-				data: components.data,
-				ToolGroup: components.ToolGroup ?? messagePartsDefaultComponents.ToolGroup,
-				ReasoningGroup: components.ReasoningGroup ?? messagePartsDefaultComponents.ReasoningGroup
-			},
-			Empty: components.Empty,
-			Quote: components.Quote
+			...components,
+			Text: components.Text ?? rnDefaultComponents.Text
 		} : rnDefaultComponents;
 		$[5] = components;
 		$[6] = t0;

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

@@ -1 +1 @@
-{"version":3,"file":"MessageParts.js","names":["FC","Text","MessagePrimitiveParts","MessagePrimitivePartsBase","MessagePartComponent","MessagePartComponentBase","MessagePrimitivePartByIndex","MessagePrimitivePartByIndexBase","messagePartsDefaultComponents","rnDefaultComponents","text","Props","props","$","_c","t0","children","components","rest","Image","Reasoning","Source","File","Unstable_Audio","ChainOfThought","tools","data","ToolGroup","ReasoningGroup","Empty","Quote","merged","t1","t2","displayName"],"sources":["../../../src/primitives/message/MessageParts.tsx"],"sourcesContent":["import type { FC } from \"react\";\nimport { Text } from \"react-native\";\nimport {\n  MessagePrimitiveParts as MessagePrimitivePartsBase,\n  MessagePartComponent as MessagePartComponentBase,\n  MessagePrimitivePartByIndex as MessagePrimitivePartByIndexBase,\n  messagePartsDefaultComponents,\n} from \"@assistant-ui/core/react\";\n\nconst rnDefaultComponents = {\n  ...messagePartsDefaultComponents,\n  Text: ({ text }: { text: string }) => <Text>{text}</Text>,\n} satisfies MessagePrimitiveParts.Props[\"components\"];\n\nexport namespace MessagePrimitiveParts {\n  export type Props = MessagePrimitivePartsBase.Props;\n}\n\n/**\n * Renders the parts of a message with React Native-specific default components.\n */\nexport const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props> = (\n  props,\n) => {\n  if (\"children\" in props) {\n    return (\n      <MessagePrimitivePartsBase>{props.children}</MessagePrimitivePartsBase>\n    );\n  }\n\n  const { components, ...rest } = props;\n  const merged = components\n    ? {\n        Text: components.Text ?? rnDefaultComponents.Text,\n        Image: components.Image ?? messagePartsDefaultComponents.Image,\n        Reasoning:\n          components.Reasoning ?? messagePartsDefaultComponents.Reasoning,\n        Source: components.Source ?? messagePartsDefaultComponents.Source,\n        File: components.File ?? messagePartsDefaultComponents.File,\n        Unstable_Audio:\n          components.Unstable_Audio ??\n          messagePartsDefaultComponents.Unstable_Audio,\n        ...(\"ChainOfThought\" in components\n          ? { ChainOfThought: components.ChainOfThought }\n          : {\n              tools: components.tools,\n              data: components.data,\n              ToolGroup:\n                components.ToolGroup ?? messagePartsDefaultComponents.ToolGroup,\n              ReasoningGroup:\n                components.ReasoningGroup ??\n                messagePartsDefaultComponents.ReasoningGroup,\n            }),\n        Empty: components.Empty,\n        Quote: components.Quote,\n      }\n    : rnDefaultComponents;\n\n  return <MessagePrimitivePartsBase components={merged as any} {...rest} />;\n};\n\nMessagePrimitiveParts.displayName = \"MessagePrimitive.Parts\";\n\nexport {\n  MessagePartComponentBase as MessagePartComponent,\n  MessagePrimitivePartByIndexBase as MessagePrimitivePartByIndex,\n};\n"],"mappings":";;;;;AASA,MAAMS,sBAAsB;CAC1B,GAAGD;CACHP,OAAO,EAAES,WAA6B,oBAAC,MAAD,EAAA,UAAOA,KAAW,CAAA;AAC1D;;;;AASA,MAAaR,yBAAyDU,UAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAGpE,IAAI,cAAcF,OAAK;EAAA,IAAAG;EAAA,IAAAF,EAAA,OAAAD,MAAAI,UAAA;GAEnBD,KAAA,oBAAC,yBAAD,EAAA,UAA4BH,MAAKI,SAAP,CAAA;GAA6CH,EAAA,KAAAD,MAAAI;GAAAH,EAAA,KAAAE;EAAA,OAAAA,KAAAF,EAAA;EAAA,OAAvEE;CAAuE;CAE1E,IAAAE;CAAA,IAAAC;CAAA,IAAAL,EAAA,OAAAD,OAAA;EAED,CAAA,CAAAK,YAAA,GAAAC,QAAgCN;EAAMC,EAAA,KAAAD;EAAAC,EAAA,KAAAI;EAAAJ,EAAA,KAAAK;CAAA,OAAA;EAAAD,aAAAJ,EAAA;EAAAK,OAAAL,EAAA;CAAA;CAAA,IAAAE;CAAA,IAAAF,EAAA,OAAAI,YAAA;EACvBF,KAAAE,aAAA;GAAAhB,MAEHgB,WAAUhB,QAASQ,oBAAmBR;GAAKkB,OAC1CF,WAAUE,SAAUX,8BAA6BW;GAAMC,WAE5DH,WAAUG,aAAcZ,8BAA6BY;GAAUC,QACzDJ,WAAUI,UAAWb,8BAA6Ba;GAAOC,MAC3DL,WAAUK,QAASd,8BAA6Bc;GAAKC,gBAEzDN,WAAUM,kBACVf,8BAA6Be;GAAe,GAC1C,oBAAoBN,aAApB,EAAAO,gBACkBP,WAAUO,eAS5B,IAVA;IAAAC,OAGSR,WAAUQ;IAAMC,MACjBT,WAAUS;IAAKC,WAEnBV,WAAUU,aAAcnB,8BAA6BmB;IAAUC,gBAE/DX,WAAUW,kBACVpB,8BAA6BoB;GACjC;GAACC,OACEZ,WAAUY;GAAMC,OAChBb,WAAUa;EAED,IAzBPrB;EAyBQI,EAAA,KAAAI;EAAAJ,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAEuB,MAAAmB,KAAAD;CAAa,IAAAE;CAAA,IAAApB,EAAA,OAAAK,QAAAL,EAAA,OAAAmB,IAAA;EAApDC,KAAA,oBAAC,yBAAD;GAAuC,YAAAD;GAAa,GAAMd;EAAI,CAAA;EAAIL,EAAA,KAAAK;EAAAL,EAAA,KAAAmB;EAAAnB,EAAA,KAAAoB;CAAA,OAAAA,KAAApB,EAAA;CAAA,OAAlEoB;AAAkE;AAG3E/B,sBAAsBgC,cAAc"}
+{"version":3,"file":"MessageParts.js","names":["FC","Text","MessagePrimitiveParts","MessagePrimitivePartsBase","MessagePartComponent","MessagePartComponentBase","MessagePrimitivePartByIndex","MessagePrimitivePartByIndexBase","messagePartsDefaultComponents","rnDefaultComponents","text","Props","props","$","_c","t0","children","components","rest","merged","t1","t2","displayName"],"sources":["../../../src/primitives/message/MessageParts.tsx"],"sourcesContent":["import type { FC } from \"react\";\nimport { Text } from \"react-native\";\nimport {\n  MessagePrimitiveParts as MessagePrimitivePartsBase,\n  MessagePartComponent as MessagePartComponentBase,\n  MessagePrimitivePartByIndex as MessagePrimitivePartByIndexBase,\n  messagePartsDefaultComponents,\n} from \"@assistant-ui/core/react\";\n\nconst rnDefaultComponents = {\n  ...messagePartsDefaultComponents,\n  Text: ({ text }: { text: string }) => <Text>{text}</Text>,\n} satisfies MessagePrimitiveParts.Props[\"components\"];\n\nexport namespace MessagePrimitiveParts {\n  export type Props = MessagePrimitivePartsBase.Props;\n}\n\n/**\n * Renders the parts of a message with React Native-specific default components.\n */\nexport const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props> = (\n  props,\n) => {\n  if (\"children\" in props) {\n    return (\n      <MessagePrimitivePartsBase>{props.children}</MessagePrimitivePartsBase>\n    );\n  }\n\n  const { components, ...rest } = props;\n  const merged = components\n    ? { ...components, Text: components.Text ?? rnDefaultComponents.Text }\n    : rnDefaultComponents;\n\n  return <MessagePrimitivePartsBase components={merged as any} {...rest} />;\n};\n\nMessagePrimitiveParts.displayName = \"MessagePrimitive.Parts\";\n\nexport {\n  MessagePartComponentBase as MessagePartComponent,\n  MessagePrimitivePartByIndexBase as MessagePrimitivePartByIndex,\n};\n"],"mappings":";;;;;AASA,MAAMS,sBAAsB;CAC1B,GAAGD;CACHP,OAAO,EAAES,WAA6B,oBAAC,MAAD,EAAA,UAAOA,KAAW,CAAA;AAC1D;;;;AASA,MAAaR,yBAAyDU,UAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAGpE,IAAI,cAAcF,OAAK;EAAA,IAAAG;EAAA,IAAAF,EAAA,OAAAD,MAAAI,UAAA;GAEnBD,KAAA,oBAAC,yBAAD,EAAA,UAA4BH,MAAKI,SAAP,CAAA;GAA6CH,EAAA,KAAAD,MAAAI;GAAAH,EAAA,KAAAE;EAAA,OAAAA,KAAAF,EAAA;EAAA,OAAvEE;CAAuE;CAE1E,IAAAE;CAAA,IAAAC;CAAA,IAAAL,EAAA,OAAAD,OAAA;EAED,CAAA,CAAAK,YAAA,GAAAC,QAAgCN;EAAMC,EAAA,KAAAD;EAAAC,EAAA,KAAAI;EAAAJ,EAAA,KAAAK;CAAA,OAAA;EAAAD,aAAAJ,EAAA;EAAAK,OAAAL,EAAA;CAAA;CAAA,IAAAE;CAAA,IAAAF,EAAA,OAAAI,YAAA;EACvBF,KAAAE,aAAA;GAAA,GACNA;GAAUhB,MAAQgB,WAAUhB,QAASQ,oBAAmBR;EAC3C,IAFPQ;EAEQI,EAAA,KAAAI;EAAAJ,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAEuB,MAAAO,KAAAD;CAAa,IAAAE;CAAA,IAAAR,EAAA,OAAAK,QAAAL,EAAA,OAAAO,IAAA;EAApDC,KAAA,oBAAC,yBAAD;GAAuC,YAAAD;GAAa,GAAMF;EAAI,CAAA;EAAIL,EAAA,KAAAK;EAAAL,EAAA,KAAAO;EAAAP,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAAA,OAAlEQ;AAAkE;AAG3EnB,sBAAsBoB,cAAc"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-native",
-  "version": "0.1.40",
+  "version": "0.1.41",
   "description": "React Native bindings for assistant-ui",
   "keywords": [
     "assistant",
@@ -31,10 +31,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.17",
-    "@assistant-ui/store": "^0.3.12",
-    "@assistant-ui/tap": "^0.9.16",
-    "assistant-stream": "^0.3.41"
+    "@assistant-ui/core": "^0.3.18",
+    "@assistant-ui/store": "^0.3.13",
+    "@assistant-ui/tap": "^0.9.17",
+    "assistant-stream": "^0.3.42"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -47,10 +47,10 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.25",
+    "@assistant-ui/x-buildutils": "0.0.26",
     "@babel/core": "^7.29.7",
     "@types/react": "^19.2.18",
-    "@types/react-dom": "^19.2.5",
+    "@types/react-dom": "^19.2.7",
     "jsdom": "^30.0.1",
     "react": "^19.2.8",
     "react-dom": "^19.2.8",

package/src/primitives/message/MessageContent.runtime.test.tsx

@@ -0,0 +1,81 @@
+import { act } from "react";
+import { createRoot } from "react-dom/client";
+import { describe, expect, it } from "vitest";
+import type { MessagePartState, ThreadMessageLike } from "@assistant-ui/core";
+import {
+  AssistantRuntimeProvider,
+  MessageByIndexProvider,
+  useAssistantDataUI,
+  useAssistantToolUI,
+  useExternalStoreRuntime,
+} from "@assistant-ui/core/react";
+import { MessageContent } from "./MessageContent";
+
+(globalThis as Record<string, unknown>).IS_REACT_ACT_ENVIRONMENT = true;
+
+describe("MessageContent with a runtime", () => {
+  it.each(["tool-call", "data"] as const)(
+    "renders derived status for registered %s UIs",
+    async (type) => {
+      const Status = ({ status }: Pick<MessagePartState, "status">) => (
+        <span>{status.type}</span>
+      );
+      const Registrations = () => {
+        useAssistantToolUI({ toolName: "search", render: Status });
+        useAssistantDataUI({ name: "chart", render: Status });
+        return <MessageContent />;
+      };
+      const App = ({ message }: { message: ThreadMessageLike }) => {
+        const runtime = useExternalStoreRuntime({
+          messages: [message],
+          convertMessage: (value) => value,
+          onNew: async () => {},
+        });
+        return (
+          <AssistantRuntimeProvider runtime={runtime}>
+            <MessageByIndexProvider index={0}>
+              <Registrations />
+            </MessageByIndexProvider>
+          </AssistantRuntimeProvider>
+        );
+      };
+      const container = document.createElement("div");
+      const root = createRoot(container);
+      try {
+        for (const status of [
+          "running",
+          "requires-action",
+          "complete",
+        ] as const) {
+          if (type === "data" && status === "requires-action") continue;
+          const message: ThreadMessageLike = {
+            id: "m1",
+            role: "assistant",
+            status:
+              status === "running"
+                ? { type: status }
+                : status === "complete"
+                  ? { type: status, reason: "stop" }
+                  : { type: status, reason: "tool-calls" },
+            content: [
+              type === "data"
+                ? { type, name: "chart", data: { value: 1 } }
+                : {
+                    type,
+                    toolName: "search",
+                    toolCallId: "c1",
+                    args: {},
+                    argsText: "{}",
+                    ...(status === "complete" ? { result: "done" } : {}),
+                  },
+            ],
+          };
+          await act(async () => root.render(<App message={message} />));
+          expect(container.textContent).toBe(status);
+        }
+      } finally {
+        await act(async () => root.unmount());
+      }
+    },
+  );
+});

package/src/primitives/message/MessageContent.test.tsx

@@ -10,7 +10,12 @@ const h = vi.hoisted(() => ({
   resumeToolCall: vi.fn(),
   respondToToolApproval: vi.fn(),
   state: {
-    message: { content: [] as AnyPart[] },
+    message: {
+      content: [] as AnyPart[],
+      get parts() {
+        return this.content;
+      },
+    },
     tools: { toolUIs: {} as Record<string, unknown> },
     dataRenderers: { renderers: {} as Record<string, unknown> },
   },

package/src/primitives/message/MessageContent.tsx

@@ -3,8 +3,7 @@ import { Text } from "react-native";
 import type {
   ThreadUserMessagePart,
   ThreadAssistantMessagePart,
-  ToolCallMessagePart,
-  DataMessagePart,
+  MessagePartState,
 } from "@assistant-ui/core";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import type {
@@ -13,6 +12,7 @@ import type {
 } from "@assistant-ui/core/react";
 
 type MessageContentPart = ThreadUserMessagePart | ThreadAssistantMessagePart;
+type MessageContentStatePart = MessagePartState;
 
 export type MessageContentProps = {
   renderText?: (props: {
@@ -59,9 +59,12 @@ const ToolUIDisplay = ({
   index,
 }: {
   Fallback:
-    | ((props: { part: ToolCallMessagePart; index: number }) => ReactElement)
+    | ((props: {
+        part: Extract<MessageContentPart, { type: "tool-call" }>;
+        index: number;
+      }) => ReactElement)
     | undefined;
-  part: ToolCallMessagePart;
+  part: Extract<MessageContentStatePart, { type: "tool-call" }>;
   index: number;
 }) => {
   const aui = useAui();
@@ -91,9 +94,12 @@ const DataUIDisplay = ({
   index,
 }: {
   Fallback:
-    | ((props: { part: DataMessagePart; index: number }) => ReactElement)
+    | ((props: {
+        part: Extract<MessageContentPart, { type: "data" }>;
+        index: number;
+      }) => ReactElement)
     | undefined;
-  part: DataMessagePart;
+  part: Extract<MessageContentStatePart, { type: "data" }>;
   index: number;
 }) => {
   const Render = useAuiState((s) => {
@@ -115,7 +121,7 @@ export const MessageContent = ({
   renderFile,
   renderData,
 }: MessageContentProps) => {
-  const content = useAuiState((s) => s.message.content);
+  const content = useAuiState((s) => s.message.parts);
 
   return (
     <>

package/src/primitives/message/MessageParts.test.tsx

@@ -0,0 +1,116 @@
+import { act } from "react";
+import { createRoot, type Root } from "react-dom/client";
+import { Text } from "react-native";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import type { ThreadMessageLike } from "@assistant-ui/core";
+import {
+  AssistantRuntimeProvider,
+  MessageByIndexProvider,
+  useExternalStoreRuntime,
+} from "@assistant-ui/core/react";
+import { MessagePrimitiveParts } from "./MessageParts";
+
+(globalThis as Record<string, unknown>).IS_REACT_ACT_ENVIRONMENT = true;
+
+const messages: ThreadMessageLike[] = [
+  {
+    role: "assistant",
+    content: [
+      { type: "text", text: "Answer" },
+      { type: "reasoning", text: "Thinking" },
+      {
+        type: "generative-ui",
+        spec: {
+          root: [
+            { component: "Card", props: { title: "Result" } },
+            { component: "Unknown" },
+          ],
+        },
+      },
+      { type: "data", name: "chart", data: 42 },
+      { type: "data", name: "other", data: 7 },
+    ],
+  },
+];
+
+const components = {
+  generativeUI: {
+    components: {
+      Card: ({ title }: { title: string }) => <Text>{title}</Text>,
+    },
+    Fallback: ({ component }: { component: string }) => (
+      <Text>Unavailable: {component}</Text>
+    ),
+  },
+  data: {
+    by_name: {
+      chart: ({ data }: { data: unknown }) => (
+        <Text>Chart: {String(data)}</Text>
+      ),
+    },
+    Fallback: ({ data }: { data: unknown }) => (
+      <Text>Data: {String(data)}</Text>
+    ),
+  },
+};
+
+const App = (props: MessagePrimitiveParts.Props) => {
+  const runtime = useExternalStoreRuntime({
+    messages,
+    convertMessage: (message) => message,
+    onNew: async () => {
+      throw new Error("This thread is read-only");
+    },
+  });
+  return (
+    <AssistantRuntimeProvider runtime={runtime}>
+      <MessageByIndexProvider index={0}>
+        <MessagePrimitiveParts {...props} />
+      </MessageByIndexProvider>
+    </AssistantRuntimeProvider>
+  );
+};
+
+const ChainOfThought = () => <Text>Thought group</Text>;
+
+describe("MessagePrimitiveParts", () => {
+  let container: HTMLDivElement;
+  let root: Root;
+
+  beforeEach(() => {
+    container = document.createElement("div");
+    document.body.appendChild(container);
+    root = createRoot(container);
+  });
+
+  afterEach(async () => {
+    await act(async () => root.unmount());
+    container.remove();
+  });
+
+  it("renders generative UI and its fallback beside native text", async () => {
+    await act(async () => root.render(<App components={components} />));
+    expect(container.textContent).toBe(
+      "AnswerResultUnavailable: UnknownChart: 42Data: 7",
+    );
+  });
+
+  it("keeps data and generative UI when chain-of-thought grouping changes", async () => {
+    await act(async () => root.render(<App components={components} />));
+    expect(container.textContent).toBe(
+      "AnswerResultUnavailable: UnknownChart: 42Data: 7",
+    );
+
+    await act(async () =>
+      root.render(<App components={{ ...components, ChainOfThought }} />),
+    );
+    expect(container.textContent).toBe(
+      "AnswerThought groupResultUnavailable: UnknownChart: 42Data: 7",
+    );
+
+    await act(async () => root.render(<App components={components} />));
+    expect(container.textContent).toBe(
+      "AnswerResultUnavailable: UnknownChart: 42Data: 7",
+    );
+  });
+});

package/src/primitives/message/MessageParts.tsx

@@ -30,30 +30,7 @@ export const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props> = (
 
   const { components, ...rest } = props;
   const merged = components
-    ? {
-        Text: components.Text ?? rnDefaultComponents.Text,
-        Image: components.Image ?? messagePartsDefaultComponents.Image,
-        Reasoning:
-          components.Reasoning ?? messagePartsDefaultComponents.Reasoning,
-        Source: components.Source ?? messagePartsDefaultComponents.Source,
-        File: components.File ?? messagePartsDefaultComponents.File,
-        Unstable_Audio:
-          components.Unstable_Audio ??
-          messagePartsDefaultComponents.Unstable_Audio,
-        ...("ChainOfThought" in components
-          ? { ChainOfThought: components.ChainOfThought }
-          : {
-              tools: components.tools,
-              data: components.data,
-              ToolGroup:
-                components.ToolGroup ?? messagePartsDefaultComponents.ToolGroup,
-              ReasoningGroup:
-                components.ReasoningGroup ??
-                messagePartsDefaultComponents.ReasoningGroup,
-            }),
-        Empty: components.Empty,
-        Quote: components.Quote,
-      }
+    ? { ...components, Text: components.Text ?? rnDefaultComponents.Text }
     : rnDefaultComponents;
 
   return <MessagePrimitivePartsBase components={merged as any} {...rest} />;