@assistant-ui/core 0.1.14 → 0.1.15

package/src/adapters/attachment.ts

@@ -3,10 +3,8 @@ import type {
   PendingAttachment,
   CompleteAttachment,
 } from "../types/attachment";
-
-// =============================================================================
-// Adapter Type
-// =============================================================================
+import type { ThreadUserMessagePart } from "../types/message";
+import { generateId } from "../utils/id";
 
 export type AttachmentAdapter = {
   accept: string;
@@ -17,10 +15,6 @@ export type AttachmentAdapter = {
   send(attachment: PendingAttachment): Promise<CompleteAttachment>;
 };
 
-// =============================================================================
-// Simple Image Attachment Adapter
-// =============================================================================
-
 export class SimpleImageAttachmentAdapter implements AttachmentAdapter {
   public accept = "image/*";
 
@@ -63,10 +57,6 @@ const getFileDataURL = (file: File) =>
     reader.readAsDataURL(file);
   });
 
-// =============================================================================
-// Simple Text Attachment Adapter
-// =============================================================================
-
 export class SimpleTextAttachmentAdapter implements AttachmentAdapter {
   public accept =
     "text/plain,text/html,text/markdown,text/csv,text/xml,text/json,text/css";
@@ -110,10 +100,6 @@ const getFileText = (file: File) =>
     reader.readAsText(file);
   });
 
-// =============================================================================
-// Composite Attachment Adapter
-// =============================================================================
-
 export function fileMatchesAccept(
   file: { name: string; type: string },
   acceptString: string,
@@ -149,6 +135,72 @@ export function fileMatchesAccept(
   return false;
 }
 
+export function attachmentsEqual(
+  a: readonly CompleteAttachment[],
+  b: readonly CompleteAttachment[],
+): boolean {
+  if (a.length !== b.length) return false;
+  return a.every((att, i) => att.id === b[i]!.id);
+}
+
+export function partToCompleteAttachment(
+  part: Exclude<ThreadUserMessagePart, { type: "text" }>,
+): CompleteAttachment {
+  const id = generateId();
+
+  if (part.type === "image") {
+    return {
+      id,
+      type: "image",
+      name: part.filename ?? "image",
+      content: [part],
+      status: { type: "complete" },
+    };
+  }
+
+  if (part.type === "file") {
+    return {
+      id,
+      type: "document",
+      name: part.filename ?? "document",
+      contentType: part.mimeType,
+      content: [part],
+      status: { type: "complete" },
+    };
+  }
+
+  if (part.type === "audio") {
+    return {
+      id,
+      type: "audio",
+      name: `audio.${part.audio.format}`,
+      contentType: `audio/${part.audio.format}`,
+      content: [part],
+      status: { type: "complete" },
+    };
+  }
+
+  return {
+    id,
+    type: "data",
+    name: part.name,
+    content: [part],
+    status: { type: "complete" },
+  };
+}
+
+export function liftNonTextParts(
+  content: readonly ThreadUserMessagePart[],
+): CompleteAttachment[] {
+  const result: CompleteAttachment[] = [];
+  for (const part of content) {
+    if (part.type !== "text") {
+      result.push(partToCompleteAttachment(part));
+    }
+  }
+  return result;
+}
+
 export class CompositeAttachmentAdapter implements AttachmentAdapter {
   private _adapters: AttachmentAdapter[];
 

package/src/adapters/{mention.ts → directive-formatter.ts}

@@ -1,21 +1,11 @@
 import type {
-  Unstable_MentionItem,
-  Unstable_DirectiveSegment,
   Unstable_DirectiveFormatter,
-} from "../types/mention";
-import type { Unstable_TriggerAdapter } from "./trigger";
-
-// =============================================================================
-// Mention Adapter — alias of the generic TriggerAdapter
-// =============================================================================
-
-export type Unstable_MentionAdapter = Unstable_TriggerAdapter;
-
-// =============================================================================
-// Default Directive Formatter
-// =============================================================================
+  Unstable_DirectiveSegment,
+} from "../types/directive";
+import type { Unstable_TriggerItem } from "../types/trigger";
 
-const DIRECTIVE_RE = /:([\w-]+)\[([^\]]+)\](?:\{name=([^}]+)\})?/g;
+const DIRECTIVE_RE =
+  /:([\w-]{1,64})\[([^\]\n]{1,1024})\](?:\{name=([^}\n]{1,1024})\})?/gu;
 
 /**
  * Default directive formatter using the `:type[label]{name=id}` syntax.
@@ -23,7 +13,7 @@ const DIRECTIVE_RE = /:([\w-]+)\[([^\]]+)\](?:\{name=([^}]+)\})?/g;
  * When `id` equals `label`, the `{name=…}` attribute is omitted for brevity.
  */
 export const unstable_defaultDirectiveFormatter: Unstable_DirectiveFormatter = {
-  serialize(item: Unstable_MentionItem): string {
+  serialize(item: Unstable_TriggerItem): string {
     const attrs = item.id !== item.label ? `{name=${item.id}}` : "";
     return `:${item.type}[${item.label}]${attrs}`;
   },
@@ -32,9 +22,7 @@ export const unstable_defaultDirectiveFormatter: Unstable_DirectiveFormatter = {
     const segments: Unstable_DirectiveSegment[] = [];
     let lastIndex = 0;
 
-    DIRECTIVE_RE.lastIndex = 0;
-    let match: RegExpExecArray | null;
-    while ((match = DIRECTIVE_RE.exec(text)) !== null) {
+    for (const match of text.matchAll(DIRECTIVE_RE)) {
       if (match.index > lastIndex) {
         segments.push({
           kind: "text",
@@ -48,7 +36,7 @@ export const unstable_defaultDirectiveFormatter: Unstable_DirectiveFormatter = {
         label,
         id: match[3] ?? label,
       });
-      lastIndex = DIRECTIVE_RE.lastIndex;
+      lastIndex = match.index + match[0].length;
     }
 
     if (lastIndex < text.length) {

package/src/adapters/index.ts

@@ -27,15 +27,8 @@ export type { SuggestionAdapter } from "./suggestion";
 // Trigger adapter (generic)
 export type { Unstable_TriggerAdapter } from "./trigger";
 
-// Slash command adapter
-export type {
-  Unstable_SlashCommandAdapter,
-  Unstable_SlashCommandItem,
-} from "./trigger";
-
-// Mention adapter
-export type { Unstable_MentionAdapter } from "./mention";
-export { unstable_defaultDirectiveFormatter } from "./mention";
+// Directive formatter
+export { unstable_defaultDirectiveFormatter } from "./directive-formatter";
 
 // Thread history adapters
 export type {

package/src/adapters/speech.ts

@@ -1,9 +1,5 @@
 import type { Unsubscribe } from "../types/unsubscribe";
 
-// =============================================================================
-// Speech Synthesis Adapter
-// =============================================================================
-
 export namespace SpeechSynthesisAdapter {
   export type Status =
     | {
@@ -26,10 +22,6 @@ export type SpeechSynthesisAdapter = {
   speak: (text: string) => SpeechSynthesisAdapter.Utterance;
 };
 
-// =============================================================================
-// Dictation Adapter
-// =============================================================================
-
 export namespace DictationAdapter {
   export type Status =
     | {
@@ -60,10 +52,6 @@ export type DictationAdapter = {
   disableInputDuringDictation?: boolean;
 };
 
-// =============================================================================
-// Web Speech Synthesis Adapter
-// =============================================================================
-
 export class WebSpeechSynthesisAdapter implements SpeechSynthesisAdapter {
   speak(text: string): SpeechSynthesisAdapter.Utterance {
     const utterance = new SpeechSynthesisUtterance(text);
@@ -76,6 +64,7 @@ export class WebSpeechSynthesisAdapter implements SpeechSynthesisAdapter {
       if (res.status.type === "ended") return;
 
       res.status = { type: "ended", reason, error };
+      // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
       subscribers.forEach((handler) => handler());
     };
 
@@ -111,10 +100,6 @@ export class WebSpeechSynthesisAdapter implements SpeechSynthesisAdapter {
   }
 }
 
-// =============================================================================
-// Web Speech Dictation Adapter
-// =============================================================================
-
 interface SpeechRecognitionResultItem {
   transcript: string;
   confidence: number;

package/src/adapters/thread-history.ts

@@ -7,10 +7,6 @@ import type {
   ExportedMessageRepositoryItem,
 } from "../runtime/utils/message-repository";
 
-// =============================================================================
-// Message Format Adapter
-// =============================================================================
-
 export interface MessageStorageEntry<TPayload> {
   id: string;
   parent_id: string | null;
@@ -40,10 +36,6 @@ export interface MessageFormatAdapter<
   getId(message: TMessage): string;
 }
 
-// =============================================================================
-// Thread History Adapter
-// =============================================================================
-
 export type GenericThreadHistoryAdapter<TMessage> = {
   load(): Promise<MessageFormatRepository<TMessage>>;
   append(item: MessageFormatItem<TMessage>): Promise<void>;
@@ -66,6 +58,7 @@ export type ThreadHistoryAdapter = {
     options: ChatModelRunOptions,
   ): AsyncGenerator<ChatModelRunResult, void, unknown>;
   append(item: ExportedMessageRepositoryItem): Promise<void>;
+  /** Required when used with `useAISDKRuntime` / `useChatRuntime`. */
   withFormat?<TMessage, TStorageFormat extends Record<string, unknown>>(
     formatAdapter: MessageFormatAdapter<TMessage, TStorageFormat>,
   ): GenericThreadHistoryAdapter<TMessage>;

package/src/adapters/trigger.ts

@@ -3,18 +3,7 @@ import type {
   Unstable_TriggerItem,
 } from "../types/trigger";
 
-// =============================================================================
-// Trigger Adapter — generic adapter for any trigger-based popover
-// =============================================================================
-
-/**
- * Adapter for providing categories and items to a trigger popover.
- *
- * All methods are synchronous by design — the adapter drives UI display and
- * must return data immediately. Use external state management (e.g. React
- * Query, SWR, or local state) to handle async data fetching, then expose
- * the loaded results synchronously through this adapter.
- */
+/** Adapter providing synchronous categories and items to a trigger popover. */
 export type Unstable_TriggerAdapter = {
   /** Return the top-level categories for the trigger popover. */
   categories(): readonly Unstable_TriggerCategory[];
@@ -25,28 +14,3 @@ export type Unstable_TriggerAdapter = {
   /** Global search across all categories (optional). */
   search?(query: string): readonly Unstable_TriggerItem[];
 };
-
-// =============================================================================
-// Slash Command Item — extends TriggerItem with an execute callback
-// =============================================================================
-
-export type Unstable_SlashCommandItem = Unstable_TriggerItem & {
-  /** Action to execute when the command is selected. */
-  readonly execute?: (() => void) | undefined;
-};
-
-// =============================================================================
-// Slash Command Adapter — extends TriggerAdapter with narrower return types
-// so consumers get SlashCommandItem directly without casting.
-// =============================================================================
-
-export type Unstable_SlashCommandAdapter = Omit<
-  Unstable_TriggerAdapter,
-  "categoryItems" | "search"
-> & {
-  /** Return commands within a category. */
-  categoryItems(categoryId: string): readonly Unstable_SlashCommandItem[];
-
-  /** Global search across all categories (optional). */
-  search?(query: string): readonly Unstable_SlashCommandItem[];
-};

package/src/index.ts

@@ -1,7 +1,5 @@
 // @assistant-ui/core - Framework-agnostic core runtime (public API)
 
-// === types ===
-
 export type {
   // Message parts
   TextMessagePart,
@@ -44,23 +42,17 @@ export type { Unsubscribe } from "./types/unsubscribe";
 export type { QuoteInfo } from "./types/quote";
 
 export type {
-  Unstable_MentionItem,
-  Unstable_MentionCategory,
   Unstable_DirectiveSegment,
   Unstable_DirectiveFormatter,
-} from "./types/mention";
+} from "./types/directive";
 
 export type {
   Unstable_TriggerItem,
   Unstable_TriggerCategory,
 } from "./types/trigger";
 
-// === trigger adapter ===
-
 export type { Unstable_TriggerAdapter } from "./adapters/trigger";
 
-// === model-context ===
-
 export type {
   // Language model settings
   LanguageModelV1CallSettings,
@@ -94,8 +86,6 @@ export type {
 } from "./model-context/frame/types";
 export { FRAME_MESSAGE_CHANNEL } from "./model-context/frame/types";
 
-// === adapters ===
-
 // Attachment adapters
 export type { AttachmentAdapter } from "./adapters/attachment";
 export {
@@ -128,15 +118,8 @@ export type { FeedbackAdapter } from "./adapters/feedback";
 // Suggestion adapter
 export type { SuggestionAdapter } from "./adapters/suggestion";
 
-// Mention adapter
-export type { Unstable_MentionAdapter } from "./adapters/mention";
-export { unstable_defaultDirectiveFormatter } from "./adapters/mention";
-
-// Slash command adapter
-export type {
-  Unstable_SlashCommandAdapter,
-  Unstable_SlashCommandItem,
-} from "./adapters/trigger";
+// Directive formatter
+export { unstable_defaultDirectiveFormatter } from "./adapters/directive-formatter";
 
 // Thread history adapters
 export type {
@@ -148,8 +131,6 @@ export type {
   MessageStorageEntry,
 } from "./adapters/thread-history";
 
-// === runtime ===
-
 // Path Types
 export type {
   ThreadListItemRuntimePath,
@@ -262,8 +243,6 @@ export {
 export type { ExportedMessageRepositoryItem } from "./runtime/utils/message-repository";
 export { ExportedMessageRepository } from "./runtime/utils/message-repository";
 
-// === runtimes ===
-
 // Local Runtime Options
 export type { LocalRuntimeOptionsBase } from "./runtimes/local/local-runtime-options";
 

package/src/model-context/frame/host.ts

@@ -1,11 +1,11 @@
-import { ModelContextProvider, ModelContext } from "../types";
+import type { ModelContextProvider, ModelContext } from "../types";
 import type { Unsubscribe } from "../../types/unsubscribe";
-import { Tool } from "assistant-stream";
+import type { Tool } from "assistant-stream";
 import {
-  FrameMessage,
+  type FrameMessage,
   FRAME_MESSAGE_CHANNEL,
-  SerializedModelContext,
-  SerializedTool,
+  type SerializedModelContext,
+  type SerializedTool,
 } from "./types";
 
 /**
@@ -175,6 +175,7 @@ export class AssistantFrameHost implements ModelContextProvider {
   }
 
   private notifySubscribers() {
+    // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
     this._subscribers.forEach((callback) => callback());
   }
 

package/src/model-context/frame/provider.ts

@@ -1,11 +1,11 @@
-import { ModelContextProvider, ModelContext } from "../types";
+import type { ModelContextProvider, ModelContext } from "../types";
 import type { Unsubscribe } from "../../types/unsubscribe";
-import { Tool, toJSONSchema } from "assistant-stream";
+import { type Tool, toJSONSchema } from "assistant-stream";
 import {
-  FrameMessage,
+  type FrameMessage,
   FRAME_MESSAGE_CHANNEL,
-  SerializedModelContext,
-  SerializedTool,
+  type SerializedModelContext,
+  type SerializedTool,
 } from "./types";
 
 const serializeTool = (tool: Tool<any, any>): SerializedTool => ({
@@ -179,6 +179,7 @@ export class AssistantFrameProvider {
       const instance = AssistantFrameProvider._instance;
       window.removeEventListener("message", instance.handleMessage);
 
+      // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
       instance._providerUnsubscribes.forEach((unsubscribe) => unsubscribe?.());
       instance._providerUnsubscribes.clear();
       instance._providers.clear();

package/src/model-context/registry.ts

@@ -1,13 +1,13 @@
-import { Tool } from "assistant-stream";
+import type { Tool } from "assistant-stream";
 import {
-  ModelContext,
-  ModelContextProvider,
+  type ModelContext,
+  type ModelContextProvider,
   mergeModelContexts,
-  AssistantToolProps,
-  AssistantInstructionsConfig,
+  type AssistantToolProps,
+  type AssistantInstructionsConfig,
 } from "./types";
 import type { Unsubscribe } from "../types/unsubscribe";
-import {
+import type {
   ModelContextRegistryToolHandle,
   ModelContextRegistryInstructionHandle,
   ModelContextRegistryProviderHandle,

package/src/model-context/tool.ts

@@ -1,4 +1,4 @@
-import { Tool } from "assistant-stream";
+import type { Tool } from "assistant-stream";
 
 // TODO re-add the inferrence of the parameters
 

package/src/model-context/types.ts

@@ -1,9 +1,5 @@
 import type { Unsubscribe } from "../types/unsubscribe";
-import { Tool } from "assistant-stream";
-
-// =============================================================================
-// Language Model Settings
-// =============================================================================
+import type { Tool } from "assistant-stream";
 
 export type LanguageModelV1CallSettings = {
   maxTokens?: number;
@@ -21,10 +17,6 @@ export type LanguageModelConfig = {
   modelName?: string;
 };
 
-// =============================================================================
-// Model Context
-// =============================================================================
-
 export type ModelContext = {
   priority?: number | undefined;
   system?: string | undefined;
@@ -38,10 +30,6 @@ export type ModelContextProvider = {
   subscribe?: (callback: () => void) => Unsubscribe;
 };
 
-// =============================================================================
-// Tool & Instruction Config
-// =============================================================================
-
 export type AssistantToolProps<
   TArgs extends Record<string, unknown>,
   TResult,
@@ -60,10 +48,6 @@ export type AssistantContextConfig = {
   disabled?: boolean | undefined;
 };
 
-// =============================================================================
-// Merging
-// =============================================================================
-
 export const mergeModelContexts = (
   configSet: Set<ModelContextProvider>,
 ): ModelContext => {

package/src/react/AssistantRuntimeProvider.tsx

@@ -1,5 +1,5 @@
 import { type ReactNode, memo } from "react";
-import { type AssistantClient } from "@assistant-ui/store";
+import type { AssistantClient } from "@assistant-ui/store";
 import type { AssistantRuntime } from "../runtime/api/assistant-runtime";
 import { AssistantProviderBase } from "./AssistantProvider";
 

package/src/react/adapters/LocalStorageThreadListAdapter.tsx

@@ -1,5 +1,5 @@
 import { type AssistantStream, createAssistantStream } from "assistant-stream";
-import { FC, PropsWithChildren, useMemo } from "react";
+import { type FC, type PropsWithChildren, useMemo } from "react";
 import { useAui } from "@assistant-ui/store";
 import type {
   RemoteThreadInitializeResponse,

package/src/react/client/DataRenderers.ts

@@ -1,11 +1,12 @@
 import { resource, tapState, tapCallback } from "@assistant-ui/tap";
-import { type ClientOutput } from "@assistant-ui/store";
-import { DataRenderersState } from "../types/scopes/dataRenderers";
-import { DataMessagePartComponent } from "../types/MessagePartComponentTypes";
+import type { ClientOutput } from "@assistant-ui/store";
+import type { DataRenderersState } from "../types/scopes/dataRenderers";
+import type { DataMessagePartComponent } from "../types/MessagePartComponentTypes";
 
 export const DataRenderers = resource((): ClientOutput<"dataRenderers"> => {
   const [state, setState] = tapState<DataRenderersState>(() => ({
     renderers: {},
+    fallbacks: [],
   }));
 
   const setDataUI = tapCallback(
@@ -35,8 +36,23 @@ export const DataRenderers = resource((): ClientOutput<"dataRenderers"> => {
     [],
   );
 
+  const setFallbackDataUI = tapCallback((render: DataMessagePartComponent) => {
+    setState((prev) => ({
+      ...prev,
+      fallbacks: [...prev.fallbacks, render],
+    }));
+
+    return () => {
+      setState((prev) => ({
+        ...prev,
+        fallbacks: prev.fallbacks.filter((r) => r !== render),
+      }));
+    };
+  }, []);
+
   return {
     getState: () => state,
     setDataUI,
+    setFallbackDataUI,
   };
 });

package/src/react/client/Tools.ts

@@ -4,10 +4,10 @@ import {
   type ClientOutput,
   attachTransformScopes,
 } from "@assistant-ui/store";
-import { ToolsState } from "../types/scopes/tools";
+import type { ToolsState } from "../types/scopes/tools";
 import type { Tool } from "assistant-stream";
-import { type Toolkit } from "../model-context/toolbox";
-import { ToolCallMessagePartComponent } from "../types/MessagePartComponentTypes";
+import type { Toolkit } from "../model-context/toolbox";
+import type { ToolCallMessagePartComponent } from "../types/MessagePartComponentTypes";
 import { ModelContext } from "../../store";
 
 export const Tools = resource(
@@ -78,6 +78,7 @@ export const Tools = resource(
       );
 
       return () => {
+        // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
         unsubscribes.forEach((fn) => fn());
       };
     }, [toolkit, setToolUI, clientRef]);

package/src/react/model-context/makeAssistantTool.ts

@@ -1,4 +1,4 @@
-import { FC } from "react";
+import type { FC } from "react";
 import { type AssistantToolProps, useAssistantTool } from "./useAssistantTool";
 
 export type AssistantTool = FC & {

package/src/react/model-context/makeAssistantToolUI.ts

@@ -1,4 +1,4 @@
-import { FC } from "react";
+import type { FC } from "react";
 import {
   type AssistantToolUIProps,
   useAssistantToolUI,

package/src/react/model-context/useInlineRender.ts

@@ -1,5 +1,5 @@
-import { FC, useCallback, useEffect, useState } from "react";
-import { ToolCallMessagePartProps } from "../types/MessagePartComponentTypes";
+import { type FC, useCallback, useEffect, useState } from "react";
+import type { ToolCallMessagePartProps } from "../types/MessagePartComponentTypes";
 import { create } from "zustand";
 
 export const useInlineRender = <TArgs, TResult>(
@@ -17,6 +17,7 @@ export const useInlineRender = <TArgs, TResult>(
 
   return useCallback(
     function ToolUI(args) {
+      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
       const store = useToolUIStore();
       return store.toolUI(args);
     },

package/src/react/primitives/chainOfThought/ChainOfThoughtParts.tsx

@@ -87,6 +87,7 @@ export const ChainOfThoughtPrimitiveParts: FC<
   }
 
   // eslint-disable-next-line react-hooks/exhaustive-deps
+  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const messageComponents = useMemo(
     () => ({
       Reasoning: components?.Reasoning,

package/src/react/primitives/composer/ComposerAttachments.tsx

@@ -1,5 +1,11 @@
 import type { Attachment } from "../../../types/attachment";
-import { ComponentType, type FC, type ReactNode, memo, useMemo } from "react";
+import {
+  type ComponentType,
+  type FC,
+  type ReactNode,
+  memo,
+  useMemo,
+} from "react";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import { ComposerAttachmentByIndexProvider } from "../../providers/AttachmentByIndexProvider";
 

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

@@ -1,5 +1,11 @@
 import type { CompleteAttachment } from "../../../types/attachment";
-import { ComponentType, type FC, type ReactNode, memo, useMemo } from "react";
+import {
+  type ComponentType,
+  type FC,
+  type ReactNode,
+  memo,
+  useMemo,
+} from "react";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
 import { MessageAttachmentByIndexProvider } from "../../providers/AttachmentByIndexProvider";