@lobehub/chat/src/utils/zustand.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import { StoreApi } from 'zustand';

export interface StoreApiWithSelector<Store> extends Omit<StoreApi<Store>, 'subscribe'> {
  subscribe: <T extends keyof Store>(
    selector: (state: Store, prevState: Store) => void,
    listener?: (state: Store[T]) => void,
  ) => () => void;
}