@assistant-ui/react-native 0.1.3 → 0.1.6

package/src/primitives/error/ErrorMessage.tsx

@@ -0,0 +1,17 @@
+import type { ReactNode } from "react";
+import { Text, type TextProps } from "react-native";
+import { useMessageError } from "@assistant-ui/core/react";
+
+export type ErrorMessageProps = TextProps & {
+  children?: ReactNode;
+};
+
+export const ErrorMessage = ({ children, ...textProps }: ErrorMessageProps) => {
+  const error = useMessageError();
+
+  if (error === undefined) return null;
+
+  return <Text {...textProps}>{children ?? String(error)}</Text>;
+};
+
+ErrorMessage.displayName = "ErrorPrimitive.Message";

package/src/primitives/error/ErrorRoot.tsx

@@ -0,0 +1,16 @@
+import type { ReactNode } from "react";
+import { View, type ViewProps } from "react-native";
+import { useMessageError } from "@assistant-ui/core/react";
+
+export type ErrorRootProps = ViewProps & {
+  children: ReactNode;
+};
+
+export const ErrorRoot = ({ children, ...viewProps }: ErrorRootProps) => {
+  const error = useMessageError();
+  if (error === undefined) return null;
+
+  return <View {...viewProps}>{children}</View>;
+};
+
+ErrorRoot.displayName = "ErrorPrimitive.Root";

package/src/primitives/error/index.ts

@@ -0,0 +1,8 @@
+export {
+  ErrorRoot as Root,
+  type ErrorRootProps as RootProps,
+} from "./ErrorRoot";
+export {
+  ErrorMessage as Message,
+  type ErrorMessageProps as MessageProps,
+} from "./ErrorMessage";

package/src/primitives/message/MessageContent.tsx

@@ -10,7 +10,7 @@ import { useAui, useAuiState } from "@assistant-ui/store";
 import type {
   ToolCallMessagePartProps,
   DataMessagePartProps,
-} from "../../types";
+} from "@assistant-ui/core/react";
 
 type MessageContentPart = ThreadUserMessagePart | ThreadAssistantMessagePart;
 

package/src/primitives/message/MessageParts.tsx

@@ -19,10 +19,16 @@ export namespace MessagePrimitiveParts {
 /**
  * Renders the parts of a message with React Native-specific default components.
  */
-export const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props> = ({
-  components,
-  ...rest
-}) => {
+export const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props> = (
+  props,
+) => {
+  if ("children" in props) {
+    return (
+      <MessagePrimitivePartsBase>{props.children}</MessagePrimitivePartsBase>
+    );
+  }
+
+  const { components, ...rest } = props;
   const merged = components
     ? {
         Text: components.Text ?? rnDefaultComponents.Text,
@@ -46,6 +52,7 @@ export const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props> = ({
                 messagePartsDefaultComponents.ReasoningGroup,
             }),
         Empty: components.Empty,
+        Quote: components.Quote,
       }
     : rnDefaultComponents;
 

package/src/primitives/{message/index.ts → message.ts}

@@ -1,17 +1,20 @@
 export {
   MessageRoot as Root,
   type MessageRootProps as RootProps,
-} from "./MessageRoot";
+} from "./message/MessageRoot";
 export {
   MessageContent as Content,
   type MessageContentProps as ContentProps,
-} from "./MessageContent";
+} from "./message/MessageContent";
 export {
   MessagePrimitiveParts as Parts,
   MessagePrimitivePartByIndex as PartByIndex,
-} from "./MessageParts";
-export { MessageIf as If, type MessageIfProps as IfProps } from "./MessageIf";
+} from "./message/MessageParts";
+export {
+  MessageIf as If,
+  type MessageIfProps as IfProps,
+} from "./message/MessageIf";
 export {
   MessageAttachments as Attachments,
   MessageAttachmentByIndex as AttachmentByIndex,
-} from "./MessageAttachments";
+} from "./message/MessageAttachments";

package/src/primitives/suggestion/SuggestionTrigger.tsx

@@ -1,6 +1,7 @@
-import { useCallback, type ReactNode } from "react";
+import type { ReactNode } from "react";
 import { Pressable, type PressableProps } from "react-native";
-import { useAuiState, useAui } from "@assistant-ui/store";
+import { useAuiState } from "@assistant-ui/store";
+import { useSuggestionTrigger } from "@assistant-ui/core/react";
 
 export type SuggestionTriggerProps = Omit<PressableProps, "onPress"> & {
   children: ReactNode;
@@ -24,38 +25,17 @@ export const SuggestionTrigger = ({
   disabled: disabledProp,
   ...pressableProps
 }: SuggestionTriggerProps) => {
-  const aui = useAui();
-  const isDisabled = useAuiState((s) => s.thread.isDisabled);
   const prompt = useAuiState((s) => s.suggestion.prompt);
-  const resolvedSend = send ?? false;
-
-  const onPress = useCallback(() => {
-    const isRunning = aui.thread().getState().isRunning;
-
-    if (resolvedSend && !isRunning) {
-      aui.thread().append({
-        content: [{ type: "text", text: prompt }],
-        runConfig: aui.composer().getState().runConfig,
-      });
-      if (clearComposer) {
-        aui.composer().setText("");
-      }
-    } else {
-      if (clearComposer) {
-        aui.composer().setText(prompt);
-      } else {
-        const currentText = aui.composer().getState().text;
-        aui
-          .composer()
-          .setText(currentText.trim() ? `${currentText} ${prompt}` : prompt);
-      }
-    }
-  }, [aui, resolvedSend, clearComposer, prompt]);
+  const { trigger, disabled } = useSuggestionTrigger({
+    prompt,
+    send,
+    clearComposer,
+  });
 
   return (
     <Pressable
-      onPress={onPress}
-      disabled={disabledProp ?? isDisabled}
+      onPress={trigger}
+      disabled={disabledProp ?? disabled}
       {...pressableProps}
     >
       {children}

package/src/primitives/{suggestion/index.ts → suggestion.ts}

@@ -1,12 +1,12 @@
 export {
   SuggestionTitle as Title,
   type SuggestionTitleProps as TitleProps,
-} from "./SuggestionTitle";
+} from "./suggestion/SuggestionTitle";
 export {
   SuggestionDescription as Description,
   type SuggestionDescriptionProps as DescriptionProps,
-} from "./SuggestionDescription";
+} from "./suggestion/SuggestionDescription";
 export {
   SuggestionTrigger as Trigger,
   type SuggestionTriggerProps as TriggerProps,
-} from "./SuggestionTrigger";
+} from "./suggestion/SuggestionTrigger";

package/src/primitives/thread/ThreadEmpty.tsx

@@ -1,5 +1,5 @@
 import type { ReactNode } from "react";
-import { useThreadIsEmpty } from "../../primitive-hooks/useThreadIsEmpty";
+import { useThreadIsEmpty } from "@assistant-ui/core/react";
 
 export type ThreadEmptyProps = {
   children: ReactNode;

package/src/primitives/thread/ThreadSuggestion.tsx

@@ -1,6 +1,6 @@
-import { useCallback, type ReactNode } from "react";
+import type { ReactNode } from "react";
 import { Pressable, type PressableProps } from "react-native";
-import { useAuiState, useAui } from "@assistant-ui/store";
+import { useSuggestionTrigger } from "@assistant-ui/core/react";
 
 export type ThreadSuggestionProps = Omit<PressableProps, "onPress"> & {
   children: ReactNode;
@@ -27,37 +27,16 @@ export const ThreadSuggestion = ({
   disabled: disabledProp,
   ...pressableProps
 }: ThreadSuggestionProps) => {
-  const aui = useAui();
-  const isDisabled = useAuiState((s) => s.thread.isDisabled);
-  const resolvedSend = send ?? false;
-
-  const onPress = useCallback(() => {
-    const isRunning = aui.thread().getState().isRunning;
-
-    if (resolvedSend && !isRunning) {
-      aui.thread().append({
-        content: [{ type: "text", text: prompt }],
-        runConfig: aui.composer().getState().runConfig,
-      });
-      if (clearComposer) {
-        aui.composer().setText("");
-      }
-    } else {
-      if (clearComposer) {
-        aui.composer().setText(prompt);
-      } else {
-        const currentText = aui.composer().getState().text;
-        aui
-          .composer()
-          .setText(currentText.trim() ? `${currentText} ${prompt}` : prompt);
-      }
-    }
-  }, [aui, resolvedSend, clearComposer, prompt]);
+  const { trigger, disabled } = useSuggestionTrigger({
+    prompt,
+    send,
+    clearComposer,
+  });
 
   return (
     <Pressable
-      onPress={onPress}
-      disabled={disabledProp ?? isDisabled}
+      onPress={trigger}
+      disabled={disabledProp ?? disabled}
       {...pressableProps}
     >
       {children}

package/src/primitives/{thread/index.ts → thread.ts}

@@ -1,21 +1,24 @@
 export {
   ThreadRoot as Root,
   type ThreadRootProps as RootProps,
-} from "./ThreadRoot";
+} from "./thread/ThreadRoot";
 export {
   ThreadMessages as Messages,
   type ThreadMessagesProps as MessagesProps,
-} from "./ThreadMessages";
+} from "./thread/ThreadMessages";
 export { ThreadPrimitiveMessageByIndex as MessageByIndex } from "@assistant-ui/core/react";
 export {
   ThreadEmpty as Empty,
   type ThreadEmptyProps as EmptyProps,
-} from "./ThreadEmpty";
-export { ThreadIf as If, type ThreadIfProps as IfProps } from "./ThreadIf";
+} from "./thread/ThreadEmpty";
+export {
+  ThreadIf as If,
+  type ThreadIfProps as IfProps,
+} from "./thread/ThreadIf";
 export {
   ThreadSuggestion as Suggestion,
   type ThreadSuggestionProps as SuggestionProps,
-} from "./ThreadSuggestion";
+} from "./thread/ThreadSuggestion";
 export {
   ThreadPrimitiveSuggestions as Suggestions,
   ThreadPrimitiveSuggestionByIndex as SuggestionByIndex,

package/src/primitives/threadList/ThreadListNew.tsx

@@ -1,6 +1,6 @@
-import { useCallback, type ReactNode } from "react";
+import type { ReactNode } from "react";
 import { Pressable, type PressableProps } from "react-native";
-import { useAui } from "@assistant-ui/store";
+import { useThreadListNew } from "@assistant-ui/core/react";
 
 export type ThreadListNewProps = Omit<PressableProps, "onPress"> & {
   children: ReactNode;
@@ -10,14 +10,10 @@ export const ThreadListNew = ({
   children,
   ...pressableProps
 }: ThreadListNewProps) => {
-  const aui = useAui();
-
-  const onPress = useCallback(() => {
-    aui.threads().switchToNewThread();
-  }, [aui]);
+  const { switchToNewThread } = useThreadListNew();
 
   return (
-    <Pressable onPress={onPress} {...pressableProps}>
+    <Pressable onPress={switchToNewThread} {...pressableProps}>
       {children}
     </Pressable>
   );

package/src/primitives/{threadList/index.ts → threadList.ts}

@@ -1,12 +1,12 @@
 export {
   ThreadListRoot as Root,
   type ThreadListRootProps as RootProps,
-} from "./ThreadListRoot";
+} from "./threadList/ThreadListRoot";
 export {
   ThreadListItems as Items,
   type ThreadListItemsProps as ItemsProps,
-} from "./ThreadListItems";
+} from "./threadList/ThreadListItems";
 export {
   ThreadListNew as New,
   type ThreadListNewProps as NewProps,
-} from "./ThreadListNew";
+} from "./threadList/ThreadListNew";

package/src/primitives/threadListItem/ThreadListItemArchive.tsx

@@ -1,6 +1,6 @@
-import { useCallback, type ReactNode } from "react";
+import type { ReactNode } from "react";
 import { Pressable, type PressableProps } from "react-native";
-import { useAui } from "@assistant-ui/store";
+import { useThreadListItemArchive } from "@assistant-ui/core/react";
 
 export type ThreadListItemArchiveProps = Omit<PressableProps, "onPress"> & {
   children: ReactNode;
@@ -10,14 +10,10 @@ export const ThreadListItemArchive = ({
   children,
   ...pressableProps
 }: ThreadListItemArchiveProps) => {
-  const aui = useAui();
-
-  const onPress = useCallback(() => {
-    aui.threadListItem().archive();
-  }, [aui]);
+  const { archive } = useThreadListItemArchive();
 
   return (
-    <Pressable onPress={onPress} {...pressableProps}>
+    <Pressable onPress={archive} {...pressableProps}>
       {children}
     </Pressable>
   );

package/src/primitives/threadListItem/ThreadListItemDelete.tsx

@@ -1,6 +1,6 @@
-import { useCallback, type ReactNode } from "react";
+import type { ReactNode } from "react";
 import { Pressable, type PressableProps } from "react-native";
-import { useAui } from "@assistant-ui/store";
+import { useThreadListItemDelete } from "@assistant-ui/core/react";
 
 export type ThreadListItemDeleteProps = Omit<PressableProps, "onPress"> & {
   children: ReactNode;
@@ -10,14 +10,10 @@ export const ThreadListItemDelete = ({
   children,
   ...pressableProps
 }: ThreadListItemDeleteProps) => {
-  const aui = useAui();
-
-  const onPress = useCallback(() => {
-    aui.threadListItem().delete();
-  }, [aui]);
+  const { delete: deleteThread } = useThreadListItemDelete();
 
   return (
-    <Pressable onPress={onPress} {...pressableProps}>
+    <Pressable onPress={deleteThread} {...pressableProps}>
       {children}
     </Pressable>
   );

package/src/primitives/threadListItem/ThreadListItemTrigger.tsx

@@ -1,6 +1,6 @@
-import { useCallback, type ReactNode } from "react";
+import type { ReactNode } from "react";
 import { Pressable, type PressableProps } from "react-native";
-import { useAui } from "@assistant-ui/store";
+import { useThreadListItemTrigger } from "@assistant-ui/core/react";
 
 export type ThreadListItemTriggerProps = Omit<PressableProps, "onPress"> & {
   children: ReactNode;
@@ -10,14 +10,10 @@ export const ThreadListItemTrigger = ({
   children,
   ...pressableProps
 }: ThreadListItemTriggerProps) => {
-  const aui = useAui();
-
-  const onPress = useCallback(() => {
-    aui.threadListItem().switchTo();
-  }, [aui]);
+  const { switchTo } = useThreadListItemTrigger();
 
   return (
-    <Pressable onPress={onPress} {...pressableProps}>
+    <Pressable onPress={switchTo} {...pressableProps}>
       {children}
     </Pressable>
   );

package/src/primitives/threadListItem/ThreadListItemUnarchive.tsx

@@ -1,6 +1,6 @@
-import { useCallback, type ReactNode } from "react";
+import type { ReactNode } from "react";
 import { Pressable, type PressableProps } from "react-native";
-import { useAui } from "@assistant-ui/store";
+import { useThreadListItemUnarchive } from "@assistant-ui/core/react";
 
 export type ThreadListItemUnarchiveProps = Omit<PressableProps, "onPress"> & {
   children: ReactNode;
@@ -10,14 +10,10 @@ export const ThreadListItemUnarchive = ({
   children,
   ...pressableProps
 }: ThreadListItemUnarchiveProps) => {
-  const aui = useAui();
-
-  const onPress = useCallback(() => {
-    aui.threadListItem().unarchive();
-  }, [aui]);
+  const { unarchive } = useThreadListItemUnarchive();
 
   return (
-    <Pressable onPress={onPress} {...pressableProps}>
+    <Pressable onPress={unarchive} {...pressableProps}>
       {children}
     </Pressable>
   );

package/src/primitives/{threadListItem/index.ts → threadListItem.ts}

@@ -1,21 +1,21 @@
 export {
   ThreadListItemRoot as Root,
   type ThreadListItemRootProps as RootProps,
-} from "./ThreadListItemRoot";
+} from "./threadListItem/ThreadListItemRoot";
 export { ThreadListItemPrimitiveTitle as Title } from "@assistant-ui/core/react";
 export {
   ThreadListItemTrigger as Trigger,
   type ThreadListItemTriggerProps as TriggerProps,
-} from "./ThreadListItemTrigger";
+} from "./threadListItem/ThreadListItemTrigger";
 export {
   ThreadListItemDelete as Delete,
   type ThreadListItemDeleteProps as DeleteProps,
-} from "./ThreadListItemDelete";
+} from "./threadListItem/ThreadListItemDelete";
 export {
   ThreadListItemArchive as Archive,
   type ThreadListItemArchiveProps as ArchiveProps,
-} from "./ThreadListItemArchive";
+} from "./threadListItem/ThreadListItemArchive";
 export {
   ThreadListItemUnarchive as Unarchive,
   type ThreadListItemUnarchiveProps as UnarchiveProps,
-} from "./ThreadListItemUnarchive";
+} from "./threadListItem/ThreadListItemUnarchive";

package/src/runtimes/useLocalRuntime.ts

@@ -1,105 +1,4 @@
-import { useEffect, useMemo, useRef, useState } from "react";
-import type {
-  AssistantRuntime,
-  ChatModelAdapter,
-  ThreadMessageLike,
-  RemoteThreadListAdapter,
-} from "@assistant-ui/core";
-import { InMemoryThreadListAdapter } from "@assistant-ui/core";
-import type { LocalRuntimeOptionsBase } from "@assistant-ui/core/internal";
-import {
-  AssistantRuntimeImpl,
-  LocalRuntimeCore,
-} from "@assistant-ui/core/internal";
-import { useAuiState } from "@assistant-ui/store";
-import type { TitleGenerationAdapter } from "../adapters/TitleGenerationAdapter";
-import { useRemoteThreadListRuntime } from "./useRemoteThreadListRuntime";
-import { createLocalStorageAdapter } from "../adapters/LocalStorageThreadListAdapter";
-import { useRuntimeAdapters } from "../context/providers/RuntimeAdapterProvider";
-
-type AsyncStorageLike = {
-  getItem(key: string): Promise<string | null>;
-  setItem(key: string, value: string): Promise<void>;
-  removeItem(key: string): Promise<void>;
-};
-
-export type LocalRuntimeOptions = Omit<LocalRuntimeOptionsBase, "adapters"> & {
-  adapters?: Omit<LocalRuntimeOptionsBase["adapters"], "chatModel"> | undefined;
-  initialMessages?: readonly ThreadMessageLike[] | undefined;
-  titleGenerator?: TitleGenerationAdapter | undefined;
-  storage?: AsyncStorageLike | undefined;
-  storagePrefix?: string | undefined;
-};
-
-const useInnerLocalRuntime = (
-  chatModel: ChatModelAdapter,
-  options: Omit<
-    LocalRuntimeOptions,
-    "titleGenerator" | "storage" | "storagePrefix"
-  >,
-): AssistantRuntime => {
-  const { initialMessages, ...restOptions } = options;
-  const { modelContext, ...threadListAdapters } = useRuntimeAdapters() ?? {};
-
-  const opt: LocalRuntimeOptionsBase = {
-    ...restOptions,
-    adapters: {
-      ...threadListAdapters,
-      ...restOptions.adapters,
-      chatModel,
-    },
-  };
-
-  const [runtime] = useState(() => new LocalRuntimeCore(opt, initialMessages));
-
-  const threadIdRef = useRef<string | undefined>(undefined);
-  threadIdRef.current = useAuiState((s) => s.threadListItem.remoteId);
-
-  useEffect(() => {
-    runtime.threads
-      .getMainThreadRuntimeCore()
-      .__internal_setGetThreadId(() => threadIdRef.current);
-  }, [runtime]);
-
-  useEffect(() => {
-    return () => {
-      runtime.threads.getMainThreadRuntimeCore().detach();
-    };
-  }, [runtime]);
-
-  useEffect(() => {
-    runtime.threads.getMainThreadRuntimeCore().__internal_setOptions(opt);
-    runtime.threads.getMainThreadRuntimeCore().__internal_load();
-  });
-
-  useEffect(() => {
-    if (!modelContext) return undefined;
-    return runtime.registerModelContextProvider(modelContext);
-  }, [modelContext, runtime]);
-
-  return useMemo(() => new AssistantRuntimeImpl(runtime), [runtime]);
-};
-
-export const useLocalRuntime = (
-  chatModel: ChatModelAdapter,
-  options: LocalRuntimeOptions = {},
-): AssistantRuntime => {
-  const { titleGenerator, storage, storagePrefix, ...innerOptions } = options;
-
-  const adapter: RemoteThreadListAdapter = useMemo(() => {
-    if (!storage) return new InMemoryThreadListAdapter();
-    return createLocalStorageAdapter({
-      storage,
-      prefix: storagePrefix,
-      titleGenerator,
-    });
-  }, [storage, storagePrefix, titleGenerator]);
-
-  return useRemoteThreadListRuntime({
-    runtimeHook: function RuntimeHook() {
-      return useInnerLocalRuntime(chatModel, innerOptions);
-    },
-    adapter,
-    allowNesting: true,
-  });
-};
+export {
+  useLocalRuntime,
+  type LocalRuntimeOptions,
+} from "@assistant-ui/core/react";

package/dist/adapters/LocalStorageThreadListAdapter.d.ts

@@ -1,15 +0,0 @@
-import type { RemoteThreadListAdapter } from "@assistant-ui/core";
-import type { TitleGenerationAdapter } from "./TitleGenerationAdapter.js";
-type AsyncStorageLike = {
-    getItem(key: string): Promise<string | null>;
-    setItem(key: string, value: string): Promise<void>;
-    removeItem(key: string): Promise<void>;
-};
-type LocalStorageAdapterOptions = {
-    storage: AsyncStorageLike;
-    prefix?: string | undefined;
-    titleGenerator?: TitleGenerationAdapter | undefined;
-};
-export declare const createLocalStorageAdapter: (options: LocalStorageAdapterOptions) => RemoteThreadListAdapter;
-export {};
-//# sourceMappingURL=LocalStorageThreadListAdapter.d.ts.map

package/dist/adapters/LocalStorageThreadListAdapter.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"LocalStorageThreadListAdapter.d.ts","sourceRoot":"","sources":["../../src/adapters/LocalStorageThreadListAdapter.tsx"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAEV,uBAAuB,EAKxB,MAAM,oBAAoB,CAAC;AAM5B,OAAO,KAAK,EAAE,sBAAsB,EAAE,oCAAiC;AAEvE,KAAK,gBAAgB,GAAG;IACtB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACxC,CAAC;AAEF,KAAK,0BAA0B,GAAG;IAChC,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,cAAc,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;CACrD,CAAC;AAyEF,eAAO,MAAM,yBAAyB,GACpC,SAAS,0BAA0B,KAClC,uBA2HF,CAAC"}