@assistant-ui/react 0.12.10 → 0.12.11

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

@@ -1,227 +0,0 @@
-import { parsePartialJsonObject } from "assistant-stream/utils";
-import { generateId } from "../../../utils/idUtils";
-import {
-  MessageStatus,
-  TextMessagePart,
-  ImageMessagePart,
-  ThreadMessage,
-  ThreadAssistantMessagePart,
-  ThreadAssistantMessage,
-  ThreadUserMessagePart,
-  ThreadUserMessage,
-  ThreadSystemMessage,
-  CompleteAttachment,
-  FileMessagePart,
-  DataMessagePart,
-  Unstable_AudioMessagePart,
-} from "../../../types";
-import {
-  ReasoningMessagePart,
-  SourceMessagePart,
-  ThreadStep,
-} from "../../../types/AssistantTypes";
-import { ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
-
-export type ThreadMessageLike = {
-  readonly role: "assistant" | "user" | "system";
-  readonly content:
-    | string
-    | readonly (
-        | TextMessagePart
-        | ReasoningMessagePart
-        | SourceMessagePart
-        | ImageMessagePart
-        | FileMessagePart
-        | DataMessagePart
-        | Unstable_AudioMessagePart
-        | {
-            readonly type: "tool-call";
-            readonly toolCallId?: string;
-            readonly toolName: string;
-            readonly args?: ReadonlyJSONObject;
-            readonly argsText?: string;
-            readonly artifact?: any;
-            readonly result?: any | undefined;
-            readonly isError?: boolean | undefined;
-            readonly parentId?: string | undefined;
-            readonly messages?: readonly ThreadMessage[] | undefined;
-          }
-      )[];
-  readonly id?: string | undefined;
-  readonly createdAt?: Date | undefined;
-  readonly status?: MessageStatus | undefined;
-  readonly attachments?: readonly CompleteAttachment[] | undefined;
-  readonly metadata?:
-    | {
-        readonly unstable_state?: ReadonlyJSONValue;
-        readonly unstable_annotations?:
-          | readonly ReadonlyJSONValue[]
-          | undefined;
-        readonly unstable_data?: readonly ReadonlyJSONValue[] | undefined;
-        readonly steps?: readonly ThreadStep[] | undefined;
-        readonly submittedFeedback?: { readonly type: "positive" | "negative" };
-        readonly custom?: Record<string, unknown> | undefined;
-      }
-    | undefined;
-};
-
-export const fromThreadMessageLike = (
-  like: ThreadMessageLike,
-  fallbackId: string,
-  fallbackStatus: MessageStatus,
-): ThreadMessage => {
-  const { role, id, createdAt, attachments, status, metadata } = like;
-  const common = {
-    id: id ?? fallbackId,
-    createdAt: createdAt ?? new Date(),
-  };
-
-  const content =
-    typeof like.content === "string"
-      ? [{ type: "text" as const, text: like.content }]
-      : like.content;
-
-  const sanitizeImageContent = ({
-    image,
-    ...rest
-  }: ImageMessagePart): ImageMessagePart | null => {
-    const match = image.match(
-      /^data:image\/(png|jpeg|jpg|gif|webp);base64,(.*)$/,
-    );
-    if (match) {
-      return { ...rest, image };
-    }
-    console.warn(`Invalid image data format detected`);
-    return null;
-  };
-
-  if (role !== "user" && attachments?.length)
-    throw new Error("attachments are only supported for user messages");
-
-  if (role !== "assistant" && status)
-    throw new Error("status is only supported for assistant messages");
-
-  if (role !== "assistant" && metadata?.steps)
-    throw new Error("metadata.steps is only supported for assistant messages");
-
-  switch (role) {
-    case "assistant":
-      return {
-        ...common,
-        role,
-        content: content
-          .map((part): ThreadAssistantMessagePart | null => {
-            const type = part.type;
-            switch (type) {
-              case "text":
-              case "reasoning":
-                if (part.text.trim().length === 0) return null;
-                return part;
-
-              case "file":
-              case "source":
-                return part;
-
-              case "image":
-                return sanitizeImageContent(part);
-
-              case "data":
-                return part;
-
-              case "tool-call": {
-                const { parentId, messages, ...basePart } = part;
-                const commonProps = {
-                  ...basePart,
-                  toolCallId: part.toolCallId ?? `tool-${generateId()}`,
-                  ...(parentId !== undefined && { parentId }),
-                  ...(messages !== undefined && { messages }),
-                };
-
-                if (part.args) {
-                  return {
-                    ...commonProps,
-                    args: part.args,
-                    argsText: part.argsText ?? JSON.stringify(part.args),
-                  };
-                }
-                return {
-                  ...commonProps,
-                  args: parsePartialJsonObject(part.argsText ?? "") ?? {},
-                  argsText: part.argsText ?? "",
-                };
-              }
-
-              default: {
-                const unhandledType: "audio" = type;
-                throw new Error(
-                  `Unsupported assistant message part type: ${unhandledType}`,
-                );
-              }
-            }
-          })
-          .filter((c) => !!c),
-        status: status ?? fallbackStatus,
-        metadata: {
-          unstable_state: metadata?.unstable_state ?? null,
-          unstable_annotations: metadata?.unstable_annotations ?? [],
-          unstable_data: metadata?.unstable_data ?? [],
-          custom: metadata?.custom ?? {},
-          steps: metadata?.steps ?? [],
-          ...(metadata?.submittedFeedback && {
-            submittedFeedback: metadata.submittedFeedback,
-          }),
-        },
-      } satisfies ThreadAssistantMessage;
-
-    case "user":
-      return {
-        ...common,
-        role,
-        content: content.map((part): ThreadUserMessagePart => {
-          const type = part.type;
-          switch (type) {
-            case "text":
-            case "image":
-            case "audio":
-            case "file":
-              return part;
-
-            default: {
-              const unhandledType:
-                | "tool-call"
-                | "reasoning"
-                | "source"
-                | "data" = type;
-              throw new Error(
-                `Unsupported user message part type: ${unhandledType}`,
-              );
-            }
-          }
-        }),
-        attachments: attachments ?? [],
-        metadata: {
-          custom: metadata?.custom ?? {},
-        },
-      } satisfies ThreadUserMessage;
-
-    case "system":
-      if (content.length !== 1 || content[0]!.type !== "text")
-        throw new Error(
-          "System messages must have exactly one text message part.",
-        );
-
-      return {
-        ...common,
-        role,
-        content: content as [TextMessagePart],
-        metadata: {
-          custom: metadata?.custom ?? {},
-        },
-      } satisfies ThreadSystemMessage;
-
-    default: {
-      const unsupportedRole: never = role;
-      throw new Error(`Unknown message role: ${unsupportedRole}`);
-    }
-  }
-};

package/src/legacy-runtime/runtime-cores/external-store/auto-status.tsx

@@ -1,67 +0,0 @@
-import { ReadonlyJSONValue } from "assistant-stream/utils";
-import { MessageStatus } from "../../../types";
-
-const symbolAutoStatus = Symbol("autoStatus");
-
-const AUTO_STATUS_RUNNING = Object.freeze(
-  Object.assign({ type: "running" as const }, { [symbolAutoStatus]: true }),
-);
-const AUTO_STATUS_COMPLETE = Object.freeze(
-  Object.assign(
-    {
-      type: "complete" as const,
-      reason: "unknown" as const,
-    },
-    { [symbolAutoStatus]: true },
-  ),
-);
-
-const AUTO_STATUS_PENDING = Object.freeze(
-  Object.assign(
-    {
-      type: "requires-action" as const,
-      reason: "tool-calls" as const,
-    },
-    { [symbolAutoStatus]: true },
-  ),
-);
-
-const AUTO_STATUS_INTERRUPT = Object.freeze(
-  Object.assign(
-    {
-      type: "requires-action" as const,
-      reason: "interrupt" as const,
-    },
-    { [symbolAutoStatus]: true },
-  ),
-);
-
-export const isAutoStatus = (status: MessageStatus) =>
-  (status as any)[symbolAutoStatus] === true;
-
-export const getAutoStatus = (
-  isLast: boolean,
-  isRunning: boolean,
-  hasInterruptedToolCalls: boolean,
-  hasPendingToolCalls: boolean,
-  error?: ReadonlyJSONValue,
-): MessageStatus => {
-  if (isLast && error) {
-    return Object.assign(
-      {
-        type: "incomplete" as const,
-        reason: "error" as const,
-        error: error,
-      },
-      { [symbolAutoStatus]: true },
-    );
-  }
-
-  return isLast && isRunning
-    ? AUTO_STATUS_RUNNING
-    : hasInterruptedToolCalls
-      ? AUTO_STATUS_INTERRUPT
-      : hasPendingToolCalls
-        ? AUTO_STATUS_PENDING
-        : AUTO_STATUS_COMPLETE;
-};

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

@@ -1,37 +0,0 @@
-import { ThreadState } from "../../runtime";
-import { ThreadMessage } from "../../../types";
-
-export const symbolInnerMessage = Symbol("innerMessage");
-const symbolInnerMessages = Symbol("innerMessages");
-
-type WithInnerMessages<T> = {
-  [symbolInnerMessage]?: T | T[];
-  [symbolInnerMessages]?: T[];
-};
-
-/**
- * @deprecated Use `getExternalStoreMessages` (plural) instead. This function will be removed in 0.12.0.
- */
-export const getExternalStoreMessage = <T,>(input: ThreadMessage) => {
-  const withInnerMessages = input as WithInnerMessages<T>;
-  return withInnerMessages[symbolInnerMessage];
-};
-
-const EMPTY_ARRAY: never[] = [];
-
-export const getExternalStoreMessages = <T,>(
-  input: ThreadState | ThreadMessage | ThreadMessage["content"][number],
-) => {
-  // TODO temp until 0.12.0 (migrate useExternalStoreRuntime to always set an array)
-
-  const container = (
-    "messages" in input ? input.messages : input
-  ) as WithInnerMessages<T>;
-  const value = container[symbolInnerMessages] || container[symbolInnerMessage];
-  if (!value) return EMPTY_ARRAY;
-  if (Array.isArray(value)) {
-    return value;
-  }
-  container[symbolInnerMessages] = [value];
-  return container[symbolInnerMessages];
-};

package/src/legacy-runtime/runtime-cores/local/ChatModelAdapter.tsx

@@ -1,64 +0,0 @@
-import type {
-  FileMessagePart,
-  MessageStatus,
-  ReasoningMessagePart,
-  RunConfig,
-  SourceMessagePart,
-  TextMessagePart,
-  ThreadAssistantMessagePart,
-  ThreadMessage,
-  ThreadStep,
-  ToolCallMessagePart,
-} from "../../../types/AssistantTypes";
-import type { ModelContext } from "../../../model-context/ModelContextTypes";
-import { ReadonlyJSONValue } from "assistant-stream/utils";
-
-export type ChatModelRunUpdate = {
-  readonly content: readonly ThreadAssistantMessagePart[];
-  readonly metadata?: Record<string, unknown>;
-};
-
-export type ChatModelRunResult = {
-  readonly content?: readonly ThreadAssistantMessagePart[] | undefined;
-  readonly status?: MessageStatus | undefined;
-  readonly metadata?: {
-    readonly unstable_state?: ReadonlyJSONValue;
-    readonly unstable_annotations?: readonly ReadonlyJSONValue[] | undefined;
-    readonly unstable_data?: readonly ReadonlyJSONValue[] | undefined;
-    readonly steps?: readonly ThreadStep[] | undefined;
-    readonly custom?: Record<string, unknown> | undefined;
-  };
-};
-
-export type CoreChatModelRunResult = Omit<ChatModelRunResult, "content"> & {
-  readonly content: readonly (
-    | TextMessagePart
-    | ReasoningMessagePart
-    | ToolCallMessagePart
-    | SourceMessagePart
-    | FileMessagePart
-  )[];
-};
-
-export type ChatModelRunOptions = {
-  readonly messages: readonly ThreadMessage[];
-  readonly runConfig: RunConfig;
-  readonly abortSignal: AbortSignal;
-  readonly context: ModelContext;
-
-  /**
-   * @deprecated This field was renamed to `context`.
-   */
-  readonly config: ModelContext;
-
-  readonly unstable_assistantMessageId?: string | undefined;
-  readonly unstable_threadId?: string | undefined;
-  readonly unstable_parentId?: string | null | undefined;
-  unstable_getMessage(): ThreadMessage;
-};
-
-export type ChatModelAdapter = {
-  run(
-    options: ChatModelRunOptions,
-  ): Promise<ChatModelRunResult> | AsyncGenerator<ChatModelRunResult, void>;
-};

package/src/legacy-runtime/runtime-cores/local/LocalRuntimeCore.tsx

@@ -1,32 +0,0 @@
-import { BaseAssistantRuntimeCore } from "../core/BaseAssistantRuntimeCore";
-import { LocalThreadRuntimeCore } from "./LocalThreadRuntimeCore";
-import { LocalRuntimeOptionsBase } from "./LocalRuntimeOptions";
-import { LocalThreadListRuntimeCore } from "./LocalThreadListRuntimeCore";
-import { ExportedMessageRepository } from "../utils/MessageRepository";
-import { ThreadMessageLike } from "../external-store";
-
-export class LocalRuntimeCore extends BaseAssistantRuntimeCore {
-  public readonly threads;
-  public readonly Provider = undefined;
-
-  private _options: LocalRuntimeOptionsBase;
-
-  constructor(
-    options: LocalRuntimeOptionsBase,
-    initialMessages: readonly ThreadMessageLike[] | undefined,
-  ) {
-    super();
-
-    this._options = options;
-
-    this.threads = new LocalThreadListRuntimeCore(() => {
-      return new LocalThreadRuntimeCore(this._contextProvider, this._options);
-    });
-
-    if (initialMessages) {
-      this.threads
-        .getMainThreadRuntimeCore()
-        .import(ExportedMessageRepository.fromArray(initialMessages));
-    }
-  }
-}

package/src/legacy-runtime/runtime-cores/local/LocalRuntimeOptions.tsx

@@ -1,60 +0,0 @@
-import type { ThreadHistoryAdapter } from "../adapters/thread-history/ThreadHistoryAdapter";
-import type { AttachmentAdapter } from "../adapters/attachment/AttachmentAdapter";
-import type { ThreadMessageLike } from "../external-store";
-import type { FeedbackAdapter } from "../adapters/feedback/FeedbackAdapter";
-import type {
-  SpeechSynthesisAdapter,
-  DictationAdapter,
-} from "../adapters/speech/SpeechAdapterTypes";
-import type { ChatModelAdapter } from "./ChatModelAdapter";
-import type { AssistantCloud } from "assistant-cloud";
-import type { SuggestionAdapter } from "../adapters";
-
-export type LocalRuntimeOptionsBase = {
-  maxSteps?: number | undefined;
-  adapters: {
-    chatModel: ChatModelAdapter;
-    history?: ThreadHistoryAdapter | undefined;
-    attachments?: AttachmentAdapter | undefined;
-    speech?: SpeechSynthesisAdapter | undefined;
-    dictation?: DictationAdapter | undefined;
-    feedback?: FeedbackAdapter | undefined;
-    suggestion?: SuggestionAdapter | undefined;
-  };
-
-  /**
-   * Names of tools that are allowed to interrupt the run in order to wait for human/external approval.
-   */
-  unstable_humanToolNames?: string[] | undefined;
-};
-
-// TODO align LocalRuntimeOptions with LocalRuntimeOptionsBase
-export type LocalRuntimeOptions = Omit<LocalRuntimeOptionsBase, "adapters"> & {
-  cloud?: AssistantCloud | undefined;
-  initialMessages?: readonly ThreadMessageLike[] | undefined;
-  adapters?: Omit<LocalRuntimeOptionsBase["adapters"], "chatModel"> | undefined;
-};
-
-export const splitLocalRuntimeOptions = <T extends LocalRuntimeOptions>(
-  options: T,
-) => {
-  const {
-    cloud,
-    initialMessages,
-    maxSteps,
-    adapters,
-    unstable_humanToolNames,
-    ...rest
-  } = options;
-
-  return {
-    localRuntimeOptions: {
-      cloud,
-      initialMessages,
-      maxSteps,
-      adapters,
-      unstable_humanToolNames,
-    },
-    otherOptions: rest,
-  };
-};

package/src/legacy-runtime/runtime-cores/local/LocalThreadListRuntimeCore.tsx

@@ -1,116 +0,0 @@
-import { ThreadListRuntimeCore } from "../core/ThreadListRuntimeCore";
-import { BaseSubscribable } from "../remote-thread-list/BaseSubscribable";
-import { LocalThreadRuntimeCore } from "./LocalThreadRuntimeCore";
-
-export type LocalThreadFactory = () => LocalThreadRuntimeCore;
-
-const EMPTY_ARRAY = Object.freeze([]);
-const DEFAULT_THREAD_ID = "__DEFAULT_ID__";
-const DEFAULT_THREAD_DATA = Object.freeze({
-  [DEFAULT_THREAD_ID]: {
-    id: DEFAULT_THREAD_ID,
-    remoteId: undefined,
-    externalId: undefined,
-    status: "regular" as const,
-    title: undefined,
-  },
-});
-export class LocalThreadListRuntimeCore
-  extends BaseSubscribable
-  implements ThreadListRuntimeCore
-{
-  private _mainThread: LocalThreadRuntimeCore;
-  constructor(_threadFactory: LocalThreadFactory) {
-    super();
-
-    this._mainThread = _threadFactory();
-  }
-
-  public get isLoading() {
-    return false;
-  }
-
-  public getMainThreadRuntimeCore() {
-    return this._mainThread;
-  }
-
-  public get newThreadId(): string {
-    throw new Error("Method not implemented.");
-  }
-
-  public get threadIds(): readonly string[] {
-    throw EMPTY_ARRAY;
-  }
-
-  public get archivedThreadIds(): readonly string[] {
-    throw EMPTY_ARRAY;
-  }
-
-  public get mainThreadId(): string {
-    return DEFAULT_THREAD_ID;
-  }
-
-  public get threadItems() {
-    return DEFAULT_THREAD_DATA;
-  }
-
-  public getThreadRuntimeCore(): never {
-    throw new Error("Method not implemented.");
-  }
-
-  public getLoadThreadsPromise(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public getItemById(threadId: string) {
-    if (threadId === this.mainThreadId) {
-      return {
-        status: "regular" as const,
-        id: this.mainThreadId,
-        remoteId: this.mainThreadId,
-        externalId: undefined,
-        title: undefined,
-        isMain: true,
-      };
-    }
-    throw new Error("Method not implemented");
-  }
-
-  public async switchToThread(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public switchToNewThread(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public rename(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public archive(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public detach(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public unarchive(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public delete(): Promise<void> {
-    throw new Error("Method not implemented.");
-  }
-
-  public initialize(
-    threadId: string,
-  ): Promise<{ remoteId: string; externalId: string | undefined }> {
-    return Promise.resolve({ remoteId: threadId, externalId: undefined });
-  }
-
-  public generateTitle(): never {
-    throw new Error("Method not implemented.");
-  }
-}