@assistant-ui/react/dist/context/react/hooks/useAssistantState.js

TypeScript/React library for AI Chat

import { useMemo, useSyncExternalStore, useDebugValue } from "react";
import { useAssistantApi, } from "../AssistantApiContext.js";
class ProxiedAssistantState {
    #api;
    constructor(api) {
        this.#api = api;
    }
    get threads() {
        return this.#api.threads().getState();
    }
    get tools() {
        return this.#api.tools().getState();
    }
    get threadListItem() {
        return this.#api.threadListItem().getState();
    }
    get thread() {
        return this.#api.thread().getState();
    }
    get composer() {
        return this.#api.composer().getState();
    }
    get message() {
        return this.#api.message().getState();
    }
    get part() {
        return this.#api.part().getState();
    }
    get attachment() {
        return this.#api.attachment().getState();
    }
}
export const useAssistantState = (selector) => {
    const api = useAssistantApi();
    const proxiedState = useMemo(() => new ProxiedAssistantState(api), [api]);
    const slice = useSyncExternalStore(api.subscribe, () => selector(proxiedState), () => selector(proxiedState));
    useDebugValue(slice);
    if (slice instanceof ProxiedAssistantState)
        throw new Error("You tried to return the entire AssistantState. This is not supported due to technical limitations.");
    return slice;
};
//# sourceMappingURL=useAssistantState.js.map