@assistant-ui/react-native 0.1.32 → 0.1.33

package/dist/primitives/attachment/AttachmentRemove.js

@@ -19,11 +19,11 @@ const AttachmentRemove = (t0) => {
 	}
 	const aui = useAui();
 	let t1;
-	if ($[3] !== aui) {
+	if ($[3] !== aui.attachment) {
 		t1 = () => {
-			aui.attachment().remove();
+			aui.attachment.remove();
 		};
-		$[3] = aui;
+		$[3] = aui.attachment;
 		$[4] = t1;
 	} else t1 = $[4];
 	const handleRemove = t1;

package/dist/primitives/attachment/AttachmentRemove.js.map

@@ -1 +1 @@
-{"version":3,"file":"AttachmentRemove.js","names":["ReactNode","useCallback","Pressable","PressableProps","useAui","AttachmentRemoveProps","Omit","children","AttachmentRemove","t0","$","_c","pressableProps","aui","t1","attachment","remove","handleRemove","t2"],"sources":["../../../src/primitives/attachment/AttachmentRemove.tsx"],"sourcesContent":["import { type ReactNode, useCallback } from \"react\";\nimport { Pressable, type PressableProps } from \"react-native\";\n\nimport { useAui } from \"@assistant-ui/store\";\n\nexport type AttachmentRemoveProps = Omit<PressableProps, \"onPress\"> & {\n  children: ReactNode;\n};\n\nexport const AttachmentRemove = ({\n  children,\n  ...pressableProps\n}: AttachmentRemoveProps) => {\n  const aui = useAui();\n\n  const handleRemove = useCallback(() => {\n    aui.attachment().remove();\n  }, [aui]);\n\n  return (\n    <Pressable onPress={handleRemove} {...pressableProps}>\n      {children}\n    </Pressable>\n  );\n};\n"],"mappings":";;;;;;AASA,MAAaQ,oBAAmBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAA,IAAAJ;CAAA,IAAAK;CAAA,IAAAF,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAF,aAAAK,kBAAAH;EAGTC,EAAA,KAAAD;EAAAC,EAAA,KAAAH;EAAAG,EAAA,KAAAE;CAAA,OAAA;EAAAL,WAAAG,EAAA;EAAAE,iBAAAF,EAAA;CAAA;CACtB,MAAAG,MAAYT,OAAO;CAAE,IAAAU;CAAA,IAAAJ,EAAA,OAAAG,KAAA;EAEYC,WAAA;GAC/BD,IAAGE,WAAY,CAAC,CAAAC,OAAQ;EAAC;EAC1BN,EAAA,KAAAG;EAAAH,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAFD,MAAAO,eAAqBH;CAEX,IAAAI;CAAA,IAAAR,EAAA,OAAAH,YAAAG,EAAA,OAAAO,gBAAAP,EAAA,OAAAE,gBAAA;EAGRM,KAAA,oBAAC,WAAD;GAAoBD,SAAAA;GAAY,GAAML;GACnCL;EADO,CAAA;EAEEG,EAAA,KAAAH;EAAAG,EAAA,KAAAO;EAAAP,EAAA,KAAAE;EAAAF,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAAA,OAFZQ;AAEY"}
+{"version":3,"file":"AttachmentRemove.js","names":["ReactNode","useCallback","Pressable","PressableProps","useAui","AttachmentRemoveProps","Omit","children","AttachmentRemove","t0","$","_c","pressableProps","aui","t1","attachment","remove","handleRemove","t2"],"sources":["../../../src/primitives/attachment/AttachmentRemove.tsx"],"sourcesContent":["import { type ReactNode, useCallback } from \"react\";\nimport { Pressable, type PressableProps } from \"react-native\";\n\nimport { useAui } from \"@assistant-ui/store\";\n\nexport type AttachmentRemoveProps = Omit<PressableProps, \"onPress\"> & {\n  children: ReactNode;\n};\n\nexport const AttachmentRemove = ({\n  children,\n  ...pressableProps\n}: AttachmentRemoveProps) => {\n  const aui = useAui();\n\n  const handleRemove = useCallback(() => {\n    aui.attachment.remove();\n  }, [aui]);\n\n  return (\n    <Pressable onPress={handleRemove} {...pressableProps}>\n      {children}\n    </Pressable>\n  );\n};\n"],"mappings":";;;;;;AASA,MAAaQ,oBAAmBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAA,IAAAJ;CAAA,IAAAK;CAAA,IAAAF,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAF,aAAAK,kBAAAH;EAGTC,EAAA,KAAAD;EAAAC,EAAA,KAAAH;EAAAG,EAAA,KAAAE;CAAA,OAAA;EAAAL,WAAAG,EAAA;EAAAE,iBAAAF,EAAA;CAAA;CACtB,MAAAG,MAAYT,OAAO;CAAE,IAAAU;CAAA,IAAAJ,EAAA,OAAAG,IAAAE,YAAA;EAEYD,WAAA;GAC/BD,IAAGE,WAAWC,OAAQ;EAAC;EACxBN,EAAA,KAAAG,IAAAE;EAAAL,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAFD,MAAAO,eAAqBH;CAEX,IAAAI;CAAA,IAAAR,EAAA,OAAAH,YAAAG,EAAA,OAAAO,gBAAAP,EAAA,OAAAE,gBAAA;EAGRM,KAAA,oBAAC,WAAD;GAAoBD,SAAAA;GAAY,GAAML;GACnCL;EADO,CAAA;EAEEG,EAAA,KAAAH;EAAAG,EAAA,KAAAO;EAAAP,EAAA,KAAAE;EAAAF,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAAA,OAFZQ;AAEY"}

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.js

@@ -20,11 +20,11 @@ const ChainOfThoughtAccordionTrigger = (t0) => {
 	const aui = useAui();
 	const collapsed = useAuiState(_temp);
 	let t1;
-	if ($[3] !== aui || $[4] !== collapsed) {
+	if ($[3] !== aui.chainOfThought || $[4] !== collapsed) {
 		t1 = () => {
-			aui.chainOfThought().setCollapsed(!collapsed);
+			aui.chainOfThought.setCollapsed(!collapsed);
 		};
-		$[3] = aui;
+		$[3] = aui.chainOfThought;
 		$[4] = collapsed;
 		$[5] = t1;
 	} else t1 = $[5];

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.js.map

@@ -1 +1 @@
-{"version":3,"file":"ChainOfThoughtAccordionTrigger.js","names":["useCallback","ReactNode","Pressable","PressableProps","useAuiState","useAui","ChainOfThoughtAccordionTriggerProps","Omit","children","ChainOfThoughtAccordionTrigger","t0","$","_c","pressableProps","aui","collapsed","_temp","t1","chainOfThought","setCollapsed","onPress","t2","s"],"sources":["../../../src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.tsx"],"sourcesContent":["import { useCallback, type ReactNode } from \"react\";\nimport { Pressable, type PressableProps } from \"react-native\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nexport type ChainOfThoughtAccordionTriggerProps = Omit<\n  PressableProps,\n  \"onPress\"\n> & {\n  children: ReactNode;\n};\n\nexport const ChainOfThoughtAccordionTrigger = ({\n  children,\n  ...pressableProps\n}: ChainOfThoughtAccordionTriggerProps) => {\n  const aui = useAui();\n  const collapsed = useAuiState((s) => s.chainOfThought.collapsed);\n\n  const onPress = useCallback(() => {\n    aui.chainOfThought().setCollapsed(!collapsed);\n  }, [aui, collapsed]);\n\n  return (\n    <Pressable onPress={onPress} {...pressableProps}>\n      {children}\n    </Pressable>\n  );\n};\n"],"mappings":";;;;;;AAWA,MAAaS,kCAAiCC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAA,IAAAJ;CAAA,IAAAK;CAAA,IAAAF,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAF,aAAAK,kBAAAH;EAGTC,EAAA,KAAAD;EAAAC,EAAA,KAAAH;EAAAG,EAAA,KAAAE;CAAA,OAAA;EAAAL,WAAAG,EAAA;EAAAE,iBAAAF,EAAA;CAAA;CACpC,MAAAG,MAAYT,OAAO;CACnB,MAAAU,YAAkBX,YAAYY,KAAiC;CAAE,IAAAC;CAAA,IAAAN,EAAA,OAAAG,OAAAH,EAAA,OAAAI,WAAA;EAErCE,WAAA;GAC1BH,IAAGI,eAAgB,CAAC,CAAAC,aAAc,CAACJ,SAAS;EAAC;EAC9CJ,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAFD,MAAAS,UAAgBH;CAEK,IAAAI;CAAA,IAAAV,EAAA,OAAAH,YAAAG,EAAA,OAAAS,WAAAT,EAAA,OAAAE,gBAAA;EAGnBQ,KAAA,oBAAC,WAAD;GAAoBD;GAAO,GAAMP;GAC9BL;EADO,CAAA;EAEEG,EAAA,KAAAH;EAAAG,EAAA,KAAAS;EAAAT,EAAA,KAAAE;EAAAF,EAAA,KAAAU;CAAA,OAAAA,KAAAV,EAAA;CAAA,OAFZU;AAEY;AAd8B,SAAAL,MAAAM,GAAA;CAAA,OAKPA,EAACJ,eAAeH;AAAU"}
+{"version":3,"file":"ChainOfThoughtAccordionTrigger.js","names":["useCallback","ReactNode","Pressable","PressableProps","useAuiState","useAui","ChainOfThoughtAccordionTriggerProps","Omit","children","ChainOfThoughtAccordionTrigger","t0","$","_c","pressableProps","aui","collapsed","_temp","t1","chainOfThought","setCollapsed","onPress","t2","s"],"sources":["../../../src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.tsx"],"sourcesContent":["import { useCallback, type ReactNode } from \"react\";\nimport { Pressable, type PressableProps } from \"react-native\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nexport type ChainOfThoughtAccordionTriggerProps = Omit<\n  PressableProps,\n  \"onPress\"\n> & {\n  children: ReactNode;\n};\n\nexport const ChainOfThoughtAccordionTrigger = ({\n  children,\n  ...pressableProps\n}: ChainOfThoughtAccordionTriggerProps) => {\n  const aui = useAui();\n  const collapsed = useAuiState((s) => s.chainOfThought.collapsed);\n\n  const onPress = useCallback(() => {\n    aui.chainOfThought.setCollapsed(!collapsed);\n  }, [aui, collapsed]);\n\n  return (\n    <Pressable onPress={onPress} {...pressableProps}>\n      {children}\n    </Pressable>\n  );\n};\n"],"mappings":";;;;;;AAWA,MAAaS,kCAAiCC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAA,IAAAJ;CAAA,IAAAK;CAAA,IAAAF,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAF,aAAAK,kBAAAH;EAGTC,EAAA,KAAAD;EAAAC,EAAA,KAAAH;EAAAG,EAAA,KAAAE;CAAA,OAAA;EAAAL,WAAAG,EAAA;EAAAE,iBAAAF,EAAA;CAAA;CACpC,MAAAG,MAAYT,OAAO;CACnB,MAAAU,YAAkBX,YAAYY,KAAiC;CAAE,IAAAC;CAAA,IAAAN,EAAA,OAAAG,IAAAI,kBAAAP,EAAA,OAAAI,WAAA;EAErCE,WAAA;GAC1BH,IAAGI,eAAeC,aAAc,CAACJ,SAAS;EAAC;EAC5CJ,EAAA,KAAAG,IAAAI;EAAAP,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAFD,MAAAS,UAAgBH;CAEK,IAAAI;CAAA,IAAAV,EAAA,OAAAH,YAAAG,EAAA,OAAAS,WAAAT,EAAA,OAAAE,gBAAA;EAGnBQ,KAAA,oBAAC,WAAD;GAAoBD;GAAO,GAAMP;GAC9BL;EADO,CAAA;EAEEG,EAAA,KAAAH;EAAAG,EAAA,KAAAS;EAAAT,EAAA,KAAAE;EAAAF,EAAA,KAAAU;CAAA,OAAAA,KAAAV,EAAA;CAAA,OAFZU;AAEY;AAd8B,SAAAL,MAAAM,GAAA;CAAA,OAKPA,EAACJ,eAAeH;AAAU"}

package/dist/primitives/composer/ComposerInput.js

@@ -58,11 +58,11 @@ const ComposerInput = (t0) => {
 		t2 = (value) => {
 			if (Platform.OS === "web") {
 				flushTapSync(() => {
-					aui.composer().setText(value);
+					aui.composer.setText(value);
 				});
 				return;
 			}
-			aui.composer().setText(value);
+			aui.composer.setText(value);
 		};
 		$[6] = aui;
 		$[7] = t2;
@@ -92,7 +92,7 @@ const ComposerInput = (t0) => {
 			if (nativeEvent.isComposing || nativeEvent.keyCode === 229) return;
 			if (nativeEvent.key === "Enter" && !nativeEvent.shiftKey) {
 				e.preventDefault?.();
-				aui.composer().send();
+				aui.composer.send();
 			}
 		};
 		$[11] = aui;

package/dist/primitives/composer/ComposerInput.js.map

@@ -1 +1 @@
-{"version":3,"file":"ComposerInput.js","names":["useCallback","useEffect","useRef","Platform","TextInput","NativeSyntheticEvent","TextInputKeyPressEventData","TextInputProps","useAui","useAuiState","flushTapSync","ComposerInputProps","Omit","submitMode","getWebTextareaMaxHeight","el","HTMLTextAreaElement","computedMaxHeight","globalThis","getComputedStyle","maxHeight","parsedComputedMaxHeight","Number","parseFloat","isFinite","inlineMaxHeight","style","adjustWebTextareaHeight","ref","React","RefObject","OS","current","rows","overflowX","overflowY","height","scrollHeight","needsScrollbar","nextHeight","Math","min","ComposerInput","t0","$","_c","numberOfLines","onKeyPressProp","props","t1","onKeyPress","undefined","aui","text","_temp","inputRef","t2","value","composer","setText","onChangeText","t3","Symbol","for","t4","t5","e","nativeEvent","shiftKey","isComposing","keyCode","key","Event","preventDefault","send","t6","t7","s"],"sources":["../../../src/primitives/composer/ComposerInput.tsx"],"sourcesContent":["import { useCallback, useEffect, useRef } from \"react\";\nimport {\n  Platform,\n  TextInput,\n  type NativeSyntheticEvent,\n  type TextInputKeyPressEventData,\n  type TextInputProps,\n} from \"react-native\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport { flushTapSync } from \"@assistant-ui/tap\";\n\nexport type ComposerInputProps = Omit<\n  TextInputProps,\n  \"value\" | \"onChangeText\"\n> & {\n  /**\n   * Controls how the Enter key submits messages (web only).\n   * - \"enter\": Plain Enter submits (Shift+Enter for newline)\n   * - \"none\": Enter inserts newline (no keyboard submission)\n   * @default \"enter\"\n   */\n  submitMode?: \"enter\" | \"none\";\n};\n\nconst getWebTextareaMaxHeight = (el: HTMLTextAreaElement) => {\n  const computedMaxHeight = globalThis.getComputedStyle?.(el).maxHeight;\n  const parsedComputedMaxHeight = Number.parseFloat(computedMaxHeight ?? \"\");\n  if (Number.isFinite(parsedComputedMaxHeight)) return parsedComputedMaxHeight;\n\n  const inlineMaxHeight = Number.parseFloat(el.style.maxHeight);\n  return Number.isFinite(inlineMaxHeight) ? inlineMaxHeight : null;\n};\n\nconst adjustWebTextareaHeight = (ref: React.RefObject<TextInput | null>) => {\n  if (Platform.OS !== \"web\") return;\n  // On web, the TextInput ref IS the DOM element\n  const el = ref.current as unknown as HTMLTextAreaElement | null;\n  if (!el) return;\n  // Ensure rows=1 so scrollHeight reflects actual content\n  if (el.rows !== 1) el.rows = 1;\n\n  el.style.overflowX = \"hidden\";\n  el.style.overflowY = \"hidden\";\n  el.style.height = \"auto\";\n\n  const scrollHeight = el.scrollHeight;\n  const maxHeight = getWebTextareaMaxHeight(el);\n  const needsScrollbar = maxHeight !== null && scrollHeight > maxHeight;\n  const nextHeight =\n    maxHeight === null ? scrollHeight : Math.min(scrollHeight, maxHeight);\n\n  el.style.overflowY = needsScrollbar ? \"auto\" : \"hidden\";\n  el.style.height = `${nextHeight}px`;\n};\n\nexport const ComposerInput = ({\n  submitMode = \"enter\",\n  onKeyPress: onKeyPressProp,\n  numberOfLines,\n  style,\n  ...props\n}: ComposerInputProps) => {\n  const aui = useAui();\n  const text = useAuiState((s) => s.composer.text);\n  const inputRef = useRef<TextInput>(null);\n\n  const onChangeText = useCallback(\n    (value: string) => {\n      if (Platform.OS === \"web\") {\n        // Keep the controlled value in sync mid-IME so react-dom does not reset the textarea to a stale value\n        flushTapSync(() => {\n          aui.composer().setText(value);\n        });\n        return;\n      }\n      aui.composer().setText(value);\n    },\n    [aui],\n  );\n\n  // Auto-resize textarea on web when text changes\n  useEffect(() => {\n    void text; // trigger re-run when text changes\n    adjustWebTextareaHeight(inputRef);\n  }, [text]);\n\n  const onKeyPress = useCallback(\n    (e: NativeSyntheticEvent<TextInputKeyPressEventData>) => {\n      onKeyPressProp?.(e);\n\n      if (Platform.OS !== \"web\") return;\n      if (submitMode !== \"enter\") return;\n\n      const nativeEvent = e.nativeEvent as TextInputKeyPressEventData & {\n        shiftKey?: boolean;\n        isComposing?: boolean;\n        keyCode?: number;\n      };\n      // react-native-web forwards the raw DOM keydown before its own composition guard, so re-check it here (mirrors isEventComposing)\n      if (nativeEvent.isComposing || nativeEvent.keyCode === 229) return;\n      if (nativeEvent.key === \"Enter\" && !nativeEvent.shiftKey) {\n        (e as unknown as Event).preventDefault?.();\n        aui.composer().send();\n      }\n    },\n    [aui, submitMode, onKeyPressProp],\n  );\n\n  return (\n    <TextInput\n      ref={inputRef}\n      value={text}\n      onChangeText={onChangeText}\n      onKeyPress={onKeyPress}\n      numberOfLines={numberOfLines ?? (Platform.OS === \"web\" ? 1 : undefined)}\n      style={style}\n      {...props}\n    />\n  );\n};\n"],"mappings":";;;;;;;AAwBA,MAAMc,2BAA2BC,OAA4B;CAC3D,MAAME,oBAAoBC,WAAWC,mBAAmBJ,EAAE,CAAC,CAACK;CAC5D,MAAMC,0BAA0BC,OAAOC,WAAWN,qBAAqB,EAAE;CACzE,IAAIK,OAAOE,SAASH,uBAAuB,GAAG,OAAOA;CAErD,MAAMI,kBAAkBH,OAAOC,WAAWR,GAAGW,MAAMN,SAAS;CAC5D,OAAOE,OAAOE,SAASC,eAAe,IAAIA,kBAAkB;AAC9D;AAEA,MAAME,2BAA2BC,QAA2C;CAC1E,IAAIzB,SAAS4B,OAAO,OAAO;CAE3B,MAAMhB,KAAKa,IAAII;CACf,IAAI,CAACjB,IAAI;CAET,IAAIA,GAAGkB,SAAS,GAAGlB,GAAGkB,OAAO;CAE7BlB,GAAGW,MAAMQ,YAAY;CACrBnB,GAAGW,MAAMS,YAAY;CACrBpB,GAAGW,MAAMU,SAAS;CAElB,MAAMC,eAAetB,GAAGsB;CACxB,MAAMjB,YAAYN,wBAAwBC,EAAE;CAC5C,MAAMuB,iBAAiBlB,cAAc,QAAQiB,eAAejB;CAC5D,MAAMmB,aACJnB,cAAc,OAAOiB,eAAeG,KAAKC,IAAIJ,cAAcjB,SAAS;CAEtEL,GAAGW,MAAMS,YAAYG,iBAAiB,SAAS;CAC/CvB,GAAGW,MAAMU,SAAS,GAAGG,WAAU;AACjC;AAEA,MAAaG,iBAAgBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAtB;CAAA,IAAAuB;CAAA,IAAAL,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAA9B,YAAAoC,IAAAC,YAAAH,gBAAAD,eAAApB,UAAAsB,SAAAL;EAMTC,EAAA,KAAAD;EAAAC,EAAA,KAAAE;EAAAF,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAlB;EAAAkB,EAAA,KAAAK;CAAA,OAAA;EAAAH,gBAAAF,EAAA;EAAAG,iBAAAH,EAAA;EAAAI,QAAAJ,EAAA;EAAAlB,QAAAkB,EAAA;EAAAK,KAAAL,EAAA;CAAA;CALnB,MAAA/B,aAAAoC,OAAAE,KAAAA,IAAA,UAAAF;CAMA,MAAAG,MAAY5C,OAAO;CACnB,MAAA6C,OAAa5C,YAAY6C,KAAsB;CAC/C,MAAAC,WAAiBrD,OAAkB,IAAI;CAAE,IAAAsD;CAAA,IAAAZ,EAAA,OAAAQ,KAAA;EAGvCI,MAAAC,UAAA;GACE,IAAItD,SAAQ4B,OAAQ,OAAK;IAEvBrB,mBAAa;KACX0C,IAAGM,SAAU,CAAC,CAAAC,QAASF,KAAK;IAAC,CAC9B;IAAC;GAAA;GAGJL,IAAGM,SAAU,CAAC,CAAAC,QAASF,KAAK;EAAC;EAC9Bb,EAAA,KAAAQ;EAAAR,EAAA,KAAAY;CAAA,OAAAA,KAAAZ,EAAA;CAVH,MAAAgB,eAAqBJ;CAYnB,IAAAK;CAAA,IAAAjB,EAAA,OAAAkB,OAAAC,IAAA,2BAAA,GAAA;EAGQF,WAAA;GAERlC,wBAAwB4B,QAAQ;EAAC;EAClCX,EAAA,KAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAAA,IAAAoB;CAAA,IAAApB,EAAA,OAAAS,MAAA;EAAEW,KAAA,CAACX,IAAI;EAACT,EAAA,KAAAS;EAAAT,EAAA,MAAAoB;CAAA,OAAAA,KAAApB,EAAA;CAHT3C,UAAU4D,IAGPG,EAAM;CAAC,IAAAC;CAAA,IAAArB,EAAA,QAAAQ,OAAAR,EAAA,QAAAG,kBAAAH,EAAA,QAAA/B,YAAA;EAGRoD,MAAAC,MAAA;GACEnB,iBAAiBmB,CAAC;GAElB,IAAI/D,SAAQ4B,OAAQ,OAAK;GACzB,IAAIlB,eAAe,SAAO;GAE1B,MAAAsD,cAAoBD,EAACC;GAMrB,IAAIA,YAAWE,eAAgBF,YAAWG,YAAa,KAAG;GAC1D,IAAIH,YAAWI,QAAS,WAApB,CAAgCJ,YAAWC,UAAS;IACtD,EAAsBK,iBAAmB;IACzCrB,IAAGM,SAAU,CAAC,CAAAgB,KAAM;GAAC;EACtB;EACF9B,EAAA,MAAAQ;EAAAR,EAAA,MAAAG;EAAAH,EAAA,MAAA/B;EAAA+B,EAAA,MAAAqB;CAAA,OAAAA,KAAArB,EAAA;CAlBH,MAAAM,aAAmBe;CA4BA,MAAAU,KAAA7B,kBAAkB3C,SAAQ4B,OAAQ,QAAhB,IAAAoB,KAAAA;CAAsC,IAAAyB;CAAA,IAAAhC,EAAA,QAAAgB,gBAAAhB,EAAA,QAAAM,cAAAN,EAAA,QAAAI,SAAAJ,EAAA,QAAAlB,SAAAkB,EAAA,QAAA+B,MAAA/B,EAAA,QAAAS,MAAA;EALzEuB,KAAA,oBAAC,WAAD;GACOrB,KAAAA;GACEF,OAAAA;GACOO;GACFV;GACG,eAAAyB;GACRjD;GAAK,GACRsB;EAAK,CAAA;EACTJ,EAAA,MAAAgB;EAAAhB,EAAA,MAAAM;EAAAN,EAAA,MAAAI;EAAAJ,EAAA,MAAAlB;EAAAkB,EAAA,MAAA+B;EAAA/B,EAAA,MAAAS;EAAAT,EAAA,MAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAAA,OARFgC;AAQE;AA9DuB,SAAAtB,MAAAuB,GAAA;CAAA,OAQKA,EAACnB,SAASL;AAAK"}
+{"version":3,"file":"ComposerInput.js","names":["useCallback","useEffect","useRef","Platform","TextInput","NativeSyntheticEvent","TextInputKeyPressEventData","TextInputProps","useAui","useAuiState","flushTapSync","ComposerInputProps","Omit","submitMode","getWebTextareaMaxHeight","el","HTMLTextAreaElement","computedMaxHeight","globalThis","getComputedStyle","maxHeight","parsedComputedMaxHeight","Number","parseFloat","isFinite","inlineMaxHeight","style","adjustWebTextareaHeight","ref","React","RefObject","OS","current","rows","overflowX","overflowY","height","scrollHeight","needsScrollbar","nextHeight","Math","min","ComposerInput","t0","$","_c","numberOfLines","onKeyPressProp","props","t1","onKeyPress","undefined","aui","text","_temp","inputRef","t2","value","composer","setText","onChangeText","t3","Symbol","for","t4","t5","e","nativeEvent","shiftKey","isComposing","keyCode","key","Event","preventDefault","send","t6","t7","s"],"sources":["../../../src/primitives/composer/ComposerInput.tsx"],"sourcesContent":["import { useCallback, useEffect, useRef } from \"react\";\nimport {\n  Platform,\n  TextInput,\n  type NativeSyntheticEvent,\n  type TextInputKeyPressEventData,\n  type TextInputProps,\n} from \"react-native\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport { flushTapSync } from \"@assistant-ui/tap\";\n\nexport type ComposerInputProps = Omit<\n  TextInputProps,\n  \"value\" | \"onChangeText\"\n> & {\n  /**\n   * Controls how the Enter key submits messages (web only).\n   * - \"enter\": Plain Enter submits (Shift+Enter for newline)\n   * - \"none\": Enter inserts newline (no keyboard submission)\n   * @default \"enter\"\n   */\n  submitMode?: \"enter\" | \"none\";\n};\n\nconst getWebTextareaMaxHeight = (el: HTMLTextAreaElement) => {\n  const computedMaxHeight = globalThis.getComputedStyle?.(el).maxHeight;\n  const parsedComputedMaxHeight = Number.parseFloat(computedMaxHeight ?? \"\");\n  if (Number.isFinite(parsedComputedMaxHeight)) return parsedComputedMaxHeight;\n\n  const inlineMaxHeight = Number.parseFloat(el.style.maxHeight);\n  return Number.isFinite(inlineMaxHeight) ? inlineMaxHeight : null;\n};\n\nconst adjustWebTextareaHeight = (ref: React.RefObject<TextInput | null>) => {\n  if (Platform.OS !== \"web\") return;\n  // On web, the TextInput ref IS the DOM element\n  const el = ref.current as unknown as HTMLTextAreaElement | null;\n  if (!el) return;\n  // Ensure rows=1 so scrollHeight reflects actual content\n  if (el.rows !== 1) el.rows = 1;\n\n  el.style.overflowX = \"hidden\";\n  el.style.overflowY = \"hidden\";\n  el.style.height = \"auto\";\n\n  const scrollHeight = el.scrollHeight;\n  const maxHeight = getWebTextareaMaxHeight(el);\n  const needsScrollbar = maxHeight !== null && scrollHeight > maxHeight;\n  const nextHeight =\n    maxHeight === null ? scrollHeight : Math.min(scrollHeight, maxHeight);\n\n  el.style.overflowY = needsScrollbar ? \"auto\" : \"hidden\";\n  el.style.height = `${nextHeight}px`;\n};\n\nexport const ComposerInput = ({\n  submitMode = \"enter\",\n  onKeyPress: onKeyPressProp,\n  numberOfLines,\n  style,\n  ...props\n}: ComposerInputProps) => {\n  const aui = useAui();\n  const text = useAuiState((s) => s.composer.text);\n  const inputRef = useRef<TextInput>(null);\n\n  const onChangeText = useCallback(\n    (value: string) => {\n      if (Platform.OS === \"web\") {\n        // Keep the controlled value in sync mid-IME so react-dom does not reset the textarea to a stale value\n        flushTapSync(() => {\n          aui.composer.setText(value);\n        });\n        return;\n      }\n      aui.composer.setText(value);\n    },\n    [aui],\n  );\n\n  // Auto-resize textarea on web when text changes\n  useEffect(() => {\n    void text; // trigger re-run when text changes\n    adjustWebTextareaHeight(inputRef);\n  }, [text]);\n\n  const onKeyPress = useCallback(\n    (e: NativeSyntheticEvent<TextInputKeyPressEventData>) => {\n      onKeyPressProp?.(e);\n\n      if (Platform.OS !== \"web\") return;\n      if (submitMode !== \"enter\") return;\n\n      const nativeEvent = e.nativeEvent as TextInputKeyPressEventData & {\n        shiftKey?: boolean;\n        isComposing?: boolean;\n        keyCode?: number;\n      };\n      // react-native-web forwards the raw DOM keydown before its own composition guard, so re-check it here (mirrors isEventComposing)\n      if (nativeEvent.isComposing || nativeEvent.keyCode === 229) return;\n      if (nativeEvent.key === \"Enter\" && !nativeEvent.shiftKey) {\n        (e as unknown as Event).preventDefault?.();\n        aui.composer.send();\n      }\n    },\n    [aui, submitMode, onKeyPressProp],\n  );\n\n  return (\n    <TextInput\n      ref={inputRef}\n      value={text}\n      onChangeText={onChangeText}\n      onKeyPress={onKeyPress}\n      numberOfLines={numberOfLines ?? (Platform.OS === \"web\" ? 1 : undefined)}\n      style={style}\n      {...props}\n    />\n  );\n};\n"],"mappings":";;;;;;;AAwBA,MAAMc,2BAA2BC,OAA4B;CAC3D,MAAME,oBAAoBC,WAAWC,mBAAmBJ,EAAE,CAAC,CAACK;CAC5D,MAAMC,0BAA0BC,OAAOC,WAAWN,qBAAqB,EAAE;CACzE,IAAIK,OAAOE,SAASH,uBAAuB,GAAG,OAAOA;CAErD,MAAMI,kBAAkBH,OAAOC,WAAWR,GAAGW,MAAMN,SAAS;CAC5D,OAAOE,OAAOE,SAASC,eAAe,IAAIA,kBAAkB;AAC9D;AAEA,MAAME,2BAA2BC,QAA2C;CAC1E,IAAIzB,SAAS4B,OAAO,OAAO;CAE3B,MAAMhB,KAAKa,IAAII;CACf,IAAI,CAACjB,IAAI;CAET,IAAIA,GAAGkB,SAAS,GAAGlB,GAAGkB,OAAO;CAE7BlB,GAAGW,MAAMQ,YAAY;CACrBnB,GAAGW,MAAMS,YAAY;CACrBpB,GAAGW,MAAMU,SAAS;CAElB,MAAMC,eAAetB,GAAGsB;CACxB,MAAMjB,YAAYN,wBAAwBC,EAAE;CAC5C,MAAMuB,iBAAiBlB,cAAc,QAAQiB,eAAejB;CAC5D,MAAMmB,aACJnB,cAAc,OAAOiB,eAAeG,KAAKC,IAAIJ,cAAcjB,SAAS;CAEtEL,GAAGW,MAAMS,YAAYG,iBAAiB,SAAS;CAC/CvB,GAAGW,MAAMU,SAAS,GAAGG,WAAU;AACjC;AAEA,MAAaG,iBAAgBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAtB;CAAA,IAAAuB;CAAA,IAAAL,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAA9B,YAAAoC,IAAAC,YAAAH,gBAAAD,eAAApB,UAAAsB,SAAAL;EAMTC,EAAA,KAAAD;EAAAC,EAAA,KAAAE;EAAAF,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAlB;EAAAkB,EAAA,KAAAK;CAAA,OAAA;EAAAH,gBAAAF,EAAA;EAAAG,iBAAAH,EAAA;EAAAI,QAAAJ,EAAA;EAAAlB,QAAAkB,EAAA;EAAAK,KAAAL,EAAA;CAAA;CALnB,MAAA/B,aAAAoC,OAAAE,KAAAA,IAAA,UAAAF;CAMA,MAAAG,MAAY5C,OAAO;CACnB,MAAA6C,OAAa5C,YAAY6C,KAAsB;CAC/C,MAAAC,WAAiBrD,OAAkB,IAAI;CAAE,IAAAsD;CAAA,IAAAZ,EAAA,OAAAQ,KAAA;EAGvCI,MAAAC,UAAA;GACE,IAAItD,SAAQ4B,OAAQ,OAAK;IAEvBrB,mBAAa;KACX0C,IAAGM,SAASC,QAASF,KAAK;IAAC,CAC5B;IAAC;GAAA;GAGJL,IAAGM,SAASC,QAASF,KAAK;EAAC;EAC5Bb,EAAA,KAAAQ;EAAAR,EAAA,KAAAY;CAAA,OAAAA,KAAAZ,EAAA;CAVH,MAAAgB,eAAqBJ;CAYnB,IAAAK;CAAA,IAAAjB,EAAA,OAAAkB,OAAAC,IAAA,2BAAA,GAAA;EAGQF,WAAA;GAERlC,wBAAwB4B,QAAQ;EAAC;EAClCX,EAAA,KAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAAA,IAAAoB;CAAA,IAAApB,EAAA,OAAAS,MAAA;EAAEW,KAAA,CAACX,IAAI;EAACT,EAAA,KAAAS;EAAAT,EAAA,MAAAoB;CAAA,OAAAA,KAAApB,EAAA;CAHT3C,UAAU4D,IAGPG,EAAM;CAAC,IAAAC;CAAA,IAAArB,EAAA,QAAAQ,OAAAR,EAAA,QAAAG,kBAAAH,EAAA,QAAA/B,YAAA;EAGRoD,MAAAC,MAAA;GACEnB,iBAAiBmB,CAAC;GAElB,IAAI/D,SAAQ4B,OAAQ,OAAK;GACzB,IAAIlB,eAAe,SAAO;GAE1B,MAAAsD,cAAoBD,EAACC;GAMrB,IAAIA,YAAWE,eAAgBF,YAAWG,YAAa,KAAG;GAC1D,IAAIH,YAAWI,QAAS,WAApB,CAAgCJ,YAAWC,UAAS;IACtD,EAAsBK,iBAAmB;IACzCrB,IAAGM,SAASgB,KAAM;GAAC;EACpB;EACF9B,EAAA,MAAAQ;EAAAR,EAAA,MAAAG;EAAAH,EAAA,MAAA/B;EAAA+B,EAAA,MAAAqB;CAAA,OAAAA,KAAArB,EAAA;CAlBH,MAAAM,aAAmBe;CA4BA,MAAAU,KAAA7B,kBAAkB3C,SAAQ4B,OAAQ,QAAhB,IAAAoB,KAAAA;CAAsC,IAAAyB;CAAA,IAAAhC,EAAA,QAAAgB,gBAAAhB,EAAA,QAAAM,cAAAN,EAAA,QAAAI,SAAAJ,EAAA,QAAAlB,SAAAkB,EAAA,QAAA+B,MAAA/B,EAAA,QAAAS,MAAA;EALzEuB,KAAA,oBAAC,WAAD;GACOrB,KAAAA;GACEF,OAAAA;GACOO;GACFV;GACG,eAAAyB;GACRjD;GAAK,GACRsB;EAAK,CAAA;EACTJ,EAAA,MAAAgB;EAAAhB,EAAA,MAAAM;EAAAN,EAAA,MAAAI;EAAAJ,EAAA,MAAAlB;EAAAkB,EAAA,MAAA+B;EAAA/B,EAAA,MAAAS;EAAAT,EAAA,MAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAAA,OARFgC;AAQE;AA9DuB,SAAAtB,MAAAuB,GAAA;CAAA,OAQKA,EAACnB,SAASL;AAAK"}

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;;cAqEK,mBAAkB,YAAA,gBAAA,aAAA,iBAAA,cAAA,YAAA,cAQ5B,wCAAmB,IAAA"}
+{"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"}

package/dist/primitives/message/MessageContent.js

@@ -21,19 +21,15 @@ const ToolUIDisplay = (t0) => {
 	const aui = useAui();
 	let t1;
 	if ($[0] !== part.toolName) {
-		t1 = (s) => {
-			const renders = s.tools.tools[part.toolName];
-			if (Array.isArray(renders)) return renders[0];
-			return renders;
-		};
+		t1 = (s) => s.tools.toolUIs[part.toolName]?.[0]?.render;
 		$[0] = part.toolName;
 		$[1] = t1;
 	} else t1 = $[1];
 	const Render = useAuiState(t1);
 	let t2;
-	if ($[2] !== aui || $[3] !== index) {
-		t2 = aui.message().part({ index });
-		$[2] = aui;
+	if ($[2] !== aui.message || $[3] !== index) {
+		t2 = aui.message.part({ index });
+		$[2] = aui.message;
 		$[3] = index;
 		$[4] = t2;
 	} else t2 = $[4];

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","renders","tools","Array","isArray","Render","t2","message","partMethods","t3","t4","addToolResult","respondToToolApproval","resumeToolCall","DataUIDisplay","name","dataRenderers","renderers","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((s) => {\n    const renders = s.tools.tools[part.toolName];\n    if (Array.isArray(renders)) return renders[0];\n    return renders;\n  });\n\n  const partMethods = useMemo(\n    () => aui.message().part({ index }),\n    [aui, index],\n  );\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;EACML,MAAAM,MAAA;GACzB,MAAAC,UAAgBD,EAACE,MAAMA,MAAOtB,KAAImB;GAClC,IAAII,MAAKC,QAASH,OAAO,GAAC,OAASA,QAAO;GAAI,OACvCA;EAAO;EACfT,EAAA,KAAAZ,KAAAmB;EAAAP,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAJD,MAAAa,SAAehC,YAAYqB,EAI1B;CAAE,IAAAY;CAAA,IAAAd,EAAA,OAAAM,OAAAN,EAAA,OAAAT,OAAA;EAGKuB,KAAAR,IAAGS,QAAS,CAAC,CAAA3B,KAAM,EAAAG,MAAQ,CAAC;EAACS,EAAA,KAAAM;EAAAN,EAAA,KAAAT;EAAAS,EAAA,KAAAc;CAAA,OAAAA,KAAAd,EAAA;CADrC,MAAAgB,cACQF;CAIR,IAAID,QAAM;EAGC,MAAAI,KAAA7B;EAAgC,IAAA8B;EAAA,IAAAlB,EAAA,OAAAa,UAAAb,EAAA,OAAAgB,YAAAG,iBAAAnB,EAAA,OAAAgB,YAAAI,yBAAApB,EAAA,OAAAgB,YAAAK,kBAAArB,EAAA,OAAAiB,IAAA;GADvCC,KAAA,oBAAC,QAAD;IAAO,GACAD;IACM,WAAAD,YAAWG;IACd,QAAAH,YAAWK;IACA,mBAAAL,YAAWI;GAAsB,CAAA;GACpDpB,EAAA,KAAAa;GAAAb,EAAA,KAAAgB,YAAAG;GAAAnB,EAAA,KAAAgB,YAAAI;GAAApB,EAAA,KAAAgB,YAAAK;GAAArB,EAAA,KAAAiB;GAAAjB,EAAA,MAAAkB;EAAA,OAAAA,KAAAlB,EAAA;EAAA,OALFkB;CAKE;CAGN,IAAIb,UAAQ;EAAA,IAAAY;EAAA,IAAAjB,EAAA,QAAAK,YAAAL,EAAA,QAAAT,SAAAS,EAAA,QAAAZ,MAAA;GAAS6B,KAAA,oBAAC,UAAD;IAAgB7B;IAAaG;GAAK,CAAA;GAAIS,EAAA,MAAAK;GAAAL,EAAA,MAAAT;GAAAS,EAAA,MAAAZ;GAAAY,EAAA,MAAAiB;EAAA,OAAAA,KAAAjB,EAAA;EAAA,OAAtCiB;CAAsC;CAAC,OACrD;AAAI;AAGb,MAAMK,iBAAgBvB,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAI,UAAAjB,MAAAG,UAAAQ;CAUtB,IAAAG;CAAA,IAAAF,EAAA,OAAAZ,KAAAmC,MAAA;EAC4BrB,MAAAM,MAAA;GACzB,MAAAC,UAAgBD,EAACgB,cAAcC,UAAWrC,KAAImC;GAC9C,IAAIZ,MAAKC,QAASH,OAAO,GAAC,OAASA,QAAO;GAAI,OACvCA;EAAO;EACfT,EAAA,KAAAZ,KAAAmC;EAAAvB,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAJD,MAAAa,SAAehC,YAAYqB,EAI1B;CACD,IAAIW,QAAM;EAAsB,MAAAC,KAAA1B;EAA4B,IAAA6B;EAAA,IAAAjB,EAAA,OAAAa,UAAAb,EAAA,OAAAc,IAAA;GAAzCG,KAAA,oBAAC,QAAD,EAAO,GAAMH,GAA4B,CAAA;GAAKd,EAAA,KAAAa;GAAAb,EAAA,KAAAc;GAAAd,EAAA,KAAAiB;EAAA,OAAAA,KAAAjB,EAAA;EAAA,OAA9CiB;CAA8C;CACjE,IAAIZ,UAAQ;EAAA,IAAAS;EAAA,IAAAd,EAAA,OAAAK,YAAAL,EAAA,OAAAT,SAAAS,EAAA,OAAAZ,MAAA;GAAS0B,KAAA,oBAAC,UAAD;IAAgB1B;IAAaG;GAAK,CAAA;GAAIS,EAAA,KAAAK;GAAAL,EAAA,KAAAT;GAAAS,EAAA,KAAAZ;GAAAY,EAAA,KAAAc;EAAA,OAAAA,KAAAd,EAAA;EAAA,OAAtCc;CAAsC;CAAC,OACrD;AAAI;AAGb,MAAaY,kBAAiB3B,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAf,YAAAM,gBAAAC,aAAAC,iBAAAC,cAAAC,YAAAC,eAAAE;CAS7B,MAAA4B,UAAgB9C,YAAY+C,KAAwB;CAAE,IAAA1B;CAAA,IAAAF,EAAA,OAAA2B,WAAA3B,EAAA,OAAAH,cAAAG,EAAA,OAAAJ,cAAAI,EAAA,OAAAP,eAAAO,EAAA,OAAAN,mBAAAM,EAAA,OAAAL,gBAAAK,EAAA,OAAAd,cAAAc,EAAA,OAAAR,gBAAA;EAAA,IAAAsB;EAAA,IAAAd,EAAA,OAAAH,cAAAG,EAAA,QAAAJ,cAAAI,EAAA,QAAAP,eAAAO,EAAA,QAAAN,mBAAAM,EAAA,QAAAL,gBAAAK,EAAA,QAAAd,cAAAc,EAAA,QAAAR,gBAAA;GAIrCsB,MAAA1B,MAAAG,UAAA;IACX,MAAAsC,MAAY,GAAGzC,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,GAAMyC,GAAN;KAME,KAEV,aAAW,OAEZ,oBAAC,YAAD,EAAA,UACE,oBAAC,eAAD;MACYrC,UAAAA;MACJJ;MACCG;KAAK,CAAA,EAJP,GAAMsC,GAAN;KAME,KAEV;MACH,IAAI,CAACpC,aAAW,OAAS;MAAK,OAE5B,oBAAC,YAAD,EAAA,UAAqBA,YAAY;OAAAL;OAAAG;MAAc,CAAC,EAAvC,GAAMsC,GAAN;KAAoD,KAE5D;MACH,IAAI,CAACnC,iBAAe,OAAS;MAAK,OAEhC,oBAAC,YAAD,EAAA,UAAqBA,gBAAgB;OAAAN;OAAAG;MAAc,CAAC,EAA3C,GAAMsC,GAAN;KAAwD,KAEhE;MACH,IAAI,CAAClC,cAAY,OAAS;MAAK,OAE7B,oBAAC,YAAD,EAAA,UAAqBA,aAAa;OAAAP;OAAAG;MAAc,CAAC,EAAxC,GAAMsC,GAAN;KAAqD,KAE7D;MACH,IAAI,CAACjC,YAAU,OAAS;MAAK,OACtB,oBAAC,YAAD,EAAA,UAAqBA,WAAW;OAAAR;OAAAG;MAAc,CAAC,EAAtC,GAAMsC,GAAN;KAAmD,KAChE,QAAM,OAEP,oBAAC,YAAD,EAAA,UACE,oBAAC,eAAD;MACYhC,UAAAA;MACJT;MACCG;KAAK,CAAA,EAJP,GAAMsC,GAAN;KAME,SAAA,OAGN;IACX;GAAC;GACF7B,EAAA,KAAAH;GAAAG,EAAA,MAAAJ;GAAAI,EAAA,MAAAP;GAAAO,EAAA,MAAAN;GAAAM,EAAA,MAAAL;GAAAK,EAAA,MAAAd;GAAAc,EAAA,MAAAR;GAAAQ,EAAA,MAAAc;EAAA,OAAAA,KAAAd,EAAA;EAtDAE,KAAAyB,QAAOG,IAAKhB,EAsDZ;EAACd,EAAA,KAAA2B;EAAA3B,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,IAAAc;CAAA,IAAAd,EAAA,QAAAE,IAAA;EAvDJY,KAAA,oBAAA,UAAA,EAAA,UACGZ,GAsDC,CAAA;EACDF,EAAA,MAAAE;EAAAF,EAAA,MAAAc;CAAA,OAAAA,KAAAd,EAAA;CAAA,OAxDHc;AAwDG;AApEuB,SAAAc,MAAApB,GAAA;CAAA,OASOA,EAACO,QAAQY;AAAQ"}
+{"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"}

package/dist/primitives/thread/ThreadMessages.js

@@ -66,7 +66,7 @@ const ThreadMessageByChildren = memo(({ index, children }) => {
 	return /* @__PURE__ */ jsx(MessageByIndexProvider, {
 		index,
 		children: /* @__PURE__ */ jsx(RenderChildrenWithAccessor, {
-			getItemState: (aui) => aui.thread().message({ index }).getState(),
+			getItemState: (aui) => aui.thread.message({ index }).getState(),
 			children: (getItem) => children({ get message() {
 				return getItem();
 			} })

package/dist/primitives/thread/ThreadMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"ThreadMessages.js","names":["ComponentType","FC","ForwardedRef","ReactNode","RefObject","forwardRef","memo","useCallback","useEffect","useRef","FlatList","FlatListProps","LayoutChangeEvent","NativeScrollEvent","NativeSyntheticEvent","MessageState","ThreadMessage","RenderChildrenWithAccessor","useAuiEvent","useAuiState","MessageByIndexProvider","MessageComponents","Message","EditComposer","UserEditComposer","AssistantEditComposer","SystemEditComposer","UserMessage","AssistantMessage","SystemMessage","MessagesContent","components","children","value","message","ThreadMessagesFlatListProps","Omit","autoScroll","scrollToBottomOnRunStart","scrollToBottomOnInitialize","scrollToBottomOnThreadSwitch","ThreadMessagesProps","DEFAULT_SYSTEM_MESSAGE","AT_BOTTOM_THRESHOLD","getComponent","role","isEditing","_exhaustiveCheck","Error","ThreadMessageComponent","t0","$","_c","_temp","_temp2","t1","Component","t2","ThreadMessageByIndex","index","prev","next","displayName","ThreadMessageByChildren","aui","thread","getState","getItem","setForwardedRef","ref","T","current","useComposedFlatListRef","forwardedRef","flatListRef","node","setFlatListRef","const","useThreadMessagesFlatListAutoScroll","hasMessages","t3","t4","undefined","t5","Symbol","for","contentHeight","viewportHeight","scrollY","metricsRef","isAtBottomRef","lastScrollEventYRef","initializeScrollRequestedRef","pendingScrollToBottomRef","t6","updateIsAtBottom","t7","animated","contentHeight_0","viewportHeight_0","Math","max","scrollToEnd","scrollToBottom","t8","event","wasAtBottom","previousViewportHeight","viewportHeight_1","nativeEvent","layout","height","pending","handleLayout","t9","event_0","contentOffset","contentSize","layoutMeasurement","previousEventY","wasPinnedToBottom","y","upwardMove","handleScroll","t10","_width","metrics","previousContentHeight","wasAtBottom_0","pendingScroll","handleContentSizeChange","t11","t12","t13","t14","t15","ThreadMessagesFlatList","flatListProps","onContentSizeChange","onLayout","onScroll","scrollEventThrottle","messages","_temp3","length","handleAutoScrollContentSizeChange","handleAutoScrollLayout","handleAutoScrollScroll","renderItem","keyExtractor","_temp4","scrollTracking","width","ThreadMessages","props","s","s_0","composer","item","id"],"sources":["../../../src/primitives/thread/ThreadMessages.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type ForwardedRef,\n  type ReactNode,\n  type RefObject,\n  forwardRef,\n  memo,\n  useCallback,\n  useEffect,\n  useRef,\n} from \"react\";\nimport {\n  FlatList,\n  type FlatListProps,\n  type LayoutChangeEvent,\n  type NativeScrollEvent,\n  type NativeSyntheticEvent,\n} from \"react-native\";\nimport type { MessageState, ThreadMessage } from \"@assistant-ui/core\";\nimport {\n  RenderChildrenWithAccessor,\n  useAuiEvent,\n  useAuiState,\n} from \"@assistant-ui/store\";\nimport { MessageByIndexProvider } from \"@assistant-ui/core/react\";\n\ntype MessageComponents =\n  | {\n      Message: ComponentType;\n      EditComposer?: ComponentType | undefined;\n      UserEditComposer?: ComponentType | undefined;\n      AssistantEditComposer?: ComponentType | undefined;\n      SystemEditComposer?: ComponentType | undefined;\n      UserMessage?: ComponentType | undefined;\n      AssistantMessage?: ComponentType | undefined;\n      SystemMessage?: ComponentType | undefined;\n    }\n  | {\n      Message?: ComponentType | undefined;\n      EditComposer?: ComponentType | undefined;\n      UserEditComposer?: ComponentType | undefined;\n      AssistantEditComposer?: ComponentType | undefined;\n      SystemEditComposer?: ComponentType | undefined;\n      UserMessage: ComponentType;\n      AssistantMessage: ComponentType;\n      SystemMessage?: ComponentType | undefined;\n    };\n\ntype MessagesContent =\n  | {\n      /** @deprecated Use the children render function instead. */\n      components: MessageComponents;\n      children?: never;\n    }\n  | {\n      children: (value: { message: MessageState }) => ReactNode;\n      components?: never;\n    };\n\nexport type ThreadMessagesFlatListProps = Omit<\n  FlatListProps<ThreadMessage>,\n  \"data\" | \"renderItem\" | \"children\"\n> &\n  MessagesContent & {\n    autoScroll?: boolean | undefined;\n    scrollToBottomOnRunStart?: boolean | undefined;\n    scrollToBottomOnInitialize?: boolean | undefined;\n    scrollToBottomOnThreadSwitch?: boolean | undefined;\n  };\n\n/** @deprecated Use ThreadMessagesFlatListProps instead. */\nexport type ThreadMessagesProps = ThreadMessagesFlatListProps;\n\nconst DEFAULT_SYSTEM_MESSAGE = () => null;\nconst AT_BOTTOM_THRESHOLD = 4;\n\nconst getComponent = (\n  components: MessageComponents,\n  role: ThreadMessage[\"role\"],\n  isEditing: boolean,\n) => {\n  switch (role) {\n    case \"user\":\n      if (isEditing) {\n        return (\n          components.UserEditComposer ??\n          components.EditComposer ??\n          components.UserMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return components.UserMessage ?? (components.Message as ComponentType);\n      }\n    case \"assistant\":\n      if (isEditing) {\n        return (\n          components.AssistantEditComposer ??\n          components.EditComposer ??\n          components.AssistantMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return (\n          components.AssistantMessage ?? (components.Message as ComponentType)\n        );\n      }\n    case \"system\":\n      if (isEditing) {\n        return (\n          components.SystemEditComposer ??\n          components.EditComposer ??\n          components.SystemMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return (\n          components.SystemMessage ??\n          (components.Message as ComponentType) ??\n          DEFAULT_SYSTEM_MESSAGE\n        );\n      }\n    default: {\n      const _exhaustiveCheck: never = role;\n      throw new Error(`Unknown message role: ${_exhaustiveCheck}`);\n    }\n  }\n};\n\nconst ThreadMessageComponent: FC<{ components: MessageComponents }> = ({\n  components,\n}) => {\n  const role = useAuiState((s) => s.message.role);\n  const isEditing = useAuiState((s) => s.message.composer.isEditing);\n  const Component = getComponent(components, role, isEditing);\n\n  return <Component />;\n};\n\nconst ThreadMessageByIndex = memo(\n  ({ index, components }: { index: number; components: MessageComponents }) => {\n    return (\n      <MessageByIndexProvider index={index}>\n        <ThreadMessageComponent components={components} />\n      </MessageByIndexProvider>\n    );\n  },\n  (prev, next) =>\n    prev.index === next.index && prev.components === next.components,\n);\nThreadMessageByIndex.displayName = \"ThreadPrimitive.MessageByIndex\";\n\nconst ThreadMessageByChildren = memo(\n  ({\n    index,\n    children,\n  }: {\n    index: number;\n    children: (value: { message: MessageState }) => ReactNode;\n  }) => {\n    return (\n      <MessageByIndexProvider index={index}>\n        <RenderChildrenWithAccessor\n          getItemState={(aui) => aui.thread().message({ index }).getState()}\n        >\n          {(getItem) =>\n            children({\n              get message() {\n                return getItem();\n              },\n            })\n          }\n        </RenderChildrenWithAccessor>\n      </MessageByIndexProvider>\n    );\n  },\n  (prev, next) => prev.index === next.index && prev.children === next.children,\n);\nThreadMessageByChildren.displayName = \"ThreadPrimitive.MessageByChildren\";\n\nconst setForwardedRef = <T,>(ref: ForwardedRef<T>, value: T | null) => {\n  if (typeof ref === \"function\") {\n    ref(value);\n  } else if (ref) {\n    ref.current = value;\n  }\n};\n\nconst useComposedFlatListRef = (\n  forwardedRef: ForwardedRef<FlatList<ThreadMessage>>,\n) => {\n  const flatListRef = useRef<FlatList<ThreadMessage> | null>(null);\n\n  const setFlatListRef = useCallback(\n    (node: FlatList<ThreadMessage> | null) => {\n      flatListRef.current = node;\n      setForwardedRef(forwardedRef, node);\n    },\n    [forwardedRef],\n  );\n\n  return [flatListRef, setFlatListRef] as const;\n};\n\nconst useThreadMessagesFlatListAutoScroll = ({\n  flatListRef,\n  hasMessages,\n  autoScroll = true,\n  scrollToBottomOnRunStart = true,\n  scrollToBottomOnInitialize = true,\n  scrollToBottomOnThreadSwitch = true,\n}: {\n  flatListRef: RefObject<FlatList<ThreadMessage> | null>;\n  hasMessages: boolean;\n  autoScroll?: boolean | undefined;\n  scrollToBottomOnRunStart?: boolean | undefined;\n  scrollToBottomOnInitialize?: boolean | undefined;\n  scrollToBottomOnThreadSwitch?: boolean | undefined;\n}) => {\n  const metricsRef = useRef({\n    contentHeight: 0,\n    viewportHeight: 0,\n    scrollY: 0,\n  });\n  const isAtBottomRef = useRef(true);\n  const lastScrollEventYRef = useRef(0);\n  const initializeScrollRequestedRef = useRef(false);\n  const pendingScrollToBottomRef = useRef<false | { animated: boolean }>(false);\n\n  const updateIsAtBottom = useCallback(() => {\n    const { contentHeight, scrollY, viewportHeight } = metricsRef.current;\n    isAtBottomRef.current =\n      contentHeight <= viewportHeight ||\n      contentHeight - scrollY - viewportHeight <= AT_BOTTOM_THRESHOLD;\n  }, []);\n\n  // Commanding a scroll records the intended position immediately; the\n  // native scroll echo is bridged and throttled, so waiting for it lets a\n  // fast stream observe stale metrics and drop out of following.\n  const scrollToBottom = useCallback(\n    (animated: boolean) => {\n      const { contentHeight, viewportHeight } = metricsRef.current;\n      metricsRef.current.scrollY = Math.max(0, contentHeight - viewportHeight);\n      isAtBottomRef.current = true;\n      flatListRef.current?.scrollToEnd({ animated });\n    },\n    [flatListRef],\n  );\n\n  const handleLayout = useCallback(\n    (event: LayoutChangeEvent) => {\n      const wasAtBottom = isAtBottomRef.current;\n      const previousViewportHeight = metricsRef.current.viewportHeight;\n      const viewportHeight = event.nativeEvent.layout.height;\n      metricsRef.current.viewportHeight = viewportHeight;\n      updateIsAtBottom();\n      if (!wasAtBottom) return;\n      // Layout changes are never user gestures, so they must not unpin. Past\n      // the first measurement, a viewport change while pinned re-commands the\n      // bottom position, since no content-size event follows a bare keyboard\n      // open or close.\n      if (\n        autoScroll &&\n        previousViewportHeight !== 0 &&\n        viewportHeight !== previousViewportHeight\n      ) {\n        const pending = pendingScrollToBottomRef.current;\n        scrollToBottom(pending ? pending.animated : false);\n      } else {\n        isAtBottomRef.current = true;\n      }\n    },\n    [autoScroll, scrollToBottom, updateIsAtBottom],\n  );\n\n  const handleScroll = useCallback(\n    (event: NativeSyntheticEvent<NativeScrollEvent>) => {\n      const { contentOffset, contentSize, layoutMeasurement } =\n        event.nativeEvent;\n      const previousEventY = lastScrollEventYRef.current;\n      const wasPinnedToBottom = isAtBottomRef.current;\n      lastScrollEventYRef.current = contentOffset.y;\n      metricsRef.current = {\n        contentHeight: contentSize.height,\n        viewportHeight: layoutMeasurement.height,\n        scrollY: contentOffset.y,\n      };\n      updateIsAtBottom();\n      const upwardMove = contentOffset.y < previousEventY;\n      // Only a deliberate upward move unpins or cancels a pending scroll.\n      // Gestures are detected echo-to-echo because a commanded scroll\n      // optimistically moves the tracked position ahead of its ascending\n      // animation echoes, and those echoes must not unpin mid-flight.\n      if (wasPinnedToBottom && !upwardMove) {\n        isAtBottomRef.current = true;\n      }\n      if (!isAtBottomRef.current && upwardMove) {\n        pendingScrollToBottomRef.current = false;\n      }\n    },\n    [updateIsAtBottom],\n  );\n\n  const handleContentSizeChange = useCallback(\n    (_width: number, height: number) => {\n      const metrics = metricsRef.current;\n      const previousContentHeight = metrics.contentHeight;\n      const wasAtBottom = isAtBottomRef.current;\n      metrics.contentHeight = height;\n      updateIsAtBottom();\n\n      // FlatList.scrollToEnd is a no-op before the list has measured, so the\n      // initialize and thread-switch scrolls land on the next content-size\n      // event, once real metrics exist.\n      const pendingScroll = pendingScrollToBottomRef.current;\n      if (pendingScroll) {\n        pendingScrollToBottomRef.current = false;\n        scrollToBottom(pendingScroll.animated);\n        return;\n      }\n\n      if (!autoScroll) return;\n      if (!wasAtBottom) return;\n      if (previousContentHeight === 0) return;\n      if (height <= previousContentHeight) return;\n\n      scrollToBottom(false);\n    },\n    [autoScroll, scrollToBottom, updateIsAtBottom],\n  );\n\n  useEffect(() => {\n    if (!scrollToBottomOnInitialize) return;\n    if (!hasMessages) {\n      initializeScrollRequestedRef.current = false;\n      return;\n    }\n    if (initializeScrollRequestedRef.current) return;\n\n    initializeScrollRequestedRef.current = true;\n    pendingScrollToBottomRef.current = { animated: false };\n    scrollToBottom(false);\n  }, [hasMessages, scrollToBottom, scrollToBottomOnInitialize]);\n\n  useAuiEvent(\"thread.runStart\", () => {\n    if (!scrollToBottomOnRunStart) return;\n    pendingScrollToBottomRef.current = { animated: true };\n    scrollToBottom(true);\n  });\n\n  useAuiEvent(\"threadListItem.switchedTo\", () => {\n    if (!scrollToBottomOnThreadSwitch) return;\n    initializeScrollRequestedRef.current = false;\n    lastScrollEventYRef.current = 0;\n    pendingScrollToBottomRef.current = { animated: false };\n    scrollToBottom(false);\n  });\n\n  return {\n    handleLayout,\n    handleScroll,\n    handleContentSizeChange,\n  };\n};\n\nexport const ThreadMessagesFlatList = forwardRef<\n  FlatList<ThreadMessage>,\n  ThreadMessagesFlatListProps\n>(\n  (\n    {\n      autoScroll,\n      components,\n      children,\n      onContentSizeChange,\n      onLayout,\n      onScroll,\n      scrollEventThrottle,\n      scrollToBottomOnInitialize,\n      scrollToBottomOnRunStart,\n      scrollToBottomOnThreadSwitch,\n      ...flatListProps\n    },\n    forwardedRef,\n  ) => {\n    const messages = useAuiState((s) => s.thread.messages);\n    const [flatListRef, setFlatListRef] = useComposedFlatListRef(forwardedRef);\n    const {\n      handleContentSizeChange: handleAutoScrollContentSizeChange,\n      handleLayout: handleAutoScrollLayout,\n      handleScroll: handleAutoScrollScroll,\n    } = useThreadMessagesFlatListAutoScroll({\n      flatListRef,\n      hasMessages: messages.length > 0,\n      autoScroll,\n      scrollToBottomOnInitialize,\n      scrollToBottomOnRunStart,\n      scrollToBottomOnThreadSwitch,\n    });\n\n    const renderItem = useCallback(\n      ({ index }: { item: ThreadMessage; index: number }) => {\n        if (children) {\n          return (\n            <ThreadMessageByChildren index={index}>\n              {children}\n            </ThreadMessageByChildren>\n          );\n        }\n        return <ThreadMessageByIndex index={index} components={components!} />;\n      },\n      [components, children],\n    );\n\n    const keyExtractor = useCallback((item: ThreadMessage) => item.id, []);\n\n    const scrollTracking =\n      (autoScroll ?? true) ||\n      (scrollToBottomOnInitialize ?? true) ||\n      (scrollToBottomOnRunStart ?? true) ||\n      (scrollToBottomOnThreadSwitch ?? true);\n\n    const handleLayout = useCallback(\n      (event: LayoutChangeEvent) => {\n        handleAutoScrollLayout(event);\n        onLayout?.(event);\n      },\n      [handleAutoScrollLayout, onLayout],\n    );\n\n    const handleScroll = useCallback(\n      (event: NativeSyntheticEvent<NativeScrollEvent>) => {\n        handleAutoScrollScroll(event);\n        onScroll?.(event);\n      },\n      [handleAutoScrollScroll, onScroll],\n    );\n\n    const handleContentSizeChange = useCallback(\n      (width: number, height: number) => {\n        handleAutoScrollContentSizeChange(width, height);\n        onContentSizeChange?.(width, height);\n      },\n      [handleAutoScrollContentSizeChange, onContentSizeChange],\n    );\n\n    return (\n      <FlatList\n        ref={setFlatListRef}\n        data={messages as unknown as ThreadMessage[]}\n        renderItem={renderItem}\n        keyExtractor={keyExtractor}\n        {...(scrollTracking\n          ? {\n              onContentSizeChange: handleContentSizeChange,\n              onLayout: handleLayout,\n              onScroll: handleScroll,\n              scrollEventThrottle: scrollEventThrottle ?? 16,\n            }\n          : {\n              ...(onContentSizeChange && { onContentSizeChange }),\n              ...(onLayout && { onLayout }),\n              ...(onScroll && { onScroll }),\n              ...(scrollEventThrottle !== undefined && { scrollEventThrottle }),\n            })}\n        {...flatListProps}\n      />\n    );\n  },\n);\nThreadMessagesFlatList.displayName = \"ThreadPrimitive.MessagesFlatList\";\n\n/** @deprecated Use ThreadPrimitive.MessagesFlatList instead. */\nexport const ThreadMessages = forwardRef<\n  FlatList<ThreadMessage>,\n  ThreadMessagesProps\n>(\n  (\n    {\n      autoScroll = false,\n      scrollToBottomOnInitialize = false,\n      scrollToBottomOnRunStart = false,\n      scrollToBottomOnThreadSwitch = false,\n      ...props\n    },\n    ref,\n  ) => (\n    <ThreadMessagesFlatList\n      ref={ref}\n      autoScroll={autoScroll}\n      scrollToBottomOnInitialize={scrollToBottomOnInitialize}\n      scrollToBottomOnRunStart={scrollToBottomOnRunStart}\n      scrollToBottomOnThreadSwitch={scrollToBottomOnThreadSwitch}\n      {...props}\n    />\n  ),\n);\nThreadMessages.displayName = \"ThreadPrimitive.Messages\";\n"],"mappings":";;;;;;;AA0EA,MAAM0C,+BAA+B;AACrC,MAAMC,sBAAsB;AAE5B,MAAMC,gBACJb,YACAc,MACAC,cACG;CACH,QAAQD,MAAR;EACE,KAAK,QACH,IAAIC,WACF,OACEf,WAAWP,oBACXO,WAAWR,gBACXQ,WAAWJ,eACVI,WAAWT;OAGd,OAAOS,WAAWJ,eAAgBI,WAAWT;EAEjD,KAAK,aACH,IAAIwB,WACF,OACEf,WAAWN,yBACXM,WAAWR,gBACXQ,WAAWH,oBACVG,WAAWT;OAGd,OACES,WAAWH,oBAAqBG,WAAWT;EAGjD,KAAK,UACH,IAAIwB,WACF,OACEf,WAAWL,sBACXK,WAAWR,gBACXQ,WAAWF,iBACVE,WAAWT;OAGd,OACES,WAAWF,iBACVE,WAAWT,WACZoB;EAGN,SAEE,MAAM,IAAIM,MAAM,yBAAyBD,MAAkB;CAE/D;AACF;AAEA,MAAME,0BAAgEC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAArB,eAAAmB;CAGrE,MAAAL,OAAa1B,YAAYkC,KAAqB;CAC9C,MAAAP,YAAkB3B,YAAYmC,MAAmC;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAApB,cAAAoB,EAAA,OAAAL,aAAAK,EAAA,OAAAN,MAAA;EACjDU,KAAAX,aAAab,YAAYc,MAAMC,SAAS;EAACK,EAAA,KAAApB;EAAAoB,EAAA,KAAAL;EAAAK,EAAA,KAAAN;EAAAM,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAA3D,MAAAK,YAAkBD;CAA0C,IAAAE;CAAA,IAAAN,EAAA,OAAAK,WAAA;EAErDC,KAAA,oBAAC,WAAD,CAAU,CAAA;EAAGN,EAAA,KAAAK;EAAAL,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAAbM;AAAa;AAGtB,MAAMC,uBAAuBpD,MAC3B4C,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAO,OAAA5B,eAAAmB;CAAuE,IAAAK;CAAA,IAAAJ,EAAA,OAAApB,YAAA;EAGlEwB,KAAA,oBAAC,wBAAD,EAAoCxB,WAAU,CAAA;EAAIoB,EAAA,KAAApB;EAAAoB,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAM;CAAA,IAAAN,EAAA,OAAAQ,SAAAR,EAAA,OAAAI,IAAA;EADpDE,KAAA,oBAAC,wBAAD;GAA+BE;GAC7BJ,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAQ;EAAAR,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAFzBM;AAEyB,IAG5BG,MAAMC,SACLD,KAAKD,UAAUE,KAAKF,SAASC,KAAK7B,eAAe8B,KAAK9B,UAC1D;AACA2B,qBAAqBI,cAAc;AAEnC,MAAMC,0BAA0BzD,MAC7B,EACCqD,OACA3B,eAII;CACJ,OACE,oBAAC,wBAAD;EAA+B2B;EAC7B,UAAA,oBAAC,4BAAD;GACE,eAAeK,QAAQA,IAAIC,OAAO,CAAC,CAAC/B,QAAQ,EAAEyB,MAAM,CAAC,CAAC,CAACO,SAAS;GAE9DC,WAAAA,YACAnC,SAAS,EACP,IAAIE,UAAU;IACZ,OAAOiC,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACN,CAAA;AAE5B,IACCP,MAAMC,SAASD,KAAKD,UAAUE,KAAKF,SAASC,KAAK5B,aAAa6B,KAAK7B,QACtE;AACA+B,wBAAwBD,cAAc;AAEtC,MAAMM,mBAAuBC,KAAsBpC,UAAoB;CACrE,IAAI,OAAOoC,QAAQ,YACjBA,IAAIpC,KAAK;MACJ,IAAIoC,KACTA,IAAIE,UAAUtC;AAElB;AAEA,MAAMuC,0BAAyBC,iBAAA;CAAA,MAAAtB,IAAAC,EAAA,CAAA;CAG7B,MAAAsB,cAAoBjE,OAAuC,IAAI;CAAE,IAAAyC;CAAA,IAAAC,EAAA,OAAAsB,cAAA;EAG/DvB,MAAAyB,SAAA;GACED,YAAWH,UAAWI;GACtBP,gBAAgBK,cAAcE,IAAI;EAAC;EACpCxB,EAAA,KAAAsB;EAAAtB,EAAA,KAAAD;CAAA,OAAAA,KAAAC,EAAA;CAJH,MAAAyB,iBAAuB1B;CAMrB,IAAAK;CAAA,IAAAJ,EAAA,OAAAyB,gBAAA;EAEKrB,KAAA,CAACmB,aAAaE,cAAc;EAACzB,EAAA,KAAAyB;EAAAzB,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA7BI;AAAsC;AAG/C,MAAMuB,uCAAsC5B,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAsB,aAAAK,aAAA1C,YAAAkB,IAAAjB,0BAAAmB,IAAAlB,4BAAAyC,IAAAxC,8BAAAyC,OAAA/B;CAG3C,MAAAb,aAAAkB,OAAA2B,KAAAA,IAAA,OAAA3B;CACA,MAAAjB,2BAAAmB,OAAAyB,KAAAA,IAAA,OAAAzB;CACA,MAAAlB,6BAAAyC,OAAAE,KAAAA,IAAA,OAAAF;CACA,MAAAxC,+BAAAyC,OAAAC,KAAAA,IAAA,OAAAD;CAAmC,IAAAE;CAAA,IAAAhC,EAAA,OAAAiC,OAAAC,IAAA,2BAAA,GAAA;EASTF,KAAA;GAAAG,eACT;GAACC,gBACA;GAACC,SACR;EACX;EAACrC,EAAA,KAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAJD,MAAAsC,aAAmBhF,OAAO0E,EAIzB;CACD,MAAAO,gBAAsBjF,OAAO,IAAI;CACjC,MAAAkF,sBAA4BlF,OAAO,CAAC;CACpC,MAAAmF,+BAAqCnF,OAAO,KAAK;CACjD,MAAAoF,2BAAiCpF,OAAsC,KAAK;CAAE,IAAAqF;CAAA,IAAA3C,EAAA,OAAAiC,OAAAC,IAAA,2BAAA,GAAA;EAEzCS,WAAA;GACnC,MAAA,EAAAR,eAAAE,SAAAD,mBAAmDE,WAAUlB;GAC7DmB,cAAanB,UACXe,iBAAiBC,kBACjBD,gBAAgBE,UAAUD,kBAAkB5C;EAFzB;EAGtBQ,EAAA,KAAA2C;CAAA,OAAAA,KAAA3C,EAAA;CALD,MAAA4C,mBAAyBD;CAKlB,IAAAE;CAAA,IAAA7C,EAAA,OAAAuB,aAAA;EAMLsB,MAAAC,aAAA;GACE,MAAA,EAAAX,eAAAY,iBAAAX,gBAAAY,qBAA0CV,WAAUlB;GACpDkB,WAAUlB,QAAQiB,UAAWY,KAAIC,IAAK,GAAGf,kBAAgBC,gBAAc;GACvEG,cAAanB,UAAW;GACxBG,YAAWH,SAAqB+B,YAAC,EAAAL,SAAW,CAAC;EAAC;EAC/C9C,EAAA,KAAAuB;EAAAvB,EAAA,KAAA6C;CAAA,OAAAA,KAAA7C,EAAA;CANH,MAAAoD,iBAAuBP;CAQrB,IAAAQ;CAAA,IAAArD,EAAA,OAAAd,cAAAc,EAAA,OAAAoD,gBAAA;EAGAC,MAAAC,UAAA;GACE,MAAAC,cAAoBhB,cAAanB;GACjC,MAAAoC,yBAA+BlB,WAAUlB,QAAQgB;GACjD,MAAAqB,mBAAuBH,MAAKI,YAAYC,OAAOC;GAC/CtB,WAAUlB,QAAQgB,iBAAkBA;GACpCQ,iBAAiB;GACjB,IAAI,CAACW,aAAW;GAKhB,IACErE,cACAsE,2BAA2B,KAC3BpB,qBAAmBoB,wBAAsB;IAEzC,MAAAK,UAAgBnB,yBAAwBtB;IACxCgC,eAAeS,UAAUA,QAAOf,WAAjB,KAAkC;GAAC,OAElDP,cAAanB,UAAW;EACzB;EACFpB,EAAA,KAAAd;EAAAc,EAAA,KAAAoD;EAAApD,EAAA,KAAAqD;CAAA,OAAAA,KAAArD,EAAA;CAtBH,MAAA8D,eAAqBT;CAwBnB,IAAAU;CAAA,IAAA/D,EAAA,OAAAiC,OAAAC,IAAA,2BAAA,GAAA;EAGA6B,MAAAC,YAAA;GACE,MAAA,EAAAC,eAAAC,aAAAC,sBACEb,QAAKI;GACP,MAAAU,iBAAuB5B,oBAAmBpB;GAC1C,MAAAiD,oBAA0B9B,cAAanB;GACvCoB,oBAAmBpB,UAAW6C,cAAaK;GAC3ChC,WAAUlB,UAAW;IAAAe,eACJ+B,YAAWN;IAAOxB,gBACjB+B,kBAAiBP;IAAOvB,SAC/B4B,cAAaK;GACxB;GACA1B,iBAAiB;GACjB,MAAA2B,aAAmBN,cAAaK,IAAKF;GAKrC,IAAIC,qBAAA,CAAsBE,YACxBhC,cAAanB,UAAW;GAE1B,IAAI,CAACmB,cAAanB,WAAdmD,YACF7B,yBAAwBtB,UAAW;EACpC;EACFpB,EAAA,KAAA+D;CAAA,OAAAA,KAAA/D,EAAA;CAxBH,MAAAwE,eAAqBT;CA0BnB,IAAAU;CAAA,IAAAzE,EAAA,OAAAd,cAAAc,EAAA,OAAAoD,gBAAA;EAGAqB,OAAAC,QAAAd,WAAA;GACE,MAAAe,UAAgBrC,WAAUlB;GAC1B,MAAAwD,wBAA8BD,QAAOxC;GACrC,MAAA0C,gBAAoBtC,cAAanB;GACjCuD,QAAOxC,gBAAiByB;GACxBhB,iBAAiB;GAKjB,MAAAkC,gBAAsBpC,yBAAwBtB;GAC9C,IAAI0D,eAAa;IACfpC,yBAAwBtB,UAAW;IACnCgC,eAAe0B,cAAahC,QAAS;IAAC;GAAA;GAIxC,IAAI,CAAC5D,YAAU;GACf,IAAI,CAACqE,eAAW;GAChB,IAAIqB,0BAA0B,GAAC;GAC/B,IAAIhB,UAAUgB,uBAAqB;GAEnCxB,eAAe,KAAK;EAAC;EACtBpD,EAAA,KAAAd;EAAAc,EAAA,KAAAoD;EAAApD,EAAA,MAAAyE;CAAA,OAAAA,MAAAzE,EAAA;CAxBH,MAAA+E,0BAAgCN;CA0B9B,IAAAO;CAAA,IAAAC;CAAA,IAAAjF,EAAA,QAAA4B,eAAA5B,EAAA,QAAAoD,kBAAApD,EAAA,QAAAZ,4BAAA;EAEQ4F,YAAA;GACR,IAAI,CAAC5F,4BAA0B;GAC/B,IAAI,CAACwC,aAAW;IACda,6BAA4BrB,UAAW;IAAH;GAAA;GAGtC,IAAIqB,6BAA4BrB,SAAQ;GAExCqB,6BAA4BrB,UAAW;GACvCsB,yBAAwBtB,UAAW,EAAA0B,UAAY,MAAM;GACrDM,eAAe,KAAK;EAAC;EACpB6B,MAAA;GAACrD;GAAawB;GAAgBhE;EAA0B;EAACY,EAAA,MAAA4B;EAAA5B,EAAA,MAAAoD;EAAApD,EAAA,MAAAZ;EAAAY,EAAA,MAAAgF;EAAAhF,EAAA,MAAAiF;CAAA,OAAA;EAAAD,MAAAhF,EAAA;EAAAiF,MAAAjF,EAAA;CAAA;CAX5D3C,UAAU2H,KAWPC,GAAyD;CAAC,IAAAC;CAAA,IAAAlF,EAAA,QAAAoD,kBAAApD,EAAA,QAAAb,0BAAA;EAE9B+F,YAAA;GAC7B,IAAI,CAAC/F,0BAAwB;GAC7BuD,yBAAwBtB,UAAW,EAAA0B,UAAY,KAAK;GACpDM,eAAe,IAAI;EAAC;EACrBpD,EAAA,MAAAoD;EAAApD,EAAA,MAAAb;EAAAa,EAAA,MAAAkF;CAAA,OAAAA,MAAAlF,EAAA;CAJDjC,YAAY,mBAAmBmH,GAI9B;CAAC,IAAAC;CAAA,IAAAnF,EAAA,QAAAoD,kBAAApD,EAAA,QAAAX,8BAAA;EAEuC8F,YAAA;GACvC,IAAI,CAAC9F,8BAA4B;GACjCoD,6BAA4BrB,UAAW;GACvCoB,oBAAmBpB,UAAW;GAC9BsB,yBAAwBtB,UAAW,EAAA0B,UAAY,MAAM;GACrDM,eAAe,KAAK;EAAC;EACtBpD,EAAA,MAAAoD;EAAApD,EAAA,MAAAX;EAAAW,EAAA,MAAAmF;CAAA,OAAAA,MAAAnF,EAAA;CANDjC,YAAY,6BAA6BoH,GAMxC;CAAC,IAAAC;CAAA,IAAApF,EAAA,QAAA+E,2BAAA/E,EAAA,QAAA8D,cAAA;EAEKsB,MAAA;GAAAtB;GAAAU;GAAAO;EAIP;EAAC/E,EAAA,MAAA+E;EAAA/E,EAAA,MAAA8D;EAAA9D,EAAA,MAAAoF;CAAA,OAAAA,MAAApF,EAAA;CAAA,OAJMoF;AAIN;AAGH,MAAaC,yBAAyBnI,YAIpC6C,IAAAuB,iBAAA;CAAA,MAAAtB,IAAAC,EAAA,EAAA;CAAA,IAAAf;CAAA,IAAAL;CAAA,IAAAD;CAAA,IAAA0G;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAtG;CAAA,IAAAD;CAAA,IAAAE;CAAA,IAAAW,EAAA,OAAAD,IAAA;EACE,CAAA,CAAAb,YAAAN,YAAAC,UAAA0G,qBAAAC,UAAAC,UAAAC,qBAAAtG,4BAAAD,0BAAAE,iCAAAiG,iBAAAvF;EAYCC,EAAA,KAAAD;EAAAC,EAAA,KAAAd;EAAAc,EAAA,KAAAnB;EAAAmB,EAAA,KAAApB;EAAAoB,EAAA,KAAAsF;EAAAtF,EAAA,KAAAuF;EAAAvF,EAAA,KAAAwF;EAAAxF,EAAA,KAAAyF;EAAAzF,EAAA,KAAA0F;EAAA1F,EAAA,KAAAZ;EAAAY,EAAA,MAAAb;EAAAa,EAAA,MAAAX;CAAA,OAAA;EAAAH,aAAAc,EAAA;EAAAnB,WAAAmB,EAAA;EAAApB,aAAAoB,EAAA;EAAAsF,gBAAAtF,EAAA;EAAAuF,sBAAAvF,EAAA;EAAAwF,WAAAxF,EAAA;EAAAyF,WAAAzF,EAAA;EAAA0F,sBAAA1F,EAAA;EAAAZ,6BAAAY,EAAA;EAAAb,2BAAAa,EAAA;EAAAX,+BAAAW,EAAA;CAAA;CAGD,MAAA2F,WAAiB3H,YAAY4H,MAAwB;CACrD,MAAA,CAAArE,aAAAE,kBAAsCJ,uBAAuBC,YAAY;CAO1D,MAAAlB,KAAAuF,SAAQE,SAAU;CAAC,IAAAvF;CAAA,IAAAN,EAAA,QAAAd,cAAAc,EAAA,QAAAuB,eAAAvB,EAAA,QAAAZ,8BAAAY,EAAA,QAAAb,4BAAAa,EAAA,QAAAX,gCAAAW,EAAA,QAAAI,IAAA;EAFME,KAAA;GAAAiB;GAAAK,aAEzBxB;GAAmBlB;GAAAE;GAAAD;GAAAE;EAKlC;EAACW,EAAA,MAAAd;EAAAc,EAAA,MAAAuB;EAAAvB,EAAA,MAAAZ;EAAAY,EAAA,MAAAb;EAAAa,EAAA,MAAAX;EAAAW,EAAA,MAAAI;EAAAJ,EAAA,MAAAM;CAAA,OAAAA,KAAAN,EAAA;CAXD,MAAA,EAAA+E,yBAAAe,mCAAAhC,cAAAiC,wBAAAvB,cAAAwB,2BAIIrE,oCAAoCrB,EAOvC;CAAE,IAAAuB;CAAA,IAAA7B,EAAA,QAAAnB,YAAAmB,EAAA,QAAApB,YAAA;EAGDiD,MAAAC,OAAA;GAAC,MAAA,EAAAtB,UAAAsB;GACC,IAAIjD,UAAQ,OAER,oBAAC,yBAAD;IAAgC2B;IAC7B3B;GADqB,CAAA;GAI3B,OACM,oBAAC,sBAAD;IAA6B2B;IAAmB5B;GAAU,CAAA;EAAK;EACvEoB,EAAA,MAAAnB;EAAAmB,EAAA,MAAApB;EAAAoB,EAAA,MAAA6B;CAAA,OAAAA,KAAA7B,EAAA;CAVH,MAAAiG,aAAmBpE;CAcnB,MAAAqE,eAAqBC;CAErB,MAAAC,kBACGlH,cAAA,UACAE,8BAAA,UACAD,4BAAA,UACAE,gCAAA;CAAsC,IAAAyC;CAAA,IAAA9B,EAAA,QAAA+F,0BAAA/F,EAAA,QAAAwF,UAAA;EAGvC1D,MAAAwB,UAAA;GACEyC,uBAAuBzC,KAAK;GAC5BkC,WAAWlC,KAAK;EAAC;EAClBtD,EAAA,MAAA+F;EAAA/F,EAAA,MAAAwF;EAAAxF,EAAA,MAAA8B;CAAA,OAAAA,KAAA9B,EAAA;CAJH,MAAA8D,eAAqBhC;CAMnB,IAAAE;CAAA,IAAAhC,EAAA,QAAAgG,0BAAAhG,EAAA,QAAAyF,UAAA;EAGAzD,MAAAgC,YAAA;GACEgC,uBAAuB1C,OAAK;GAC5BmC,WAAWnC,OAAK;EAAC;EAClBtD,EAAA,MAAAgG;EAAAhG,EAAA,MAAAyF;EAAAzF,EAAA,MAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAJH,MAAAwE,eAAqBxC;CAMnB,IAAAW;CAAA,IAAA3C,EAAA,QAAA8F,qCAAA9F,EAAA,QAAAuF,qBAAA;EAGA5C,MAAA0D,OAAAzC,WAAA;GACEkC,kCAAkCO,OAAOzC,MAAM;GAC/C2B,sBAAsBc,OAAOzC,MAAM;EAAC;EACrC5D,EAAA,MAAA8F;EAAA9F,EAAA,MAAAuF;EAAAvF,EAAA,MAAA2C;CAAA,OAAAA,KAAA3C,EAAA;CAJH,MAAA+E,0BAAgCpC;CAWtB,MAAAE,KAAA8C;CAAsC,IAAAtC;CAAA,IAAArD,EAAA,QAAA+E,2BAAA/E,EAAA,QAAA8D,gBAAA9D,EAAA,QAAAwE,gBAAAxE,EAAA,QAAAuF,uBAAAvF,EAAA,QAAAwF,YAAAxF,EAAA,QAAAyF,YAAAzF,EAAA,QAAA0F,uBAAA1F,EAAA,QAAAoG,gBAAA;EAGvC/C,KAAA+C,iBAAA;GAAAb,qBAEsBR;GAAuBS,UAClC1B;GAAY2B,UACZjB;GAAYkB,qBACDA,uBAAA;EAOvB,IAZC;GAAA,GAQKH,uBAAA,EAAAA,oBAA6C;GAAC,GAC9CC,YAAA,EAAAA,SAAuB;GAAC,GACxBC,YAAA,EAAAA,SAAuB;GAAC,GACxBC,wBAAwB3D,KAAAA,KAAxB,EAAA2D,oBAA2D;EACjE;EAAC1F,EAAA,MAAA+E;EAAA/E,EAAA,MAAA8D;EAAA9D,EAAA,MAAAwE;EAAAxE,EAAA,MAAAuF;EAAAvF,EAAA,MAAAwF;EAAAxF,EAAA,MAAAyF;EAAAzF,EAAA,MAAA0F;EAAA1F,EAAA,MAAAoG;EAAApG,EAAA,MAAAqD;CAAA,OAAAA,KAAArD,EAAA;CAAA,IAAA+D;CAAA,IAAA/D,EAAA,QAAAsF,iBAAAtF,EAAA,QAAAiG,cAAAjG,EAAA,QAAAyB,kBAAAzB,EAAA,QAAA6C,MAAA7C,EAAA,QAAAqD,IAAA;EAjBPU,KAAA,oBAAC,UAAD;GACOtC,KAAAA;GACC,MAAAoB;GACMoD;GACEC;GAAY,GACrB7C;GAYA,GACDiC;EAAa,CAAA;EACjBtF,EAAA,MAAAsF;EAAAtF,EAAA,MAAAiG;EAAAjG,EAAA,MAAAyB;EAAAzB,EAAA,MAAA6C;EAAA7C,EAAA,MAAAqD;EAAArD,EAAA,MAAA+D;CAAA,OAAAA,KAAA/D,EAAA;CAAA,OAnBF+D;AAmBE,CAGR;AACAsB,uBAAuB1E,cAAc;;AAGrC,MAAa2F,iBAAiBpJ,YAI5B6C,IAAAmB,QAAA;CAAA,MAAAlB,IAAAC,EAAA,EAAA;CAAA,IAAAsG;CAAA,IAAAnG;CAAA,IAAAE;CAAA,IAAAuB;CAAA,IAAAC;CAAA,IAAA9B,EAAA,OAAAD,IAAA;EACE,CAAA,CAAAb,YAAAkB,IAAAhB,4BAAAkB,IAAAnB,0BAAA0C,IAAAxC,8BAAAyC,OAAAyE,SAAAxG;EAMCC,EAAA,KAAAD;EAAAC,EAAA,KAAAuG;EAAAvG,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;EAAAN,EAAA,KAAA6B;EAAA7B,EAAA,KAAA8B;CAAA,OAAA;EAAAyE,QAAAvG,EAAA;EAAAI,KAAAJ,EAAA;EAAAM,KAAAN,EAAA;EAAA6B,KAAA7B,EAAA;EAAA8B,KAAA9B,EAAA;CAAA;CALC,MAAAd,aAAAkB,OAAA2B,KAAAA,IAAA,QAAA3B;CACA,MAAAhB,6BAAAkB,OAAAyB,KAAAA,IAAA,QAAAzB;CACA,MAAAnB,2BAAA0C,OAAAE,KAAAA,IAAA,QAAAF;CACA,MAAAxC,+BAAAyC,OAAAC,KAAAA,IAAA,QAAAD;CAAoC,IAAAE;CAAA,IAAAhC,EAAA,OAAAd,cAAAc,EAAA,OAAAuG,SAAAvG,EAAA,OAAAkB,OAAAlB,EAAA,OAAAZ,8BAAAY,EAAA,QAAAb,4BAAAa,EAAA,QAAAX,8BAAA;EAKtC2C,KAAA,oBAAC,wBAAD;GACOd;GACOhC;GACgBE;GACFD;GACIE;GAA4B,GACtDkH;EAAK,CAAA;EACTvG,EAAA,KAAAd;EAAAc,EAAA,KAAAuG;EAAAvG,EAAA,KAAAkB;EAAAlB,EAAA,KAAAZ;EAAAY,EAAA,MAAAb;EAAAa,EAAA,MAAAX;EAAAW,EAAA,MAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAAA,OAPFgC;AAOE,CAEN;AACAsE,eAAe3F,cAAc;AAhXyC,SAAAT,MAAAsG,GAAA;CAAA,OAGpCA,EAACzH,QAAQW;AAAK;AAHsB,SAAAS,OAAAsG,KAAA;CAAA,OAI/BD,IAACzH,QAAQ2H,SAAS/G;AAAU;AA4OjE,SAAAiG,OAAAY,GAAA;CAAA,OAgBsCA,EAAC1F,OAAO6E;AAAS;AAhBvD,SAAAQ,OAAAQ,MAAA;CAAA,OA6C4DA,KAAIC;AAAG"}
+{"version":3,"file":"ThreadMessages.js","names":["ComponentType","FC","ForwardedRef","ReactNode","RefObject","forwardRef","memo","useCallback","useEffect","useRef","FlatList","FlatListProps","LayoutChangeEvent","NativeScrollEvent","NativeSyntheticEvent","MessageState","ThreadMessage","RenderChildrenWithAccessor","useAuiEvent","useAuiState","MessageByIndexProvider","MessageComponents","Message","EditComposer","UserEditComposer","AssistantEditComposer","SystemEditComposer","UserMessage","AssistantMessage","SystemMessage","MessagesContent","components","children","value","message","ThreadMessagesFlatListProps","Omit","autoScroll","scrollToBottomOnRunStart","scrollToBottomOnInitialize","scrollToBottomOnThreadSwitch","ThreadMessagesProps","DEFAULT_SYSTEM_MESSAGE","AT_BOTTOM_THRESHOLD","getComponent","role","isEditing","_exhaustiveCheck","Error","ThreadMessageComponent","t0","$","_c","_temp","_temp2","t1","Component","t2","ThreadMessageByIndex","index","prev","next","displayName","ThreadMessageByChildren","aui","thread","getState","getItem","setForwardedRef","ref","T","current","useComposedFlatListRef","forwardedRef","flatListRef","node","setFlatListRef","const","useThreadMessagesFlatListAutoScroll","hasMessages","t3","t4","undefined","t5","Symbol","for","contentHeight","viewportHeight","scrollY","metricsRef","isAtBottomRef","lastScrollEventYRef","initializeScrollRequestedRef","pendingScrollToBottomRef","t6","updateIsAtBottom","t7","animated","contentHeight_0","viewportHeight_0","Math","max","scrollToEnd","scrollToBottom","t8","event","wasAtBottom","previousViewportHeight","viewportHeight_1","nativeEvent","layout","height","pending","handleLayout","t9","event_0","contentOffset","contentSize","layoutMeasurement","previousEventY","wasPinnedToBottom","y","upwardMove","handleScroll","t10","_width","metrics","previousContentHeight","wasAtBottom_0","pendingScroll","handleContentSizeChange","t11","t12","t13","t14","t15","ThreadMessagesFlatList","flatListProps","onContentSizeChange","onLayout","onScroll","scrollEventThrottle","messages","_temp3","length","handleAutoScrollContentSizeChange","handleAutoScrollLayout","handleAutoScrollScroll","renderItem","keyExtractor","_temp4","scrollTracking","width","ThreadMessages","props","s","s_0","composer","item","id"],"sources":["../../../src/primitives/thread/ThreadMessages.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type ForwardedRef,\n  type ReactNode,\n  type RefObject,\n  forwardRef,\n  memo,\n  useCallback,\n  useEffect,\n  useRef,\n} from \"react\";\nimport {\n  FlatList,\n  type FlatListProps,\n  type LayoutChangeEvent,\n  type NativeScrollEvent,\n  type NativeSyntheticEvent,\n} from \"react-native\";\nimport type { MessageState, ThreadMessage } from \"@assistant-ui/core\";\nimport {\n  RenderChildrenWithAccessor,\n  useAuiEvent,\n  useAuiState,\n} from \"@assistant-ui/store\";\nimport { MessageByIndexProvider } from \"@assistant-ui/core/react\";\n\ntype MessageComponents =\n  | {\n      Message: ComponentType;\n      EditComposer?: ComponentType | undefined;\n      UserEditComposer?: ComponentType | undefined;\n      AssistantEditComposer?: ComponentType | undefined;\n      SystemEditComposer?: ComponentType | undefined;\n      UserMessage?: ComponentType | undefined;\n      AssistantMessage?: ComponentType | undefined;\n      SystemMessage?: ComponentType | undefined;\n    }\n  | {\n      Message?: ComponentType | undefined;\n      EditComposer?: ComponentType | undefined;\n      UserEditComposer?: ComponentType | undefined;\n      AssistantEditComposer?: ComponentType | undefined;\n      SystemEditComposer?: ComponentType | undefined;\n      UserMessage: ComponentType;\n      AssistantMessage: ComponentType;\n      SystemMessage?: ComponentType | undefined;\n    };\n\ntype MessagesContent =\n  | {\n      /** @deprecated Use the children render function instead. */\n      components: MessageComponents;\n      children?: never;\n    }\n  | {\n      children: (value: { message: MessageState }) => ReactNode;\n      components?: never;\n    };\n\nexport type ThreadMessagesFlatListProps = Omit<\n  FlatListProps<ThreadMessage>,\n  \"data\" | \"renderItem\" | \"children\"\n> &\n  MessagesContent & {\n    autoScroll?: boolean | undefined;\n    scrollToBottomOnRunStart?: boolean | undefined;\n    scrollToBottomOnInitialize?: boolean | undefined;\n    scrollToBottomOnThreadSwitch?: boolean | undefined;\n  };\n\n/** @deprecated Use ThreadMessagesFlatListProps instead. */\nexport type ThreadMessagesProps = ThreadMessagesFlatListProps;\n\nconst DEFAULT_SYSTEM_MESSAGE = () => null;\nconst AT_BOTTOM_THRESHOLD = 4;\n\nconst getComponent = (\n  components: MessageComponents,\n  role: ThreadMessage[\"role\"],\n  isEditing: boolean,\n) => {\n  switch (role) {\n    case \"user\":\n      if (isEditing) {\n        return (\n          components.UserEditComposer ??\n          components.EditComposer ??\n          components.UserMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return components.UserMessage ?? (components.Message as ComponentType);\n      }\n    case \"assistant\":\n      if (isEditing) {\n        return (\n          components.AssistantEditComposer ??\n          components.EditComposer ??\n          components.AssistantMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return (\n          components.AssistantMessage ?? (components.Message as ComponentType)\n        );\n      }\n    case \"system\":\n      if (isEditing) {\n        return (\n          components.SystemEditComposer ??\n          components.EditComposer ??\n          components.SystemMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return (\n          components.SystemMessage ??\n          (components.Message as ComponentType) ??\n          DEFAULT_SYSTEM_MESSAGE\n        );\n      }\n    default: {\n      const _exhaustiveCheck: never = role;\n      throw new Error(`Unknown message role: ${_exhaustiveCheck}`);\n    }\n  }\n};\n\nconst ThreadMessageComponent: FC<{ components: MessageComponents }> = ({\n  components,\n}) => {\n  const role = useAuiState((s) => s.message.role);\n  const isEditing = useAuiState((s) => s.message.composer.isEditing);\n  const Component = getComponent(components, role, isEditing);\n\n  return <Component />;\n};\n\nconst ThreadMessageByIndex = memo(\n  ({ index, components }: { index: number; components: MessageComponents }) => {\n    return (\n      <MessageByIndexProvider index={index}>\n        <ThreadMessageComponent components={components} />\n      </MessageByIndexProvider>\n    );\n  },\n  (prev, next) =>\n    prev.index === next.index && prev.components === next.components,\n);\nThreadMessageByIndex.displayName = \"ThreadPrimitive.MessageByIndex\";\n\nconst ThreadMessageByChildren = memo(\n  ({\n    index,\n    children,\n  }: {\n    index: number;\n    children: (value: { message: MessageState }) => ReactNode;\n  }) => {\n    return (\n      <MessageByIndexProvider index={index}>\n        <RenderChildrenWithAccessor\n          getItemState={(aui) => aui.thread.message({ index }).getState()}\n        >\n          {(getItem) =>\n            children({\n              get message() {\n                return getItem();\n              },\n            })\n          }\n        </RenderChildrenWithAccessor>\n      </MessageByIndexProvider>\n    );\n  },\n  (prev, next) => prev.index === next.index && prev.children === next.children,\n);\nThreadMessageByChildren.displayName = \"ThreadPrimitive.MessageByChildren\";\n\nconst setForwardedRef = <T,>(ref: ForwardedRef<T>, value: T | null) => {\n  if (typeof ref === \"function\") {\n    ref(value);\n  } else if (ref) {\n    ref.current = value;\n  }\n};\n\nconst useComposedFlatListRef = (\n  forwardedRef: ForwardedRef<FlatList<ThreadMessage>>,\n) => {\n  const flatListRef = useRef<FlatList<ThreadMessage> | null>(null);\n\n  const setFlatListRef = useCallback(\n    (node: FlatList<ThreadMessage> | null) => {\n      flatListRef.current = node;\n      setForwardedRef(forwardedRef, node);\n    },\n    [forwardedRef],\n  );\n\n  return [flatListRef, setFlatListRef] as const;\n};\n\nconst useThreadMessagesFlatListAutoScroll = ({\n  flatListRef,\n  hasMessages,\n  autoScroll = true,\n  scrollToBottomOnRunStart = true,\n  scrollToBottomOnInitialize = true,\n  scrollToBottomOnThreadSwitch = true,\n}: {\n  flatListRef: RefObject<FlatList<ThreadMessage> | null>;\n  hasMessages: boolean;\n  autoScroll?: boolean | undefined;\n  scrollToBottomOnRunStart?: boolean | undefined;\n  scrollToBottomOnInitialize?: boolean | undefined;\n  scrollToBottomOnThreadSwitch?: boolean | undefined;\n}) => {\n  const metricsRef = useRef({\n    contentHeight: 0,\n    viewportHeight: 0,\n    scrollY: 0,\n  });\n  const isAtBottomRef = useRef(true);\n  const lastScrollEventYRef = useRef(0);\n  const initializeScrollRequestedRef = useRef(false);\n  const pendingScrollToBottomRef = useRef<false | { animated: boolean }>(false);\n\n  const updateIsAtBottom = useCallback(() => {\n    const { contentHeight, scrollY, viewportHeight } = metricsRef.current;\n    isAtBottomRef.current =\n      contentHeight <= viewportHeight ||\n      contentHeight - scrollY - viewportHeight <= AT_BOTTOM_THRESHOLD;\n  }, []);\n\n  // Commanding a scroll records the intended position immediately; the\n  // native scroll echo is bridged and throttled, so waiting for it lets a\n  // fast stream observe stale metrics and drop out of following.\n  const scrollToBottom = useCallback(\n    (animated: boolean) => {\n      const { contentHeight, viewportHeight } = metricsRef.current;\n      metricsRef.current.scrollY = Math.max(0, contentHeight - viewportHeight);\n      isAtBottomRef.current = true;\n      flatListRef.current?.scrollToEnd({ animated });\n    },\n    [flatListRef],\n  );\n\n  const handleLayout = useCallback(\n    (event: LayoutChangeEvent) => {\n      const wasAtBottom = isAtBottomRef.current;\n      const previousViewportHeight = metricsRef.current.viewportHeight;\n      const viewportHeight = event.nativeEvent.layout.height;\n      metricsRef.current.viewportHeight = viewportHeight;\n      updateIsAtBottom();\n      if (!wasAtBottom) return;\n      // Layout changes are never user gestures, so they must not unpin. Past\n      // the first measurement, a viewport change while pinned re-commands the\n      // bottom position, since no content-size event follows a bare keyboard\n      // open or close.\n      if (\n        autoScroll &&\n        previousViewportHeight !== 0 &&\n        viewportHeight !== previousViewportHeight\n      ) {\n        const pending = pendingScrollToBottomRef.current;\n        scrollToBottom(pending ? pending.animated : false);\n      } else {\n        isAtBottomRef.current = true;\n      }\n    },\n    [autoScroll, scrollToBottom, updateIsAtBottom],\n  );\n\n  const handleScroll = useCallback(\n    (event: NativeSyntheticEvent<NativeScrollEvent>) => {\n      const { contentOffset, contentSize, layoutMeasurement } =\n        event.nativeEvent;\n      const previousEventY = lastScrollEventYRef.current;\n      const wasPinnedToBottom = isAtBottomRef.current;\n      lastScrollEventYRef.current = contentOffset.y;\n      metricsRef.current = {\n        contentHeight: contentSize.height,\n        viewportHeight: layoutMeasurement.height,\n        scrollY: contentOffset.y,\n      };\n      updateIsAtBottom();\n      const upwardMove = contentOffset.y < previousEventY;\n      // Only a deliberate upward move unpins or cancels a pending scroll.\n      // Gestures are detected echo-to-echo because a commanded scroll\n      // optimistically moves the tracked position ahead of its ascending\n      // animation echoes, and those echoes must not unpin mid-flight.\n      if (wasPinnedToBottom && !upwardMove) {\n        isAtBottomRef.current = true;\n      }\n      if (!isAtBottomRef.current && upwardMove) {\n        pendingScrollToBottomRef.current = false;\n      }\n    },\n    [updateIsAtBottom],\n  );\n\n  const handleContentSizeChange = useCallback(\n    (_width: number, height: number) => {\n      const metrics = metricsRef.current;\n      const previousContentHeight = metrics.contentHeight;\n      const wasAtBottom = isAtBottomRef.current;\n      metrics.contentHeight = height;\n      updateIsAtBottom();\n\n      // FlatList.scrollToEnd is a no-op before the list has measured, so the\n      // initialize and thread-switch scrolls land on the next content-size\n      // event, once real metrics exist.\n      const pendingScroll = pendingScrollToBottomRef.current;\n      if (pendingScroll) {\n        pendingScrollToBottomRef.current = false;\n        scrollToBottom(pendingScroll.animated);\n        return;\n      }\n\n      if (!autoScroll) return;\n      if (!wasAtBottom) return;\n      if (previousContentHeight === 0) return;\n      if (height <= previousContentHeight) return;\n\n      scrollToBottom(false);\n    },\n    [autoScroll, scrollToBottom, updateIsAtBottom],\n  );\n\n  useEffect(() => {\n    if (!scrollToBottomOnInitialize) return;\n    if (!hasMessages) {\n      initializeScrollRequestedRef.current = false;\n      return;\n    }\n    if (initializeScrollRequestedRef.current) return;\n\n    initializeScrollRequestedRef.current = true;\n    pendingScrollToBottomRef.current = { animated: false };\n    scrollToBottom(false);\n  }, [hasMessages, scrollToBottom, scrollToBottomOnInitialize]);\n\n  useAuiEvent(\"thread.runStart\", () => {\n    if (!scrollToBottomOnRunStart) return;\n    pendingScrollToBottomRef.current = { animated: true };\n    scrollToBottom(true);\n  });\n\n  useAuiEvent(\"threadListItem.switchedTo\", () => {\n    if (!scrollToBottomOnThreadSwitch) return;\n    initializeScrollRequestedRef.current = false;\n    lastScrollEventYRef.current = 0;\n    pendingScrollToBottomRef.current = { animated: false };\n    scrollToBottom(false);\n  });\n\n  return {\n    handleLayout,\n    handleScroll,\n    handleContentSizeChange,\n  };\n};\n\nexport const ThreadMessagesFlatList = forwardRef<\n  FlatList<ThreadMessage>,\n  ThreadMessagesFlatListProps\n>(\n  (\n    {\n      autoScroll,\n      components,\n      children,\n      onContentSizeChange,\n      onLayout,\n      onScroll,\n      scrollEventThrottle,\n      scrollToBottomOnInitialize,\n      scrollToBottomOnRunStart,\n      scrollToBottomOnThreadSwitch,\n      ...flatListProps\n    },\n    forwardedRef,\n  ) => {\n    const messages = useAuiState((s) => s.thread.messages);\n    const [flatListRef, setFlatListRef] = useComposedFlatListRef(forwardedRef);\n    const {\n      handleContentSizeChange: handleAutoScrollContentSizeChange,\n      handleLayout: handleAutoScrollLayout,\n      handleScroll: handleAutoScrollScroll,\n    } = useThreadMessagesFlatListAutoScroll({\n      flatListRef,\n      hasMessages: messages.length > 0,\n      autoScroll,\n      scrollToBottomOnInitialize,\n      scrollToBottomOnRunStart,\n      scrollToBottomOnThreadSwitch,\n    });\n\n    const renderItem = useCallback(\n      ({ index }: { item: ThreadMessage; index: number }) => {\n        if (children) {\n          return (\n            <ThreadMessageByChildren index={index}>\n              {children}\n            </ThreadMessageByChildren>\n          );\n        }\n        return <ThreadMessageByIndex index={index} components={components!} />;\n      },\n      [components, children],\n    );\n\n    const keyExtractor = useCallback((item: ThreadMessage) => item.id, []);\n\n    const scrollTracking =\n      (autoScroll ?? true) ||\n      (scrollToBottomOnInitialize ?? true) ||\n      (scrollToBottomOnRunStart ?? true) ||\n      (scrollToBottomOnThreadSwitch ?? true);\n\n    const handleLayout = useCallback(\n      (event: LayoutChangeEvent) => {\n        handleAutoScrollLayout(event);\n        onLayout?.(event);\n      },\n      [handleAutoScrollLayout, onLayout],\n    );\n\n    const handleScroll = useCallback(\n      (event: NativeSyntheticEvent<NativeScrollEvent>) => {\n        handleAutoScrollScroll(event);\n        onScroll?.(event);\n      },\n      [handleAutoScrollScroll, onScroll],\n    );\n\n    const handleContentSizeChange = useCallback(\n      (width: number, height: number) => {\n        handleAutoScrollContentSizeChange(width, height);\n        onContentSizeChange?.(width, height);\n      },\n      [handleAutoScrollContentSizeChange, onContentSizeChange],\n    );\n\n    return (\n      <FlatList\n        ref={setFlatListRef}\n        data={messages as unknown as ThreadMessage[]}\n        renderItem={renderItem}\n        keyExtractor={keyExtractor}\n        {...(scrollTracking\n          ? {\n              onContentSizeChange: handleContentSizeChange,\n              onLayout: handleLayout,\n              onScroll: handleScroll,\n              scrollEventThrottle: scrollEventThrottle ?? 16,\n            }\n          : {\n              ...(onContentSizeChange && { onContentSizeChange }),\n              ...(onLayout && { onLayout }),\n              ...(onScroll && { onScroll }),\n              ...(scrollEventThrottle !== undefined && { scrollEventThrottle }),\n            })}\n        {...flatListProps}\n      />\n    );\n  },\n);\nThreadMessagesFlatList.displayName = \"ThreadPrimitive.MessagesFlatList\";\n\n/** @deprecated Use ThreadPrimitive.MessagesFlatList instead. */\nexport const ThreadMessages = forwardRef<\n  FlatList<ThreadMessage>,\n  ThreadMessagesProps\n>(\n  (\n    {\n      autoScroll = false,\n      scrollToBottomOnInitialize = false,\n      scrollToBottomOnRunStart = false,\n      scrollToBottomOnThreadSwitch = false,\n      ...props\n    },\n    ref,\n  ) => (\n    <ThreadMessagesFlatList\n      ref={ref}\n      autoScroll={autoScroll}\n      scrollToBottomOnInitialize={scrollToBottomOnInitialize}\n      scrollToBottomOnRunStart={scrollToBottomOnRunStart}\n      scrollToBottomOnThreadSwitch={scrollToBottomOnThreadSwitch}\n      {...props}\n    />\n  ),\n);\nThreadMessages.displayName = \"ThreadPrimitive.Messages\";\n"],"mappings":";;;;;;;AA0EA,MAAM0C,+BAA+B;AACrC,MAAMC,sBAAsB;AAE5B,MAAMC,gBACJb,YACAc,MACAC,cACG;CACH,QAAQD,MAAR;EACE,KAAK,QACH,IAAIC,WACF,OACEf,WAAWP,oBACXO,WAAWR,gBACXQ,WAAWJ,eACVI,WAAWT;OAGd,OAAOS,WAAWJ,eAAgBI,WAAWT;EAEjD,KAAK,aACH,IAAIwB,WACF,OACEf,WAAWN,yBACXM,WAAWR,gBACXQ,WAAWH,oBACVG,WAAWT;OAGd,OACES,WAAWH,oBAAqBG,WAAWT;EAGjD,KAAK,UACH,IAAIwB,WACF,OACEf,WAAWL,sBACXK,WAAWR,gBACXQ,WAAWF,iBACVE,WAAWT;OAGd,OACES,WAAWF,iBACVE,WAAWT,WACZoB;EAGN,SAEE,MAAM,IAAIM,MAAM,yBAAyBD,MAAkB;CAE/D;AACF;AAEA,MAAME,0BAAgEC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAArB,eAAAmB;CAGrE,MAAAL,OAAa1B,YAAYkC,KAAqB;CAC9C,MAAAP,YAAkB3B,YAAYmC,MAAmC;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAApB,cAAAoB,EAAA,OAAAL,aAAAK,EAAA,OAAAN,MAAA;EACjDU,KAAAX,aAAab,YAAYc,MAAMC,SAAS;EAACK,EAAA,KAAApB;EAAAoB,EAAA,KAAAL;EAAAK,EAAA,KAAAN;EAAAM,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAA3D,MAAAK,YAAkBD;CAA0C,IAAAE;CAAA,IAAAN,EAAA,OAAAK,WAAA;EAErDC,KAAA,oBAAC,WAAD,CAAU,CAAA;EAAGN,EAAA,KAAAK;EAAAL,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAAbM;AAAa;AAGtB,MAAMC,uBAAuBpD,MAC3B4C,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAO,OAAA5B,eAAAmB;CAAuE,IAAAK;CAAA,IAAAJ,EAAA,OAAApB,YAAA;EAGlEwB,KAAA,oBAAC,wBAAD,EAAoCxB,WAAU,CAAA;EAAIoB,EAAA,KAAApB;EAAAoB,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAM;CAAA,IAAAN,EAAA,OAAAQ,SAAAR,EAAA,OAAAI,IAAA;EADpDE,KAAA,oBAAC,wBAAD;GAA+BE;GAC7BJ,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAQ;EAAAR,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAFzBM;AAEyB,IAG5BG,MAAMC,SACLD,KAAKD,UAAUE,KAAKF,SAASC,KAAK7B,eAAe8B,KAAK9B,UAC1D;AACA2B,qBAAqBI,cAAc;AAEnC,MAAMC,0BAA0BzD,MAC7B,EACCqD,OACA3B,eAII;CACJ,OACE,oBAAC,wBAAD;EAA+B2B;EAC7B,UAAA,oBAAC,4BAAD;GACE,eAAeK,QAAQA,IAAIC,OAAO/B,QAAQ,EAAEyB,MAAM,CAAC,CAAC,CAACO,SAAS;GAE5DC,WAAAA,YACAnC,SAAS,EACP,IAAIE,UAAU;IACZ,OAAOiC,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACN,CAAA;AAE5B,IACCP,MAAMC,SAASD,KAAKD,UAAUE,KAAKF,SAASC,KAAK5B,aAAa6B,KAAK7B,QACtE;AACA+B,wBAAwBD,cAAc;AAEtC,MAAMM,mBAAuBC,KAAsBpC,UAAoB;CACrE,IAAI,OAAOoC,QAAQ,YACjBA,IAAIpC,KAAK;MACJ,IAAIoC,KACTA,IAAIE,UAAUtC;AAElB;AAEA,MAAMuC,0BAAyBC,iBAAA;CAAA,MAAAtB,IAAAC,EAAA,CAAA;CAG7B,MAAAsB,cAAoBjE,OAAuC,IAAI;CAAE,IAAAyC;CAAA,IAAAC,EAAA,OAAAsB,cAAA;EAG/DvB,MAAAyB,SAAA;GACED,YAAWH,UAAWI;GACtBP,gBAAgBK,cAAcE,IAAI;EAAC;EACpCxB,EAAA,KAAAsB;EAAAtB,EAAA,KAAAD;CAAA,OAAAA,KAAAC,EAAA;CAJH,MAAAyB,iBAAuB1B;CAMrB,IAAAK;CAAA,IAAAJ,EAAA,OAAAyB,gBAAA;EAEKrB,KAAA,CAACmB,aAAaE,cAAc;EAACzB,EAAA,KAAAyB;EAAAzB,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA7BI;AAAsC;AAG/C,MAAMuB,uCAAsC5B,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAsB,aAAAK,aAAA1C,YAAAkB,IAAAjB,0BAAAmB,IAAAlB,4BAAAyC,IAAAxC,8BAAAyC,OAAA/B;CAG3C,MAAAb,aAAAkB,OAAA2B,KAAAA,IAAA,OAAA3B;CACA,MAAAjB,2BAAAmB,OAAAyB,KAAAA,IAAA,OAAAzB;CACA,MAAAlB,6BAAAyC,OAAAE,KAAAA,IAAA,OAAAF;CACA,MAAAxC,+BAAAyC,OAAAC,KAAAA,IAAA,OAAAD;CAAmC,IAAAE;CAAA,IAAAhC,EAAA,OAAAiC,OAAAC,IAAA,2BAAA,GAAA;EASTF,KAAA;GAAAG,eACT;GAACC,gBACA;GAACC,SACR;EACX;EAACrC,EAAA,KAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAJD,MAAAsC,aAAmBhF,OAAO0E,EAIzB;CACD,MAAAO,gBAAsBjF,OAAO,IAAI;CACjC,MAAAkF,sBAA4BlF,OAAO,CAAC;CACpC,MAAAmF,+BAAqCnF,OAAO,KAAK;CACjD,MAAAoF,2BAAiCpF,OAAsC,KAAK;CAAE,IAAAqF;CAAA,IAAA3C,EAAA,OAAAiC,OAAAC,IAAA,2BAAA,GAAA;EAEzCS,WAAA;GACnC,MAAA,EAAAR,eAAAE,SAAAD,mBAAmDE,WAAUlB;GAC7DmB,cAAanB,UACXe,iBAAiBC,kBACjBD,gBAAgBE,UAAUD,kBAAkB5C;EAFzB;EAGtBQ,EAAA,KAAA2C;CAAA,OAAAA,KAAA3C,EAAA;CALD,MAAA4C,mBAAyBD;CAKlB,IAAAE;CAAA,IAAA7C,EAAA,OAAAuB,aAAA;EAMLsB,MAAAC,aAAA;GACE,MAAA,EAAAX,eAAAY,iBAAAX,gBAAAY,qBAA0CV,WAAUlB;GACpDkB,WAAUlB,QAAQiB,UAAWY,KAAIC,IAAK,GAAGf,kBAAgBC,gBAAc;GACvEG,cAAanB,UAAW;GACxBG,YAAWH,SAAqB+B,YAAC,EAAAL,SAAW,CAAC;EAAC;EAC/C9C,EAAA,KAAAuB;EAAAvB,EAAA,KAAA6C;CAAA,OAAAA,KAAA7C,EAAA;CANH,MAAAoD,iBAAuBP;CAQrB,IAAAQ;CAAA,IAAArD,EAAA,OAAAd,cAAAc,EAAA,OAAAoD,gBAAA;EAGAC,MAAAC,UAAA;GACE,MAAAC,cAAoBhB,cAAanB;GACjC,MAAAoC,yBAA+BlB,WAAUlB,QAAQgB;GACjD,MAAAqB,mBAAuBH,MAAKI,YAAYC,OAAOC;GAC/CtB,WAAUlB,QAAQgB,iBAAkBA;GACpCQ,iBAAiB;GACjB,IAAI,CAACW,aAAW;GAKhB,IACErE,cACAsE,2BAA2B,KAC3BpB,qBAAmBoB,wBAAsB;IAEzC,MAAAK,UAAgBnB,yBAAwBtB;IACxCgC,eAAeS,UAAUA,QAAOf,WAAjB,KAAkC;GAAC,OAElDP,cAAanB,UAAW;EACzB;EACFpB,EAAA,KAAAd;EAAAc,EAAA,KAAAoD;EAAApD,EAAA,KAAAqD;CAAA,OAAAA,KAAArD,EAAA;CAtBH,MAAA8D,eAAqBT;CAwBnB,IAAAU;CAAA,IAAA/D,EAAA,OAAAiC,OAAAC,IAAA,2BAAA,GAAA;EAGA6B,MAAAC,YAAA;GACE,MAAA,EAAAC,eAAAC,aAAAC,sBACEb,QAAKI;GACP,MAAAU,iBAAuB5B,oBAAmBpB;GAC1C,MAAAiD,oBAA0B9B,cAAanB;GACvCoB,oBAAmBpB,UAAW6C,cAAaK;GAC3ChC,WAAUlB,UAAW;IAAAe,eACJ+B,YAAWN;IAAOxB,gBACjB+B,kBAAiBP;IAAOvB,SAC/B4B,cAAaK;GACxB;GACA1B,iBAAiB;GACjB,MAAA2B,aAAmBN,cAAaK,IAAKF;GAKrC,IAAIC,qBAAA,CAAsBE,YACxBhC,cAAanB,UAAW;GAE1B,IAAI,CAACmB,cAAanB,WAAdmD,YACF7B,yBAAwBtB,UAAW;EACpC;EACFpB,EAAA,KAAA+D;CAAA,OAAAA,KAAA/D,EAAA;CAxBH,MAAAwE,eAAqBT;CA0BnB,IAAAU;CAAA,IAAAzE,EAAA,OAAAd,cAAAc,EAAA,OAAAoD,gBAAA;EAGAqB,OAAAC,QAAAd,WAAA;GACE,MAAAe,UAAgBrC,WAAUlB;GAC1B,MAAAwD,wBAA8BD,QAAOxC;GACrC,MAAA0C,gBAAoBtC,cAAanB;GACjCuD,QAAOxC,gBAAiByB;GACxBhB,iBAAiB;GAKjB,MAAAkC,gBAAsBpC,yBAAwBtB;GAC9C,IAAI0D,eAAa;IACfpC,yBAAwBtB,UAAW;IACnCgC,eAAe0B,cAAahC,QAAS;IAAC;GAAA;GAIxC,IAAI,CAAC5D,YAAU;GACf,IAAI,CAACqE,eAAW;GAChB,IAAIqB,0BAA0B,GAAC;GAC/B,IAAIhB,UAAUgB,uBAAqB;GAEnCxB,eAAe,KAAK;EAAC;EACtBpD,EAAA,KAAAd;EAAAc,EAAA,KAAAoD;EAAApD,EAAA,MAAAyE;CAAA,OAAAA,MAAAzE,EAAA;CAxBH,MAAA+E,0BAAgCN;CA0B9B,IAAAO;CAAA,IAAAC;CAAA,IAAAjF,EAAA,QAAA4B,eAAA5B,EAAA,QAAAoD,kBAAApD,EAAA,QAAAZ,4BAAA;EAEQ4F,YAAA;GACR,IAAI,CAAC5F,4BAA0B;GAC/B,IAAI,CAACwC,aAAW;IACda,6BAA4BrB,UAAW;IAAH;GAAA;GAGtC,IAAIqB,6BAA4BrB,SAAQ;GAExCqB,6BAA4BrB,UAAW;GACvCsB,yBAAwBtB,UAAW,EAAA0B,UAAY,MAAM;GACrDM,eAAe,KAAK;EAAC;EACpB6B,MAAA;GAACrD;GAAawB;GAAgBhE;EAA0B;EAACY,EAAA,MAAA4B;EAAA5B,EAAA,MAAAoD;EAAApD,EAAA,MAAAZ;EAAAY,EAAA,MAAAgF;EAAAhF,EAAA,MAAAiF;CAAA,OAAA;EAAAD,MAAAhF,EAAA;EAAAiF,MAAAjF,EAAA;CAAA;CAX5D3C,UAAU2H,KAWPC,GAAyD;CAAC,IAAAC;CAAA,IAAAlF,EAAA,QAAAoD,kBAAApD,EAAA,QAAAb,0BAAA;EAE9B+F,YAAA;GAC7B,IAAI,CAAC/F,0BAAwB;GAC7BuD,yBAAwBtB,UAAW,EAAA0B,UAAY,KAAK;GACpDM,eAAe,IAAI;EAAC;EACrBpD,EAAA,MAAAoD;EAAApD,EAAA,MAAAb;EAAAa,EAAA,MAAAkF;CAAA,OAAAA,MAAAlF,EAAA;CAJDjC,YAAY,mBAAmBmH,GAI9B;CAAC,IAAAC;CAAA,IAAAnF,EAAA,QAAAoD,kBAAApD,EAAA,QAAAX,8BAAA;EAEuC8F,YAAA;GACvC,IAAI,CAAC9F,8BAA4B;GACjCoD,6BAA4BrB,UAAW;GACvCoB,oBAAmBpB,UAAW;GAC9BsB,yBAAwBtB,UAAW,EAAA0B,UAAY,MAAM;GACrDM,eAAe,KAAK;EAAC;EACtBpD,EAAA,MAAAoD;EAAApD,EAAA,MAAAX;EAAAW,EAAA,MAAAmF;CAAA,OAAAA,MAAAnF,EAAA;CANDjC,YAAY,6BAA6BoH,GAMxC;CAAC,IAAAC;CAAA,IAAApF,EAAA,QAAA+E,2BAAA/E,EAAA,QAAA8D,cAAA;EAEKsB,MAAA;GAAAtB;GAAAU;GAAAO;EAIP;EAAC/E,EAAA,MAAA+E;EAAA/E,EAAA,MAAA8D;EAAA9D,EAAA,MAAAoF;CAAA,OAAAA,MAAApF,EAAA;CAAA,OAJMoF;AAIN;AAGH,MAAaC,yBAAyBnI,YAIpC6C,IAAAuB,iBAAA;CAAA,MAAAtB,IAAAC,EAAA,EAAA;CAAA,IAAAf;CAAA,IAAAL;CAAA,IAAAD;CAAA,IAAA0G;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAtG;CAAA,IAAAD;CAAA,IAAAE;CAAA,IAAAW,EAAA,OAAAD,IAAA;EACE,CAAA,CAAAb,YAAAN,YAAAC,UAAA0G,qBAAAC,UAAAC,UAAAC,qBAAAtG,4BAAAD,0BAAAE,iCAAAiG,iBAAAvF;EAYCC,EAAA,KAAAD;EAAAC,EAAA,KAAAd;EAAAc,EAAA,KAAAnB;EAAAmB,EAAA,KAAApB;EAAAoB,EAAA,KAAAsF;EAAAtF,EAAA,KAAAuF;EAAAvF,EAAA,KAAAwF;EAAAxF,EAAA,KAAAyF;EAAAzF,EAAA,KAAA0F;EAAA1F,EAAA,KAAAZ;EAAAY,EAAA,MAAAb;EAAAa,EAAA,MAAAX;CAAA,OAAA;EAAAH,aAAAc,EAAA;EAAAnB,WAAAmB,EAAA;EAAApB,aAAAoB,EAAA;EAAAsF,gBAAAtF,EAAA;EAAAuF,sBAAAvF,EAAA;EAAAwF,WAAAxF,EAAA;EAAAyF,WAAAzF,EAAA;EAAA0F,sBAAA1F,EAAA;EAAAZ,6BAAAY,EAAA;EAAAb,2BAAAa,EAAA;EAAAX,+BAAAW,EAAA;CAAA;CAGD,MAAA2F,WAAiB3H,YAAY4H,MAAwB;CACrD,MAAA,CAAArE,aAAAE,kBAAsCJ,uBAAuBC,YAAY;CAO1D,MAAAlB,KAAAuF,SAAQE,SAAU;CAAC,IAAAvF;CAAA,IAAAN,EAAA,QAAAd,cAAAc,EAAA,QAAAuB,eAAAvB,EAAA,QAAAZ,8BAAAY,EAAA,QAAAb,4BAAAa,EAAA,QAAAX,gCAAAW,EAAA,QAAAI,IAAA;EAFME,KAAA;GAAAiB;GAAAK,aAEzBxB;GAAmBlB;GAAAE;GAAAD;GAAAE;EAKlC;EAACW,EAAA,MAAAd;EAAAc,EAAA,MAAAuB;EAAAvB,EAAA,MAAAZ;EAAAY,EAAA,MAAAb;EAAAa,EAAA,MAAAX;EAAAW,EAAA,MAAAI;EAAAJ,EAAA,MAAAM;CAAA,OAAAA,KAAAN,EAAA;CAXD,MAAA,EAAA+E,yBAAAe,mCAAAhC,cAAAiC,wBAAAvB,cAAAwB,2BAIIrE,oCAAoCrB,EAOvC;CAAE,IAAAuB;CAAA,IAAA7B,EAAA,QAAAnB,YAAAmB,EAAA,QAAApB,YAAA;EAGDiD,MAAAC,OAAA;GAAC,MAAA,EAAAtB,UAAAsB;GACC,IAAIjD,UAAQ,OAER,oBAAC,yBAAD;IAAgC2B;IAC7B3B;GADqB,CAAA;GAI3B,OACM,oBAAC,sBAAD;IAA6B2B;IAAmB5B;GAAU,CAAA;EAAK;EACvEoB,EAAA,MAAAnB;EAAAmB,EAAA,MAAApB;EAAAoB,EAAA,MAAA6B;CAAA,OAAAA,KAAA7B,EAAA;CAVH,MAAAiG,aAAmBpE;CAcnB,MAAAqE,eAAqBC;CAErB,MAAAC,kBACGlH,cAAA,UACAE,8BAAA,UACAD,4BAAA,UACAE,gCAAA;CAAsC,IAAAyC;CAAA,IAAA9B,EAAA,QAAA+F,0BAAA/F,EAAA,QAAAwF,UAAA;EAGvC1D,MAAAwB,UAAA;GACEyC,uBAAuBzC,KAAK;GAC5BkC,WAAWlC,KAAK;EAAC;EAClBtD,EAAA,MAAA+F;EAAA/F,EAAA,MAAAwF;EAAAxF,EAAA,MAAA8B;CAAA,OAAAA,KAAA9B,EAAA;CAJH,MAAA8D,eAAqBhC;CAMnB,IAAAE;CAAA,IAAAhC,EAAA,QAAAgG,0BAAAhG,EAAA,QAAAyF,UAAA;EAGAzD,MAAAgC,YAAA;GACEgC,uBAAuB1C,OAAK;GAC5BmC,WAAWnC,OAAK;EAAC;EAClBtD,EAAA,MAAAgG;EAAAhG,EAAA,MAAAyF;EAAAzF,EAAA,MAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAJH,MAAAwE,eAAqBxC;CAMnB,IAAAW;CAAA,IAAA3C,EAAA,QAAA8F,qCAAA9F,EAAA,QAAAuF,qBAAA;EAGA5C,MAAA0D,OAAAzC,WAAA;GACEkC,kCAAkCO,OAAOzC,MAAM;GAC/C2B,sBAAsBc,OAAOzC,MAAM;EAAC;EACrC5D,EAAA,MAAA8F;EAAA9F,EAAA,MAAAuF;EAAAvF,EAAA,MAAA2C;CAAA,OAAAA,KAAA3C,EAAA;CAJH,MAAA+E,0BAAgCpC;CAWtB,MAAAE,KAAA8C;CAAsC,IAAAtC;CAAA,IAAArD,EAAA,QAAA+E,2BAAA/E,EAAA,QAAA8D,gBAAA9D,EAAA,QAAAwE,gBAAAxE,EAAA,QAAAuF,uBAAAvF,EAAA,QAAAwF,YAAAxF,EAAA,QAAAyF,YAAAzF,EAAA,QAAA0F,uBAAA1F,EAAA,QAAAoG,gBAAA;EAGvC/C,KAAA+C,iBAAA;GAAAb,qBAEsBR;GAAuBS,UAClC1B;GAAY2B,UACZjB;GAAYkB,qBACDA,uBAAA;EAOvB,IAZC;GAAA,GAQKH,uBAAA,EAAAA,oBAA6C;GAAC,GAC9CC,YAAA,EAAAA,SAAuB;GAAC,GACxBC,YAAA,EAAAA,SAAuB;GAAC,GACxBC,wBAAwB3D,KAAAA,KAAxB,EAAA2D,oBAA2D;EACjE;EAAC1F,EAAA,MAAA+E;EAAA/E,EAAA,MAAA8D;EAAA9D,EAAA,MAAAwE;EAAAxE,EAAA,MAAAuF;EAAAvF,EAAA,MAAAwF;EAAAxF,EAAA,MAAAyF;EAAAzF,EAAA,MAAA0F;EAAA1F,EAAA,MAAAoG;EAAApG,EAAA,MAAAqD;CAAA,OAAAA,KAAArD,EAAA;CAAA,IAAA+D;CAAA,IAAA/D,EAAA,QAAAsF,iBAAAtF,EAAA,QAAAiG,cAAAjG,EAAA,QAAAyB,kBAAAzB,EAAA,QAAA6C,MAAA7C,EAAA,QAAAqD,IAAA;EAjBPU,KAAA,oBAAC,UAAD;GACOtC,KAAAA;GACC,MAAAoB;GACMoD;GACEC;GAAY,GACrB7C;GAYA,GACDiC;EAAa,CAAA;EACjBtF,EAAA,MAAAsF;EAAAtF,EAAA,MAAAiG;EAAAjG,EAAA,MAAAyB;EAAAzB,EAAA,MAAA6C;EAAA7C,EAAA,MAAAqD;EAAArD,EAAA,MAAA+D;CAAA,OAAAA,KAAA/D,EAAA;CAAA,OAnBF+D;AAmBE,CAGR;AACAsB,uBAAuB1E,cAAc;;AAGrC,MAAa2F,iBAAiBpJ,YAI5B6C,IAAAmB,QAAA;CAAA,MAAAlB,IAAAC,EAAA,EAAA;CAAA,IAAAsG;CAAA,IAAAnG;CAAA,IAAAE;CAAA,IAAAuB;CAAA,IAAAC;CAAA,IAAA9B,EAAA,OAAAD,IAAA;EACE,CAAA,CAAAb,YAAAkB,IAAAhB,4BAAAkB,IAAAnB,0BAAA0C,IAAAxC,8BAAAyC,OAAAyE,SAAAxG;EAMCC,EAAA,KAAAD;EAAAC,EAAA,KAAAuG;EAAAvG,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;EAAAN,EAAA,KAAA6B;EAAA7B,EAAA,KAAA8B;CAAA,OAAA;EAAAyE,QAAAvG,EAAA;EAAAI,KAAAJ,EAAA;EAAAM,KAAAN,EAAA;EAAA6B,KAAA7B,EAAA;EAAA8B,KAAA9B,EAAA;CAAA;CALC,MAAAd,aAAAkB,OAAA2B,KAAAA,IAAA,QAAA3B;CACA,MAAAhB,6BAAAkB,OAAAyB,KAAAA,IAAA,QAAAzB;CACA,MAAAnB,2BAAA0C,OAAAE,KAAAA,IAAA,QAAAF;CACA,MAAAxC,+BAAAyC,OAAAC,KAAAA,IAAA,QAAAD;CAAoC,IAAAE;CAAA,IAAAhC,EAAA,OAAAd,cAAAc,EAAA,OAAAuG,SAAAvG,EAAA,OAAAkB,OAAAlB,EAAA,OAAAZ,8BAAAY,EAAA,QAAAb,4BAAAa,EAAA,QAAAX,8BAAA;EAKtC2C,KAAA,oBAAC,wBAAD;GACOd;GACOhC;GACgBE;GACFD;GACIE;GAA4B,GACtDkH;EAAK,CAAA;EACTvG,EAAA,KAAAd;EAAAc,EAAA,KAAAuG;EAAAvG,EAAA,KAAAkB;EAAAlB,EAAA,KAAAZ;EAAAY,EAAA,MAAAb;EAAAa,EAAA,MAAAX;EAAAW,EAAA,MAAAgC;CAAA,OAAAA,KAAAhC,EAAA;CAAA,OAPFgC;AAOE,CAEN;AACAsE,eAAe3F,cAAc;AAhXyC,SAAAT,MAAAsG,GAAA;CAAA,OAGpCA,EAACzH,QAAQW;AAAK;AAHsB,SAAAS,OAAAsG,KAAA;CAAA,OAI/BD,IAACzH,QAAQ2H,SAAS/G;AAAU;AA4OjE,SAAAiG,OAAAY,GAAA;CAAA,OAgBsCA,EAAC1F,OAAO6E;AAAS;AAhBvD,SAAAQ,OAAAQ,MAAA;CAAA,OA6C4DA,KAAIC;AAAG"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-native",
-  "version": "0.1.32",
+  "version": "0.1.33",
   "description": "React Native bindings for assistant-ui",
   "keywords": [
     "assistant",
@@ -31,10 +31,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.23",
-    "@assistant-ui/store": "^0.2.22",
-    "@assistant-ui/tap": "^0.9.6",
-    "assistant-stream": "^0.3.28",
+    "@assistant-ui/core": "^0.3.0",
+    "@assistant-ui/store": "^0.3.0",
+    "@assistant-ui/tap": "^0.9.7",
+    "assistant-stream": "^0.3.29",
     "zustand": "^5.0.14"
   },
   "peerDependencies": {

package/src/primitives/attachment/AttachmentRemove.test.tsx

@@ -8,9 +8,8 @@ const h = vi.hoisted(() => ({
 }));
 
 vi.mock("@assistant-ui/store", () => {
-  const aui = {
-    attachment: () => ({ remove: h.remove }),
-  };
+  const attachment = Object.assign(() => attachment, { remove: h.remove });
+  const aui = { attachment };
   return {
     useAui: () => aui,
   };

package/src/primitives/attachment/AttachmentRemove.tsx

@@ -14,7 +14,7 @@ export const AttachmentRemove = ({
   const aui = useAui();
 
   const handleRemove = useCallback(() => {
-    aui.attachment().remove();
+    aui.attachment.remove();
   }, [aui]);
 
   return (

package/src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.test.tsx

@@ -10,9 +10,10 @@ const h = vi.hoisted(() => ({
 }));
 
 vi.mock("@assistant-ui/store", () => {
-  const aui = {
-    chainOfThought: () => ({ setCollapsed: h.setCollapsed }),
-  };
+  const chainOfThought = Object.assign(() => chainOfThought, {
+    setCollapsed: h.setCollapsed,
+  });
+  const aui = { chainOfThought };
   return {
     useAui: () => aui,
     useAuiState: <T,>(selector: (s: typeof h.state) => T) => selector(h.state),

package/src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.tsx

@@ -17,7 +17,7 @@ export const ChainOfThoughtAccordionTrigger = ({
   const collapsed = useAuiState((s) => s.chainOfThought.collapsed);
 
   const onPress = useCallback(() => {
-    aui.chainOfThought().setCollapsed(!collapsed);
+    aui.chainOfThought.setCollapsed(!collapsed);
   }, [aui, collapsed]);
 
   return (

package/src/primitives/composer/ComposerInput.test.tsx

@@ -12,12 +12,11 @@ const h = vi.hoisted(() => ({
 }));
 
 vi.mock("@assistant-ui/store", () => {
-  const aui = {
-    composer: () => ({
-      setText: h.setText,
-      send: h.sendSpy,
-    }),
-  };
+  const composer = Object.assign(() => composer, {
+    setText: h.setText,
+    send: h.sendSpy,
+  });
+  const aui = { composer };
   return {
     useAui: () => aui,
     useAuiState: <T,>(

package/src/primitives/composer/ComposerInput.tsx

@@ -69,11 +69,11 @@ export const ComposerInput = ({
       if (Platform.OS === "web") {
         // Keep the controlled value in sync mid-IME so react-dom does not reset the textarea to a stale value
         flushTapSync(() => {
-          aui.composer().setText(value);
+          aui.composer.setText(value);
         });
         return;
       }
-      aui.composer().setText(value);
+      aui.composer.setText(value);
     },
     [aui],
   );
@@ -100,7 +100,7 @@ export const ComposerInput = ({
       if (nativeEvent.isComposing || nativeEvent.keyCode === 229) return;
       if (nativeEvent.key === "Enter" && !nativeEvent.shiftKey) {
         (e as unknown as Event).preventDefault?.();
-        aui.composer().send();
+        aui.composer.send();
       }
     },
     [aui, submitMode, onKeyPressProp],

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

@@ -11,23 +11,21 @@ const h = vi.hoisted(() => ({
   respondToToolApproval: vi.fn(),
   state: {
     message: { content: [] as AnyPart[] },
-    tools: { tools: {} as Record<string, unknown> },
+    tools: { toolUIs: {} as Record<string, unknown> },
     dataRenderers: { renderers: {} as Record<string, unknown> },
   },
 }));
 
 vi.mock("@assistant-ui/store", () => {
-  const aui = {
-    message: () => ({
-      part: ({ index }: { index: number }) => ({
-        addToolResult: (...args: unknown[]) => h.addToolResult(index, ...args),
-        resumeToolCall: (...args: unknown[]) =>
-          h.resumeToolCall(index, ...args),
-        respondToToolApproval: (...args: unknown[]) =>
-          h.respondToToolApproval(index, ...args),
-      }),
+  const message = Object.assign(() => message, {
+    part: ({ index }: { index: number }) => ({
+      addToolResult: (...args: unknown[]) => h.addToolResult(index, ...args),
+      resumeToolCall: (...args: unknown[]) => h.resumeToolCall(index, ...args),
+      respondToToolApproval: (...args: unknown[]) =>
+        h.respondToToolApproval(index, ...args),
     }),
-  };
+  });
+  const aui = { message };
   return {
     useAui: () => aui,
     useAuiState: <T,>(selector: (s: typeof h.state) => T) => selector(h.state),
@@ -45,7 +43,7 @@ describe("MessageContent", () => {
     h.resumeToolCall.mockReset();
     h.respondToToolApproval.mockReset();
     h.state.message.content = [];
-    h.state.tools.tools = {};
+    h.state.tools.toolUIs = {};
     h.state.dataRenderers.renderers = {};
 
     container = document.createElement("div");
@@ -153,7 +151,7 @@ describe("MessageContent", () => {
         (props.respondToApproval as () => void)();
         return <span data-testid="tool">tool:{String(props.toolName)}</span>;
       });
-      h.state.tools.tools = { search: ToolRender };
+      h.state.tools.toolUIs = { search: [{ render: ToolRender }] };
 
       await mount();
 
@@ -164,13 +162,15 @@ describe("MessageContent", () => {
       expect(h.respondToToolApproval).toHaveBeenCalledWith(0);
     });
 
-    it("picks the first renderer when the registry holds an array", async () => {
+    it("picks the first registration when multiple are registered", async () => {
       h.state.message.content = [
         { type: "tool-call", toolName: "search", toolCallId: "c1" },
       ];
       const First = vi.fn(() => <span data-testid="first">first</span>);
       const Second = vi.fn(() => <span>second</span>);
-      h.state.tools.tools = { search: [First, Second] };
+      h.state.tools.toolUIs = {
+        search: [{ render: First }, { render: Second }],
+      };
 
       await mount();
 
@@ -265,8 +265,8 @@ describe("MessageContent", () => {
       { type: "tool-call", toolName: "t", toolCallId: "c" },
       { type: "data", name: "d", data: {} },
     ];
-    h.state.tools.tools = {
-      t: () => <span>[tool]</span>,
+    h.state.tools.toolUIs = {
+      t: [{ render: () => <span>[tool]</span> }],
     };
     h.state.dataRenderers.renderers = {
       d: () => <span>[data]</span>,

package/src/primitives/message/MessageContent.tsx

@@ -65,17 +65,12 @@ const ToolUIDisplay = ({
   index: number;
 }) => {
   const aui = useAui();
-  const Render = useAuiState((s) => {
-    const renders = s.tools.tools[part.toolName];
-    if (Array.isArray(renders)) return renders[0];
-    return renders;
-  });
-
-  const partMethods = useMemo(
-    () => aui.message().part({ index }),
-    [aui, index],
+  const Render = useAuiState(
+    (s) => s.tools.toolUIs[part.toolName]?.[0]?.render,
   );
 
+  const partMethods = useMemo(() => aui.message.part({ index }), [aui, index]);
+
   if (Render) {
     return (
       <Render

package/src/primitives/thread/ThreadMessages.tsx

@@ -161,7 +161,7 @@ const ThreadMessageByChildren = memo(
     return (
       <MessageByIndexProvider index={index}>
         <RenderChildrenWithAccessor
-          getItemState={(aui) => aui.thread().message({ index }).getState()}
+          getItemState={(aui) => aui.thread.message({ index }).getState()}
         >
           {(getItem) =>
             children({