@assistant-ui/react-native 0.1.42 → 0.1.43

package/src/primitives/actionBar/ActionBarSpeak.tsx

@@ -0,0 +1,42 @@
+import type { ReactNode } from "react";
+import {
+  Pressable,
+  type PressableProps,
+  type PressableStateCallbackType,
+} from "react-native";
+import { useActionBarSpeak } from "@assistant-ui/core/react";
+
+export type ActionBarSpeakProps = Omit<
+  PressableProps,
+  "children" | "onPress"
+> & {
+  children:
+    | ReactNode
+    | ((
+        state: PressableStateCallbackType & { disabled: boolean },
+      ) => ReactNode);
+};
+
+export const ActionBarSpeak = ({
+  children,
+  disabled: disabledProp,
+  ...pressableProps
+}: ActionBarSpeakProps) => {
+  const { speak, disabled } = useActionBarSpeak();
+  const isDisabled = disabledProp ?? disabled;
+
+  return (
+    <Pressable
+      onPress={() => void speak()}
+      disabled={isDisabled}
+      accessibilityRole="button"
+      {...pressableProps}
+    >
+      {typeof children === "function"
+        ? (state) => children({ ...state, disabled: isDisabled })
+        : children}
+    </Pressable>
+  );
+};
+
+ActionBarSpeak.displayName = "ActionBarPrimitive.Speak";

package/src/primitives/actionBar/ActionBarStopSpeaking.tsx

@@ -0,0 +1,42 @@
+import type { ReactNode } from "react";
+import {
+  Pressable,
+  type PressableProps,
+  type PressableStateCallbackType,
+} from "react-native";
+import { useActionBarStopSpeaking } from "@assistant-ui/core/react";
+
+export type ActionBarStopSpeakingProps = Omit<
+  PressableProps,
+  "children" | "onPress"
+> & {
+  children:
+    | ReactNode
+    | ((
+        state: PressableStateCallbackType & { disabled: boolean },
+      ) => ReactNode);
+};
+
+export const ActionBarStopSpeaking = ({
+  children,
+  disabled: disabledProp,
+  ...pressableProps
+}: ActionBarStopSpeakingProps) => {
+  const { stopSpeaking, disabled } = useActionBarStopSpeaking();
+  const isDisabled = disabledProp ?? disabled;
+
+  return (
+    <Pressable
+      onPress={() => stopSpeaking()}
+      disabled={isDisabled}
+      accessibilityRole="button"
+      {...pressableProps}
+    >
+      {typeof children === "function"
+        ? (state) => children({ ...state, disabled: isDisabled })
+        : children}
+    </Pressable>
+  );
+};
+
+ActionBarStopSpeaking.displayName = "ActionBarPrimitive.StopSpeaking";

package/src/primitives/actionBar.ts

@@ -10,6 +10,14 @@ export {
   ActionBarReload as Reload,
   type ActionBarReloadProps as ReloadProps,
 } from "./actionBar/ActionBarReload";
+export {
+  ActionBarSpeak as Speak,
+  type ActionBarSpeakProps as SpeakProps,
+} from "./actionBar/ActionBarSpeak";
+export {
+  ActionBarStopSpeaking as StopSpeaking,
+  type ActionBarStopSpeakingProps as StopSpeakingProps,
+} from "./actionBar/ActionBarStopSpeaking";
 export {
   ActionBarFeedbackPositive as FeedbackPositive,
   type ActionBarFeedbackPositiveProps as FeedbackPositiveProps,

package/src/primitives/branchPicker/BranchPickerNext.test.tsx

@@ -8,9 +8,17 @@ const h = vi.hoisted(() => ({
   state: { disabled: false },
 }));
 
-vi.mock("@assistant-ui/core/react", () => ({
-  useBranchPickerNext: () => ({ next: h.next, disabled: h.state.disabled }),
-}));
+vi.mock("@assistant-ui/core/react", async (importOriginal) => {
+  const actual =
+    await importOriginal<typeof import("@assistant-ui/core/react")>();
+  return {
+    ...actual,
+    useBranchPickerNext: () => ({
+      next: h.next,
+      disabled: h.state.disabled,
+    }),
+  };
+});
 
 (globalThis as Record<string, unknown>).IS_REACT_ACT_ENVIRONMENT = true;
 
@@ -94,4 +102,45 @@ describe("BranchPickerNext", () => {
 
     expect(h.next).toHaveBeenCalledTimes(1);
   });
+
+  it("passes the hook disabled state and an explicit override to render children", async () => {
+    h.state.disabled = true;
+    let hookState: unknown = null;
+
+    await act(async () => {
+      root.render(
+        <BranchPickerNext testID="t">
+          {(state) => {
+            hookState = state;
+            return state.disabled ? "disabled" : "enabled";
+          }}
+        </BranchPickerNext>,
+      );
+    });
+
+    expect(hookState).toMatchObject({ pressed: false, disabled: true });
+    expect(container.textContent).toBe("disabled");
+
+    let overrideState: unknown = null;
+
+    await act(async () => {
+      root.render(
+        <BranchPickerNext testID="t" disabled={false}>
+          {(state) => {
+            overrideState = state;
+            return state.disabled ? "disabled" : "enabled";
+          }}
+        </BranchPickerNext>,
+      );
+    });
+
+    expect(overrideState).toMatchObject({ pressed: false, disabled: false });
+    expect(container.textContent).toBe("enabled");
+  });
+
+  it("renders plain children", async () => {
+    const el = await mount();
+
+    expect(el.textContent).toBe("next");
+  });
 });

package/src/primitives/branchPicker/BranchPickerNext.tsx

@@ -1,11 +1,20 @@
-import { Pressable, type PressableProps } from "react-native";
+import type { ReactNode } from "react";
+import {
+  Pressable,
+  type PressableProps,
+  type PressableStateCallbackType,
+} from "react-native";
 import { useBranchPickerNext } from "@assistant-ui/core/react";
 
 export type BranchPickerNextProps = Omit<
   PressableProps,
   "onPress" | "children"
 > & {
-  children: PressableProps["children"];
+  children:
+    | ReactNode
+    | ((
+        state: PressableStateCallbackType & { disabled: boolean },
+      ) => ReactNode);
 };
 
 export const BranchPickerNext = ({
@@ -14,15 +23,18 @@ export const BranchPickerNext = ({
   ...pressableProps
 }: BranchPickerNextProps) => {
   const { next, disabled } = useBranchPickerNext();
+  const isDisabled = disabledProp ?? disabled;
 
   return (
     <Pressable
       onPress={next}
-      disabled={disabledProp ?? disabled}
+      disabled={isDisabled}
       accessibilityRole="button"
       {...pressableProps}
     >
-      {children}
+      {typeof children === "function"
+        ? (state) => children({ ...state, disabled: isDisabled })
+        : children}
     </Pressable>
   );
 };

package/src/primitives/branchPicker/BranchPickerPrevious.test.tsx

@@ -8,12 +8,17 @@ const h = vi.hoisted(() => ({
   state: { disabled: false },
 }));
 
-vi.mock("@assistant-ui/core/react", () => ({
-  useBranchPickerPrevious: () => ({
-    previous: h.previous,
-    disabled: h.state.disabled,
-  }),
-}));
+vi.mock("@assistant-ui/core/react", async (importOriginal) => {
+  const actual =
+    await importOriginal<typeof import("@assistant-ui/core/react")>();
+  return {
+    ...actual,
+    useBranchPickerPrevious: () => ({
+      previous: h.previous,
+      disabled: h.state.disabled,
+    }),
+  };
+});
 
 (globalThis as Record<string, unknown>).IS_REACT_ACT_ENVIRONMENT = true;
 
@@ -97,4 +102,45 @@ describe("BranchPickerPrevious", () => {
 
     expect(h.previous).toHaveBeenCalledTimes(1);
   });
+
+  it("passes the hook disabled state and an explicit override to render children", async () => {
+    h.state.disabled = true;
+    let hookState: unknown = null;
+
+    await act(async () => {
+      root.render(
+        <BranchPickerPrevious testID="t">
+          {(state) => {
+            hookState = state;
+            return state.disabled ? "disabled" : "enabled";
+          }}
+        </BranchPickerPrevious>,
+      );
+    });
+
+    expect(hookState).toMatchObject({ pressed: false, disabled: true });
+    expect(container.textContent).toBe("disabled");
+
+    let overrideState: unknown = null;
+
+    await act(async () => {
+      root.render(
+        <BranchPickerPrevious testID="t" disabled={false}>
+          {(state) => {
+            overrideState = state;
+            return state.disabled ? "disabled" : "enabled";
+          }}
+        </BranchPickerPrevious>,
+      );
+    });
+
+    expect(overrideState).toMatchObject({ pressed: false, disabled: false });
+    expect(container.textContent).toBe("enabled");
+  });
+
+  it("renders plain children", async () => {
+    const el = await mount();
+
+    expect(el.textContent).toBe("prev");
+  });
 });

package/src/primitives/branchPicker/BranchPickerPrevious.tsx

@@ -1,11 +1,20 @@
-import { Pressable, type PressableProps } from "react-native";
+import type { ReactNode } from "react";
+import {
+  Pressable,
+  type PressableProps,
+  type PressableStateCallbackType,
+} from "react-native";
 import { useBranchPickerPrevious } from "@assistant-ui/core/react";
 
 export type BranchPickerPreviousProps = Omit<
   PressableProps,
   "onPress" | "children"
 > & {
-  children: PressableProps["children"];
+  children:
+    | ReactNode
+    | ((
+        state: PressableStateCallbackType & { disabled: boolean },
+      ) => ReactNode);
 };
 
 export const BranchPickerPrevious = ({
@@ -14,15 +23,18 @@ export const BranchPickerPrevious = ({
   ...pressableProps
 }: BranchPickerPreviousProps) => {
   const { previous, disabled } = useBranchPickerPrevious();
+  const isDisabled = disabledProp ?? disabled;
 
   return (
     <Pressable
       onPress={previous}
-      disabled={disabledProp ?? disabled}
+      disabled={isDisabled}
       accessibilityRole="button"
       {...pressableProps}
     >
-      {children}
+      {typeof children === "function"
+        ? (state) => children({ ...state, disabled: isDisabled })
+        : children}
     </Pressable>
   );
 };

package/src/primitives/messagePart/MessagePartImage.test.tsx

@@ -0,0 +1,91 @@
+import { act } from "react";
+import { createRoot, type Root } from "react-dom/client";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import type { ThreadMessageLike } from "@assistant-ui/core";
+import {
+  AssistantRuntimeProvider,
+  MessageByIndexProvider,
+  PartByIndexProvider,
+  useExternalStoreRuntime,
+} from "@assistant-ui/core/react";
+import { MessagePartPrimitiveImage } from "./MessagePartImage";
+
+vi.mock("react-native", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("react-native")>();
+  const React = await import("react");
+
+  const ImageMock = React.forwardRef(function ImageMock(
+    props: { source?: { uri?: string }; testID?: string },
+    ref: React.Ref<HTMLImageElement>,
+  ) {
+    return React.createElement("img", {
+      ref,
+      src: props.source?.uri,
+      "data-testid": props.testID,
+    });
+  });
+
+  return { ...actual, Image: ImageMock };
+});
+
+(globalThis as Record<string, unknown>).IS_REACT_ACT_ENVIRONMENT = true;
+
+const messages: ThreadMessageLike[] = [
+  {
+    role: "assistant",
+    content: [
+      { type: "text", text: "Answer" },
+      { type: "image", image: "https://example.com/image.png" },
+    ],
+  },
+];
+
+const App = ({ partIndex }: { partIndex: number }) => {
+  const runtime = useExternalStoreRuntime({
+    messages,
+    convertMessage: (message) => message,
+    onNew: async () => {
+      throw new Error("This thread is read-only");
+    },
+  });
+
+  return (
+    <AssistantRuntimeProvider runtime={runtime}>
+      <MessageByIndexProvider index={0}>
+        <PartByIndexProvider index={partIndex}>
+          <MessagePartPrimitiveImage testID="image" />
+        </PartByIndexProvider>
+      </MessageByIndexProvider>
+    </AssistantRuntimeProvider>
+  );
+};
+
+describe("MessagePartPrimitiveImage", () => {
+  let container: HTMLDivElement;
+  let root: Root;
+
+  beforeEach(() => {
+    container = document.createElement("div");
+    document.body.appendChild(container);
+    root = createRoot(container);
+  });
+
+  afterEach(async () => {
+    await act(async () => root.unmount());
+    container.remove();
+  });
+
+  const mount = async (partIndex: number) => {
+    await act(async () => root.render(<App partIndex={partIndex} />));
+    return container.querySelector<HTMLImageElement>('[data-testid="image"]');
+  };
+
+  it("renders the current image part URI", async () => {
+    const image = await mount(1);
+    expect(image?.getAttribute("src")).toBe("https://example.com/image.png");
+  });
+
+  it("renders nothing for a non-image part", async () => {
+    expect(await mount(0)).toBeNull();
+  });
+});

package/src/primitives/messagePart/MessagePartImage.tsx

@@ -0,0 +1,33 @@
+import {
+  type ComponentRef,
+  type ForwardRefExoticComponent,
+  forwardRef,
+  type RefAttributes,
+} from "react";
+import { Image, type ImageProps } from "react-native";
+import { useAuiState } from "@assistant-ui/store";
+
+export namespace MessagePartPrimitiveImage {
+  export type Element = ComponentRef<typeof Image>;
+  export type Props = Omit<ImageProps, "source">;
+}
+
+export const MessagePartPrimitiveImage: ForwardRefExoticComponent<
+  MessagePartPrimitiveImage.Props &
+    RefAttributes<MessagePartPrimitiveImage.Element>
+> = forwardRef<
+  MessagePartPrimitiveImage.Element,
+  MessagePartPrimitiveImage.Props
+>((props, forwardedRef) => {
+  const uri = useAuiState((s) => {
+    const part = s.part;
+    return part.type === "image" ? part.image : undefined;
+  });
+
+  if (uri === undefined) return null;
+  return <Image source={{ uri }} {...props} ref={forwardedRef} />;
+});
+
+MessagePartPrimitiveImage.displayName = "MessagePartPrimitive.Image";
+
+export type MessagePartImageProps = MessagePartPrimitiveImage.Props;

package/src/primitives/messagePart/MessagePartInProgress.ts

@@ -0,0 +1 @@
+export { MessagePartPrimitiveInProgress } from "@assistant-ui/core/react";

package/src/primitives/messagePart/MessagePartText.test.tsx

@@ -0,0 +1,80 @@
+import { act } from "react";
+import { createRoot, type Root } from "react-dom/client";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import type { ThreadMessageLike } from "@assistant-ui/core";
+import {
+  AssistantRuntimeProvider,
+  MessageByIndexProvider,
+  PartByIndexProvider,
+  useExternalStoreRuntime,
+} from "@assistant-ui/core/react";
+import { MessagePartPrimitiveText } from "./MessagePartText";
+
+(globalThis as Record<string, unknown>).IS_REACT_ACT_ENVIRONMENT = true;
+
+const messages: ThreadMessageLike[] = [
+  {
+    role: "assistant",
+    content: [
+      { type: "text", text: "Answer" },
+      { type: "reasoning", text: "Thinking" },
+      { type: "image", image: "https://example.com/image.png" },
+    ],
+  },
+];
+
+const App = ({ partIndex }: { partIndex: number }) => {
+  const runtime = useExternalStoreRuntime({
+    messages,
+    convertMessage: (message) => message,
+    onNew: async () => {
+      throw new Error("This thread is read-only");
+    },
+  });
+
+  return (
+    <AssistantRuntimeProvider runtime={runtime}>
+      <MessageByIndexProvider index={0}>
+        <PartByIndexProvider index={partIndex}>
+          <MessagePartPrimitiveText testID="text" />
+        </PartByIndexProvider>
+      </MessageByIndexProvider>
+    </AssistantRuntimeProvider>
+  );
+};
+
+describe("MessagePartPrimitiveText", () => {
+  let container: HTMLDivElement;
+  let root: Root;
+
+  beforeEach(() => {
+    container = document.createElement("div");
+    document.body.appendChild(container);
+    root = createRoot(container);
+  });
+
+  afterEach(async () => {
+    await act(async () => root.unmount());
+    container.remove();
+  });
+
+  const mount = async (partIndex: number) => {
+    await act(async () => root.render(<App partIndex={partIndex} />));
+    return container.querySelector('[data-testid="text"]') as HTMLElement;
+  };
+
+  it("renders text from the current text part", async () => {
+    const text = await mount(0);
+    expect(text.textContent).toBe("Answer");
+  });
+
+  it("renders text from the current reasoning part", async () => {
+    const text = await mount(1);
+    expect(text.textContent).toBe("Thinking");
+  });
+
+  it("renders empty text for a non-text part", async () => {
+    const text = await mount(2);
+    expect(text.textContent).toBe("");
+  });
+});

package/src/primitives/messagePart/MessagePartText.tsx

@@ -0,0 +1,36 @@
+import {
+  type ComponentRef,
+  type ForwardRefExoticComponent,
+  forwardRef,
+  type RefAttributes,
+} from "react";
+import { Text, type TextProps } from "react-native";
+import { useAuiState } from "@assistant-ui/store";
+
+export namespace MessagePartPrimitiveText {
+  export type Element = ComponentRef<typeof Text>;
+  export type Props = Omit<TextProps, "children">;
+}
+
+export const MessagePartPrimitiveText: ForwardRefExoticComponent<
+  MessagePartPrimitiveText.Props &
+    RefAttributes<MessagePartPrimitiveText.Element>
+> = forwardRef<
+  MessagePartPrimitiveText.Element,
+  MessagePartPrimitiveText.Props
+>((props, forwardedRef) => {
+  const text = useAuiState((s) => {
+    const part = s.part;
+    return part.type === "text" || part.type === "reasoning" ? part.text : "";
+  });
+
+  return (
+    <Text {...props} ref={forwardedRef}>
+      {text}
+    </Text>
+  );
+});
+
+MessagePartPrimitiveText.displayName = "MessagePartPrimitive.Text";
+
+export type MessagePartTextProps = MessagePartPrimitiveText.Props;

package/src/primitives/messagePart.ts

@@ -0,0 +1,10 @@
+export {
+  MessagePartPrimitiveText as Text,
+  type MessagePartTextProps as TextProps,
+} from "./messagePart/MessagePartText";
+export {
+  MessagePartPrimitiveImage as Image,
+  type MessagePartImageProps as ImageProps,
+} from "./messagePart/MessagePartImage";
+export { MessagePartPrimitiveInProgress as InProgress } from "./messagePart/MessagePartInProgress";
+export { PartPrimitiveMessages as Messages } from "@assistant-ui/core/react";

package/src/primitives/thread/ThreadMessages.test.tsx

@@ -1,7 +1,7 @@
 import { act, createRef } from "react";
 import { createRoot, type Root } from "react-dom/client";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
-import type { FlatList } from "react-native";
+import type { FlatList, FlatListProps } from "react-native";
 import type { ThreadMessage } from "@assistant-ui/core";
 import { ThreadMessages, ThreadMessagesFlatList } from "./ThreadMessages";
 
@@ -154,6 +154,10 @@ describe("ThreadMessages", () => {
       onContentSizeChange?: (width: number, height: number) => void;
       onLayout?: (event: unknown) => void;
       onScroll?: (event: unknown) => void;
+      onStartReached?: NonNullable<
+        FlatListProps<ThreadMessage>["onStartReached"]
+      >;
+      onStartReachedThreshold?: number;
       scrollEventThrottle?: number;
     } | null;
     if (!props) throw new Error("FlatList was not rendered");
@@ -392,6 +396,80 @@ describe("ThreadMessages", () => {
     expect(h.scrollToOffset).not.toHaveBeenCalled();
   });
 
+  describe("MessagesFlatList history", () => {
+    it("wires loadMore only while history can load more", async () => {
+      const loadMore = vi.fn();
+
+      await mountFlatList({
+        components: messageComponents,
+        history: { hasMore: true, isLoadingMore: false, loadMore },
+      });
+      getFlatListProps().onStartReached?.({ distanceFromStart: 0 });
+      expect(loadMore).toHaveBeenCalledOnce();
+
+      await mountFlatList({
+        components: messageComponents,
+        history: { hasMore: false, isLoadingMore: false, loadMore },
+      });
+      expect(getFlatListProps().onStartReached).toBeUndefined();
+
+      await mountFlatList({
+        components: messageComponents,
+        history: { hasMore: true, isLoadingMore: true, loadMore },
+      });
+      expect(getFlatListProps().onStartReached).toBeUndefined();
+    });
+
+    it("defaults the history threshold and preserves a caller override", async () => {
+      const history = {
+        hasMore: true,
+        isLoadingMore: false,
+        loadMore: vi.fn(),
+      };
+
+      await mountFlatList({ components: messageComponents, history });
+      expect(getFlatListProps().onStartReachedThreshold).toBe(1);
+
+      await mountFlatList({
+        components: messageComponents,
+        history,
+        onStartReachedThreshold: 0.5,
+      });
+      expect(getFlatListProps().onStartReachedThreshold).toBe(0.5);
+    });
+
+    it("runs a caller onStartReached before loading more history", async () => {
+      const calls: string[] = [];
+      const info = { distanceFromStart: 42 };
+      const onStartReached = vi.fn(() => calls.push("onStartReached"));
+
+      await mountFlatList({
+        components: messageComponents,
+        history: {
+          hasMore: true,
+          isLoadingMore: false,
+          loadMore: () => calls.push("loadMore"),
+        },
+        onStartReached,
+      });
+
+      getFlatListProps().onStartReached?.(info);
+
+      expect(onStartReached).toHaveBeenCalledWith(info);
+      expect(calls).toEqual(["onStartReached", "loadMore"]);
+    });
+
+    it("leaves start-reached props untouched without history", async () => {
+      const onStartReached = vi.fn();
+
+      await mountFlatList({ components: messageComponents, onStartReached });
+
+      const props = getFlatListProps();
+      expect(props.onStartReached).toBe(onStartReached);
+      expect(props).not.toHaveProperty("onStartReachedThreshold");
+    });
+  });
+
   describe("MessagesFlatList auto-scroll", () => {
     it("scrolls to the bottom when messages first appear", async () => {
       h.state.thread.messages = [{ id: "1", role: "user" }];