@assistant-ui/react 0.12.16 → 0.12.19

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/legacy-runtime/runtime-cores/local/LocalRuntimeOptions.ts

@@ -1,36 +1,3 @@
 export type { LocalRuntimeOptionsBase } from "@assistant-ui/core";
-
-import type { LocalRuntimeOptionsBase } from "@assistant-ui/core";
-import type { ThreadMessageLike } from "@assistant-ui/core";
-import type { AssistantCloud } from "assistant-cloud";
-
-// TODO align LocalRuntimeOptions with LocalRuntimeOptionsBase
-export type LocalRuntimeOptions = Omit<LocalRuntimeOptionsBase, "adapters"> & {
-  cloud?: AssistantCloud | undefined;
-  initialMessages?: readonly ThreadMessageLike[] | undefined;
-  adapters?: Omit<LocalRuntimeOptionsBase["adapters"], "chatModel"> | undefined;
-};
-
-export const splitLocalRuntimeOptions = <T extends LocalRuntimeOptions>(
-  options: T,
-) => {
-  const {
-    cloud,
-    initialMessages,
-    maxSteps,
-    adapters,
-    unstable_humanToolNames,
-    ...rest
-  } = options;
-
-  return {
-    localRuntimeOptions: {
-      cloud,
-      initialMessages,
-      maxSteps,
-      adapters,
-      unstable_humanToolNames,
-    },
-    otherOptions: rest,
-  };
-};
+export type { LocalRuntimeOptions } from "@assistant-ui/core/react";
+export { splitLocalRuntimeOptions } from "@assistant-ui/core/react";

package/src/legacy-runtime/runtime-cores/local/useLocalRuntime.ts

@@ -1,69 +1,3 @@
 "use client";
 
-import { useEffect, useMemo, useRef, useState } from "react";
-import type { ChatModelAdapter } from "./ChatModelAdapter";
-import { LocalRuntimeCore } from "./LocalRuntimeCore";
-import type { LocalRuntimeOptions } from "./LocalRuntimeOptions";
-import { useRuntimeAdapters } from "../adapters/RuntimeAdapterProvider";
-import { useRemoteThreadListRuntime } from "../remote-thread-list/useRemoteThreadListRuntime";
-import { useCloudThreadListAdapter } from "../remote-thread-list/adapter/cloud";
-import { AssistantRuntimeImpl } from "../../../internal";
-import { useAuiState } from "@assistant-ui/store";
-
-const useLocalThreadRuntime = (
-  adapter: ChatModelAdapter,
-  { initialMessages, ...options }: LocalRuntimeOptions,
-) => {
-  const { modelContext, ...threadListAdapters } = useRuntimeAdapters() ?? {};
-  const opt = {
-    ...options,
-    adapters: {
-      ...threadListAdapters,
-      ...options.adapters,
-      chatModel: adapter,
-    },
-  };
-
-  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 = (
-  adapter: ChatModelAdapter,
-  { cloud, ...options }: LocalRuntimeOptions = {},
-) => {
-  const cloudAdapter = useCloudThreadListAdapter({ cloud });
-  return useRemoteThreadListRuntime({
-    runtimeHook: function RuntimeHook() {
-      return useLocalThreadRuntime(adapter, options);
-    },
-    adapter: cloudAdapter,
-    allowNesting: true,
-  });
-};
+export { useLocalRuntime } from "@assistant-ui/core/react";

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
 ```