@assistant-ui/react 0.12.17 → 0.12.19

package/src/primitives/actionBar/ActionBarStopSpeaking.tsx

@@ -5,20 +5,12 @@ import { ActionButtonProps } from "../../utils/createActionButton";
 import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
 import { Primitive } from "@radix-ui/react-primitive";
 import { composeEventHandlers } from "@radix-ui/primitive";
-import { useCallback } from "react";
-import { useAuiState, useAui } from "@assistant-ui/store";
+import { useActionBarStopSpeaking as useActionBarStopSpeakingBehavior } from "@assistant-ui/core/react";
 
 const useActionBarStopSpeaking = () => {
-  const aui = useAui();
-  const isSpeaking = useAuiState((s) => s.message.speech != null);
-
-  const callback = useCallback(() => {
-    aui.message().stopSpeaking();
-  }, [aui]);
-
-  if (!isSpeaking) return null;
-
-  return callback;
+  const { disabled, stopSpeaking } = useActionBarStopSpeakingBehavior();
+  if (disabled) return null;
+  return stopSpeaking;
 };
 
 export namespace ActionBarPrimitiveStopSpeaking {

package/src/primitives/actionBar.ts

@@ -0,0 +1,9 @@
+export { ActionBarPrimitiveRoot as Root } from "./actionBar/ActionBarRoot";
+export { ActionBarPrimitiveCopy as Copy } from "./actionBar/ActionBarCopy";
+export { ActionBarPrimitiveReload as Reload } from "./actionBar/ActionBarReload";
+export { ActionBarPrimitiveEdit as Edit } from "./actionBar/ActionBarEdit";
+export { ActionBarPrimitiveSpeak as Speak } from "./actionBar/ActionBarSpeak";
+export { ActionBarPrimitiveStopSpeaking as StopSpeaking } from "./actionBar/ActionBarStopSpeaking";
+export { ActionBarPrimitiveFeedbackPositive as FeedbackPositive } from "./actionBar/ActionBarFeedbackPositive";
+export { ActionBarPrimitiveFeedbackNegative as FeedbackNegative } from "./actionBar/ActionBarFeedbackNegative";
+export { ActionBarPrimitiveExportMarkdown as ExportMarkdown } from "./actionBar/ActionBarExportMarkdown";

package/src/primitives/actionBarMore.ts

@@ -0,0 +1,5 @@
+export { ActionBarMorePrimitiveRoot as Root } from "./actionBarMore/ActionBarMoreRoot";
+export { ActionBarMorePrimitiveTrigger as Trigger } from "./actionBarMore/ActionBarMoreTrigger";
+export { ActionBarMorePrimitiveContent as Content } from "./actionBarMore/ActionBarMoreContent";
+export { ActionBarMorePrimitiveItem as Item } from "./actionBarMore/ActionBarMoreItem";
+export { ActionBarMorePrimitiveSeparator as Separator } from "./actionBarMore/ActionBarMoreSeparator";

package/src/primitives/assistantModal.ts

@@ -0,0 +1,4 @@
+export { AssistantModalPrimitiveRoot as Root } from "./assistantModal/AssistantModalRoot";
+export { AssistantModalPrimitiveTrigger as Trigger } from "./assistantModal/AssistantModalTrigger";
+export { AssistantModalPrimitiveContent as Content } from "./assistantModal/AssistantModalContent";
+export { AssistantModalPrimitiveAnchor as Anchor } from "./assistantModal/AssistantModalAnchor";

package/src/primitives/attachment/AttachmentThumb.tsx

@@ -26,4 +26,4 @@ export const AttachmentPrimitiveThumb = forwardRef<
   );
 });
 
-AttachmentPrimitiveThumb.displayName = "AttachmentPrimitive.Thumb";
+AttachmentPrimitiveThumb.displayName = "AttachmentPrimitive.unstable_Thumb";

package/src/primitives/attachment.ts

@@ -0,0 +1,4 @@
+export { AttachmentPrimitiveRoot as Root } from "./attachment/AttachmentRoot";
+export { AttachmentPrimitiveThumb as unstable_Thumb } from "./attachment/AttachmentThumb";
+export { AttachmentPrimitiveName as Name } from "./attachment/AttachmentName";
+export { AttachmentPrimitiveRemove as Remove } from "./attachment/AttachmentRemove";

package/src/primitives/branchPicker/BranchPickerNext.ts

@@ -5,29 +5,12 @@ import {
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
-import { useCallback } from "react";
-import { useAuiState, useAui } from "@assistant-ui/store";
+import { useBranchPickerNext as useBranchPickerNextBehavior } from "@assistant-ui/core/react";
 
 const useBranchPickerNext = () => {
-  const aui = useAui();
-  const disabled = useAuiState((s) => {
-    // Disabled if no next branch
-    if (s.message.branchNumber >= s.message.branchCount) return true;
-
-    // Disabled if running and capability not supported
-    if (s.thread.isRunning && !s.thread.capabilities.switchBranchDuringRun) {
-      return true;
-    }
-
-    return false;
-  });
-
-  const callback = useCallback(() => {
-    aui.message().switchToBranch({ position: "next" });
-  }, [aui]);
-
+  const { disabled, next } = useBranchPickerNextBehavior();
   if (disabled) return null;
-  return callback;
+  return next;
 };
 
 export namespace BranchPickerPrimitiveNext {

package/src/primitives/branchPicker/BranchPickerPrevious.ts

@@ -5,50 +5,12 @@ import {
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
-import { useCallback } from "react";
-import { useAuiState, useAui } from "@assistant-ui/store";
+import { useBranchPickerPrevious as useBranchPickerPreviousBehavior } from "@assistant-ui/core/react";
 
-/**
- * Hook that provides navigation to the previous branch functionality.
- *
- * This hook returns a callback function that switches to the previous branch
- * in the message branch tree, or null if there is no previous branch available.
- *
- * @returns A previous branch callback function, or null if navigation is disabled
- *
- * @example
- * ```tsx
- * function CustomPreviousButton() {
- *   const previous = useBranchPickerPrevious();
- *
- *   return (
- *     <button onClick={previous} disabled={!previous}>
- *       {previous ? "Previous Branch" : "No Previous Branch"}
- *     </button>
- *   );
- * }
- * ```
- */
 const useBranchPickerPrevious = () => {
-  const aui = useAui();
-  const disabled = useAuiState((s) => {
-    // Disabled if no previous branch
-    if (s.message.branchNumber <= 1) return true;
-
-    // Disabled if running and capability not supported
-    if (s.thread.isRunning && !s.thread.capabilities.switchBranchDuringRun) {
-      return true;
-    }
-
-    return false;
-  });
-
-  const callback = useCallback(() => {
-    aui.message().switchToBranch({ position: "previous" });
-  }, [aui]);
-
+  const { disabled, previous } = useBranchPickerPreviousBehavior();
   if (disabled) return null;
-  return callback;
+  return previous;
 };
 
 export namespace BranchPickerPrimitivePrevious {

package/src/primitives/branchPicker/BranchPickerRoot.tsx

@@ -2,7 +2,7 @@
 
 import { Primitive } from "@radix-ui/react-primitive";
 import { type ComponentRef, forwardRef, ComponentPropsWithoutRef } from "react";
-import { If } from "../message";
+import { MessagePrimitiveIf as If } from "../message/MessageIf";
 
 export namespace BranchPickerPrimitiveRoot {
   export type Element = ComponentRef<typeof Primitive.div>;

package/src/primitives/branchPicker.ts

@@ -0,0 +1,5 @@
+export { BranchPickerPrimitiveNext as Next } from "./branchPicker/BranchPickerNext";
+export { BranchPickerPrimitivePrevious as Previous } from "./branchPicker/BranchPickerPrevious";
+export { BranchPickerPrimitiveCount as Count } from "./branchPicker/BranchPickerCount";
+export { BranchPickerPrimitiveNumber as Number } from "./branchPicker/BranchPickerNumber";
+export { BranchPickerPrimitiveRoot as Root } from "./branchPicker/BranchPickerRoot";

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

@@ -1,14 +1,14 @@
 export {
   ChainOfThoughtPrimitiveRoot as Root,
   type ChainOfThoughtPrimitiveRoot as RootPrimitive,
-} from "./ChainOfThoughtRoot";
+} from "./chainOfThought/ChainOfThoughtRoot";
 
 export {
   ChainOfThoughtPrimitiveAccordionTrigger as AccordionTrigger,
   type ChainOfThoughtPrimitiveAccordionTrigger as AccordionTriggerPrimitive,
-} from "./ChainOfThoughtAccordionTrigger";
+} from "./chainOfThought/ChainOfThoughtAccordionTrigger";
 
 export {
   ChainOfThoughtPrimitiveParts as Parts,
   type ChainOfThoughtPrimitiveParts as PartsPrimitive,
-} from "./ChainOfThoughtParts";
+} from "./chainOfThought/ChainOfThoughtParts";

package/src/primitives/composer/ComposerDictate.ts

@@ -1,28 +1,16 @@
 "use client";
 
-import { useCallback } from "react";
-import { useAui, useAuiState } from "@assistant-ui/store";
 import type {
   ActionButtonElement,
   ActionButtonProps,
 } from "../../utils/createActionButton";
 import { createActionButton } from "../../utils/createActionButton";
+import { useComposerDictate as useComposerDictateBehavior } from "@assistant-ui/core/react";
 
 const useComposerDictate = () => {
-  const aui = useAui();
-  const disabled = useAuiState(
-    (s) =>
-      s.composer.dictation != null ||
-      !s.thread.capabilities.dictation ||
-      !s.composer.isEditing,
-  );
-
-  const callback = useCallback(() => {
-    aui.composer().startDictation();
-  }, [aui]);
-
+  const { disabled, startDictation } = useComposerDictateBehavior();
   if (disabled) return null;
-  return callback;
+  return startDictation;
 };
 
 export namespace ComposerPrimitiveDictate {

package/src/primitives/composer.ts

@@ -0,0 +1,15 @@
+export { ComposerPrimitiveRoot as Root } from "./composer/ComposerRoot";
+export { ComposerPrimitiveInput as Input } from "./composer/ComposerInput";
+export { ComposerPrimitiveSend as Send } from "./composer/ComposerSend";
+export { ComposerPrimitiveCancel as Cancel } from "./composer/ComposerCancel";
+export { ComposerPrimitiveAddAttachment as AddAttachment } from "./composer/ComposerAddAttachment";
+export { ComposerPrimitiveAttachments as Attachments } from "./composer/ComposerAttachments";
+export { ComposerPrimitiveAttachmentByIndex as AttachmentByIndex } from "./composer/ComposerAttachments";
+export { ComposerPrimitiveAttachmentDropzone as AttachmentDropzone } from "./composer/ComposerAttachmentDropzone";
+export { ComposerPrimitiveDictate as Dictate } from "./composer/ComposerDictate";
+export { ComposerPrimitiveStopDictation as StopDictation } from "./composer/ComposerStopDictation";
+export { ComposerPrimitiveDictationTranscript as DictationTranscript } from "./composer/ComposerDictationTranscript";
+export { ComposerPrimitiveIf as If } from "./composer/ComposerIf";
+export { ComposerPrimitiveQuote as Quote } from "./composer/ComposerQuote";
+export { ComposerPrimitiveQuoteText as QuoteText } from "./composer/ComposerQuote";
+export { ComposerPrimitiveQuoteDismiss as QuoteDismiss } from "./composer/ComposerQuote";

package/src/primitives/error/ErrorMessage.tsx

@@ -2,7 +2,7 @@
 
 import { Primitive } from "@radix-ui/react-primitive";
 import { type ComponentRef, forwardRef, ComponentPropsWithoutRef } from "react";
-import { useAuiState } from "@assistant-ui/store";
+import { useMessageError } from "@assistant-ui/core/react";
 
 export namespace ErrorPrimitiveMessage {
   export type Element = ComponentRef<typeof Primitive.span>;
@@ -13,12 +13,7 @@ export const ErrorPrimitiveMessage = forwardRef<
   ErrorPrimitiveMessage.Element,
   ErrorPrimitiveMessage.Props
 >(({ children, ...props }, forwardRef) => {
-  const error = useAuiState((s) => {
-    return s.message.status?.type === "incomplete" &&
-      s.message.status.reason === "error"
-      ? s.message.status.error
-      : undefined;
-  });
+  const error = useMessageError();
 
   if (error === undefined) return null;
 

package/src/primitives/error.ts

@@ -0,0 +1,2 @@
+export { ErrorPrimitiveRoot as Root } from "./error/ErrorRoot";
+export { ErrorPrimitiveMessage as Message } from "./error/ErrorMessage";

package/src/primitives/message/MessageError.ts

@@ -1,15 +1,11 @@
 "use client";
 
 import { FC, PropsWithChildren } from "react";
-import { useAuiState } from "@assistant-ui/store";
+import { useMessageError } from "@assistant-ui/core/react";
 
 export const MessagePrimitiveError: FC<PropsWithChildren> = ({ children }) => {
-  const hasError = useAuiState(
-    (s) =>
-      s.message.status?.type === "incomplete" &&
-      s.message.status.reason === "error",
-  );
-  return hasError ? children : null;
+  const error = useMessageError();
+  return error !== undefined ? children : null;
 };
 
 MessagePrimitiveError.displayName = "MessagePrimitive.Error";

package/src/primitives/message/MessageParts.tsx

@@ -31,10 +31,16 @@ export namespace MessagePrimitiveParts {
 /**
  * Renders the parts of a message with web-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 ?? webDefaultComponents.Text,
@@ -58,6 +64,7 @@ export const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props> = ({
                 messagePartsDefaultComponents.ReasoningGroup,
             }),
         Empty: components.Empty,
+        Quote: components.Quote,
       }
     : webDefaultComponents;
 

package/src/primitives/message/MessagePartsGrouped.tsx

@@ -8,10 +8,8 @@ import {
   useMemo,
 } from "react";
 import { useAuiState, useAui } from "@assistant-ui/store";
-import {
-  PartByIndexProvider,
-  TextMessagePartProvider,
-} from "../../context/providers";
+import { PartByIndexProvider } from "../../context/providers/PartByIndexProvider";
+import { TextMessagePartProvider } from "../../context/providers/TextMessagePartProvider";
 import { MessagePartPrimitiveText } from "../messagePart/MessagePartText";
 import { MessagePartPrimitiveImage } from "../messagePart/MessagePartImage";
 import type {
@@ -26,7 +24,7 @@ import type {
   ToolCallMessagePartProps,
   FileMessagePartComponent,
   ReasoningMessagePartComponent,
-} from "../../types";
+} from "@assistant-ui/core/react";
 import { MessagePartPrimitiveInProgress } from "../messagePart/MessagePartInProgress";
 import type { MessagePartStatus } from "@assistant-ui/core";
 

package/src/primitives/message.ts

@@ -0,0 +1,13 @@
+export { MessagePrimitiveRoot as Root } from "./message/MessageRoot";
+export { MessagePrimitiveParts as Parts } from "./message/MessageParts";
+export { MessagePrimitivePartByIndex as PartByIndex } from "./message/MessageParts";
+export { MessagePrimitiveParts as Content } from "./message/MessageParts";
+export { MessagePrimitiveIf as If } from "./message/MessageIf";
+export { MessagePrimitiveAttachments as Attachments } from "./message/MessageAttachments";
+export { MessagePrimitiveAttachmentByIndex as AttachmentByIndex } from "./message/MessageAttachments";
+export { MessagePrimitiveQuote as Quote } from "@assistant-ui/core/react";
+export { MessagePrimitiveError as Error } from "./message/MessageError";
+export {
+  MessagePrimitiveUnstable_PartsGrouped as Unstable_PartsGrouped,
+  MessagePrimitiveUnstable_PartsGroupedByParentId as Unstable_PartsGroupedByParentId,
+} from "./message/MessagePartsGrouped";

package/src/primitives/messagePart.ts

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

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

@@ -1,8 +1,8 @@
 export {
   SelectionToolbarPrimitiveRoot as Root,
   type SelectionToolbarPrimitiveRoot,
-} from "./SelectionToolbarRoot";
+} from "./selectionToolbar/SelectionToolbarRoot";
 export {
   SelectionToolbarPrimitiveQuote as Quote,
   type SelectionToolbarPrimitiveQuote,
-} from "./SelectionToolbarQuote";
+} from "./selectionToolbar/SelectionToolbarQuote";

package/src/primitives/suggestion.ts

@@ -0,0 +1,3 @@
+export { SuggestionPrimitiveTitle as Title } from "./suggestion/SuggestionTitle";
+export { SuggestionPrimitiveDescription as Description } from "./suggestion/SuggestionDescription";
+export { SuggestionPrimitiveTrigger as Trigger } from "./suggestion/SuggestionTrigger";

package/src/primitives/thread/ThreadRoot.tsx

@@ -22,7 +22,9 @@ export namespace ThreadPrimitiveRoot {
  * ```tsx
  * <ThreadPrimitive.Root>
  *   <ThreadPrimitive.Viewport>
- *     <ThreadPrimitive.Messages components={{ Message: MyMessage }} />
+ *     <ThreadPrimitive.Messages>
+ *       {() => <MyMessage />}
+ *     </ThreadPrimitive.Messages>
  *   </ThreadPrimitive.Viewport>
  * </ThreadPrimitive.Root>
  * ```

package/src/primitives/thread/ThreadSuggestion.ts

@@ -5,13 +5,12 @@ import {
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
-import { useCallback } from "react";
-import { useAuiState, useAui } from "@assistant-ui/store";
+import { useSuggestionTrigger as useSuggestionTriggerBehavior } from "@assistant-ui/core/react";
 
 const useThreadSuggestion = ({
   prompt,
   send,
-  clearComposer = true,
+  clearComposer,
   autoSend,
   method: _method,
 }: {
@@ -39,38 +38,17 @@ const useThreadSuggestion = ({
   /** @deprecated Use `clearComposer` instead. */
   method?: "replace";
 }) => {
-  const aui = useAui();
-  const disabled = useAuiState((s) => s.thread.isDisabled);
-
   // ========== Deprecation Mapping ==========
   const resolvedSend = send ?? autoSend ?? false;
   // ==========================================
 
-  const callback = 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 { disabled, trigger } = useSuggestionTriggerBehavior({
+    prompt,
+    send: resolvedSend,
+    clearComposer,
+  });
   if (disabled) return null;
-  return callback;
+  return trigger;
 };
 
 export namespace ThreadPrimitiveSuggestion {

package/src/primitives/thread/ThreadViewport.tsx

@@ -105,7 +105,9 @@ ThreadPrimitiveViewportScrollable.displayName =
  * @example
  * ```tsx
  * <ThreadPrimitive.Viewport turnAnchor="top">
- *   <ThreadPrimitive.Messages components={{ Message: MyMessage }} />
+ *   <ThreadPrimitive.Messages>
+ *     {() => <MyMessage />}
+ *   </ThreadPrimitive.Messages>
  * </ThreadPrimitive.Viewport>
  * ```
  */

package/src/primitives/thread/ThreadViewportFooter.tsx

@@ -30,7 +30,9 @@ export namespace ThreadPrimitiveViewportFooter {
  * @example
  * ```tsx
  * <ThreadPrimitive.Viewport>
- *   <ThreadPrimitive.Messages components={{ ... }} />
+ *   <ThreadPrimitive.Messages>
+ *     {() => <MyMessage />}
+ *   </ThreadPrimitive.Messages>
  *   <ThreadPrimitive.ViewportFooter className="sticky bottom-0">
  *     <Composer />
  *   </ThreadPrimitive.ViewportFooter>

package/src/primitives/thread.ts

@@ -0,0 +1,15 @@
+export { ThreadPrimitiveRoot as Root } from "./thread/ThreadRoot";
+export { ThreadPrimitiveEmpty as Empty } from "./thread/ThreadEmpty";
+export { ThreadPrimitiveIf as If } from "./thread/ThreadIf";
+export { ThreadPrimitiveViewport as Viewport } from "./thread/ThreadViewport";
+export { ThreadPrimitiveViewportProvider as ViewportProvider } from "../context/providers/ThreadViewportProvider";
+export { ThreadPrimitiveViewportFooter as ViewportFooter } from "./thread/ThreadViewportFooter";
+export { ThreadPrimitiveViewportSlack as ViewportSlack } from "./thread/ThreadViewportSlack";
+export { ThreadPrimitiveMessages as Messages } from "./thread/ThreadMessages";
+export { ThreadPrimitiveMessageByIndex as MessageByIndex } from "./thread/ThreadMessages";
+export { ThreadPrimitiveScrollToBottom as ScrollToBottom } from "./thread/ThreadScrollToBottom";
+export { ThreadPrimitiveSuggestion as Suggestion } from "./thread/ThreadSuggestion";
+export {
+  ThreadPrimitiveSuggestions as Suggestions,
+  ThreadPrimitiveSuggestionByIndex as SuggestionByIndex,
+} from "./thread/ThreadSuggestions";

package/src/primitives/threadList/ThreadListNew.tsx

@@ -7,7 +7,8 @@ import {
 import { forwardRef } from "react";
 import { Primitive } from "@radix-ui/react-primitive";
 import { composeEventHandlers } from "@radix-ui/primitive";
-import { useAuiState, useAui } from "@assistant-ui/store";
+import { useAuiState } from "@assistant-ui/store";
+import { useThreadListNew as useThreadListNewBehavior } from "@assistant-ui/core/react";
 
 export namespace ThreadListPrimitiveNew {
   export type Element = ActionButtonElement;
@@ -22,7 +23,7 @@ export const ThreadListPrimitiveNew = forwardRef<
     (s) => s.threads.newThreadId === s.threads.mainThreadId,
   );
 
-  const aui = useAui();
+  const { switchToNewThread } = useThreadListNewBehavior();
 
   return (
     <Primitive.button
@@ -31,9 +32,7 @@ export const ThreadListPrimitiveNew = forwardRef<
       {...props}
       ref={forwardedRef}
       disabled={disabled}
-      onClick={composeEventHandlers(onClick, () => {
-        aui.threads().switchToNewThread();
-      })}
+      onClick={composeEventHandlers(onClick, switchToNewThread)}
     />
   );
 });

package/src/primitives/threadList.ts

@@ -0,0 +1,4 @@
+export { ThreadListPrimitiveNew as New } from "./threadList/ThreadListNew";
+export { ThreadListPrimitiveItems as Items } from "./threadList/ThreadListItems";
+export { ThreadListPrimitiveItemByIndex as ItemByIndex } from "./threadList/ThreadListItems";
+export { ThreadListPrimitiveRoot as Root } from "./threadList/ThreadListRoot";

package/src/primitives/threadListItem/ThreadListItemArchive.ts

@@ -5,14 +5,11 @@ import {
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
-import { useAui } from "@assistant-ui/store";
-import { useCallback } from "react";
+import { useThreadListItemArchive as useThreadListItemArchiveBehavior } from "@assistant-ui/core/react";
 
 const useThreadListItemArchive = () => {
-  const aui = useAui();
-  return useCallback(() => {
-    aui.threadListItem().archive();
-  }, [aui]);
+  const { archive } = useThreadListItemArchiveBehavior();
+  return archive;
 };
 
 export namespace ThreadListItemPrimitiveArchive {

package/src/primitives/threadListItem/ThreadListItemDelete.ts

@@ -5,14 +5,11 @@ import {
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
-import { useAui } from "@assistant-ui/store";
-import { useCallback } from "react";
+import { useThreadListItemDelete as useThreadListItemDeleteBehavior } from "@assistant-ui/core/react";
 
 const useThreadListItemDelete = () => {
-  const aui = useAui();
-  return useCallback(() => {
-    aui.threadListItem().delete();
-  }, [aui]);
+  const { delete: deleteThread } = useThreadListItemDeleteBehavior();
+  return deleteThread;
 };
 
 export namespace ThreadListItemPrimitiveDelete {

package/src/primitives/threadListItem/ThreadListItemTrigger.ts

@@ -5,14 +5,11 @@ import {
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
-import { useAui } from "@assistant-ui/store";
-import { useCallback } from "react";
+import { useThreadListItemTrigger as useThreadListItemTriggerBehavior } from "@assistant-ui/core/react";
 
 const useThreadListItemTrigger = () => {
-  const aui = useAui();
-  return useCallback(() => {
-    aui.threadListItem().switchTo();
-  }, [aui]);
+  const { switchTo } = useThreadListItemTriggerBehavior();
+  return switchTo;
 };
 
 export namespace ThreadListItemPrimitiveTrigger {

package/src/primitives/threadListItem/ThreadListItemUnarchive.ts

@@ -5,14 +5,11 @@ import {
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
-import { useAui } from "@assistant-ui/store";
-import { useCallback } from "react";
+import { useThreadListItemUnarchive as useThreadListItemUnarchiveBehavior } from "@assistant-ui/core/react";
 
 const useThreadListItemUnarchive = () => {
-  const aui = useAui();
-  return useCallback(() => {
-    aui.threadListItem().unarchive();
-  }, [aui]);
+  const { unarchive } = useThreadListItemUnarchiveBehavior();
+  return unarchive;
 };
 
 export namespace ThreadListItemPrimitiveUnarchive {

package/src/primitives/threadListItem.ts

@@ -0,0 +1,6 @@
+export { ThreadListItemPrimitiveRoot as Root } from "./threadListItem/ThreadListItemRoot";
+export { ThreadListItemPrimitiveArchive as Archive } from "./threadListItem/ThreadListItemArchive";
+export { ThreadListItemPrimitiveUnarchive as Unarchive } from "./threadListItem/ThreadListItemUnarchive";
+export { ThreadListItemPrimitiveDelete as Delete } from "./threadListItem/ThreadListItemDelete";
+export { ThreadListItemPrimitiveTrigger as Trigger } from "./threadListItem/ThreadListItemTrigger";
+export { ThreadListItemPrimitiveTitle as Title } from "./threadListItem/ThreadListItemTitle";

package/src/primitives/threadListItemMore.ts

@@ -0,0 +1,5 @@
+export { ThreadListItemMorePrimitiveRoot as Root } from "./threadListItemMore/ThreadListItemMoreRoot";
+export { ThreadListItemMorePrimitiveTrigger as Trigger } from "./threadListItemMore/ThreadListItemMoreTrigger";
+export { ThreadListItemMorePrimitiveContent as Content } from "./threadListItemMore/ThreadListItemMoreContent";
+export { ThreadListItemMorePrimitiveItem as Item } from "./threadListItemMore/ThreadListItemMoreItem";
+export { ThreadListItemMorePrimitiveSeparator as Separator } from "./threadListItemMore/ThreadListItemMoreSeparator";

package/dist/client/index.d.ts

@@ -1,7 +0,0 @@
-export { Tools, DataRenderers } from "@assistant-ui/core/react";
-export { ModelContext as ModelContextClient } from "@assistant-ui/core/store";
-export { Suggestions, type SuggestionConfig } from "@assistant-ui/core/store";
-export { ChainOfThoughtClient } from "@assistant-ui/core/store";
-export { ExternalThread, type ExternalThreadProps, type ExternalThreadMessage, } from "./ExternalThread.js";
-export { InMemoryThreadList, type InMemoryThreadListProps, } from "./InMemoryThreadList.js";
-//# sourceMappingURL=index.d.ts.map

package/dist/client/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAGhE,OAAO,EAAE,YAAY,IAAI,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9E,OAAO,EAAE,WAAW,EAAE,KAAK,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC9E,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAGhE,OAAO,EACL,cAAc,EACd,KAAK,mBAAmB,EACxB,KAAK,qBAAqB,GAC3B,4BAAyB;AAC1B,OAAO,EACL,kBAAkB,EAClB,KAAK,uBAAuB,GAC7B,gCAA6B"}