@llamaindex/ui/src/indexes/hooks/use-index-store.ts

A comprehensive UI component library built with React, TypeScript, and Tailwind CSS for LlamaIndex applications

import { useMemo } from "react";
import { useCloudApiClient, useProject } from "../../lib/api-provider";
import { createIndexStore, type IndexStoreState } from "../store/index-store";

let globalStore: ReturnType<typeof createIndexStore> | null = null;

export function useIndexStore(): IndexStoreState;
export function useIndexStore<T>(selector: (state: IndexStoreState) => T): T;

export function useIndexStore<T>(selector?: (state: IndexStoreState) => T) {
  const client = useCloudApiClient();
  const project = useProject();
  const store = useMemo(() => {
    if (!globalStore) {
      globalStore = createIndexStore(client, {
        getProject: () => project,
      });
    }
    return globalStore;
  }, [client, project]);

  return selector ? store(selector) : store();
}

export function __resetIndexStore() {
  globalStore = null;
}