@assistant-ui/react 0.10.50 → 0.11.1

package/src/cloud/AssistantCloudThreadHistoryAdapter.tsx

@@ -1,183 +0,0 @@
-import { RefObject, useState } from "react";
-import { useThreadListItemRuntime } from "../context";
-import { ThreadHistoryAdapter } from "../runtimes/adapters/thread-history/ThreadHistoryAdapter";
-import { ExportedMessageRepositoryItem } from "../runtimes/utils/MessageRepository";
-import { AssistantCloud } from "assistant-cloud";
-import { auiV0Decode, auiV0Encode } from "./auiV0";
-import { ThreadListItemRuntime } from "../api";
-import {
-  MessageFormatAdapter,
-  MessageFormatItem,
-  MessageFormatRepository,
-  MessageStorageEntry,
-} from "../runtimes/adapters/thread-history/MessageFormatAdapter";
-import { GenericThreadHistoryAdapter } from "../runtimes/adapters/thread-history/ThreadHistoryAdapter";
-import { ReadonlyJSONObject } from "assistant-stream/utils";
-
-// Global WeakMap to store message ID mappings across adapter instances
-const globalMessageIdMapping = new WeakMap<
-  ThreadListItemRuntime,
-  Record<string, string | Promise<string>>
->();
-
-class FormattedThreadHistoryAdapter<TMessage, TStorageFormat>
-  implements GenericThreadHistoryAdapter<TMessage>
-{
-  constructor(
-    private parent: AssistantCloudThreadHistoryAdapter,
-    private formatAdapter: MessageFormatAdapter<TMessage, TStorageFormat>,
-  ) {}
-
-  async append(item: MessageFormatItem<TMessage>) {
-    // Encode the message using the format adapter
-    const encoded = this.formatAdapter.encode(item);
-    const messageId = this.formatAdapter.getId(item.message);
-
-    // Delegate to parent's internal append method with the encoded format
-    return this.parent._appendWithFormat(
-      item.parentId,
-      messageId,
-      this.formatAdapter.format,
-      encoded,
-    );
-  }
-
-  async load(): Promise<MessageFormatRepository<TMessage>> {
-    // Delegate to parent's internal load method with format filter
-    return this.parent._loadWithFormat(
-      this.formatAdapter.format,
-      (message: MessageStorageEntry<TStorageFormat>) =>
-        this.formatAdapter.decode(message),
-    );
-  }
-}
-
-class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
-  constructor(
-    private cloudRef: RefObject<AssistantCloud>,
-    private threadListItemRuntime: ThreadListItemRuntime,
-  ) {}
-
-  private get _getIdForLocalId(): Record<string, string | Promise<string>> {
-    if (!globalMessageIdMapping.has(this.threadListItemRuntime)) {
-      globalMessageIdMapping.set(this.threadListItemRuntime, {});
-    }
-    return globalMessageIdMapping.get(this.threadListItemRuntime)!;
-  }
-
-  withFormat<TMessage, TStorageFormat>(
-    formatAdapter: MessageFormatAdapter<TMessage, TStorageFormat>,
-  ): GenericThreadHistoryAdapter<TMessage> {
-    return new FormattedThreadHistoryAdapter(this, formatAdapter);
-  }
-
-  async append({ parentId, message }: ExportedMessageRepositoryItem) {
-    const { remoteId } = await this.threadListItemRuntime.initialize();
-    const task = this.cloudRef.current.threads.messages
-      .create(remoteId, {
-        parent_id: parentId
-          ? ((await this._getIdForLocalId[parentId]) ?? parentId)
-          : null,
-        format: "aui/v0",
-        content: auiV0Encode(message),
-      })
-      .then(({ message_id }) => {
-        this._getIdForLocalId[message.id] = message_id;
-        return message_id;
-      });
-
-    this._getIdForLocalId[message.id] = task;
-
-    return task.then(() => {});
-  }
-
-  async load() {
-    const remoteId = this.threadListItemRuntime.getState().remoteId;
-    if (!remoteId) return { messages: [] };
-    const { messages } = await this.cloudRef.current.threads.messages.list(
-      remoteId,
-      {
-        format: "aui/v0",
-      },
-    );
-    const payload = {
-      messages: messages
-        .filter(
-          (m): m is typeof m & { format: "aui/v0" } => m.format === "aui/v0",
-        )
-        .map(auiV0Decode)
-        .reverse(),
-    };
-    return payload;
-  }
-
-  // Internal methods for FormattedThreadHistoryAdapter
-  async _appendWithFormat<T>(
-    parentId: string | null,
-    messageId: string,
-    format: string,
-    content: T,
-  ) {
-    const { remoteId } = await this.threadListItemRuntime.initialize();
-
-    const task = this.cloudRef.current.threads.messages
-      .create(remoteId, {
-        parent_id: parentId
-          ? ((await this._getIdForLocalId[parentId]) ?? parentId)
-          : null,
-        format,
-        content: content as ReadonlyJSONObject,
-      })
-      .then(({ message_id }) => {
-        this._getIdForLocalId[messageId] = message_id;
-        return message_id;
-      });
-
-    this._getIdForLocalId[messageId] = task;
-
-    return task.then(() => {});
-  }
-
-  async _loadWithFormat<TMessage, TStorageFormat>(
-    format: string,
-    decoder: (
-      message: MessageStorageEntry<TStorageFormat>,
-    ) => MessageFormatItem<TMessage>,
-  ): Promise<MessageFormatRepository<TMessage>> {
-    const remoteId = this.threadListItemRuntime.getState().remoteId;
-    if (!remoteId) return { messages: [] };
-
-    const { messages } = await this.cloudRef.current.threads.messages.list(
-      remoteId,
-      {
-        format,
-      },
-    );
-
-    return {
-      messages: messages
-        .filter((m) => m.format === format)
-        .map((m) =>
-          decoder({
-            id: m.id,
-            parent_id: m.parent_id,
-            format: m.format,
-            content: m.content as TStorageFormat,
-          }),
-        )
-        .reverse(),
-    };
-  }
-}
-
-export const useAssistantCloudThreadHistoryAdapter = (
-  cloudRef: RefObject<AssistantCloud>,
-): ThreadHistoryAdapter => {
-  const threadListItemRuntime = useThreadListItemRuntime();
-  const [adapter] = useState(
-    () =>
-      new AssistantCloudThreadHistoryAdapter(cloudRef, threadListItemRuntime),
-  );
-
-  return adapter;
-};

package/src/cloud/auiV0.ts

@@ -1,158 +0,0 @@
-import { ThreadMessage } from "../types";
-import { MessageStatus } from "../types/AssistantTypes";
-import { fromThreadMessageLike } from "../runtimes/external-store/ThreadMessageLike";
-import { CloudMessage } from "assistant-cloud";
-import { isJSONValue } from "../utils/json/is-json";
-import { ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
-import { ExportedMessageRepositoryItem } from "../runtimes/utils/MessageRepository";
-
-type AuiV0MessageMessagePart =
-  | {
-      readonly type: "text";
-      readonly text: string;
-    }
-  | {
-      readonly type: "reasoning";
-      readonly text: string;
-    }
-  | {
-      readonly type: "source";
-      readonly sourceType: "url";
-      readonly id: string;
-      readonly url: string;
-      readonly title?: string;
-    }
-  | {
-      readonly type: "tool-call";
-      readonly toolCallId: string;
-      readonly toolName: string;
-      readonly args: ReadonlyJSONObject;
-      readonly result?: ReadonlyJSONValue;
-      readonly isError?: true;
-    }
-  | {
-      readonly type: "tool-call";
-      readonly toolCallId: string;
-      readonly toolName: string;
-      readonly argsText: string;
-      readonly result?: ReadonlyJSONValue;
-      readonly isError?: true;
-    };
-
-type AuiV0Message = {
-  readonly role: "assistant" | "user" | "system";
-  readonly status?: MessageStatus;
-  readonly content: readonly AuiV0MessageMessagePart[];
-  readonly metadata: {
-    readonly unstable_state?: ReadonlyJSONValue;
-    readonly unstable_annotations: readonly ReadonlyJSONValue[];
-    readonly unstable_data: readonly ReadonlyJSONValue[];
-    readonly steps: readonly {
-      readonly usage?: {
-        readonly promptTokens: number;
-        readonly completionTokens: number;
-      };
-    }[];
-    readonly custom: ReadonlyJSONObject;
-  };
-};
-
-export const auiV0Encode = (message: ThreadMessage): AuiV0Message => {
-  // TODO attachments are currently intentionally ignored
-  // info: ID and createdAt are ignored (we use the server value instead)
-  return {
-    role: message.role,
-    content: message.content.map((part) => {
-      const type = part.type;
-      switch (type) {
-        case "text": {
-          return {
-            type: "text",
-            text: part.text,
-          };
-        }
-
-        case "reasoning": {
-          return {
-            type: "reasoning",
-            text: part.text,
-          };
-        }
-
-        case "source": {
-          return {
-            type: "source",
-            sourceType: part.sourceType,
-            id: part.id,
-            url: part.url,
-            ...(part.title ? { title: part.title } : undefined),
-          };
-        }
-
-        case "tool-call": {
-          if (!isJSONValue(part.result)) {
-            console.warn(
-              "tool-call result is not JSON! " + JSON.stringify(part),
-            );
-          }
-          return {
-            type: "tool-call",
-            toolCallId: part.toolCallId,
-            toolName: part.toolName,
-            ...(JSON.stringify(part.args) === part.argsText
-              ? {
-                  args: part.args,
-                }
-              : { argsText: part.argsText }),
-            ...(part.result
-              ? { result: part.result as ReadonlyJSONValue }
-              : {}),
-            ...(part.isError ? { isError: true } : {}),
-          };
-        }
-
-        default: {
-          const unhandledType: "image" | "file" | "audio" = type;
-          throw new Error(
-            `Message part type not supported by aui/v0: ${unhandledType}`,
-          );
-        }
-      }
-    }),
-    metadata: message.metadata as AuiV0Message["metadata"],
-    ...(message.status
-      ? {
-          status:
-            message.status?.type === "running"
-              ? {
-                  type: "incomplete",
-                  reason: "cancelled",
-                }
-              : message.status,
-        }
-      : undefined),
-  };
-};
-
-export const auiV0Decode = (
-  cloudMessage: CloudMessage & { format: "aui/v0" },
-): ExportedMessageRepositoryItem => {
-  const payload = cloudMessage.content as unknown as AuiV0Message;
-  const message = fromThreadMessageLike(
-    {
-      id: cloudMessage.id,
-      createdAt: cloudMessage.created_at,
-      ...payload,
-    },
-    cloudMessage.id,
-    {
-      type: "complete",
-      reason: "unknown",
-    },
-  );
-
-  return {
-    parentId: cloudMessage.parent_id,
-    message,
-  };
-};

package/src/cloud/useCloudThreadListRuntime.tsx

@@ -1,32 +0,0 @@
-"use client";
-
-import { AssistantCloud } from "assistant-cloud";
-import { AssistantRuntime } from "../api";
-import { useRemoteThreadListRuntime } from "../runtimes/remote-thread-list/useRemoteThreadListRuntime";
-import { useCloudThreadListAdapter } from "../runtimes/remote-thread-list/adapter/cloud";
-
-type ThreadData = {
-  externalId: string;
-};
-
-type CloudThreadListAdapter = {
-  cloud: AssistantCloud;
-
-  runtimeHook: () => AssistantRuntime;
-
-  create?(): Promise<ThreadData>;
-  delete?(threadId: string): Promise<void>;
-};
-
-export const useCloudThreadListRuntime = ({
-  runtimeHook,
-  ...adapterOptions
-}: CloudThreadListAdapter) => {
-  const adapter = useCloudThreadListAdapter(adapterOptions);
-  const runtime = useRemoteThreadListRuntime({
-    runtimeHook: runtimeHook,
-    adapter,
-  });
-
-  return runtime;
-};

package/src/context/providers/AssistantRuntimeProvider.tsx

@@ -1,77 +0,0 @@
-"use client";
-
-import {
-  FC,
-  PropsWithChildren,
-  memo,
-  useEffect,
-  useMemo,
-  useState,
-} from "react";
-import { AssistantContext } from "../react/AssistantContext";
-import { makeAssistantToolUIsStore } from "../stores/AssistantToolUIs";
-import { ThreadRuntimeProvider } from "./ThreadRuntimeProvider";
-import { AssistantRuntime } from "../../api/AssistantRuntime";
-import { create } from "zustand";
-import { writableStore } from "../ReadonlyStore";
-import { AssistantRuntimeCore } from "../../runtimes/core/AssistantRuntimeCore";
-import { ensureBinding } from "../react/utils/ensureBinding";
-
-export namespace AssistantRuntimeProvider {
-  export type Props = PropsWithChildren<{
-    /**
-     * The runtime to provide to the rest of your app.
-     */
-    runtime: AssistantRuntime;
-  }>;
-}
-
-const useAssistantRuntimeStore = (runtime: AssistantRuntime) => {
-  const [store] = useState(() => create(() => runtime));
-
-  useEffect(() => {
-    ensureBinding(runtime);
-    ensureBinding(runtime.threads);
-
-    writableStore(store).setState(runtime, true);
-  }, [runtime, store]);
-
-  return store;
-};
-
-const useAssistantToolUIsStore = () => {
-  return useMemo(() => makeAssistantToolUIsStore(), []);
-};
-
-const getRenderComponent = (runtime: AssistantRuntime) => {
-  return (runtime as { _core?: AssistantRuntimeCore })._core?.RenderComponent;
-};
-
-export const AssistantRuntimeProviderImpl: FC<
-  AssistantRuntimeProvider.Props
-> = ({ children, runtime }) => {
-  const useAssistantRuntime = useAssistantRuntimeStore(runtime);
-  const useToolUIs = useAssistantToolUIsStore();
-  const [context] = useState(() => {
-    return {
-      useToolUIs,
-      useAssistantRuntime,
-    };
-  });
-
-  const RenderComponent = getRenderComponent(runtime);
-
-  return (
-    <AssistantContext.Provider value={context}>
-      {RenderComponent && <RenderComponent />}
-      <ThreadRuntimeProvider
-        runtime={runtime.thread}
-        listItemRuntime={runtime.threads.mainItem}
-      >
-        {children}
-      </ThreadRuntimeProvider>
-    </AssistantContext.Provider>
-  );
-};
-
-export const AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);

package/src/context/providers/AttachmentRuntimeProvider.tsx

@@ -1,44 +0,0 @@
-"use client";
-
-import { type FC, type PropsWithChildren, useEffect, useState } from "react";
-import { create } from "zustand";
-import { AttachmentContext } from "../react/AttachmentContext";
-import { writableStore } from "../ReadonlyStore";
-import { AttachmentRuntime } from "../../api/AttachmentRuntime";
-import { ensureBinding } from "../react/utils/ensureBinding";
-
-export namespace AttachmentRuntimeProvider {
-  export type Props = PropsWithChildren<{
-    runtime: AttachmentRuntime;
-  }>;
-}
-
-const useAttachmentRuntimeStore = (runtime: AttachmentRuntime) => {
-  const [store] = useState(() => create(() => runtime));
-
-  useEffect(() => {
-    ensureBinding(runtime);
-
-    writableStore(store).setState(runtime, true);
-  }, [runtime, store]);
-
-  return store;
-};
-
-export const AttachmentRuntimeProvider: FC<AttachmentRuntimeProvider.Props> = ({
-  runtime,
-  children,
-}) => {
-  const useAttachmentRuntime = useAttachmentRuntimeStore(runtime);
-  const [context] = useState(() => {
-    return {
-      useAttachmentRuntime,
-    };
-  });
-
-  return (
-    <AttachmentContext.Provider value={context}>
-      {children}
-    </AttachmentContext.Provider>
-  );
-};

package/src/context/providers/MessagePartRuntimeProvider.tsx

@@ -1,42 +0,0 @@
-"use client";
-
-import { type FC, type PropsWithChildren, useEffect, useState } from "react";
-import { create } from "zustand";
-import { MessagePartContext } from "../react/MessagePartContext";
-import type { MessagePartContextValue } from "../react/MessagePartContext";
-import { writableStore } from "../ReadonlyStore";
-import { MessagePartRuntime } from "../../api/MessagePartRuntime";
-import { ensureBinding } from "../react/utils/ensureBinding";
-
-export namespace MessagePartRuntimeProvider {
-  export type Props = PropsWithChildren<{
-    runtime: MessagePartRuntime;
-  }>;
-}
-
-const useMessagePartRuntimeStore = (runtime: MessagePartRuntime) => {
-  const [store] = useState(() => create(() => runtime));
-
-  useEffect(() => {
-    ensureBinding(runtime);
-
-    writableStore(store).setState(runtime, true);
-  }, [runtime, store]);
-
-  return store;
-};
-
-export const MessagePartRuntimeProvider: FC<
-  MessagePartRuntimeProvider.Props
-> = ({ runtime, children }) => {
-  const useMessagePartRuntime = useMessagePartRuntimeStore(runtime);
-  const [context] = useState<MessagePartContextValue>(() => {
-    return { useMessagePartRuntime };
-  });
-
-  return (
-    <MessagePartContext.Provider value={context}>
-      {children}
-    </MessagePartContext.Provider>
-  );
-};

package/src/context/providers/MessageRuntimeProvider.tsx

@@ -1,50 +0,0 @@
-"use client";
-
-import { type FC, type PropsWithChildren, useEffect, useState } from "react";
-import { create } from "zustand";
-import { MessageContext } from "../react/MessageContext";
-import type { MessageContextValue } from "../react/MessageContext";
-import { makeMessageUtilsStore } from "../stores/MessageUtils";
-import { writableStore } from "../ReadonlyStore";
-import { MessageRuntime } from "../../api/MessageRuntime";
-import { ensureBinding } from "../react/utils/ensureBinding";
-
-export namespace MessageRuntimeProvider {
-  export type Props = PropsWithChildren<{
-    runtime: MessageRuntime;
-  }>;
-}
-
-const useMessageRuntimeStore = (runtime: MessageRuntime) => {
-  const [store] = useState(() => create(() => runtime));
-
-  useEffect(() => {
-    ensureBinding(runtime);
-
-    writableStore(store).setState(runtime, true);
-  }, [runtime, store]);
-
-  return store;
-};
-
-const useMessageUtilsStore = () => {
-  const [store] = useState(() => makeMessageUtilsStore());
-  return store;
-};
-
-export const MessageRuntimeProvider: FC<MessageRuntimeProvider.Props> = ({
-  runtime,
-  children,
-}) => {
-  const useMessageRuntime = useMessageRuntimeStore(runtime);
-  const useMessageUtils = useMessageUtilsStore();
-  const [context] = useState<MessageContextValue>(() => {
-    return { useMessageRuntime, useMessageUtils };
-  });
-
-  return (
-    <MessageContext.Provider value={context}>
-      {children}
-    </MessageContext.Provider>
-  );
-};

package/src/context/providers/ThreadListItemRuntimeProvider.tsx

@@ -1,43 +0,0 @@
-"use client";
-
-import { type FC, type PropsWithChildren, useEffect, useState } from "react";
-import { create } from "zustand";
-import {
-  ThreadListItemContext,
-  type ThreadListItemContextValue,
-} from "../react/ThreadListItemContext";
-import { writableStore } from "../ReadonlyStore";
-import { ThreadListItemRuntime } from "../../api/ThreadListItemRuntime";
-import { ensureBinding } from "../react/utils/ensureBinding";
-
-export namespace ThreadListItemRuntimeProvider {
-  export type Props = PropsWithChildren<{
-    runtime: ThreadListItemRuntime;
-  }>;
-}
-
-const useThreadListItemRuntimeStore = (runtime: ThreadListItemRuntime) => {
-  const [store] = useState(() => create(() => runtime));
-
-  useEffect(() => {
-    ensureBinding(runtime);
-    writableStore(store).setState(runtime, true);
-  }, [runtime, store]);
-
-  return store;
-};
-
-export const ThreadListItemRuntimeProvider: FC<
-  ThreadListItemRuntimeProvider.Props
-> = ({ runtime, children }) => {
-  const useThreadListItemRuntime = useThreadListItemRuntimeStore(runtime);
-  const [context] = useState<ThreadListItemContextValue>(() => {
-    return { useThreadListItemRuntime };
-  });
-
-  return (
-    <ThreadListItemContext.Provider value={context}>
-      {children}
-    </ThreadListItemContext.Provider>
-  );
-};

package/src/context/providers/ThreadRuntimeProvider.tsx

@@ -1,53 +0,0 @@
-"use client";
-
-import type { FC, PropsWithChildren } from "react";
-import { useEffect, useState } from "react";
-import type { ThreadContextValue } from "../react/ThreadContext";
-import { ThreadContext } from "../react/ThreadContext";
-import { writableStore } from "../ReadonlyStore";
-import { ThreadRuntime } from "../../api/ThreadRuntime";
-import { create } from "zustand";
-import { ThreadListItemRuntime } from "../../api/ThreadListItemRuntime";
-import { ThreadListItemRuntimeProvider } from "./ThreadListItemRuntimeProvider";
-import { ensureBinding } from "../react/utils/ensureBinding";
-import { ThreadViewportProvider } from "./ThreadViewportProvider";
-
-type ThreadProviderProps = {
-  listItemRuntime: ThreadListItemRuntime;
-  runtime: ThreadRuntime;
-};
-
-const useThreadRuntimeStore = (runtime: ThreadRuntime) => {
-  const [store] = useState(() => create(() => runtime));
-
-  useEffect(() => {
-    ensureBinding(runtime);
-    ensureBinding(runtime.composer);
-
-    writableStore(store).setState(runtime, true);
-  }, [runtime, store]);
-
-  return store;
-};
-
-export const ThreadRuntimeProvider: FC<
-  PropsWithChildren<ThreadProviderProps>
-> = ({ children, listItemRuntime: threadListItemRuntime, runtime }) => {
-  const useThreadRuntime = useThreadRuntimeStore(runtime);
-
-  const [context] = useState<ThreadContextValue>(() => {
-    return {
-      useThreadRuntime,
-    };
-  });
-
-  return (
-    <ThreadListItemRuntimeProvider runtime={threadListItemRuntime}>
-      <ThreadContext.Provider value={context}>
-        {/* TODO temporarily allow accessing viewport state from outside the viewport */}
-        {/* TODO figure out if this behavior should be deprecated, since it is quite hacky */}
-        <ThreadViewportProvider>{children}</ThreadViewportProvider>
-      </ThreadContext.Provider>
-    </ThreadListItemRuntimeProvider>
-  );
-};