@assistant-ui/react 0.11.0 → 0.11.2

package/src/runtimes/core/BaseThreadRuntimeCore.tsx

@@ -1,228 +0,0 @@
-import type { AppendMessage, Unsubscribe } from "../../types";
-import {
-  ExportedMessageRepository,
-  MessageRepository,
-} from "../utils/MessageRepository";
-import { DefaultThreadComposerRuntimeCore } from "../composer/DefaultThreadComposerRuntimeCore";
-import {
-  AddToolResultOptions,
-  ThreadSuggestion,
-  SubmitFeedbackOptions,
-  ThreadRuntimeCore,
-  SpeechState,
-  RuntimeCapabilities,
-  SubmittedFeedback,
-  ThreadRuntimeEventType,
-  StartRunConfig,
-  ResumeRunConfig,
-} from "../core/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 cancelRun(): 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);
-  }
-
-  private _submittedFeedback: Record<string, SubmittedFeedback> = {};
-
-  public getSubmittedFeedback(messageId: string) {
-    return this._submittedFeedback[messageId];
-  }
-
-  public submitFeedback({ messageId, type }: SubmitFeedbackOptions) {
-    const adapter = this.adapters?.feedback;
-    if (!adapter) throw new Error("Feedback adapter not configured");
-
-    const { message } = this.repository.getMessage(messageId);
-    adapter.submit({ message, type });
-
-    this._submittedFeedback[messageId] = { type };
-    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 === "model-context-update") {
-      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/runtimes/core/ComposerRuntimeCore.tsx

@@ -1,44 +0,0 @@
-import type { Attachment, PendingAttachment, Unsubscribe } from "../../types";
-import { MessageRole, RunConfig } from "../../types/AssistantTypes";
-
-export type ComposerRuntimeEventType = "send" | "attachment-add";
-
-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;
-
-  reset: () => Promise<void>;
-  clearAttachments: () => Promise<void>;
-
-  send: () => void;
-  cancel: () => void;
-
-  subscribe: (callback: () => void) => Unsubscribe;
-
-  unstable_on: (
-    event: ComposerRuntimeEventType,
-    callback: () => void,
-  ) => Unsubscribe;
-}>;
-
-export type ThreadComposerRuntimeCore = ComposerRuntimeCore &
-  Readonly<{
-    attachments: readonly PendingAttachment[];
-  }>;

package/src/runtimes/core/ThreadListRuntimeCore.tsx

@@ -1,49 +0,0 @@
-import { Unsubscribe } from "../../types";
-import { ThreadRuntimeCore } from "./ThreadRuntimeCore";
-import type { ThreadListItemStatus } from "../../api/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 threadData: 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/runtimes/core/ThreadRuntimeCore.tsx

@@ -1,125 +0,0 @@
-import { 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 edit: boolean;
-  readonly reload: boolean;
-  readonly cancel: boolean;
-  readonly unstable_copy: boolean;
-  readonly speech: 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 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 =
-  | "run-start"
-  | "run-end"
-  | "initialize"
-  | "model-context-update";
-
-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;
-      }
-    | 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;
-
-  speak: (messageId: string) => void;
-  stopSpeaking: () => void;
-
-  getSubmittedFeedback: (messageId: string) => SubmittedFeedback | undefined;
-  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.8.0.
-  //  * Please migrate to using `AssistantRuntimeCore.Provider` instead.
-  //  */
-  extras: unknown;
-
-  subscribe: (callback: () => void) => Unsubscribe;
-
-  import(repository: ExportedMessageRepository): void;
-  export(): ExportedMessageRepository;
-
-  reset(initialMessages?: readonly ThreadMessageLike[]): void;
-
-  unstable_on(event: ThreadRuntimeEventType, callback: () => void): Unsubscribe;
-}>;

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

@@ -1,98 +0,0 @@
-import { AppendMessage, ThreadMessage } from "../../types";
-import { AttachmentAdapter } from "../adapters/attachment";
-import {
-  AddToolResultOptions,
-  StartRunConfig,
-  ThreadSuggestion,
-} from "../core/ThreadRuntimeCore";
-import { FeedbackAdapter } from "../adapters/feedback/FeedbackAdapter";
-import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
-import { ThreadMessageLike } from "./ThreadMessageLike";
-import { ExportedMessageRepository } from "../utils/MessageRepository";
-
-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;
-  extras?: unknown;
-
-  setMessages?: ((messages: readonly T[]) => void) | undefined;
-  onImport?: ((messages: readonly ThreadMessage[]) => void) | undefined;
-  onNew: (message: AppendMessage) => Promise<void>;
-  onEdit?: ((message: AppendMessage) => Promise<void>) | undefined;
-  onReload?: // TODO: remove parentId in 0.8.0
-  | ((parentId: string | null, config: StartRunConfig) => Promise<void>)
-    | undefined;
-  onCancel?: (() => Promise<void>) | undefined;
-  onAddToolResult?:
-    | ((options: AddToolResultOptions) => Promise<void> | void)
-    | undefined;
-  convertMessage?: ExternalStoreMessageConverter<T> | undefined;
-  adapters?:
-    | {
-        attachments?: AttachmentAdapter | undefined;
-        speech?: SpeechSynthesisAdapter | 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>);