@assistant-ui/core 0.1.14 → 0.1.16

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/bindings.ts

@@ -47,4 +47,5 @@ export type ThreadListItemState = {
   readonly externalId: string | undefined;
   readonly status: import("../interfaces/thread-list-runtime-core").ThreadListItemStatus;
   readonly title?: string | undefined;
+  readonly custom?: Record<string, unknown> | undefined;
 };

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

@@ -136,8 +136,10 @@ export type ComposerRuntime = {
   /**
    * Add an attachment to the composer. Accepts either a standard File object
    * (processed through the AttachmentAdapter) or a CreateAttachment descriptor
-   * for external-source attachments (URLs, API data, CMS references) that
-   * bypasses the adapter entirely.
+   * for external-source attachments (URLs, API data, CMS references). External
+   * descriptors bypass the adapter's `add()` step but still respect
+   * `adapter.accept` when an adapter is configured; without an adapter they
+   * are added as-is.
    * @param fileOrAttachment The file or attachment descriptor to add.
    */
   addAttachment(fileOrAttachment: File | CreateAttachment): Promise<void>;

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({

package/src/runtime/api/thread-list-runtime.ts

@@ -9,17 +9,19 @@ import {
 } from "../../subscribable/subscribable";
 import type { ThreadListRuntimeCore } from "../interfaces/thread-list-runtime-core";
 import {
-  ThreadListItemRuntime,
+  type ThreadListItemRuntime,
   ThreadListItemRuntimeImpl,
-  ThreadListItemState,
+  type ThreadListItemState,
 } from "./thread-list-item-runtime";
 import {
-  ThreadListItemRuntimeBinding,
-  ThreadRuntime,
-  ThreadRuntimeCoreBinding,
+  type ThreadListItemRuntimeBinding,
+  type ThreadRuntime,
+  type ThreadRuntimeCoreBinding,
   ThreadRuntimeImpl,
 } from "./thread-runtime";
 
+const RESOLVED_PROMISE = Promise.resolve();
+
 export type ThreadListState = {
   readonly mainThreadId: string;
   readonly newThreadId: string | undefined;
@@ -48,6 +50,7 @@ export type ThreadListRuntime = {
   switchToNewThread(): Promise<void>;
 
   getLoadThreadsPromise(): Promise<void>;
+  reload(): Promise<void>;
 };
 
 const getThreadListState = (
@@ -77,6 +80,7 @@ const getThreadListItemState = (
     externalId: threadData.externalId,
     title: threadData.title,
     status: threadData.status,
+    custom: threadData.custom,
     isMain: threadData.id === threadList.mainThreadId,
   };
 };
@@ -133,6 +137,7 @@ export class ThreadListRuntimeImpl implements ThreadListRuntime {
     this.switchToThread = this.switchToThread.bind(this);
     this.switchToNewThread = this.switchToNewThread.bind(this);
     this.getLoadThreadsPromise = this.getLoadThreadsPromise.bind(this);
+    this.reload = this.reload.bind(this);
     this.getState = this.getState.bind(this);
     this.subscribe = this.subscribe.bind(this);
     this.getById = this.getById.bind(this);
@@ -153,6 +158,10 @@ export class ThreadListRuntimeImpl implements ThreadListRuntime {
     return this._core.getLoadThreadsPromise();
   }
 
+  public reload(): Promise<void> {
+    return this._core.reload?.() ?? RESOLVED_PROMISE;
+  }
+
   public getState(): ThreadListState {
     return this._getState();
   }

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

@@ -1,4 +1,4 @@
-import {
+import type {
   ThreadSuggestion,
   RuntimeCapabilities,
   ThreadRuntimeCore,
@@ -8,12 +8,12 @@ import {
   StartRunConfig,
   ResumeRunConfig,
 } from "../interfaces/thread-runtime-core";
-import { ExportedMessageRepository } from "../utils/message-repository";
-import { ThreadMessageLike } from "../utils/thread-message-like";
+import type { ExportedMessageRepository } from "../utils/message-repository";
+import type { ThreadMessageLike } from "../utils/thread-message-like";
 import {
-  MessageRuntime,
+  type MessageRuntime,
   MessageRuntimeImpl,
-  MessageState,
+  type MessageState,
 } from "./message-runtime";
 import { NestedSubscriptionSubject } from "../../subscribable/subscribable";
 import {
@@ -22,10 +22,10 @@ import {
 } from "../../subscribable/subscribable";
 import type { SubscribableWithState } from "../../subscribable/subscribable";
 import {
-  ThreadComposerRuntime,
+  type ThreadComposerRuntime,
   ThreadComposerRuntimeImpl,
 } from "./composer-runtime";
-import {
+import type {
   MessageRuntimePath,
   ThreadListItemRuntimePath,
   ThreadRuntimePath,
@@ -36,12 +36,12 @@ import type { Unsubscribe } from "../../types/unsubscribe";
 import type { RunConfig } from "../../types/message";
 import { EventSubscriptionSubject } from "../../subscribable/subscribable";
 import { symbolInnerMessage } from "../utils/external-store-message";
-import { ModelContext } from "../../model-context/types";
-import {
+import type { ModelContext } from "../../model-context/types";
+import type {
   ChatModelRunOptions,
   ChatModelRunResult,
 } from "../utils/chat-model-adapter";
-import { ReadonlyJSONValue } from "assistant-stream/utils";
+import type { ReadonlyJSONValue } from "assistant-stream/utils";
 
 export type CreateStartRunConfig = {
   parentId: string | null;
@@ -204,9 +204,10 @@ export const getThreadState = (
     isDisabled: runtime.isDisabled,
     isLoading: runtime.isLoading,
     isRunning:
-      lastMessage?.role !== "assistant"
+      runtime.isRunning ??
+      (lastMessage?.role !== "assistant"
         ? false
-        : lastMessage.status.type === "running",
+        : lastMessage.status.type === "running"),
     messages: runtime.messages,
     state: runtime.state,
     suggestions: runtime.suggestions,

package/src/runtime/base/base-composer-runtime-core.ts

@@ -204,6 +204,27 @@ export abstract class BaseComposerRuntimeCore
 
   async addAttachment(fileOrAttachment: File | CreateAttachment) {
     if (!(fileOrAttachment instanceof File)) {
+      const adapter = this.getAttachmentAdapter();
+      if (
+        adapter &&
+        !fileMatchesAccept(
+          {
+            name: fileOrAttachment.name,
+            type: fileOrAttachment.contentType ?? "",
+          },
+          adapter.accept,
+        )
+      ) {
+        try {
+          this._notifyEventSubscribers("attachmentAddError");
+        } catch {
+          // prevent subscriber errors from masking the original error
+        }
+        throw new Error(
+          `File type ${fileOrAttachment.contentType || "unknown"} is not accepted. Accepted types: ${adapter.accept}`,
+        );
+      }
+
       const a: CompleteAttachment = {
         id: fileOrAttachment.id ?? generateId(),
         type: fileOrAttachment.type ?? "document",
@@ -213,25 +234,11 @@ export abstract class BaseComposerRuntimeCore
         status: { type: "complete" },
       };
       this._attachments = [...this._attachments, a];
-      this._notifyEventSubscribers("attachmentAdd");
       this._notifySubscribers();
+      this._notifyEventSubscribers("attachmentAdd");
       return;
     }
 
-    const adapter = this.getAttachmentAdapter();
-    if (!adapter) throw new Error("Attachments are not supported");
-
-    if (
-      !fileMatchesAccept(
-        { name: fileOrAttachment.name, type: fileOrAttachment.type },
-        adapter.accept,
-      )
-    ) {
-      throw new Error(
-        `File type ${fileOrAttachment.type || "unknown"} is not accepted. Accepted types: ${adapter.accept}`,
-      );
-    }
-
     const upsertAttachment = (a: PendingAttachment) => {
       const idx = this._attachments.findIndex(
         (attachment) => attachment.id === a.id,
@@ -251,6 +258,20 @@ export abstract class BaseComposerRuntimeCore
 
     let lastAttachment: PendingAttachment | undefined;
     try {
+      const adapter = this.getAttachmentAdapter();
+      if (!adapter) throw new Error("Attachments are not supported");
+
+      if (
+        !fileMatchesAccept(
+          { name: fileOrAttachment.name, type: fileOrAttachment.type },
+          adapter.accept,
+        )
+      ) {
+        throw new Error(
+          `File type ${fileOrAttachment.type || "unknown"} is not accepted. Accepted types: ${adapter.accept}`,
+        );
+      }
+
       const promiseOrGenerator = adapter.add({ file: fileOrAttachment });
       if (Symbol.asyncIterator in promiseOrGenerator) {
         for await (const r of promiseOrGenerator) {
@@ -271,7 +292,7 @@ export abstract class BaseComposerRuntimeCore
       try {
         this._notifyEventSubscribers("attachmentAddError");
       } catch {
-        // prevent subscriber errors from masking the adapter error
+        // prevent subscriber errors from masking the original error
       }
       throw e;
     }

package/src/runtime/base/base-thread-runtime-core.ts

@@ -93,7 +93,7 @@ export abstract class BaseThreadRuntimeCore implements ThreadRuntimeCore {
   }
 
   public get state() {
-    let mostRecentAssistantMessage;
+    let mostRecentAssistantMessage: (typeof this.messages)[number] | undefined;
     for (const message of this.messages) {
       if (message.role === "assistant") {
         mostRecentAssistantMessage = message;

package/src/runtime/base/default-edit-composer-runtime-core.ts

@@ -1,5 +1,7 @@
 import type { AppendMessage, ThreadMessage } from "../../types/message";
+import type { CompleteAttachment } from "../../types/attachment";
 import { getThreadMessageText } from "../../utils/text";
+import { attachmentsEqual, liftNonTextParts } from "../../adapters/attachment";
 import type { AttachmentAdapter } from "../../adapters/attachment";
 import type { DictationAdapter } from "../../adapters/speech";
 import type { SendOptions } from "../interfaces/composer-runtime-core";
@@ -19,8 +21,9 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
     return this.runtime.adapters?.dictation;
   }
 
-  private _nonTextParts;
   private _previousText;
+  private _previousAttachments: readonly CompleteAttachment[];
+  private _nonTextPassthrough: readonly ThreadMessage["content"][number][];
   private _parentId: string | null;
   private _sourceId: string | null;
   constructor(
@@ -42,9 +45,20 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
     this.setText(this._previousText);
 
     this.setRole(message.role);
-    this.setAttachments(message.attachments ?? []);
 
-    this._nonTextParts = message.content.filter((part) => part.type !== "text");
+    if (message.role === "user") {
+      this._previousAttachments = [
+        ...(message.attachments ?? []),
+        ...liftNonTextParts(message.content),
+      ];
+      this._nonTextPassthrough = [];
+    } else {
+      this._previousAttachments = message.attachments ?? [];
+      this._nonTextPassthrough = message.content.filter(
+        (p) => p.type !== "text",
+      );
+    }
+    this.setAttachments(this._previousAttachments);
 
     this.setRunConfig({ ...runtime.composer.runConfig });
   }
@@ -62,10 +76,26 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
     options?: SendOptions,
   ) {
     const text = getThreadMessageText(message as AppendMessage);
-    if (text !== this._previousText || options?.startRun) {
+    const attachmentsChanged = !attachmentsEqual(
+      message.attachments ?? [],
+      this._previousAttachments,
+    );
+
+    if (
+      text !== this._previousText ||
+      attachmentsChanged ||
+      options?.startRun
+    ) {
+      const content =
+        this._nonTextPassthrough.length > 0
+          ? ([
+              ...message.content,
+              ...this._nonTextPassthrough,
+            ] as AppendMessage["content"])
+          : message.content;
       this.runtime.append({
         ...message,
-        content: [...message.content, ...this._nonTextParts] as any,
+        content,
         parentId: this._parentId,
         sourceId: this._sourceId,
         startRun: options?.startRun,

package/src/runtime/interfaces/thread-list-runtime-core.ts

@@ -10,6 +10,7 @@ export type ThreadListItemCoreState = {
 
   readonly status: ThreadListItemStatus;
   readonly title?: string | undefined;
+  readonly custom?: Record<string, unknown> | undefined;
 
   readonly runtime?: ThreadRuntimeCore | undefined;
 };
@@ -33,6 +34,7 @@ export type ThreadListRuntimeCore = {
   switchToNewThread(): Promise<void>;
 
   getLoadThreadsPromise(): Promise<void>;
+  reload?(): Promise<void>;
 
   detach(threadId: string): Promise<void>;
   rename(threadId: string, newTitle: string): Promise<void>;

package/src/runtime/interfaces/thread-runtime-core.ts

@@ -129,6 +129,13 @@ export type ThreadRuntimeCore = Readonly<{
   capabilities: Readonly<RuntimeCapabilities>;
   isDisabled: boolean;
   isLoading: boolean;
+  /**
+   * Optional explicit thread-level running flag. When provided, takes
+   * precedence over the last-message-status heuristic. When omitted, falls
+   * back to the legacy behavior. External-store runtimes surface this via
+   * `ExternalStoreAdapter.isRunning`.
+   */
+  isRunning?: boolean | undefined;
   messages: readonly ThreadMessage[];
   state: ReadonlyJSONValue;
   suggestions: readonly ThreadSuggestion[];

package/src/runtime/internal.ts

@@ -1,7 +1,4 @@
-// =============================================================================
-// Internal API — implementation details used by framework bindings
-// Not part of the public API surface.
-// =============================================================================
+// Internal API — implementation details used by framework bindings, not part of the public API surface.
 
 // Binding Types
 export type {

package/src/runtime/utils/auto-status.ts

@@ -1,5 +1,5 @@
 import type { MessageStatus } from "../../types/message";
-import { ReadonlyJSONValue } from "assistant-stream/utils";
+import type { ReadonlyJSONValue } from "assistant-stream/utils";
 
 const symbolAutoStatus = Symbol("autoStatus");
 

package/src/runtime/utils/chat-model-adapter.ts

@@ -14,7 +14,7 @@ import type {
   TextMessagePart,
 } from "../../types/message";
 import type { ModelContext } from "../../model-context/types";
-import { ReadonlyJSONValue } from "assistant-stream/utils";
+import type { ReadonlyJSONValue } from "assistant-stream/utils";
 
 export type ChatModelRunUpdate = {
   readonly content: readonly ThreadAssistantMessagePart[];

package/src/runtime/utils/thread-message-like.ts

@@ -18,7 +18,10 @@ import type {
 } from "../../types/message";
 import type { CompleteAttachment } from "../../types/attachment";
 import type { MessageTiming, TextMessagePart } from "../../types/message";
-import { ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
+import type {
+  ReadonlyJSONObject,
+  ReadonlyJSONValue,
+} from "assistant-stream/utils";
 
 type DataPrefixedPart = {
   readonly type: `data-${string}`;

package/src/runtimes/assistant-transport/utils.ts

@@ -1,4 +1,8 @@
-import { Tool, toToolsJSONSchema, type ToolJSONSchema } from "assistant-stream";
+import {
+  type Tool,
+  toToolsJSONSchema,
+  type ToolJSONSchema,
+} from "assistant-stream";
 
 /**
  * @deprecated Use `toToolsJSONSchema` from `assistant-stream` instead.

package/src/runtimes/external-store/external-store-adapter.ts

@@ -60,6 +60,14 @@ type ExternalStoreMessageConverterAdapter<T> = {
 
 type ExternalStoreAdapterBase<T> = {
   isDisabled?: boolean | undefined;
+  /**
+   * Whether the thread is running. When provided, this value flows directly
+   * to `thread.isRunning`, letting the application keep the thread in a
+   * running state even after the last assistant message has completed (for
+   * example while non-message stream chunks like suggestions or metadata
+   * updates are still arriving). When omitted, `thread.isRunning` falls back
+   * to the last-message-status heuristic.
+   */
   isRunning?: boolean | undefined;
   isLoading?: boolean | undefined;
   messages?: readonly T[];