Function useChat

src/chat/useChat.ts:29–281

(props?: IUseChatProps<C, P, M>)

const updateReducer = (num: number): number => (num + 1) % 1_000_000;

export default function useChat<
    // eslint-disable-next-line space-before-function-paren
    C extends { new (...params: ConstructorParameters<typeof Conversation>): Conversation },
    P extends { new (...params: ConstructorParameters<typeof Prompt>): Prompt },
    M extends { new (...params: ConstructorParameters<typeof Message>): Message }
>(props?: IUseChatProps<C, P, M>) {
    const [, update] = useReducer(updateReducer, 0);

    const state = useRef<Conversation | undefined>(undefined);
    const typing = useTyping({
        onTyping(post) {
            _updateMessage(typingIds.current.promptId, typingIds.current.messageId, {
                content: post,
            });
        },
    });
    const typingIds = useRef<{ promptId: Id; messageId: Id }>({
        promptId: '',
        messageId: '',
    });
    const closing = useRef<boolean>(false);

    // ================================== Typing ==================================
    function _start(promptId: Id, messageId: Id) {
        typing.start();
        _updateMessage(promptId, messageId, { status: MessageStatus.PENDING });
        typingIds.current = { promptId, messageId };
    }

    function _push(promptId: Id, messageId: Id, str: string) {
        if (promptId === typingIds.current.promptId && messageId === typingIds.current.messageId) {
            typing.push(str);
            if (_getMessage(promptId, messageId)?.status !== MessageStatus.GENERATING) {
                _updateMessage(promptId, messageId, { status: MessageStatus.GENERATING });
            }
        }
    }

    function _close(promptId: Id, messageId: Id) {
        closing.current = true;
        typing.close().then(() => {
            closing.current = false;
            _updateMessage(promptId, messageId, { status: MessageStatus.DONE });
        });
    }

    // ================================== Conversation ==================================
    function _createConversation(data: ConversationProperties) {
        const ConversationCtor = props?.ctor?.Conversation || BaseConversation;
        state.current = new ConversationCtor(data);
        update();
    }

    function _removeConversation() {
        state.current = undefined;
        update();
    }