@assistant-ui/react 0.10.49 → 0.11.0

package/src/context/react/MessagePartContext.ts

@@ -1,37 +0,0 @@
-"use client";
-
-import { createContext } from "react";
-import { ReadonlyStore } from "../ReadonlyStore";
-import { createContextHook } from "./utils/createContextHook";
-import { UseBoundStore } from "zustand";
-import { MessagePartRuntime } from "../../api/MessagePartRuntime";
-import { createStateHookForRuntime } from "./utils/createStateHookForRuntime";
-
-export type MessagePartContextValue = {
-  useMessagePartRuntime: UseBoundStore<ReadonlyStore<MessagePartRuntime>>;
-};
-
-export const MessagePartContext = createContext<MessagePartContextValue | null>(
-  null,
-);
-
-const useMessagePartContext = createContextHook(
-  MessagePartContext,
-  "a component passed to <MessagePrimitive.Parts components={...}>",
-);
-
-export function useMessagePartRuntime(options?: {
-  optional?: false | undefined;
-}): MessagePartRuntime;
-export function useMessagePartRuntime(options?: {
-  optional?: boolean | undefined;
-}): MessagePartRuntime | null;
-export function useMessagePartRuntime(options?: {
-  optional?: boolean | undefined;
-}) {
-  const context = useMessagePartContext(options);
-  if (!context) return null;
-  return context.useMessagePartRuntime();
-}
-
-export const useMessagePart = createStateHookForRuntime(useMessagePartRuntime);

package/src/context/react/ThreadListItemContext.ts

@@ -1,38 +0,0 @@
-"use client";
-
-import { createContext } from "react";
-import { ReadonlyStore } from "../ReadonlyStore";
-import { createContextHook } from "./utils/createContextHook";
-import { UseBoundStore } from "zustand";
-import { ThreadListItemRuntime } from "../../api/ThreadListItemRuntime";
-import { createStateHookForRuntime } from "./utils/createStateHookForRuntime";
-
-export type ThreadListItemContextValue = {
-  useThreadListItemRuntime: UseBoundStore<ReadonlyStore<ThreadListItemRuntime>>;
-};
-
-export const ThreadListItemContext =
-  createContext<ThreadListItemContextValue | null>(null);
-
-const useThreadListItemContext = createContextHook(
-  ThreadListItemContext,
-  "a component passed to <ThreadListPrimitive.Items components={...}>",
-);
-
-export function useThreadListItemRuntime(options?: {
-  optional?: false | undefined;
-}): ThreadListItemRuntime;
-export function useThreadListItemRuntime(options?: {
-  optional?: boolean | undefined;
-}): ThreadListItemRuntime | null;
-export function useThreadListItemRuntime(options?: {
-  optional?: boolean | undefined;
-}) {
-  const context = useThreadListItemContext(options);
-  if (!context) return null;
-  return context.useThreadListItemRuntime();
-}
-
-export const useThreadListItem = createStateHookForRuntime(
-  useThreadListItemRuntime,
-);

package/src/context/stores/AssistantToolUIs.ts

@@ -1,51 +0,0 @@
-import { create } from "zustand";
-import type { ToolCallMessagePartComponent } from "../../types/MessagePartComponentTypes";
-import { Unsubscribe } from "../../types";
-
-export type AssistantToolUIsState = {
-  /**
-   * Get the tool UI configured for a given tool name.
-   */
-  getToolUI: (toolName: string) => ToolCallMessagePartComponent | null;
-
-  /**
-   * Registers a tool UI for a given tool name. Returns an unsubscribe function to remove the tool UI.
-   */
-  setToolUI: (
-    toolName: string,
-    render: ToolCallMessagePartComponent,
-  ) => Unsubscribe;
-};
-
-export const makeAssistantToolUIsStore = () =>
-  create<AssistantToolUIsState>((set) => {
-    const renderers = new Map<string, ToolCallMessagePartComponent[]>();
-
-    return Object.freeze({
-      getToolUI: (name) => {
-        const arr = renderers.get(name);
-        const last = arr?.at(-1);
-        if (last) return last;
-        return null;
-      },
-      setToolUI: (name, render) => {
-        let arr = renderers.get(name);
-        if (!arr) {
-          arr = [];
-          renderers.set(name, arr);
-        }
-        arr.push(render);
-        set({}); // notify the store listeners
-
-        return () => {
-          const index = arr.indexOf(render);
-          if (index !== -1) {
-            arr.splice(index, 1);
-          }
-          if (index === arr.length) {
-            set({}); // notify the store listeners
-          }
-        };
-      },
-    }) satisfies AssistantToolUIsState;
-  });

package/src/context/stores/MessageUtils.ts

@@ -1,23 +0,0 @@
-import { create } from "zustand";
-
-export type MessageUtilsState = {
-  readonly isCopied: boolean;
-  readonly setIsCopied: (value: boolean) => void;
-
-  readonly isHovering: boolean;
-  readonly setIsHovering: (value: boolean) => void;
-};
-
-export const makeMessageUtilsStore = () =>
-  create<MessageUtilsState>((set) => {
-    return {
-      isCopied: false,
-      setIsCopied: (value) => {
-        set({ isCopied: value });
-      },
-      isHovering: false,
-      setIsHovering: (value) => {
-        set({ isHovering: value });
-      },
-    };
-  });

package/src/utils/combined/createCombinedStore.ts

@@ -1,31 +0,0 @@
-"use client";
-
-import { useSyncExternalStore } from "react";
-import type { Unsubscribe } from "../../types/Unsubscribe";
-
-export type StoreOrRuntime<T> = {
-  getState: () => T;
-  subscribe: (callback: () => void) => Unsubscribe;
-};
-
-export type CombinedSelector<T extends Array<unknown>, R> = (...args: T) => R;
-
-export const createCombinedStore = <T extends Array<unknown>, R>(stores: {
-  [K in keyof T]: StoreOrRuntime<T[K]>;
-}) => {
-  const subscribe = (callback: () => void): Unsubscribe => {
-    const unsubscribes = stores.map((store) => store.subscribe(callback));
-    return () => {
-      for (const unsub of unsubscribes) {
-        unsub();
-      }
-    };
-  };
-
-  return (selector: CombinedSelector<T, R>): R => {
-    const getSnapshot = (): R =>
-      selector(...(stores.map((store) => store.getState()) as T));
-
-    return useSyncExternalStore(subscribe, getSnapshot, getSnapshot);
-  };
-};

package/src/utils/combined/useCombinedStore.ts

@@ -1,17 +0,0 @@
-"use client";
-
-import { useMemo } from "react";
-import {
-  type CombinedSelector,
-  createCombinedStore,
-  StoreOrRuntime,
-} from "./createCombinedStore";
-
-export const useCombinedStore = <T extends Array<unknown>, R>(
-  stores: { [K in keyof T]: StoreOrRuntime<T[K]> },
-  selector: CombinedSelector<T, R>,
-): R => {
-  // eslint-disable-next-line react-hooks/exhaustive-deps -- shallow-compare the store array
-  const useCombined = useMemo(() => createCombinedStore<T, R>(stores), stores);
-  return useCombined(selector);
-};