@assistant-ui/react/src/utils/combined/useCombinedStore.ts

Typescript/React library for AI Chat

"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);
};