@assistant-ui/react 0.11.0 → 0.11.2

package/src/runtimes/external-store/ExternalStoreThreadListRuntimeCore.tsx

@@ -1,233 +0,0 @@
-import type { Unsubscribe } from "../../types";
-import { ExternalStoreThreadRuntimeCore } from "./ExternalStoreThreadRuntimeCore";
-import {
-  ThreadListItemCoreState,
-  ThreadListRuntimeCore,
-} from "../core/ThreadListRuntimeCore";
-import { ExternalStoreThreadListAdapter } from "./ExternalStoreAdapter";
-
-export type ExternalStoreThreadFactory = () => ExternalStoreThreadRuntimeCore;
-
-const EMPTY_ARRAY = Object.freeze([]);
-const DEFAULT_THREAD_ID = "DEFAULT_THREAD_ID";
-const DEFAULT_THREADS = Object.freeze([DEFAULT_THREAD_ID]);
-const DEFAULT_THREAD = Object.freeze({
-  id: DEFAULT_THREAD_ID,
-  remoteId: undefined,
-  externalId: undefined,
-  status: "regular",
-});
-const RESOLVED_PROMISE = Promise.resolve();
-const DEFAULT_THREAD_DATA = Object.freeze({
-  [DEFAULT_THREAD_ID]: DEFAULT_THREAD,
-});
-
-export class ExternalStoreThreadListRuntimeCore
-  implements ThreadListRuntimeCore
-{
-  private _mainThreadId: string = DEFAULT_THREAD_ID;
-  private _threads: readonly string[] = DEFAULT_THREADS;
-  private _archivedThreads: readonly string[] = EMPTY_ARRAY;
-  private _threadData: Readonly<Record<string, ThreadListItemCoreState>> =
-    DEFAULT_THREAD_DATA;
-
-  public get isLoading() {
-    return this.adapter.isLoading ?? false;
-  }
-
-  public get newThreadId() {
-    return undefined;
-  }
-
-  public get threadIds() {
-    return this._threads;
-  }
-
-  public get archivedThreadIds() {
-    return this._archivedThreads;
-  }
-
-  public get threadData() {
-    return this._threadData;
-  }
-
-  public getLoadThreadsPromise() {
-    return RESOLVED_PROMISE;
-  }
-
-  private _mainThread: ExternalStoreThreadRuntimeCore;
-
-  public get mainThreadId() {
-    return this._mainThreadId;
-  }
-
-  constructor(
-    private adapter: ExternalStoreThreadListAdapter = {},
-    private threadFactory: ExternalStoreThreadFactory,
-  ) {
-    this._mainThread = this.threadFactory();
-    this.__internal_setAdapter(adapter, true);
-  }
-
-  public getMainThreadRuntimeCore() {
-    return this._mainThread;
-  }
-
-  public getThreadRuntimeCore(): never {
-    throw new Error("Method not implemented.");
-  }
-
-  public getItemById(threadId: string) {
-    for (const thread of this.adapter.threads ?? []) {
-      if (thread.id === threadId) return thread as any;
-    }
-    for (const thread of this.adapter.archivedThreads ?? []) {
-      if (thread.id === threadId) return thread as any;
-    }
-    if (threadId === DEFAULT_THREAD_ID) return DEFAULT_THREAD;
-    return undefined;
-  }
-
-  public __internal_setAdapter(
-    adapter: ExternalStoreThreadListAdapter,
-    initialLoad = false,
-  ) {
-    const previousAdapter = this.adapter;
-    this.adapter = adapter;
-
-    const newThreadId = adapter.threadId ?? DEFAULT_THREAD_ID;
-    const newThreads = adapter.threads ?? EMPTY_ARRAY;
-    const newArchivedThreads = adapter.archivedThreads ?? EMPTY_ARRAY;
-
-    const previousThreadId = previousAdapter.threadId ?? DEFAULT_THREAD_ID;
-    const previousThreads = previousAdapter.threads ?? EMPTY_ARRAY;
-    const previousArchivedThreads =
-      previousAdapter.archivedThreads ?? EMPTY_ARRAY;
-
-    if (
-      !initialLoad &&
-      previousThreadId === newThreadId &&
-      previousThreads === newThreads &&
-      previousArchivedThreads === newArchivedThreads
-    ) {
-      return;
-    }
-
-    this._threadData = {
-      ...DEFAULT_THREAD_DATA,
-      ...Object.fromEntries(
-        adapter.threads?.map((t) => [
-          t.id,
-          {
-            ...t,
-            remoteId: t.remoteId,
-            externalId: t.externalId,
-            status: "regular",
-          },
-        ]) ?? [],
-      ),
-      ...Object.fromEntries(
-        adapter.archivedThreads?.map((t) => [
-          t.id,
-          {
-            ...t,
-            remoteId: t.remoteId,
-            externalId: t.externalId,
-            status: "archived",
-          },
-        ]) ?? [],
-      ),
-    };
-
-    if (previousThreads !== newThreads) {
-      this._threads = this.adapter.threads?.map((t) => t.id) ?? EMPTY_ARRAY;
-    }
-
-    if (previousArchivedThreads !== newArchivedThreads) {
-      this._archivedThreads =
-        this.adapter.archivedThreads?.map((t) => t.id) ?? EMPTY_ARRAY;
-    }
-
-    if (previousThreadId !== newThreadId) {
-      this._mainThreadId = newThreadId;
-      this._mainThread = this.threadFactory();
-    }
-
-    this._notifySubscribers();
-  }
-
-  public async switchToThread(threadId: string): Promise<void> {
-    if (this._mainThreadId === threadId) return;
-    const onSwitchToThread = this.adapter.onSwitchToThread;
-    if (!onSwitchToThread)
-      throw new Error(
-        "External store adapter does not support switching to thread",
-      );
-    onSwitchToThread(threadId);
-  }
-
-  public async switchToNewThread(): Promise<void> {
-    const onSwitchToNewThread = this.adapter.onSwitchToNewThread;
-    if (!onSwitchToNewThread)
-      throw new Error(
-        "External store adapter does not support switching to new thread",
-      );
-
-    onSwitchToNewThread();
-  }
-
-  public async rename(threadId: string, newTitle: string): Promise<void> {
-    const onRename = this.adapter.onRename;
-    if (!onRename)
-      throw new Error("External store adapter does not support renaming");
-
-    onRename(threadId, newTitle);
-  }
-
-  public async detach(): Promise<void> {
-    // no-op
-  }
-
-  public async archive(threadId: string): Promise<void> {
-    const onArchive = this.adapter.onArchive;
-    if (!onArchive)
-      throw new Error("External store adapter does not support archiving");
-
-    onArchive(threadId);
-  }
-
-  public async unarchive(threadId: string): Promise<void> {
-    const onUnarchive = this.adapter.onUnarchive;
-    if (!onUnarchive)
-      throw new Error("External store adapter does not support unarchiving");
-
-    onUnarchive(threadId);
-  }
-
-  public async delete(threadId: string): Promise<void> {
-    const onDelete = this.adapter.onDelete;
-    if (!onDelete)
-      throw new Error("External store adapter does not support deleting");
-
-    onDelete(threadId);
-  }
-
-  public initialize(): never {
-    throw new Error("Method not implemented.");
-  }
-
-  public generateTitle(): never {
-    throw new Error("Method not implemented.");
-  }
-
-  private _subscriptions = new Set<() => void>();
-
-  public subscribe(callback: () => void): Unsubscribe {
-    this._subscriptions.add(callback);
-    return () => this._subscriptions.delete(callback);
-  }
-
-  private _notifySubscribers() {
-    for (const callback of this._subscriptions) callback();
-  }
-}

package/src/runtimes/external-store/ExternalStoreThreadRuntimeCore.tsx

@@ -1,317 +0,0 @@
-import {
-  AddToolResultOptions,
-  StartRunConfig,
-  ThreadSuggestion,
-} from "../core/ThreadRuntimeCore";
-
-import { AppendMessage, ThreadMessage } from "../../types";
-import { ExternalStoreAdapter } from "./ExternalStoreAdapter";
-import {
-  getExternalStoreMessage,
-  symbolInnerMessage,
-} from "./getExternalStoreMessage";
-import { ThreadMessageConverter } from "./ThreadMessageConverter";
-import { getAutoStatus, isAutoStatus } from "./auto-status";
-import { fromThreadMessageLike, ThreadMessageLike } from "./ThreadMessageLike";
-import { getThreadMessageText } from "../../utils/getThreadMessageText";
-import {
-  RuntimeCapabilities,
-  ThreadRuntimeCore,
-} from "../core/ThreadRuntimeCore";
-import { BaseThreadRuntimeCore } from "../core/BaseThreadRuntimeCore";
-import { ModelContextProvider } from "../../model-context";
-import {
-  ExportedMessageRepository,
-  MessageRepository,
-} from "../utils/MessageRepository";
-
-const EMPTY_ARRAY = Object.freeze([]);
-
-export const hasUpcomingMessage = (
-  isRunning: boolean,
-  messages: readonly ThreadMessage[],
-) => {
-  return isRunning && messages[messages.length - 1]?.role !== "assistant";
-};
-
-export class ExternalStoreThreadRuntimeCore
-  extends BaseThreadRuntimeCore
-  implements ThreadRuntimeCore
-{
-  private _assistantOptimisticId: string | null = null;
-
-  private _capabilities: RuntimeCapabilities = {
-    switchToBranch: false,
-    edit: false,
-    reload: false,
-    cancel: false,
-    unstable_copy: false,
-    speech: false,
-    attachments: false,
-    feedback: false,
-  };
-
-  public get capabilities() {
-    return this._capabilities;
-  }
-
-  private _messages!: readonly ThreadMessage[];
-  public isDisabled!: boolean;
-  public get isLoading() {
-    return this._store.isLoading ?? false;
-  }
-
-  public override get messages() {
-    return this._messages;
-  }
-
-  public get adapters() {
-    return this._store.adapters;
-  }
-
-  public suggestions: readonly ThreadSuggestion[] = [];
-  public extras: unknown = undefined;
-
-  private _converter = new ThreadMessageConverter();
-
-  private _store!: ExternalStoreAdapter<any>;
-
-  public override beginEdit(messageId: string) {
-    if (!this._store.onEdit)
-      throw new Error("Runtime does not support editing.");
-
-    super.beginEdit(messageId);
-  }
-
-  constructor(
-    contextProvider: ModelContextProvider,
-    store: ExternalStoreAdapter<any>,
-  ) {
-    super(contextProvider);
-    this.__internal_setAdapter(store);
-  }
-
-  public __internal_setAdapter(store: ExternalStoreAdapter<any>) {
-    if (this._store === store) return;
-
-    const isRunning = store.isRunning ?? false;
-    this.isDisabled = store.isDisabled ?? false;
-
-    const oldStore = this._store as ExternalStoreAdapter<any> | undefined;
-    this._store = store;
-    this.extras = store.extras;
-    this.suggestions = store.suggestions ?? EMPTY_ARRAY;
-    this._capabilities = {
-      switchToBranch: this._store.setMessages !== undefined,
-      edit: this._store.onEdit !== undefined,
-      reload: this._store.onReload !== undefined,
-      cancel: this._store.onCancel !== undefined,
-      speech: this._store.adapters?.speech !== undefined,
-      unstable_copy: this._store.unstable_capabilities?.copy !== false, // default true
-      attachments: !!this._store.adapters?.attachments,
-      feedback: !!this._store.adapters?.feedback,
-    };
-
-    let messages: readonly ThreadMessage[];
-
-    if (store.messageRepository) {
-      // Handle messageRepository
-      if (
-        oldStore &&
-        oldStore.isRunning === store.isRunning &&
-        oldStore.messageRepository === store.messageRepository
-      ) {
-        this._notifySubscribers();
-        return;
-      }
-
-      // Clear and import the message repository
-      this.repository.clear();
-      this._assistantOptimisticId = null;
-      this.repository.import(store.messageRepository);
-
-      messages = this.repository.getMessages();
-    } else if (store.messages) {
-      // Handle messages array
-
-      if (oldStore) {
-        // flush the converter cache when the convertMessage prop changes
-        if (oldStore.convertMessage !== store.convertMessage) {
-          this._converter = new ThreadMessageConverter();
-        } else if (
-          oldStore.isRunning === store.isRunning &&
-          oldStore.messages === store.messages
-        ) {
-          this._notifySubscribers();
-          // no conversion update
-          return;
-        }
-      }
-
-      messages = !store.convertMessage
-        ? store.messages
-        : this._converter.convertMessages(store.messages, (cache, m, idx) => {
-            if (!store.convertMessage) return m;
-
-            const isLast = idx === store.messages!.length - 1;
-            const autoStatus = getAutoStatus(isLast, isRunning, false);
-
-            if (
-              cache &&
-              (cache.role !== "assistant" ||
-                !isAutoStatus(cache.status) ||
-                cache.status === autoStatus)
-            )
-              return cache;
-
-            const messageLike = store.convertMessage(m, idx);
-            const newMessage = fromThreadMessageLike(
-              messageLike,
-              idx.toString(),
-              autoStatus,
-            );
-            (newMessage as any)[symbolInnerMessage] = m;
-            return newMessage;
-          });
-
-      for (let i = 0; i < messages.length; i++) {
-        const message = messages[i]!;
-        const parent = messages[i - 1];
-        this.repository.addOrUpdateMessage(parent?.id ?? null, message);
-      }
-    } else {
-      throw new Error(
-        "ExternalStoreAdapter must provide either 'messages' or 'messageRepository'",
-      );
-    }
-
-    // Common logic for both paths
-    if (messages.length > 0) this.ensureInitialized();
-
-    if ((oldStore?.isRunning ?? false) !== (store.isRunning ?? false)) {
-      if (store.isRunning) {
-        this._notifyEventSubscribers("run-start");
-      } else {
-        this._notifyEventSubscribers("run-end");
-      }
-    }
-
-    if (this._assistantOptimisticId) {
-      this.repository.deleteMessage(this._assistantOptimisticId);
-      this._assistantOptimisticId = null;
-    }
-
-    if (hasUpcomingMessage(isRunning, messages)) {
-      this._assistantOptimisticId = this.repository.appendOptimisticMessage(
-        messages.at(-1)?.id ?? null,
-        {
-          role: "assistant",
-          content: [],
-        },
-      );
-    }
-
-    this.repository.resetHead(
-      this._assistantOptimisticId ?? messages.at(-1)?.id ?? null,
-    );
-
-    this._messages = this.repository.getMessages();
-    this._notifySubscribers();
-  }
-
-  public override switchToBranch(branchId: string): void {
-    if (!this._store.setMessages)
-      throw new Error("Runtime does not support switching branches.");
-
-    this.repository.switchToBranch(branchId);
-    this.updateMessages(this.repository.getMessages());
-  }
-
-  public async append(message: AppendMessage): Promise<void> {
-    if (message.parentId !== (this.messages.at(-1)?.id ?? null)) {
-      if (!this._store.onEdit)
-        throw new Error("Runtime does not support editing messages.");
-      await this._store.onEdit(message);
-    } else {
-      await this._store.onNew(message);
-    }
-  }
-
-  public async startRun(config: StartRunConfig): Promise<void> {
-    if (!this._store.onReload)
-      throw new Error("Runtime does not support reloading messages.");
-
-    await this._store.onReload(config.parentId, config);
-  }
-
-  public async resumeRun(): Promise<void> {
-    throw new Error("Runtime does not support resuming runs.");
-  }
-
-  public cancelRun(): void {
-    if (!this._store.onCancel)
-      throw new Error("Runtime does not support cancelling runs.");
-
-    this._store.onCancel();
-
-    if (this._assistantOptimisticId) {
-      this.repository.deleteMessage(this._assistantOptimisticId);
-      this._assistantOptimisticId = null;
-    }
-
-    let messages = this.repository.getMessages();
-    const previousMessage = messages[messages.length - 1];
-    if (
-      previousMessage?.role === "user" &&
-      previousMessage.id === messages.at(-1)?.id // ensure the previous message is a leaf node
-    ) {
-      this.repository.deleteMessage(previousMessage.id);
-      if (!this.composer.text.trim()) {
-        this.composer.setText(getThreadMessageText(previousMessage));
-      }
-
-      messages = this.repository.getMessages();
-    } else {
-      this._notifySubscribers();
-    }
-
-    // resync messages (for reloading, to restore the previous branch)
-    setTimeout(() => {
-      this.updateMessages(messages);
-    }, 0);
-  }
-
-  public addToolResult(options: AddToolResultOptions) {
-    if (!this._store.onAddToolResult && !this._store.onAddToolResult)
-      throw new Error("Runtime does not support tool results.");
-    this._store.onAddToolResult?.(options);
-  }
-
-  public override reset(initialMessages?: readonly ThreadMessageLike[]) {
-    const repo = new MessageRepository();
-    repo.import(ExportedMessageRepository.fromArray(initialMessages ?? []));
-    this.updateMessages(repo.getMessages());
-  }
-
-  public override import(data: ExportedMessageRepository) {
-    this._assistantOptimisticId = null;
-
-    super.import(data);
-
-    if (this._store.onImport) {
-      this._store.onImport(this.repository.getMessages());
-    }
-  }
-
-  private updateMessages = (messages: readonly ThreadMessage[]) => {
-    const hasConverter = this._store.convertMessage !== undefined;
-    if (hasConverter) {
-      this._store.setMessages?.(
-        messages.flatMap(getExternalStoreMessage).filter((m) => m != null),
-      );
-    } else {
-      // TODO mark this as readonly in v0.8.0
-      this._store.setMessages?.(messages as ThreadMessage[]);
-    }
-  };
-}

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

@@ -1,23 +0,0 @@
-import { ThreadMessage } from "../../types";
-
-export type ConverterCallback<TIn> = (
-  cache: ThreadMessage | undefined,
-  message: TIn,
-  idx: number,
-) => ThreadMessage;
-
-export class ThreadMessageConverter {
-  private readonly cache = new WeakMap<WeakKey, ThreadMessage>();
-
-  convertMessages<TIn extends WeakKey>(
-    messages: readonly TIn[],
-    converter: ConverterCallback<TIn>,
-  ): ThreadMessage[] {
-    return messages.map((m, idx) => {
-      const cached = this.cache.get(m);
-      const newMessage = converter(cached, m, idx);
-      this.cache.set(m, newMessage);
-      return newMessage;
-    });
-  }
-}