@assistant-ui/core 0.1.14 → 0.1.15

package/src/react/primitives/message/MessageParts.tsx

@@ -3,7 +3,7 @@ import {
   type FC,
   type ReactNode,
   memo,
-  PropsWithChildren,
+  type PropsWithChildren,
   useMemo,
 } from "react";
 import {
@@ -32,6 +32,7 @@ import type {
   QuoteMessagePartComponent,
 } from "../../types/MessagePartComponentTypes";
 import type { MessagePartStatus } from "../../../types/message";
+import type { DataRenderersState } from "../../types/scopes/dataRenderers";
 import { useShallow } from "zustand/shallow";
 
 type MessagePartRange =
@@ -284,17 +285,25 @@ const ToolUIDisplay = ({
   return <Render {...props} />;
 };
 
+const getDataRenderer = (
+  dataRenderers: DataRenderersState,
+  name: string,
+  inlineFallback: DataMessagePartComponent | undefined,
+): DataMessagePartComponent | undefined => {
+  const named = dataRenderers.renderers[name]?.[0];
+  if (named) return named;
+  return dataRenderers.fallbacks[0] ?? inlineFallback;
+};
+
 const DataUIDisplay = ({
   Fallback,
   ...props
 }: {
   Fallback: DataMessagePartComponent | undefined;
 } & DataMessagePartProps) => {
-  const Render = useAuiState((s) => {
-    const Render = s.dataRenderers.renderers[props.name] ?? Fallback;
-    if (Array.isArray(Render)) return Render[0] ?? Fallback;
-    return Render;
-  });
+  const Render = useAuiState((s) =>
+    getDataRenderer(s.dataRenderers, props.name, Fallback),
+  );
   if (!Render) return null;
   return <Render {...props} />;
 };
@@ -522,12 +531,11 @@ const RegisteredToolUI: FC = () => {
  */
 const RegisteredDataRendererUI: FC = () => {
   const part = useAuiState((s) => s.part);
-  const Render = useAuiState((s) => {
-    if (s.part.type !== "data") return null;
-    const entry = s.dataRenderers.renderers[s.part.name];
-    if (Array.isArray(entry)) return entry[0] ?? null;
-    return entry ?? null;
-  });
+  const Render = useAuiState((s) =>
+    s.part.type === "data"
+      ? (getDataRenderer(s.dataRenderers, s.part.name, undefined) ?? null)
+      : null,
+  );
 
   if (!Render || part.type !== "data") return null;
 
@@ -584,6 +592,9 @@ const MessagePrimitivePartsInner: FC<{
 }> = ({ children }) => {
   const aui = useAui();
   const contentLength = useAuiState((s) => s.message.parts.length);
+  // Subscribed (not snapshotted like `tools`) so fallbacks registered after
+  // the first render trigger a re-render and `hasUI` re-evaluates.
+  const dataRenderers = useAuiState((s) => s.dataRenderers);
 
   // biome-ignore lint/correctness/useExhaustiveDependencies: aui accessors are stable refs
   return useMemo(
@@ -609,10 +620,9 @@ const MessagePrimitivePartsInner: FC<{
                     };
                   }
                   if (state.type === "data") {
-                    const entry = aui.dataRenderers().getState().renderers[
-                      state.name
-                    ];
-                    const hasUI = Array.isArray(entry) ? !!entry[0] : !!entry;
+                    const hasUI =
+                      getDataRenderer(dataRenderers, state.name, undefined) !==
+                      undefined;
                     return {
                       ...state,
                       dataRendererUI: hasUI ? (
@@ -629,7 +639,7 @@ const MessagePrimitivePartsInner: FC<{
           </RenderChildrenWithAccessor>
         </PartByIndexProvider>
       )),
-    [contentLength, children],
+    [contentLength, children, dataRenderers],
   );
 };
 

package/src/react/primitives/part/PartMessages.tsx

@@ -3,7 +3,7 @@ import { useAuiState } from "@assistant-ui/store";
 import type { ThreadMessage } from "../../../types/message";
 import { ReadonlyThreadProvider } from "../../providers/ReadonlyThreadProvider";
 import {
-  ThreadPrimitiveMessages,
+  type ThreadPrimitiveMessages,
   ThreadPrimitiveMessagesImpl,
 } from "../thread/ThreadMessages";
 

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

@@ -7,7 +7,7 @@ import {
 } from "react";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import { MessageByIndexProvider } from "../../providers/MessageByIndexProvider";
-import { MessageState } from "../../../store";
+import type { MessageState } from "../../../store";
 
 type MessagesComponentConfig =
   | {
@@ -124,9 +124,10 @@ const getComponent = (
           DEFAULT_SYSTEM_MESSAGE
         );
       }
-    default:
+    default: {
       const _exhaustiveCheck: never = role;
       throw new Error(`Unknown message role: ${_exhaustiveCheck}`);
+    }
   }
 };
 

package/src/react/primitives/threadList/ThreadListItems.tsx

@@ -1,4 +1,10 @@
-import { ComponentType, FC, ReactNode, memo, useMemo } from "react";
+import {
+  type ComponentType,
+  type FC,
+  type ReactNode,
+  memo,
+  useMemo,
+} from "react";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import type { ThreadListItemState } from "../../../store/scopes/thread-list-item";
 import { ThreadListItemByIndexProvider } from "../../providers/ThreadListItemByIndexProvider";

package/src/react/providers/AttachmentByIndexProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
 
 export const MessageAttachmentByIndexProvider: FC<

package/src/react/providers/ChainOfThoughtByIndicesProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, useAuiState, AuiProvider } from "@assistant-ui/store";
 import { ChainOfThoughtClient } from "../../store/clients/chain-of-thought-client";
 import type { ChainOfThoughtPart } from "../../store/scopes/chain-of-thought";

package/src/react/providers/ChainOfThoughtPartByIndexProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
 
 export const ChainOfThoughtPartByIndexProvider: FC<

package/src/react/providers/MessageByIndexProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
 
 export const MessageByIndexProvider: FC<

package/src/react/providers/PartByIndexProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
 
 export const PartByIndexProvider: FC<

package/src/react/providers/QueueItemByIndexProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { AuiProvider, Derived, useAui } from "@assistant-ui/store";
 
 export type QueueItemByIndexProviderProps = PropsWithChildren<{

package/src/react/providers/SuggestionByIndexProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { AuiProvider, Derived, useAui } from "@assistant-ui/store";
 
 export type SuggestionByIndexProviderProps = PropsWithChildren<{

package/src/react/providers/TextMessagePartProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, type ClientOutput } from "@assistant-ui/store";
 import type { PartState } from "../../store/scopes/part";
 import { resource, tapMemo } from "@assistant-ui/tap";

package/src/react/providers/ThreadListItemByIndexProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
 
 export const ThreadListItemByIndexProvider: FC<

package/src/react/providers/ThreadListItemRuntimeProvider.tsx

@@ -1,4 +1,4 @@
-import { type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider } from "@assistant-ui/store";
 import type { ThreadListItemRuntime } from "../..";
 import { ThreadListItemClient } from "../../store/internal";

package/src/react/runtimes/RemoteThreadListHookInstanceManager.tsx

@@ -1,21 +1,21 @@
 import {
-  FC,
+  type FC,
   useCallback,
   useRef,
   useEffect,
   memo,
-  PropsWithChildren,
-  ComponentType,
+  type PropsWithChildren,
+  type ComponentType,
   useMemo,
 } from "react";
-import { UseBoundStore, StoreApi, create } from "zustand";
+import { type UseBoundStore, type StoreApi, create } from "zustand";
 import { useAui } from "@assistant-ui/store";
 import { ThreadListItemRuntimeProvider } from "../providers/ThreadListItemRuntimeProvider";
 import type { ThreadRuntimeCore } from "../../runtime/interfaces/thread-runtime-core";
 import type { ThreadListRuntimeCore } from "../../runtime/interfaces/thread-list-runtime-core";
 import type { AssistantRuntime } from "../../runtime/api/assistant-runtime";
 import { BaseSubscribable } from "../../subscribable/subscribable";
-import { ThreadRuntimeImpl } from "../../runtime/api/thread-runtime";
+import type { ThreadRuntimeImpl } from "../../runtime/api/thread-runtime";
 import { ThreadListRuntimeImpl } from "../../runtime/api/thread-list-runtime";
 
 type RemoteThreadListHook = () => AssistantRuntime;
@@ -116,9 +116,8 @@ export class RemoteThreadListHookInstanceManager extends BaseSubscribable {
 
     useEffect(() => {
       const runtimeCore = threadBinding.getState();
-      const setGetInitializePromise = (runtimeCore as Record<string, unknown>)[
-        "__internal_setGetInitializePromise"
-      ];
+      const setGetInitializePromise = (runtimeCore as Record<string, unknown>)
+        .__internal_setGetInitializePromise;
       if (typeof setGetInitializePromise === "function") {
         setGetInitializePromise.call(runtimeCore, () => initPromiseRef.current);
       }

package/src/react/runtimes/RemoteThreadListThreadListRuntimeCore.tsx

@@ -16,10 +16,10 @@ import {
 import type { RemoteThreadListOptions } from "../../runtimes/remote-thread-list/types";
 import { RemoteThreadListHookInstanceManager } from "./RemoteThreadListHookInstanceManager";
 import {
-  ComponentType,
-  FC,
+  type ComponentType,
+  type FC,
   Fragment,
-  PropsWithChildren,
+  type PropsWithChildren,
   useEffect,
   useId,
 } from "react";
@@ -36,6 +36,7 @@ export class RemoteThreadListThreadListRuntimeCore
   private readonly _hookManager: RemoteThreadListHookInstanceManager;
 
   private _loadThreadsPromise: Promise<void> | undefined;
+  private _loadGeneration = 0;
 
   private _mainThreadId!: string;
   private readonly _state = new OptimisticState<RemoteThreadState>({
@@ -54,6 +55,7 @@ export class RemoteThreadListThreadListRuntimeCore
   public getLoadThreadsPromise() {
     // TODO this needs to be cached in case this promise is loaded during suspense
     if (!this._loadThreadsPromise) {
+      const generation = this._loadGeneration;
       this._loadThreadsPromise = this._state
         .optimisticUpdate({
           execute: () => this._options.adapter.list(),
@@ -63,7 +65,9 @@ export class RemoteThreadListThreadListRuntimeCore
               isLoading: true,
             };
           },
+          // biome-ignore lint/suspicious/noThenProperty: OptimisticState reducer pattern
           then: (state, l) => {
+            if (generation !== this._loadGeneration) return state;
             const newThreadIds = [];
             const newArchivedThreadIds = [];
             const newThreadIdMap = {} as Record<string, THREAD_MAPPING_ID>;
@@ -118,6 +122,7 @@ export class RemoteThreadListThreadListRuntimeCore
           },
         })
         .catch(() => {
+          if (generation !== this._loadGeneration) return;
           this._loadThreadsPromise = undefined;
           this._state.update({
             ...this._state.baseValue,
@@ -179,6 +184,12 @@ export class RemoteThreadListThreadListRuntimeCore
     }
   }
 
+  public reload() {
+    this._loadGeneration++;
+    this._loadThreadsPromise = undefined;
+    return this.getLoadThreadsPromise();
+  }
+
   public get isLoading() {
     return this._state.value.isLoading;
   }
@@ -247,15 +258,23 @@ export class RemoteThreadListThreadListRuntimeCore
         [remoteMetadata.remoteId]: mappingId,
       };
 
+      // Filter both arrays first so a concurrent `list()` can't leave the id
+      // duplicated or under the wrong status.
+      const threadIdsWithoutRemote = state.threadIds.filter(
+        (id) => id !== remoteMetadata.remoteId,
+      );
+      const archivedThreadIdsWithoutRemote = state.archivedThreadIds.filter(
+        (id) => id !== remoteMetadata.remoteId,
+      );
+
       const newThreadIds =
         remoteMetadata.status === "regular"
-          ? [...state.threadIds, remoteMetadata.remoteId]
-          : state.threadIds;
-
+          ? [...threadIdsWithoutRemote, remoteMetadata.remoteId]
+          : threadIdsWithoutRemote;
       const newArchivedThreadIds =
         remoteMetadata.status === "archived"
-          ? [...state.archivedThreadIds, remoteMetadata.remoteId]
-          : state.archivedThreadIds;
+          ? [...archivedThreadIdsWithoutRemote, remoteMetadata.remoteId]
+          : archivedThreadIdsWithoutRemote;
 
       this._state.update({
         ...state,
@@ -352,6 +371,7 @@ export class RemoteThreadListThreadListRuntimeCore
           },
         };
       },
+      // biome-ignore lint/suspicious/noThenProperty: OptimisticState reducer pattern
       then: (state, { remoteId, externalId }) => {
         const data = getThreadData(state, threadId);
         if (!data) return state;

package/src/react/runtimes/RuntimeAdapterProvider.tsx

@@ -1,4 +1,4 @@
-import { createContext, FC, ReactNode, useContext } from "react";
+import { createContext, type FC, type ReactNode, useContext } from "react";
 import type { ThreadHistoryAdapter } from "../../adapters/thread-history";
 import type { AttachmentAdapter } from "../../adapters/attachment";
 import type { ModelContextProvider } from "../../model-context/types";

package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.ts

@@ -1,4 +1,4 @@
-import { RefObject, useState } from "react";
+import { type RefObject, useState } from "react";
 import type {
   GenericThreadHistoryAdapter,
   ThreadHistoryAdapter,
@@ -8,12 +8,12 @@ import type {
 } from "../../../adapters/thread-history";
 import type { ExportedMessageRepositoryItem } from "../../../runtime/utils/message-repository";
 import {
-  AssistantCloud,
+  type AssistantCloud,
   CloudMessagePersistence,
   createFormattedPersistence,
 } from "assistant-cloud";
 import { auiV0Decode, auiV0Encode } from "./auiV0";
-import { AssistantClient, useAui } from "@assistant-ui/store";
+import { type AssistantClient, useAui } from "@assistant-ui/store";
 import type { ThreadListItemMethods } from "../../../store/scopes/thread-list-item";
 
 const globalPersistence = new WeakMap<

package/src/react/runtimes/cloud/auiV0.ts

@@ -1,8 +1,11 @@
 import type { MessageStatus, ThreadMessage } from "../../../types/message";
 import { fromThreadMessageLike } from "../../../runtime/utils/thread-message-like";
-import { CloudMessage } from "assistant-cloud";
+import type { CloudMessage } from "assistant-cloud";
 import { isJSONValue } from "../../../utils/json/is-json";
-import { ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
+import type {
+  ReadonlyJSONObject,
+  ReadonlyJSONValue,
+} from "assistant-stream/utils";
 import type { ExportedMessageRepositoryItem } from "../../../runtime/utils/message-repository";
 
 type AuiV0MessagePart =

package/src/react/runtimes/cloud/useCloudThreadListAdapter.tsx

@@ -1,8 +1,8 @@
 declare const process: { env: Record<string, string | undefined> };
 
 import {
-  FC,
-  PropsWithChildren,
+  type FC,
+  type PropsWithChildren,
   useCallback,
   useEffect,
   useMemo,
@@ -28,7 +28,7 @@ type CloudThreadListAdapterOptions = {
 
 const baseUrl =
   typeof process !== "undefined" &&
-  process?.env?.["NEXT_PUBLIC_ASSISTANT_BASE_URL"];
+  process?.env?.NEXT_PUBLIC_ASSISTANT_BASE_URL;
 const autoCloud = baseUrl
   ? new AssistantCloud({ baseUrl, anonymous: true })
   : undefined;
@@ -43,17 +43,20 @@ export const useCloudThreadListAdapter = (
 
   const unstable_Provider = useCallback<FC<PropsWithChildren>>(
     function Provider({ children }) {
+      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
       const history = useAssistantCloudThreadHistoryAdapter({
         get current() {
           return adapterRef.current.cloud ?? autoCloud!;
         },
       });
       const cloudInstance = adapterRef.current.cloud ?? autoCloud!;
+      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
       const attachments = useMemo(
         () => new CloudFileAttachmentAdapter(cloudInstance),
         [cloudInstance],
       );
 
+      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
       const adapters = useMemo(
         () => ({
           history,

package/src/react/runtimes/useLocalRuntime.ts

@@ -22,6 +22,7 @@ const useLocalThreadRuntime = (
   chatModel: ChatModelAdapter,
   { initialMessages, ...options }: LocalRuntimeOptions,
 ): AssistantRuntime => {
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const { modelContext, ...threadListAdapters } = useRuntimeAdapters() ?? {};
   const opt = {
     ...options,
@@ -32,33 +33,41 @@ const useLocalThreadRuntime = (
     },
   };
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const [runtime] = useState(() => new LocalRuntimeCore(opt, initialMessages));
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const threadIdRef = useRef<string | undefined>(undefined);
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   threadIdRef.current = useAuiState((s) => s.threadListItem.remoteId);
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   useEffect(() => {
     runtime.threads
       .getMainThreadRuntimeCore()
       .__internal_setGetThreadId(() => threadIdRef.current);
   }, [runtime]);
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   useEffect(() => {
     return () => {
       runtime.threads.getMainThreadRuntimeCore().detach();
     };
   }, [runtime]);
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   useEffect(() => {
     runtime.threads.getMainThreadRuntimeCore().__internal_setOptions(opt);
     runtime.threads.getMainThreadRuntimeCore().__internal_load();
   });
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   useEffect(() => {
     if (!modelContext) return undefined;
     return runtime.registerModelContextProvider(modelContext);
   }, [modelContext, runtime]);
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   return useMemo(() => new AssistantRuntimeImpl(runtime), [runtime]);
 };
 
@@ -93,6 +102,7 @@ export const useLocalRuntime = (
   const cloudAdapter = useCloudThreadListAdapter({ cloud });
   return useRemoteThreadListRuntime({
     runtimeHook: function RuntimeHook() {
+      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
       return useLocalThreadRuntime(chatModel, options);
     },
     adapter: cloudAdapter,

package/src/react/runtimes/useRemoteThreadListRuntime.ts

@@ -29,12 +29,15 @@ class RemoteThreadListRuntimeCore
 const useRemoteThreadListRuntimeImpl = (
   options: RemoteThreadListOptions,
 ): AssistantRuntime => {
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const [runtime] = useState(() => new RemoteThreadListRuntimeCore(options));
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   useEffect(() => {
     runtime.threads.__internal_setOptions(options);
     runtime.threads.__internal_load();
   }, [runtime, options]);
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   return useMemo(() => new AssistantRuntimeImpl(runtime), [runtime]);
 };
 
@@ -77,9 +80,12 @@ export const useRemoteThreadListRuntime = (
     return stableRuntimeHook();
   }
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const runtime = useRemoteThreadListRuntimeImpl(stableOptions);
 
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const prevThreadIdRef = useRef(options.threadId);
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   useEffect(() => {
     if (options.threadId === prevThreadIdRef.current) return;
     prevThreadIdRef.current = options.threadId;

package/src/react/runtimes/useToolInvocations.ts

@@ -197,6 +197,7 @@ export function useToolInvocations({
             if (wasStarted) {
               executingCountRef.current--;
               if (executingCountRef.current === 0) {
+                // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
                 settledResolversRef.current.forEach((resolve) => resolve());
                 settledResolversRef.current = [];
               }
@@ -215,6 +216,7 @@ export function useToolInvocations({
           });
           // Resolve any waiting abort promises when all tools have settled
           if (executingCountRef.current === 0) {
+            // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
             settledResolversRef.current.forEach((resolve) => resolve());
             settledResolversRef.current = [];
           }

package/src/react/types/MessagePartComponentTypes.ts

@@ -12,7 +12,7 @@ import type {
   QuoteInfo,
 } from "../..";
 import type { MessagePartState } from "../..";
-import { ToolResponse } from "assistant-stream";
+import type { ToolResponse } from "assistant-stream";
 
 export type EmptyMessagePartProps = {
   status: MessagePartStatus;

package/src/react/types/scopes/dataRenderers.ts

@@ -3,11 +3,13 @@ import type { Unsubscribe } from "../../..";
 
 export type DataRenderersState = {
   renderers: Record<string, DataMessagePartComponent[]>;
+  fallbacks: DataMessagePartComponent[];
 };
 
 export type DataRenderersMethods = {
   getState(): DataRenderersState;
   setDataUI(name: string, render: DataMessagePartComponent): Unsubscribe;
+  setFallbackDataUI(render: DataMessagePartComponent): Unsubscribe;
 };
 
 export type DataRenderersClientSchema = {

package/src/runtime/api/assistant-runtime.ts

@@ -3,7 +3,7 @@ import type { ModelContextProvider } from "../../model-context/types";
 import type { AssistantRuntimeCore } from "../interfaces/assistant-runtime-core";
 import type { ThreadRuntime } from "./thread-runtime";
 import {
-  ThreadListRuntime,
+  type ThreadListRuntime,
   ThreadListRuntimeImpl,
 } from "./thread-list-runtime";
 import { ExportedMessageRepository } from "../utils/message-repository";

package/src/runtime/api/attachment-runtime.ts

@@ -105,10 +105,6 @@ export class MessageAttachmentRuntimeImpl extends AttachmentRuntimeImpl<"message
     return "message";
   }
 
-  constructor(core: AttachmentSnapshotBinding<"message">) {
-    super(core);
-  }
-
   public remove(): never {
     throw new Error("Message attachments cannot be removed");
   }

package/src/runtime/api/message-runtime.ts

@@ -1,4 +1,4 @@
-import {
+import type {
   SpeechState,
   SubmittedFeedback,
 } from "../interfaces/thread-runtime-core";
@@ -18,21 +18,21 @@ import {
   ShallowMemoizeSubject,
 } from "../../subscribable/subscribable";
 import {
-  AttachmentRuntime,
-  AttachmentState,
+  type AttachmentRuntime,
+  type AttachmentState,
   MessageAttachmentRuntimeImpl,
 } from "./attachment-runtime";
 import {
-  EditComposerRuntime,
+  type EditComposerRuntime,
   EditComposerRuntimeImpl,
 } from "./composer-runtime";
 import {
-  MessagePartRuntime,
+  type MessagePartRuntime,
   MessagePartRuntimeImpl,
-  MessagePartState,
+  type MessagePartState,
 } from "./message-part-runtime";
-import { MessageRuntimePath } from "./paths";
-import { ThreadRuntimeCoreBinding } from "./thread-runtime";
+import type { MessageRuntimePath } from "./paths";
+import type { ThreadRuntimeCoreBinding } from "./thread-runtime";
 import type { MessageStateBinding } from "./bindings";
 
 const COMPLETE_STATUS: MessagePartStatus = Object.freeze({