@assistant-ui/react 0.12.10 → 0.12.12

package/src/legacy-runtime/runtime-cores/core/BaseThreadRuntimeCore.tsx

@@ -1,237 +0,0 @@
-import type { AppendMessage, ThreadMessage, Unsubscribe } from "../../../types";
-import {
-  ExportedMessageRepository,
-  MessageRepository,
-} from "../utils/MessageRepository";
-import { DefaultThreadComposerRuntimeCore } from "../composer/DefaultThreadComposerRuntimeCore";
-import {
-  AddToolResultOptions,
-  ResumeToolCallOptions,
-  ThreadSuggestion,
-  SubmitFeedbackOptions,
-  ThreadRuntimeCore,
-  SpeechState,
-  RuntimeCapabilities,
-  ThreadRuntimeEventType,
-  StartRunConfig,
-  ResumeRunConfig,
-} from "./ThreadRuntimeCore";
-import { DefaultEditComposerRuntimeCore } from "../composer/DefaultEditComposerRuntimeCore";
-import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
-import { FeedbackAdapter } from "../adapters/feedback/FeedbackAdapter";
-import { AttachmentAdapter } from "../adapters/attachment";
-import { getThreadMessageText } from "../../../utils/getThreadMessageText";
-import { ModelContextProvider } from "../../../model-context";
-import { ThreadMessageLike } from "../external-store";
-
-type BaseThreadAdapters = {
-  speech?: SpeechSynthesisAdapter | undefined;
-  feedback?: FeedbackAdapter | undefined;
-  attachments?: AttachmentAdapter | undefined;
-};
-
-export abstract class BaseThreadRuntimeCore implements ThreadRuntimeCore {
-  private _subscriptions = new Set<() => void>();
-  private _isInitialized = false;
-
-  protected readonly repository = new MessageRepository();
-  public abstract get adapters(): BaseThreadAdapters | undefined;
-  public abstract get isDisabled(): boolean;
-  public abstract get isLoading(): boolean;
-  public abstract get suggestions(): readonly ThreadSuggestion[];
-  public abstract get extras(): unknown;
-
-  public abstract get capabilities(): RuntimeCapabilities;
-  public abstract append(message: AppendMessage): void;
-  public abstract startRun(config: StartRunConfig): void;
-  public abstract resumeRun(config: ResumeRunConfig): void;
-  public abstract addToolResult(options: AddToolResultOptions): void;
-  public abstract resumeToolCall(options: ResumeToolCallOptions): void;
-  public abstract cancelRun(): void;
-  public abstract exportExternalState(): any;
-  public abstract importExternalState(state: any): void;
-  public abstract unstable_loadExternalState(state: any): void;
-
-  public get messages() {
-    return this.repository.getMessages();
-  }
-
-  public get state() {
-    let mostRecentAssistantMessage;
-    for (const message of this.messages) {
-      if (message.role === "assistant") {
-        mostRecentAssistantMessage = message;
-        break;
-      }
-    }
-
-    return mostRecentAssistantMessage?.metadata.unstable_state ?? null;
-  }
-
-  public readonly composer = new DefaultThreadComposerRuntimeCore(this);
-
-  constructor(private readonly _contextProvider: ModelContextProvider) {}
-
-  public getModelContext() {
-    return this._contextProvider.getModelContext();
-  }
-
-  private _editComposers = new Map<string, DefaultEditComposerRuntimeCore>();
-  public getEditComposer(messageId: string) {
-    return this._editComposers.get(messageId);
-  }
-  public beginEdit(messageId: string) {
-    if (this._editComposers.has(messageId))
-      throw new Error("Edit already in progress");
-
-    this._editComposers.set(
-      messageId,
-      new DefaultEditComposerRuntimeCore(
-        this,
-        () => this._editComposers.delete(messageId),
-        this.repository.getMessage(messageId),
-      ),
-    );
-    this._notifySubscribers();
-  }
-
-  public getMessageById(messageId: string) {
-    try {
-      return this.repository.getMessage(messageId);
-    } catch {
-      return undefined;
-    }
-  }
-
-  public getBranches(messageId: string): string[] {
-    return this.repository.getBranches(messageId);
-  }
-
-  public switchToBranch(branchId: string): void {
-    this.repository.switchToBranch(branchId);
-    this._notifySubscribers();
-  }
-
-  protected _notifySubscribers() {
-    for (const callback of this._subscriptions) callback();
-  }
-
-  public _notifyEventSubscribers(event: ThreadRuntimeEventType) {
-    const subscribers = this._eventSubscribers.get(event);
-    if (!subscribers) return;
-
-    for (const callback of subscribers) callback();
-  }
-
-  public subscribe(callback: () => void): Unsubscribe {
-    this._subscriptions.add(callback);
-    return () => this._subscriptions.delete(callback);
-  }
-
-  public submitFeedback({ messageId, type }: SubmitFeedbackOptions) {
-    const adapter = this.adapters?.feedback;
-    if (!adapter) throw new Error("Feedback adapter not configured");
-
-    const { message, parentId } = this.repository.getMessage(messageId);
-    adapter.submit({ message, type });
-
-    // Update the message metadata with the feedback
-    if (message.role === "assistant") {
-      const updatedMessage: ThreadMessage = {
-        ...message,
-        metadata: {
-          ...message.metadata,
-          submittedFeedback: { type },
-        },
-      };
-      this.repository.addOrUpdateMessage(parentId, updatedMessage);
-    }
-
-    this._notifySubscribers();
-  }
-
-  private _stopSpeaking: Unsubscribe | undefined;
-  public speech: SpeechState | undefined;
-
-  public speak(messageId: string) {
-    const adapter = this.adapters?.speech;
-    if (!adapter) throw new Error("Speech adapter not configured");
-
-    const { message } = this.repository.getMessage(messageId);
-
-    this._stopSpeaking?.();
-
-    const utterance = adapter.speak(getThreadMessageText(message));
-    const unsub = utterance.subscribe(() => {
-      if (utterance.status.type === "ended") {
-        this._stopSpeaking = undefined;
-        this.speech = undefined;
-      } else {
-        this.speech = { messageId, status: utterance.status };
-      }
-      this._notifySubscribers();
-    });
-
-    this.speech = { messageId, status: utterance.status };
-    this._notifySubscribers();
-
-    this._stopSpeaking = () => {
-      utterance.cancel();
-      unsub();
-      this.speech = undefined;
-      this._stopSpeaking = undefined;
-    };
-  }
-
-  public stopSpeaking() {
-    if (!this._stopSpeaking) throw new Error("No message is being spoken");
-    this._stopSpeaking();
-    this._notifySubscribers();
-  }
-
-  protected ensureInitialized() {
-    if (!this._isInitialized) {
-      this._isInitialized = true;
-      this._notifyEventSubscribers("initialize");
-    }
-  }
-
-  // TODO import()/export() on external store doesn't make much sense
-  public export() {
-    return this.repository.export();
-  }
-
-  public import(data: ExportedMessageRepository) {
-    this.ensureInitialized();
-    this.repository.clear();
-    this.repository.import(data);
-    this._notifySubscribers();
-  }
-
-  public reset(initialMessages?: readonly ThreadMessageLike[]) {
-    this.import(ExportedMessageRepository.fromArray(initialMessages ?? []));
-  }
-
-  private _eventSubscribers = new Map<
-    ThreadRuntimeEventType,
-    Set<() => void>
-  >();
-
-  public unstable_on(event: ThreadRuntimeEventType, callback: () => void) {
-    if (event === "modelContextUpdate") {
-      return this._contextProvider.subscribe?.(callback) ?? (() => {});
-    }
-
-    const subscribers = this._eventSubscribers.get(event);
-    if (!subscribers) {
-      this._eventSubscribers.set(event, new Set([callback]));
-    } else {
-      subscribers.add(callback);
-    }
-
-    return () => {
-      const subscribers = this._eventSubscribers.get(event)!;
-      subscribers.delete(callback);
-    };
-  }
-}

package/src/legacy-runtime/runtime-cores/core/ComposerRuntimeCore.tsx

@@ -1,90 +0,0 @@
-import type {
-  Attachment,
-  PendingAttachment,
-  Unsubscribe,
-} from "../../../types";
-import type { MessageRole, RunConfig } from "../../../types/AssistantTypes";
-import type { DictationAdapter } from "../adapters/speech/SpeechAdapterTypes";
-import type { QuoteInfo } from "../../../types/QuoteTypes";
-
-export type ComposerRuntimeEventType = "send" | "attachmentAdd";
-
-/**
- * State representing an active dictation session.
- */
-export type DictationState = {
-  readonly status: DictationAdapter.Status;
-  /**
-   * The current interim (partial) transcript being recognized.
-   * This is a preview of what the user is saying and may change
-   * as dictation refines its prediction.
-   *
-   * Note: By default, interim transcripts are shown directly in the composer
-   * input field (like native dictation). This property is provided for
-   * advanced customization when you want to display or style the interim
-   * transcript separately.
-   */
-  readonly transcript?: string;
-  /** Whether text input is disabled during this dictation session. */
-  readonly inputDisabled?: boolean;
-};
-
-export type ComposerRuntimeCore = Readonly<{
-  isEditing: boolean;
-
-  canCancel: boolean;
-  isEmpty: boolean;
-
-  attachments: readonly Attachment[];
-  attachmentAccept: string;
-
-  addAttachment: (file: File) => Promise<void>;
-  removeAttachment: (attachmentId: string) => Promise<void>;
-
-  text: string;
-  setText: (value: string) => void;
-
-  role: MessageRole;
-  setRole: (role: MessageRole) => void;
-
-  runConfig: RunConfig;
-  setRunConfig: (runConfig: RunConfig) => void;
-
-  quote: QuoteInfo | undefined;
-  setQuote: (quote: QuoteInfo | undefined) => void;
-
-  reset: () => Promise<void>;
-  clearAttachments: () => Promise<void>;
-
-  send: () => void;
-  cancel: () => void;
-
-  /**
-   * The current state of dictation.
-   * Undefined when dictation is not active.
-   */
-  dictation: DictationState | undefined;
-
-  /**
-   * Start dictation to convert voice to text input.
-   * Requires a DictationAdapter to be configured.
-   */
-  startDictation: () => void;
-
-  /**
-   * Stop the current dictation session.
-   */
-  stopDictation: () => void;
-
-  subscribe: (callback: () => void) => Unsubscribe;
-
-  unstable_on: (
-    event: ComposerRuntimeEventType,
-    callback: () => void,
-  ) => Unsubscribe;
-}>;
-
-export type ThreadComposerRuntimeCore = ComposerRuntimeCore &
-  Readonly<{
-    attachments: readonly PendingAttachment[];
-  }>;

package/src/legacy-runtime/runtime-cores/core/ThreadListRuntimeCore.tsx

@@ -1,49 +0,0 @@
-import { Unsubscribe } from "../../../types";
-import { ThreadRuntimeCore } from "./ThreadRuntimeCore";
-import type { ThreadListItemStatus } from "../../runtime/RuntimeBindings";
-
-export type ThreadListItemCoreState = {
-  readonly id: string;
-  readonly remoteId: string | undefined;
-  readonly externalId: string | undefined;
-
-  readonly status: ThreadListItemStatus;
-  readonly title?: string | undefined;
-
-  readonly runtime?: ThreadRuntimeCore | undefined;
-};
-
-export type ThreadListRuntimeCore = {
-  readonly isLoading: boolean;
-  mainThreadId: string;
-  newThreadId: string | undefined;
-
-  threadIds: readonly string[];
-  archivedThreadIds: readonly string[];
-
-  readonly threadItems: Readonly<Record<string, ThreadListItemCoreState>>;
-
-  getMainThreadRuntimeCore(): ThreadRuntimeCore;
-  getThreadRuntimeCore(threadId: string): ThreadRuntimeCore;
-
-  getItemById(threadId: string): ThreadListItemCoreState | undefined;
-
-  switchToThread(threadId: string): Promise<void>;
-  switchToNewThread(): Promise<void>;
-
-  getLoadThreadsPromise(): Promise<void>;
-  // getLoadArchivedThreadsPromise(): Promise<void>;
-
-  detach(threadId: string): Promise<void>;
-  rename(threadId: string, newTitle: string): Promise<void>;
-  archive(threadId: string): Promise<void>;
-  unarchive(threadId: string): Promise<void>;
-  delete(threadId: string): Promise<void>;
-
-  initialize(
-    threadId: string,
-  ): Promise<{ remoteId: string; externalId: string | undefined }>;
-  generateTitle(threadId: string): Promise<void>;
-
-  subscribe(callback: () => void): Unsubscribe;
-};

package/src/legacy-runtime/runtime-cores/core/ThreadRuntimeCore.tsx

@@ -1,154 +0,0 @@
-import type { ReadonlyJSONValue } from "assistant-stream/utils";
-import { ModelContext } from "../../../model-context";
-import { AppendMessage, ThreadMessage } from "../../../types";
-import { RunConfig } from "../../../types/AssistantTypes";
-import type { Unsubscribe } from "../../../types/Unsubscribe";
-import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
-import { ChatModelRunOptions, ChatModelRunResult } from "../local";
-import { ExportedMessageRepository } from "../utils/MessageRepository";
-import { ThreadMessageLike } from "../external-store";
-import {
-  ComposerRuntimeCore,
-  ThreadComposerRuntimeCore,
-} from "./ComposerRuntimeCore";
-
-export type RuntimeCapabilities = {
-  readonly switchToBranch: boolean;
-  readonly switchBranchDuringRun: boolean;
-  readonly edit: boolean;
-  readonly reload: boolean;
-  readonly cancel: boolean;
-  readonly unstable_copy: boolean;
-  readonly speech: boolean;
-  readonly dictation: boolean;
-  readonly attachments: boolean;
-  readonly feedback: boolean;
-};
-
-export type AddToolResultOptions = {
-  messageId: string;
-  toolName: string;
-  toolCallId: string;
-  result: ReadonlyJSONValue;
-  isError: boolean;
-  artifact?: ReadonlyJSONValue | undefined;
-};
-
-export type ResumeToolCallOptions = {
-  toolCallId: string;
-  payload: unknown;
-};
-
-export type SubmitFeedbackOptions = {
-  messageId: string;
-  type: "negative" | "positive";
-};
-
-export type ThreadSuggestion = {
-  prompt: string;
-};
-
-export type SpeechState = {
-  readonly messageId: string;
-  readonly status: SpeechSynthesisAdapter.Status;
-};
-
-export type SubmittedFeedback = {
-  readonly type: "negative" | "positive";
-};
-
-export type ThreadRuntimeEventType =
-  | "runStart"
-  | "runEnd"
-  | "initialize"
-  | "modelContextUpdate";
-
-export type StartRunConfig = {
-  parentId: string | null;
-  sourceId: string | null;
-  runConfig: RunConfig;
-};
-
-export type ResumeRunConfig = StartRunConfig & {
-  stream?: (
-    options: ChatModelRunOptions,
-  ) => AsyncGenerator<ChatModelRunResult, void, unknown>;
-};
-
-export type ThreadRuntimeCore = Readonly<{
-  getMessageById: (messageId: string) =>
-    | {
-        parentId: string | null;
-        message: ThreadMessage;
-        index: number;
-      }
-    | undefined;
-
-  getBranches: (messageId: string) => readonly string[];
-  switchToBranch: (branchId: string) => void;
-
-  append: (message: AppendMessage) => void;
-  startRun: (config: StartRunConfig) => void;
-  resumeRun: (config: ResumeRunConfig) => void;
-  cancelRun: () => void;
-
-  addToolResult: (options: AddToolResultOptions) => void;
-  resumeToolCall: (options: ResumeToolCallOptions) => void;
-
-  speak: (messageId: string) => void;
-  stopSpeaking: () => void;
-
-  submitFeedback: (feedback: SubmitFeedbackOptions) => void;
-
-  getModelContext: () => ModelContext;
-
-  composer: ThreadComposerRuntimeCore;
-  getEditComposer: (messageId: string) => ComposerRuntimeCore | undefined;
-  beginEdit: (messageId: string) => void;
-
-  speech: SpeechState | undefined;
-
-  capabilities: Readonly<RuntimeCapabilities>;
-  isDisabled: boolean;
-  isLoading: boolean;
-  messages: readonly ThreadMessage[];
-  state: ReadonlyJSONValue;
-  suggestions: readonly ThreadSuggestion[];
-
-  // TODO deprecate for a more elegant solution
-  // /**
-  //  * @deprecated This field is deprecated and will be removed in 0.12.0.
-  //  * Please migrate to using `AssistantRuntimeCore.Provider` instead.
-  //  */
-  extras: unknown;
-
-  subscribe: (callback: () => void) => Unsubscribe;
-
-  import(repository: ExportedMessageRepository): void;
-  export(): ExportedMessageRepository;
-
-  /**
-   * Export the thread state in the external store format.
-   * For AI SDK runtimes, this returns the AI SDK message format.
-   * For other runtimes, this may return different formats or throw an error.
-   * @returns The thread state in the external format (typed as any)
-   */
-  exportExternalState(): any;
-
-  /**
-   * Import thread state from the external store format.
-   * For AI SDK runtimes, this accepts AI SDK messages.
-   * For other runtimes, this may accept different formats or throw an error.
-   * @param state The thread state in the external format (typed as any)
-   */
-  importExternalState(state: any): void;
-
-  reset(initialMessages?: readonly ThreadMessageLike[]): void;
-
-  unstable_on(event: ThreadRuntimeEventType, callback: () => void): Unsubscribe;
-
-  /**
-   * @deprecated Use importExternalState instead. This method will be removed in 0.12.0.
-   */
-  unstable_loadExternalState: (state: any) => void;
-}>;

package/src/legacy-runtime/runtime-cores/external-store/ExternalStoreAdapter.tsx

@@ -1,111 +0,0 @@
-import type { AppendMessage, ThreadMessage } from "../../../types";
-import type { AttachmentAdapter } from "../adapters/attachment";
-import type {
-  AddToolResultOptions,
-  ResumeRunConfig,
-  StartRunConfig,
-  ThreadSuggestion,
-} from "../core/ThreadRuntimeCore";
-import type { FeedbackAdapter } from "../adapters/feedback/FeedbackAdapter";
-import type {
-  SpeechSynthesisAdapter,
-  DictationAdapter,
-} from "../adapters/speech/SpeechAdapterTypes";
-import type { ThreadMessageLike } from "./ThreadMessageLike";
-import type { ExportedMessageRepository } from "../utils/MessageRepository";
-import type { ReadonlyJSONValue } from "assistant-stream/utils";
-
-export type ExternalStoreThreadData<TState extends "regular" | "archived"> = {
-  status: TState;
-  id: string;
-  remoteId?: string | undefined;
-  externalId?: string | undefined;
-  title?: string | undefined;
-};
-
-export type ExternalStoreThreadListAdapter = {
-  /**
-   * @deprecated This API is still under active development and might change without notice.
-   */
-  threadId?: string | undefined;
-  isLoading?: boolean | undefined;
-  threads?: readonly ExternalStoreThreadData<"regular">[] | undefined;
-  archivedThreads?: readonly ExternalStoreThreadData<"archived">[] | undefined;
-  /**
-   * @deprecated This API is still under active development and might change without notice.
-   */
-  onSwitchToNewThread?: (() => Promise<void> | void) | undefined;
-  /**
-   * @deprecated This API is still under active development and might change without notice.
-   */
-  onSwitchToThread?: ((threadId: string) => Promise<void> | void) | undefined;
-  onRename?: (
-    threadId: string,
-    newTitle: string,
-  ) => (Promise<void> | void) | undefined;
-  onArchive?: ((threadId: string) => Promise<void> | void) | undefined;
-  onUnarchive?: ((threadId: string) => Promise<void> | void) | undefined;
-  onDelete?: ((threadId: string) => Promise<void> | void) | undefined;
-};
-
-export type ExternalStoreMessageConverter<T> = (
-  message: T,
-  idx: number,
-) => ThreadMessageLike;
-
-type ExternalStoreMessageConverterAdapter<T> = {
-  convertMessage: ExternalStoreMessageConverter<T>;
-};
-
-type ExternalStoreAdapterBase<T> = {
-  isDisabled?: boolean | undefined;
-  isRunning?: boolean | undefined;
-  isLoading?: boolean | undefined;
-  messages?: readonly T[];
-  messageRepository?: ExportedMessageRepository;
-  suggestions?: readonly ThreadSuggestion[] | undefined;
-  state?: ReadonlyJSONValue | undefined;
-  extras?: unknown;
-
-  setMessages?: ((messages: readonly T[]) => void) | undefined;
-  onImport?: ((messages: readonly ThreadMessage[]) => void) | undefined;
-  onExportExternalState?: (() => any) | undefined;
-  onLoadExternalState?: ((state: any) => void) | undefined;
-  onNew: (message: AppendMessage) => Promise<void>;
-  onEdit?: ((message: AppendMessage) => Promise<void>) | undefined;
-  onReload?: // TODO: remove parentId in 0.12.0
-    | ((parentId: string | null, config: StartRunConfig) => Promise<void>)
-    | undefined;
-  onResume?: ((config: ResumeRunConfig) => Promise<void>) | undefined;
-  onCancel?: (() => Promise<void>) | undefined;
-  onAddToolResult?:
-    | ((options: AddToolResultOptions) => Promise<void> | void)
-    | undefined;
-  onResumeToolCall?:
-    | ((options: { toolCallId: string; payload: unknown }) => void)
-    | undefined;
-  convertMessage?: ExternalStoreMessageConverter<T> | undefined;
-  adapters?:
-    | {
-        attachments?: AttachmentAdapter | undefined;
-        speech?: SpeechSynthesisAdapter | undefined;
-        dictation?: DictationAdapter | undefined;
-        feedback?: FeedbackAdapter | undefined;
-        /**
-         * @deprecated This API is still under active development and might change without notice.
-         */
-        threadList?: ExternalStoreThreadListAdapter | undefined;
-      }
-    | undefined;
-  unstable_capabilities?:
-    | {
-        copy?: boolean | undefined;
-      }
-    | undefined;
-};
-
-export type ExternalStoreAdapter<T = ThreadMessage> =
-  ExternalStoreAdapterBase<T> &
-    (T extends ThreadMessage
-      ? object
-      : ExternalStoreMessageConverterAdapter<T>);

package/src/legacy-runtime/runtime-cores/external-store/ExternalStoreRuntimeCore.tsx

@@ -1,26 +0,0 @@
-import { BaseAssistantRuntimeCore } from "../core/BaseAssistantRuntimeCore";
-import { ExternalStoreThreadListRuntimeCore } from "./ExternalStoreThreadListRuntimeCore";
-import { ExternalStoreAdapter } from "./ExternalStoreAdapter";
-import { ExternalStoreThreadRuntimeCore } from "./ExternalStoreThreadRuntimeCore";
-
-const getThreadListAdapter = (store: ExternalStoreAdapter<any>) => {
-  return store.adapters?.threadList ?? {};
-};
-
-export class ExternalStoreRuntimeCore extends BaseAssistantRuntimeCore {
-  public readonly threads;
-
-  constructor(adapter: ExternalStoreAdapter<any>) {
-    super();
-    this.threads = new ExternalStoreThreadListRuntimeCore(
-      getThreadListAdapter(adapter),
-      () => new ExternalStoreThreadRuntimeCore(this._contextProvider, adapter),
-    );
-  }
-
-  public setAdapter(adapter: ExternalStoreAdapter<any>) {
-    // Update the thread list adapter and propagate store changes to the main thread
-    this.threads.__internal_setAdapter(getThreadListAdapter(adapter));
-    this.threads.getMainThreadRuntimeCore().__internal_setAdapter(adapter);
-  }
-}