@assistant-ui/react 0.12.17 → 0.12.19

package/dist/primitives/threadListItemMore.d.ts

@@ -0,0 +1,6 @@
+export { ThreadListItemMorePrimitiveRoot as Root } from "./threadListItemMore/ThreadListItemMoreRoot.js";
+export { ThreadListItemMorePrimitiveTrigger as Trigger } from "./threadListItemMore/ThreadListItemMoreTrigger.js";
+export { ThreadListItemMorePrimitiveContent as Content } from "./threadListItemMore/ThreadListItemMoreContent.js";
+export { ThreadListItemMorePrimitiveItem as Item } from "./threadListItemMore/ThreadListItemMoreItem.js";
+export { ThreadListItemMorePrimitiveSeparator as Separator } from "./threadListItemMore/ThreadListItemMoreSeparator.js";
+//# sourceMappingURL=threadListItemMore.d.ts.map

package/dist/primitives/threadListItemMore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"threadListItemMore.d.ts","sourceRoot":"","sources":["../../src/primitives/threadListItemMore.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,IAAI,IAAI,EAAE,uDAAoD;AACtG,OAAO,EAAE,kCAAkC,IAAI,OAAO,EAAE,0DAAuD;AAC/G,OAAO,EAAE,kCAAkC,IAAI,OAAO,EAAE,0DAAuD;AAC/G,OAAO,EAAE,+BAA+B,IAAI,IAAI,EAAE,uDAAoD;AACtG,OAAO,EAAE,oCAAoC,IAAI,SAAS,EAAE,4DAAyD"}

package/dist/primitives/threadListItemMore.js

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

package/dist/primitives/threadListItemMore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"threadListItemMore.js","sourceRoot":"","sources":["../../src/primitives/threadListItemMore.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,+BAA+B,IAAI,IAAI,EAAE,uDAAoD;AACtG,OAAO,EAAE,kCAAkC,IAAI,OAAO,EAAE,0DAAuD;AAC/G,OAAO,EAAE,kCAAkC,IAAI,OAAO,EAAE,0DAAuD;AAC/G,OAAO,EAAE,+BAA+B,IAAI,IAAI,EAAE,uDAAoD;AACtG,OAAO,EAAE,oCAAoC,IAAI,SAAS,EAAE,4DAAyD"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react",
-  "version": "0.12.17",
+  "version": "0.12.19",
   "description": "TypeScript/React library for AI Chat",
   "keywords": [
     "radix-ui",
@@ -49,17 +49,17 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.1.5",
-    "@assistant-ui/store": "^0.2.2",
-    "@assistant-ui/tap": "^0.5.2",
+    "@assistant-ui/core": "^0.1.7",
+    "@assistant-ui/store": "^0.2.3",
+    "@assistant-ui/tap": "^0.5.3",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.3",
     "@radix-ui/react-primitive": "^2.1.4",
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@radix-ui/react-use-escape-keydown": "^1.1.1",
-    "assistant-cloud": "^0.1.21",
-    "assistant-stream": "^0.3.4",
+    "assistant-cloud": "^0.1.22",
+    "assistant-stream": "^0.3.6",
     "nanoid": "^5.1.6",
     "radix-ui": "^1.4.3",
     "react-textarea-autosize": "^8.5.9",
@@ -83,14 +83,14 @@
   "devDependencies": {
     "@testing-library/react": "^16.3.2",
     "@types/json-schema": "^7.0.15",
-    "@types/node": "^25.3.3",
+    "@types/node": "^25.5.0",
     "@types/react": "^19.2.14",
     "@types/react-dom": "^19.2.3",
     "jsdom": "^28.1.0",
     "react": "^19.2.4",
     "react-dom": "^19.2.4",
-    "vitest": "^4.0.18",
-    "@assistant-ui/x-buildutils": "0.0.2"
+    "vitest": "^4.1.0",
+    "@assistant-ui/x-buildutils": "0.0.3"
   },
   "publishConfig": {
     "access": "public",

package/src/client/ExternalThread.ts

@@ -468,6 +468,7 @@ export const ExternalThread = resource(
       startRun: () => {
         onStartRun?.();
       },
+      resumeRun: () => {},
       unstable_resumeRun: () => {},
       cancelRun: handleCancelRun,
       getModelContext: () => ({ tools: {}, config: {} }),

package/src/context/index.ts

@@ -1,3 +1,67 @@
-export * from "./providers";
-export * from "./stores";
-export * from "./react";
+export { AssistantRuntimeProvider } from "../legacy-runtime/AssistantRuntimeProvider";
+export {
+  ThreadListItemByIndexProvider,
+  ThreadListItemRuntimeProvider,
+} from "./providers/ThreadListItemProvider";
+export { MessageByIndexProvider } from "./providers/MessageByIndexProvider";
+export { SuggestionByIndexProvider } from "./providers/SuggestionByIndexProvider";
+export { PartByIndexProvider } from "./providers/PartByIndexProvider";
+export {
+  MessageAttachmentByIndexProvider,
+  ComposerAttachmentByIndexProvider,
+} from "./providers/AttachmentByIndexProvider";
+export { TextMessagePartProvider } from "./providers/TextMessagePartProvider";
+export { MessageProvider } from "./providers/MessageProvider";
+export { ChainOfThoughtByIndicesProvider } from "./providers/ChainOfThoughtByIndicesProvider";
+export { ReadonlyThreadProvider } from "@assistant-ui/core/react";
+
+export type { ThreadViewportState } from "./stores/ThreadViewport";
+
+export {
+  useThreadViewport,
+  useThreadViewportStore,
+} from "./react/ThreadViewportContext";
+
+export {
+  useAssistantRuntime,
+  useThreadList,
+} from "../legacy-runtime/hooks/AssistantContext";
+
+export {
+  useAttachmentRuntime,
+  useAttachment,
+  useThreadComposerAttachmentRuntime,
+  useThreadComposerAttachment,
+  useEditComposerAttachmentRuntime,
+  useEditComposerAttachment,
+  useMessageAttachment,
+  useMessageAttachmentRuntime,
+} from "../legacy-runtime/hooks/AttachmentContext";
+
+export {
+  useComposerRuntime,
+  useComposer,
+} from "../legacy-runtime/hooks/ComposerContext";
+
+export {
+  useMessageRuntime,
+  useEditComposer,
+  useMessage,
+} from "../legacy-runtime/hooks/MessageContext";
+
+export {
+  useMessagePartRuntime,
+  useMessagePart,
+} from "../legacy-runtime/hooks/MessagePartContext";
+
+export {
+  useThreadRuntime,
+  useThread,
+  useThreadComposer,
+  useThreadModelContext,
+} from "../legacy-runtime/hooks/ThreadContext";
+
+export {
+  useThreadListItemRuntime,
+  useThreadListItem,
+} from "../legacy-runtime/hooks/ThreadListItemContext";

package/src/context/react/ThreadViewportContext.ts

@@ -5,7 +5,7 @@ import { ReadonlyStore } from "../ReadonlyStore";
 import { UseBoundStore } from "zustand";
 import { createContextHook } from "./utils/createContextHook";
 import { createContextStoreHook } from "./utils/createContextStoreHook";
-import { ThreadViewportState } from "../stores";
+import { ThreadViewportState } from "../stores/ThreadViewport";
 
 export type ThreadViewportContextValue = {
   useThreadViewport: UseBoundStore<ReadonlyStore<ThreadViewportState>>;

package/src/hooks/useMessageQuote.ts

@@ -1,7 +1,8 @@
 "use client";
 
-import { useAuiState } from "@assistant-ui/store";
 import type { QuoteInfo } from "@assistant-ui/core";
+import { getMessageQuote } from "@assistant-ui/core/react";
+import { useAuiState } from "@assistant-ui/store";
 
 /**
  * Hook that returns the quote info for the current message, if any.
@@ -18,10 +19,5 @@ import type { QuoteInfo } from "@assistant-ui/core";
  * ```
  */
 export const useMessageQuote = (): QuoteInfo | undefined => {
-  return useAuiState(
-    (s) =>
-      (s.message.metadata?.custom as Record<string, unknown>)?.quote as
-        | QuoteInfo
-        | undefined,
-  );
+  return useAuiState(getMessageQuote);
 };

package/src/index.ts

@@ -16,17 +16,347 @@ export {
   type AssistantEventCallback,
 } from "@assistant-ui/store";
 
-export * from "./legacy-runtime/runtime";
-export * from "./legacy-runtime/cloud";
-export * from "./legacy-runtime/runtime-cores";
+// ============================================================================
+// legacy-runtime/runtime
+// ============================================================================
+// Re-export public runtime types from @assistant-ui/core
+export type {
+  AssistantRuntime,
+  ThreadRuntime,
+  ThreadState,
+  CreateAppendMessage,
+  CreateStartRunConfig,
+  CreateResumeRunConfig,
+  MessageRuntime,
+  MessageState,
+  MessagePartRuntime,
+  MessagePartState,
+  ComposerRuntime,
+  ThreadComposerRuntime,
+  EditComposerRuntime,
+  EditComposerState,
+  ThreadComposerState,
+  ComposerState,
+  AttachmentRuntime,
+  AttachmentState,
+  ThreadListRuntime,
+  ThreadListState,
+  ThreadListItemRuntime,
+  ThreadListItemState,
+} from "@assistant-ui/core";
+
+// ============================================================================
+// legacy-runtime/cloud
+// ============================================================================
+export { useCloudThreadListRuntime } from "./legacy-runtime/cloud/useCloudThreadListRuntime";
+export { AssistantCloud } from "assistant-cloud";
+
+// ============================================================================
+// legacy-runtime/runtime-cores
+// ============================================================================
+
+// --- adapters/attachment ---
+export type { AttachmentAdapter } from "@assistant-ui/core";
+export {
+  SimpleImageAttachmentAdapter,
+  SimpleTextAttachmentAdapter,
+  CompositeAttachmentAdapter,
+} from "@assistant-ui/core";
+export { CloudFileAttachmentAdapter } from "./legacy-runtime/runtime-cores/adapters/attachment/CloudFileAttachmentAdapter";
+
+// --- adapters/feedback ---
+export type { FeedbackAdapter } from "@assistant-ui/core";
+
+// --- adapters/speech ---
+export type {
+  SpeechSynthesisAdapter,
+  DictationAdapter,
+} from "@assistant-ui/core";
+export {
+  WebSpeechSynthesisAdapter,
+  WebSpeechDictationAdapter,
+} from "@assistant-ui/core";
+
+// --- adapters/suggestion ---
+export type { SuggestionAdapter } from "@assistant-ui/core";
+
+// --- adapters/RuntimeAdapterProvider ---
+export {
+  RuntimeAdapterProvider,
+  useRuntimeAdapters,
+  type RuntimeAdapters,
+} from "@assistant-ui/core/react";
 
+// --- adapters/thread-history ---
+export type {
+  ThreadHistoryAdapter,
+  GenericThreadHistoryAdapter,
+  MessageFormatAdapter,
+  MessageFormatItem,
+  MessageFormatRepository,
+  MessageStorageEntry,
+} from "@assistant-ui/core";
+
+// --- assistant-transport ---
+export {
+  useAssistantTransportRuntime,
+  useAssistantTransportSendCommand,
+  useAssistantTransportState,
+} from "./legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime";
+export type {
+  AssistantTransportConnectionMetadata,
+  AssistantTransportCommand,
+  AssistantTransportProtocol,
+  SendCommandsRequestBody,
+} from "./legacy-runtime/runtime-cores/assistant-transport/types";
+
+// --- core ---
+export type {
+  AddToolResultOptions,
+  SubmitFeedbackOptions,
+  ThreadSuggestion,
+  DictationState,
+} from "@assistant-ui/core";
+
+// --- external-store ---
+export type { ThreadMessageLike } from "@assistant-ui/core";
+export {
+  getExternalStoreMessage,
+  getExternalStoreMessages,
+  bindExternalStoreMessage,
+} from "@assistant-ui/core";
+export type {
+  ExternalStoreAdapter,
+  ExternalStoreMessageConverter,
+  ExternalStoreThreadListAdapter,
+  ExternalStoreThreadData,
+} from "@assistant-ui/core";
+export { useExternalStoreRuntime } from "./legacy-runtime/runtime-cores/external-store/useExternalStoreRuntime";
+export {
+  useExternalMessageConverter,
+  convertExternalMessages as unstable_convertExternalMessages,
+} from "./legacy-runtime/runtime-cores/external-store/external-message-converter";
+export { createMessageConverter as unstable_createMessageConverter } from "./legacy-runtime/runtime-cores/external-store/createMessageConverter";
+
+// --- local ---
+export type {
+  ChatModelAdapter,
+  ChatModelRunOptions,
+  ChatModelRunResult,
+  ChatModelRunUpdate,
+  LocalRuntimeOptionsBase,
+} from "@assistant-ui/core";
+export { useLocalRuntime } from "./legacy-runtime/runtime-cores/local/useLocalRuntime";
+/**
+ * @deprecated Use `useLocalRuntime` instead.
+ */
+export { useLocalRuntime as useLocalThreadRuntime } from "./legacy-runtime/runtime-cores/local/useLocalRuntime";
+export type { LocalRuntimeOptions } from "./legacy-runtime/runtime-cores/local/LocalRuntimeOptions";
+
+// --- remote-thread-list ---
+export { useRemoteThreadListRuntime } from "./legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime";
+/** @deprecated Use `useRemoteThreadListRuntime` instead. */
+export { useRemoteThreadListRuntime as unstable_useRemoteThreadListRuntime } from "./legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime";
+export { useCloudThreadListAdapter } from "./legacy-runtime/runtime-cores/remote-thread-list/adapter/cloud";
+/** @deprecated Use `useCloudThreadListAdapter` instead. */
+export { useCloudThreadListAdapter as unstable_useCloudThreadListAdapter } from "./legacy-runtime/runtime-cores/remote-thread-list/adapter/cloud";
+export type { RemoteThreadListAdapter } from "@assistant-ui/core";
+/** @deprecated Use `RemoteThreadListAdapter` instead. */
+export type { RemoteThreadListAdapter as unstable_RemoteThreadListAdapter } from "@assistant-ui/core";
+export { InMemoryThreadListAdapter } from "@assistant-ui/core";
+/** @deprecated Use `InMemoryThreadListAdapter` instead. */
+export { InMemoryThreadListAdapter as unstable_InMemoryThreadListAdapter } from "@assistant-ui/core";
+
+// Re-export from @assistant-ui/core (runtime-cores root)
+export type { ExportedMessageRepositoryItem } from "@assistant-ui/core";
+export { ExportedMessageRepository } from "@assistant-ui/core";
+
+// ============================================================================
+// context
+// ============================================================================
 export * from "./context";
-export * from "./model-context";
-export * from "./primitives";
-export * from "./types";
-export * from "./devtools";
-export * from "./client";
 
+// ============================================================================
+// model-context
+// ============================================================================
+// Re-export shared from core/react
+export {
+  makeAssistantTool,
+  type AssistantTool,
+  makeAssistantToolUI,
+  type AssistantToolUI,
+  makeAssistantDataUI,
+  type AssistantDataUI,
+  useAssistantTool,
+  type AssistantToolProps,
+  useAssistantToolUI,
+  type AssistantToolUIProps,
+  useAssistantDataUI,
+  type AssistantDataUIProps,
+  useAssistantInstructions,
+  useInlineRender,
+  type Toolkit,
+  type ToolDefinition,
+  Tools,
+  DataRenderers,
+} from "@assistant-ui/core/react";
+
+// Core pass-through (unchanged)
+export type {
+  ModelContext,
+  ModelContextProvider,
+  LanguageModelConfig,
+  LanguageModelV1CallSettings,
+} from "@assistant-ui/core";
+
+export { mergeModelContexts } from "@assistant-ui/core";
+
+export type { Tool } from "assistant-stream";
+
+export { tool } from "@assistant-ui/core";
+
+export { Suggestions, type SuggestionConfig } from "@assistant-ui/core/store";
+
+// React-only (stays)
+export { makeAssistantVisible } from "./model-context/makeAssistantVisible";
+
+// --- model-context/registry ---
+export { ModelContextRegistry } from "@assistant-ui/core";
+export type {
+  ModelContextRegistryToolHandle,
+  ModelContextRegistryInstructionHandle,
+  ModelContextRegistryProviderHandle,
+} from "@assistant-ui/core";
+
+// --- model-context/frame ---
+export { AssistantFrameHost } from "@assistant-ui/core";
+export { AssistantFrameProvider } from "@assistant-ui/core";
+export type {
+  SerializedTool,
+  SerializedModelContext,
+  FrameMessageType,
+  FrameMessage,
+} from "@assistant-ui/core";
+export { FRAME_MESSAGE_CHANNEL } from "@assistant-ui/core";
+export { useAssistantFrameHost } from "./model-context/frame/useAssistantFrameHost";
+
+// ============================================================================
+// primitives
+// ============================================================================
+export * as ActionBarPrimitive from "./primitives/actionBar";
+export * as ActionBarMorePrimitive from "./primitives/actionBarMore";
+export * as AssistantModalPrimitive from "./primitives/assistantModal";
+export * as AttachmentPrimitive from "./primitives/attachment";
+export * as BranchPickerPrimitive from "./primitives/branchPicker";
+export * as ChainOfThoughtPrimitive from "./primitives/chainOfThought";
+export * as ComposerPrimitive from "./primitives/composer";
+export * as MessagePartPrimitive from "./primitives/messagePart";
+export * as ErrorPrimitive from "./primitives/error";
+export * as MessagePrimitive from "./primitives/message";
+export * as ThreadPrimitive from "./primitives/thread";
+export * as SuggestionPrimitive from "./primitives/suggestion";
+export * as ThreadListPrimitive from "./primitives/threadList";
+export * as ThreadListItemPrimitive from "./primitives/threadListItem";
+export * as ThreadListItemMorePrimitive from "./primitives/threadListItemMore";
+export * as SelectionToolbarPrimitive from "./primitives/selectionToolbar";
+
+export { useMessagePartText } from "./primitives/messagePart/useMessagePartText";
+export { useMessagePartReasoning } from "./primitives/messagePart/useMessagePartReasoning";
+export { useMessagePartSource } from "./primitives/messagePart/useMessagePartSource";
+export { useMessagePartFile } from "./primitives/messagePart/useMessagePartFile";
+export { useMessagePartImage } from "./primitives/messagePart/useMessagePartImage";
+export { useMessagePartData } from "./primitives/messagePart/useMessagePartData";
+export { useThreadViewportAutoScroll } from "./primitives/thread/useThreadViewportAutoScroll";
+export { useScrollLock } from "./primitives/reasoning/useScrollLock";
+export { useMessageQuote } from "./hooks/useMessageQuote";
+export { useMessageTiming } from "./hooks/useMessageTiming";
+
+// ============================================================================
+// types
+// ============================================================================
+// Re-export core types from @assistant-ui/core
+export type {
+  Attachment,
+  PendingAttachment,
+  CompleteAttachment,
+  AttachmentStatus,
+  AppendMessage,
+  TextMessagePart,
+  ReasoningMessagePart,
+  SourceMessagePart,
+  ImageMessagePart,
+  FileMessagePart,
+  DataMessagePart,
+  Unstable_AudioMessagePart,
+  ToolCallMessagePart,
+  MessageStatus,
+  MessagePartStatus,
+  ToolCallMessagePartStatus,
+  MessageTiming,
+  ThreadUserMessagePart,
+  ThreadAssistantMessagePart,
+  ThreadSystemMessage,
+  ThreadAssistantMessage,
+  ThreadUserMessage,
+  ThreadMessage,
+  Unsubscribe,
+  QuoteInfo,
+  CreateAttachment,
+} from "@assistant-ui/core";
+
+// React component types (from core/react)
+export type {
+  EmptyMessagePartComponent,
+  EmptyMessagePartProps,
+  TextMessagePartComponent,
+  TextMessagePartProps,
+  ReasoningMessagePartComponent,
+  ReasoningMessagePartProps,
+  SourceMessagePartComponent,
+  SourceMessagePartProps,
+  ImageMessagePartComponent,
+  ImageMessagePartProps,
+  FileMessagePartComponent,
+  FileMessagePartProps,
+  Unstable_AudioMessagePartComponent,
+  Unstable_AudioMessagePartProps,
+  DataMessagePartComponent,
+  DataMessagePartProps,
+  ToolCallMessagePartComponent,
+  ToolCallMessagePartProps,
+  ReasoningGroupProps,
+  ReasoningGroupComponent,
+  QuoteMessagePartComponent,
+  QuoteMessagePartProps,
+} from "@assistant-ui/core/react";
+
+// Thread list item types
+export type { ThreadListItemStatus } from "@assistant-ui/core";
+
+// ============================================================================
+// devtools
+// ============================================================================
+export { DevToolsHooks, DevToolsProviderApi } from "./devtools/DevToolsHooks";
+
+// ============================================================================
+// client
+// ============================================================================
+// Note: Tools, DataRenderers, Suggestions, SuggestionConfig already exported above from model-context
+export { ModelContext as ModelContextClient } from "@assistant-ui/core/store";
+export { ChainOfThoughtClient } from "@assistant-ui/core/store";
+export {
+  ExternalThread,
+  type ExternalThreadProps,
+  type ExternalThreadMessage,
+} from "./client/ExternalThread";
+export {
+  InMemoryThreadList,
+  type InMemoryThreadListProps,
+} from "./client/InMemoryThreadList";
+
+// ============================================================================
+// internal & augmentations
+// ============================================================================
 export * as INTERNAL from "./internal";
 export type { ToolExecutionStatus } from "./internal";
 

package/src/internal.ts

@@ -28,4 +28,8 @@ export {
   type ToolExecutionStatus,
 } from "@assistant-ui/core/react";
 
-export * from "./utils/smooth";
+export { useSmooth } from "./utils/smooth/useSmooth";
+export {
+  useSmoothStatus,
+  withSmoothContextProvider,
+} from "./utils/smooth/SmoothContext";

package/src/legacy-runtime/AssistantRuntimeProvider.tsx

@@ -5,7 +5,7 @@ import { AssistantClient, useAui } from "@assistant-ui/store";
 import { AssistantRuntime } from "./runtime/AssistantRuntime";
 import { AssistantProviderBase } from "@assistant-ui/core/react";
 import { ThreadPrimitiveViewportProvider } from "../context/providers/ThreadViewportProvider";
-import { DevToolsProviderApi } from "../devtools";
+import { DevToolsProviderApi } from "../devtools/DevToolsHooks";
 
 export namespace AssistantRuntimeProvider {
   export type Props = PropsWithChildren<{

package/src/legacy-runtime/cloud/useCloudThreadListRuntime.ts

@@ -1,7 +1,7 @@
 "use client";
 
 import { AssistantCloud } from "assistant-cloud";
-import { AssistantRuntime } from "../runtime";
+import type { AssistantRuntime } from "@assistant-ui/core";
 import { useRemoteThreadListRuntime } from "../runtime-cores/remote-thread-list/useRemoteThreadListRuntime";
 import { useCloudThreadListAdapter } from "../runtime-cores/remote-thread-list/adapter/cloud";
 

package/src/legacy-runtime/hooks/MessageContext.ts

@@ -3,7 +3,7 @@
 import { MessageRuntime } from "../runtime/MessageRuntime";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { createStateHookForRuntime } from "../../context/react/utils/createStateHookForRuntime";
-import { EditComposerRuntime } from "../runtime";
+import type { EditComposerRuntime } from "@assistant-ui/core";
 
 /**
  * @deprecated Use `useAui()` with `aui.message()` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12

package/src/legacy-runtime/hooks/ThreadContext.ts

@@ -2,9 +2,9 @@
 
 import { useState } from "react";
 import { ThreadRuntime } from "../runtime/ThreadRuntime";
-import { ModelContext } from "../../model-context";
+import type { ModelContext } from "@assistant-ui/core";
 import { createStateHookForRuntime } from "../../context/react/utils/createStateHookForRuntime";
-import { ThreadComposerRuntime } from "../runtime";
+import type { ThreadComposerRuntime } from "@assistant-ui/core";
 import { useAui, useAuiEvent, useAuiState } from "@assistant-ui/store";
 
 /**

package/src/legacy-runtime/runtime-cores/assistant-transport/types.ts

@@ -1,6 +1,9 @@
 import type { ThreadMessage } from "@assistant-ui/core";
 import { ReadonlyJSONValue } from "assistant-stream/utils";
-import { AttachmentAdapter, ThreadHistoryAdapter } from "..";
+import type {
+  AttachmentAdapter,
+  ThreadHistoryAdapter,
+} from "@assistant-ui/core";
 import { UserCommands } from "../../../augmentations";
 import type { ToolExecutionStatus } from "./useToolInvocations";
 

package/src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.ts

@@ -8,7 +8,7 @@ import {
 } from "assistant-stream/utils";
 import { useExternalStoreRuntime } from "../external-store/useExternalStoreRuntime";
 import { AssistantRuntime } from "../../runtime/AssistantRuntime";
-import { AddToolResultOptions } from "../core";
+import type { AddToolResultOptions } from "@assistant-ui/core";
 import { useState, useRef, useMemo } from "react";
 import {
   AssistantMessageAccumulator,

package/src/model-context/frame/SPEC_AssistantFrame.md

@@ -45,8 +45,8 @@ registry.addTool({
 const frameHost = new AssistantFrameHost(iframeWindow);
 
 // Register with assistant runtime
-const runtime = useAssistantRuntime();
-runtime.registerModelContextProvider(frameHost);
+const aui = useAui();
+aui.modelContext().register(frameHost);
 
 // The assistant now has access to tools from the iframe
 ```

package/src/primitives/actionBar/ActionBarSpeak.ts

@@ -1,29 +1,16 @@
 "use client";
 
-import { useCallback } from "react";
-import { useAuiState, useAui } from "@assistant-ui/store";
 import {
   ActionButtonElement,
   ActionButtonProps,
   createActionButton,
 } from "../../utils/createActionButton";
+import { useActionBarSpeak as useActionBarSpeakBehavior } from "@assistant-ui/core/react";
 
 const useActionBarSpeak = () => {
-  const aui = useAui();
-  const callback = useCallback(async () => {
-    aui.message().speak();
-  }, [aui]);
-
-  const hasSpeakableContent = useAuiState((s) => {
-    return (
-      (s.message.role !== "assistant" ||
-        s.message.status?.type !== "running") &&
-      s.message.parts.some((c) => c.type === "text" && c.text.length > 0)
-    );
-  });
-
-  if (!hasSpeakableContent) return null;
-  return callback;
+  const { disabled, speak } = useActionBarSpeakBehavior();
+  if (disabled) return null;
+  return speak;
 };
 
 export namespace ActionBarPrimitiveSpeak {