@assistant-ui/react 0.12.10 → 0.12.12

package/src/legacy-runtime/runtime/ComposerRuntime.ts

@@ -1,502 +1,20 @@
-import type {
-  Attachment,
-  PendingAttachment,
-} from "../../types/AttachmentTypes";
-import type {
-  ComposerRuntimeCore,
-  ComposerRuntimeEventType,
-  DictationState,
-  ThreadComposerRuntimeCore,
-} from "../runtime-cores/core/ComposerRuntimeCore";
-import type { Unsubscribe } from "../../types";
-import type { MessageRole, RunConfig } from "../../types/AssistantTypes";
-import type { QuoteInfo } from "../../types/QuoteTypes";
-import type {
-  ThreadComposerRuntimeCoreBinding,
-  EditComposerRuntimeCoreBinding,
-  ComposerRuntimeCoreBinding,
-} from "./RuntimeBindings";
-import type { ComposerRuntimePath } from "./RuntimePathTypes";
-
-import { LazyMemoizeSubject } from "./subscribable/LazyMemoizeSubject";
-import {
-  type AttachmentRuntime,
-  type AttachmentState,
-  EditComposerAttachmentRuntimeImpl,
-  ThreadComposerAttachmentRuntimeImpl,
-} from "./AttachmentRuntime";
-import { ShallowMemoizeSubject } from "./subscribable/ShallowMemoizeSubject";
-import { SKIP_UPDATE } from "./subscribable/SKIP_UPDATE";
-import { EventSubscriptionSubject } from "./subscribable/EventSubscriptionSubject";
+export type {
+  ThreadComposerState,
+  EditComposerState,
+  ComposerState,
+  ComposerRuntime,
+  ThreadComposerRuntime,
+  EditComposerRuntime,
+} from "@assistant-ui/core";
 
 export type {
+  ComposerRuntimeCoreBinding,
   ThreadComposerRuntimeCoreBinding,
   EditComposerRuntimeCoreBinding,
-  ComposerRuntimeCoreBinding,
-};
-
-type BaseComposerState = {
-  readonly canCancel: boolean;
-  readonly isEditing: boolean;
-  readonly isEmpty: boolean;
-
-  readonly text: string;
-  readonly role: MessageRole;
-  readonly attachments: readonly Attachment[];
-  readonly runConfig: RunConfig;
-
-  readonly attachmentAccept: string;
-
-  /**
-   * The current state of dictation.
-   * Undefined when dictation is not active.
-   */
-  readonly dictation: DictationState | undefined;
-
-  /**
-   * The currently quoted text, if any.
-   * Undefined when no quote is set.
-   */
-  readonly quote: QuoteInfo | undefined;
-};
-
-export type ThreadComposerState = BaseComposerState & {
-  readonly type: "thread";
-
-  readonly attachments: readonly PendingAttachment[];
-};
-
-export type EditComposerState = BaseComposerState & {
-  readonly type: "edit";
-};
-
-export type ComposerState = ThreadComposerState | EditComposerState;
-
-const EMPTY_ARRAY = Object.freeze([]);
-const EMPTY_OBJECT = Object.freeze({});
-const getThreadComposerState = (
-  runtime: ThreadComposerRuntimeCore | undefined,
-): ThreadComposerState => {
-  return Object.freeze({
-    type: "thread",
-
-    isEditing: runtime?.isEditing ?? false,
-    canCancel: runtime?.canCancel ?? false,
-    isEmpty: runtime?.isEmpty ?? true,
-
-    attachments: runtime?.attachments ?? EMPTY_ARRAY,
-    text: runtime?.text ?? "",
-    role: runtime?.role ?? "user",
-    runConfig: runtime?.runConfig ?? EMPTY_OBJECT,
-    attachmentAccept: runtime?.attachmentAccept ?? "",
-    dictation: runtime?.dictation,
-    quote: runtime?.quote,
-
-    value: runtime?.text ?? "",
-  });
-};
-
-const getEditComposerState = (
-  runtime: ComposerRuntimeCore | undefined,
-): EditComposerState => {
-  return Object.freeze({
-    type: "edit",
-
-    isEditing: runtime?.isEditing ?? false,
-    canCancel: runtime?.canCancel ?? false,
-    isEmpty: runtime?.isEmpty ?? true,
-
-    text: runtime?.text ?? "",
-    role: runtime?.role ?? "user",
-    attachments: runtime?.attachments ?? EMPTY_ARRAY,
-    runConfig: runtime?.runConfig ?? EMPTY_OBJECT,
-    attachmentAccept: runtime?.attachmentAccept ?? "",
-    dictation: runtime?.dictation,
-    quote: runtime?.quote,
-
-    value: runtime?.text ?? "",
-  });
-};
-
-export type ComposerRuntime = {
-  readonly path: ComposerRuntimePath;
-  readonly type: "edit" | "thread";
-
-  /**
-   * Get the current state of the composer. Includes any data that has been added to the composer.
-   */
-  getState(): ComposerState;
-
-  /**
-   * Given a standard js File object, add it to the composer. A composer can have multiple attachments.
-   * @param file The file to add to the composer.
-   */
-  addAttachment(file: File): Promise<void>;
-
-  /**
-   * Set the text of the composer.
-   * @param text The text to set in the composer.
-   */
-  setText(text: string): void;
-
-  /**
-   * Set the role of the composer. For instance, if you'd like a specific message to have the 'assistant' role, you can do so here.
-   * @param role The role to set in the composer.
-   */
-  setRole(role: MessageRole): void;
-
-  /**
-   * Set the run config of the composer. This is used to send custom configuration data to the model.
-   * Within your backend, you can access this data using the `runConfig` object.
-   * Example:
-   * ```ts
-   * composerRuntime.setRunConfig({
-   *   custom: { customField: "customValue" }
-   * });
-   * ```
-   * @param runConfig The run config to set in the composer.
-   */
-  setRunConfig(runConfig: RunConfig): void;
-
-  /**
-   * Reset the composer. This will clear the entire state of the composer, including all text and attachments.
-   */
-  reset(): Promise<void>;
-
-  /**
-   * Clear all attachments from the composer.
-   */
-  clearAttachments(): Promise<void>;
-
-  /**
-   * Send a message. This will send whatever text or attachments are in the composer.
-   */
-  send(): void;
-
-  /**
-   * Cancel the current run. In edit mode, this will exit edit mode.
-   */
-  cancel(): void;
-
-  /**
-   * Listens for changes to the composer state.
-   * @param callback The callback to call when the composer state changes.
-   */
-  subscribe(callback: () => void): Unsubscribe;
-
-  /**
-   * Get an attachment by index.
-   * @param idx The index of the attachment to get.
-   */
-  getAttachmentByIndex(idx: number): AttachmentRuntime;
-
-  /**
-   * Start dictation to convert voice to text input.
-   * Requires a DictationAdapter to be configured.
-   */
-  startDictation(): void;
-
-  /**
-   * Stop the current dictation session.
-   */
-  stopDictation(): void;
-
-  /**
-   * Set a quote for the next message. Pass undefined to clear.
-   * @param quote The quote info to set, or undefined to clear.
-   */
-  setQuote(quote: QuoteInfo | undefined): void;
-
-  /**
-   * @deprecated This API is still under active development and might change without notice.
-   */
-  unstable_on(
-    event: ComposerRuntimeEventType,
-    callback: () => void,
-  ): Unsubscribe;
-};
-
-export abstract class ComposerRuntimeImpl implements ComposerRuntime {
-  public get path() {
-    return this._core.path;
-  }
-
-  public abstract get type(): "edit" | "thread";
-
-  constructor(protected _core: ComposerRuntimeCoreBinding) {}
-
-  protected __internal_bindMethods() {
-    this.setText = this.setText.bind(this);
-    this.setRunConfig = this.setRunConfig.bind(this);
-    this.getState = this.getState.bind(this);
-    this.subscribe = this.subscribe.bind(this);
-    this.addAttachment = this.addAttachment.bind(this);
-    this.reset = this.reset.bind(this);
-    this.clearAttachments = this.clearAttachments.bind(this);
-    this.send = this.send.bind(this);
-    this.cancel = this.cancel.bind(this);
-    this.setRole = this.setRole.bind(this);
-    this.getAttachmentByIndex = this.getAttachmentByIndex.bind(this);
-    this.startDictation = this.startDictation.bind(this);
-    this.stopDictation = this.stopDictation.bind(this);
-    this.setQuote = this.setQuote.bind(this);
-    this.unstable_on = this.unstable_on.bind(this);
-  }
-
-  public abstract getState(): ComposerState;
-
-  public setText(text: string) {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.setText(text);
-  }
-
-  public setRunConfig(runConfig: RunConfig) {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.setRunConfig(runConfig);
-  }
-
-  public addAttachment(file: File) {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    return core.addAttachment(file);
-  }
-
-  public reset() {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    return core.reset();
-  }
-
-  public clearAttachments() {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    return core.clearAttachments();
-  }
-
-  public send() {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.send();
-  }
-
-  public cancel() {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.cancel();
-  }
-
-  public setRole(role: MessageRole) {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.setRole(role);
-  }
-
-  public startDictation() {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.startDictation();
-  }
-
-  public stopDictation() {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.stopDictation();
-  }
-
-  public setQuote(quote: QuoteInfo | undefined) {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    core.setQuote(quote);
-  }
-
-  public subscribe(callback: () => void) {
-    return this._core.subscribe(callback);
-  }
-
-  private _eventSubscriptionSubjects = new Map<
-    string,
-    EventSubscriptionSubject<ComposerRuntimeEventType>
-  >();
-
-  public unstable_on(
-    event: ComposerRuntimeEventType,
-    callback: () => void,
-  ): Unsubscribe {
-    let subject = this._eventSubscriptionSubjects.get(event);
-    if (!subject) {
-      subject = new EventSubscriptionSubject({
-        event: event,
-        binding: this._core,
-      });
-      this._eventSubscriptionSubjects.set(event, subject);
-    }
-    return subject.subscribe(callback);
-  }
-
-  public abstract getAttachmentByIndex(idx: number): AttachmentRuntime;
-}
-
-export type ThreadComposerRuntime = Omit<
-  ComposerRuntime,
-  "getState" | "getAttachmentByIndex"
-> & {
-  readonly path: ComposerRuntimePath & { composerSource: "thread" };
-  readonly type: "thread";
-  getState(): ThreadComposerState;
-
-  getAttachmentByIndex(
-    idx: number,
-  ): AttachmentRuntime & { source: "thread-composer" };
-};
-
-export class ThreadComposerRuntimeImpl
-  extends ComposerRuntimeImpl
-  implements ThreadComposerRuntime
-{
-  public override get path() {
-    return this._core.path as ComposerRuntimePath & {
-      composerSource: "thread";
-    };
-  }
-
-  public get type() {
-    return "thread" as const;
-  }
-
-  private _getState;
-
-  constructor(core: ThreadComposerRuntimeCoreBinding) {
-    const stateBinding = new LazyMemoizeSubject({
-      path: core.path,
-      getState: () => getThreadComposerState(core.getState()),
-      subscribe: (callback) => core.subscribe(callback),
-    });
-    super({
-      path: core.path,
-      getState: () => core.getState(),
-      subscribe: (callback) => stateBinding.subscribe(callback),
-    });
-    this._getState = stateBinding.getState.bind(stateBinding);
-
-    this.__internal_bindMethods();
-  }
-
-  public override getState(): ThreadComposerState {
-    return this._getState();
-  }
-
-  public getAttachmentByIndex(idx: number) {
-    return new ThreadComposerAttachmentRuntimeImpl(
-      new ShallowMemoizeSubject({
-        path: {
-          ...this.path,
-          attachmentSource: "thread-composer",
-          attachmentSelector: { type: "index", index: idx },
-          ref: `${this.path.ref}.attachments[${idx}]`,
-        },
-        getState: () => {
-          const attachments = this.getState().attachments;
-          const attachment = attachments[idx];
-          if (!attachment) return SKIP_UPDATE;
-
-          return {
-            ...attachment,
-            source: "thread-composer",
-          } satisfies AttachmentState & { source: "thread-composer" };
-        },
-        subscribe: (callback) => this._core.subscribe(callback),
-      }),
-      this._core,
-    );
-  }
-}
-
-export type EditComposerRuntime = Omit<
-  ComposerRuntime,
-  "getState" | "getAttachmentByIndex"
-> & {
-  readonly path: ComposerRuntimePath & { composerSource: "edit" };
-  readonly type: "edit";
-
-  getState(): EditComposerState;
-  beginEdit(): void;
-
-  getAttachmentByIndex(
-    idx: number,
-  ): AttachmentRuntime & { source: "edit-composer" };
-};
-
-export class EditComposerRuntimeImpl
-  extends ComposerRuntimeImpl
-  implements EditComposerRuntime
-{
-  public override get path() {
-    return this._core.path as ComposerRuntimePath & { composerSource: "edit" };
-  }
-
-  public get type() {
-    return "edit" as const;
-  }
-
-  private _getState;
-  constructor(
-    core: EditComposerRuntimeCoreBinding,
-    private _beginEdit: () => void,
-  ) {
-    const stateBinding = new LazyMemoizeSubject({
-      path: core.path,
-      getState: () => getEditComposerState(core.getState()),
-      subscribe: (callback) => core.subscribe(callback),
-    });
-
-    super({
-      path: core.path,
-      getState: () => core.getState(),
-      subscribe: (callback) => stateBinding.subscribe(callback),
-    });
-
-    this._getState = stateBinding.getState.bind(stateBinding);
-
-    this.__internal_bindMethods();
-  }
-
-  public override __internal_bindMethods() {
-    super.__internal_bindMethods();
-    this.beginEdit = this.beginEdit.bind(this);
-  }
-
-  public override getState(): EditComposerState {
-    return this._getState();
-  }
-
-  public beginEdit() {
-    this._beginEdit();
-  }
-
-  public getAttachmentByIndex(idx: number) {
-    return new EditComposerAttachmentRuntimeImpl(
-      new ShallowMemoizeSubject({
-        path: {
-          ...this.path,
-          attachmentSource: "edit-composer",
-          attachmentSelector: { type: "index", index: idx },
-          ref: `${this.path.ref}.attachments[${idx}]`,
-        },
-        getState: () => {
-          const attachments = this.getState().attachments;
-          const attachment = attachments[idx];
-          if (!attachment) return SKIP_UPDATE;
+} from "@assistant-ui/core/internal";
 
-          return {
-            ...attachment,
-            source: "edit-composer",
-          } satisfies AttachmentState & { source: "edit-composer" };
-        },
-        subscribe: (callback) => this._core.subscribe(callback),
-      }),
-      this._core,
-    );
-  }
-}
+export {
+  ComposerRuntimeImpl,
+  ThreadComposerRuntimeImpl,
+  EditComposerRuntimeImpl,
+} from "@assistant-ui/core/internal";

package/src/legacy-runtime/runtime/MessagePartRuntime.ts

@@ -1,117 +1,2 @@
-import {
-  ThreadAssistantMessagePart,
-  ThreadUserMessagePart,
-  MessagePartStatus,
-  ToolCallMessagePartStatus,
-} from "../../types/AssistantTypes";
-import { ThreadRuntimeCoreBinding } from "./ThreadRuntime";
-import type { MessageStateBinding } from "./RuntimeBindings";
-import { SubscribableWithState } from "./subscribable/Subscribable";
-import { Unsubscribe } from "../../types";
-import { MessagePartRuntimePath } from "./RuntimePathTypes";
-import { ToolResponse } from "assistant-stream";
-
-export type MessagePartState = (
-  | ThreadUserMessagePart
-  | ThreadAssistantMessagePart
-) & {
-  readonly status: MessagePartStatus | ToolCallMessagePartStatus;
-};
-
-type MessagePartSnapshotBinding = SubscribableWithState<
-  MessagePartState,
-  MessagePartRuntimePath
->;
-
-export type MessagePartRuntime = {
-  /**
-   * Add tool result to a tool call message part that has no tool result yet.
-   * This is useful when you are collecting a tool result via user input ("human tool calls").
-   */
-  addToolResult(result: any | ToolResponse<any>): void;
-
-  /**
-   * Resume a tool call that is waiting for human input with a payload.
-   * This is useful when a tool has requested human input and is waiting for a response.
-   */
-  resumeToolCall(payload: unknown): void;
-
-  readonly path: MessagePartRuntimePath;
-  getState(): MessagePartState;
-  subscribe(callback: () => void): Unsubscribe;
-};
-
-export class MessagePartRuntimeImpl implements MessagePartRuntime {
-  public get path() {
-    return this.contentBinding.path;
-  }
-
-  constructor(
-    private contentBinding: MessagePartSnapshotBinding,
-    private messageApi?: MessageStateBinding,
-    private threadApi?: ThreadRuntimeCoreBinding,
-  ) {
-    this.__internal_bindMethods();
-  }
-
-  protected __internal_bindMethods() {
-    this.addToolResult = this.addToolResult.bind(this);
-    this.resumeToolCall = this.resumeToolCall.bind(this);
-    this.getState = this.getState.bind(this);
-    this.subscribe = this.subscribe.bind(this);
-  }
-
-  public getState() {
-    return this.contentBinding.getState();
-  }
-
-  public addToolResult(result: any | ToolResponse<any>) {
-    const state = this.contentBinding.getState();
-    if (!state) throw new Error("Message part is not available");
-
-    if (state.type !== "tool-call")
-      throw new Error("Tried to add tool result to non-tool message part");
-
-    if (!this.messageApi)
-      throw new Error(
-        "Message API is not available. This is likely a bug in assistant-ui.",
-      );
-    if (!this.threadApi) throw new Error("Thread API is not available");
-
-    const message = this.messageApi.getState();
-    if (!message) throw new Error("Message is not available");
-
-    const toolName = state.toolName;
-    const toolCallId = state.toolCallId;
-
-    const response = ToolResponse.toResponse(result);
-    this.threadApi.getState().addToolResult({
-      messageId: message.id,
-      toolName,
-      toolCallId,
-      result: response.result,
-      artifact: response.artifact,
-      isError: response.isError,
-    });
-  }
-
-  public resumeToolCall(payload: unknown) {
-    const state = this.contentBinding.getState();
-    if (!state) throw new Error("Message part is not available");
-
-    if (state.type !== "tool-call")
-      throw new Error("Tried to resume tool call on non-tool message part");
-
-    if (!this.threadApi) throw new Error("Thread API is not available");
-
-    const toolCallId = state.toolCallId;
-    this.threadApi.getState().resumeToolCall({
-      toolCallId,
-      payload,
-    });
-  }
-
-  public subscribe(callback: () => void) {
-    return this.contentBinding.subscribe(callback);
-  }
-}
+export type { MessagePartState, MessagePartRuntime } from "@assistant-ui/core";
+export { MessagePartRuntimeImpl } from "@assistant-ui/core/internal";