AutocompleteServer for ChatGPT

index.tsx

type CacheEntry<T> = {
    value: T,
    timeout: ReturnType<typeof setTimeout>,
}

class ExpiringCache<T> {
    private readonly cache: Record<string, CacheEntry<T>>;
    private readonly expirationTime: number;

    constructor(expirationTime: number = 5000) {
        this.cache = {};
        this.expirationTime = expirationTime;
    }

    set(key: string, value: T): void {
        // If there's an existing timeout for this key, clear it
        if (this.cache[key] && this.cache[key].timeout) {
            clearTimeout(this.cache[key].timeout);
        }

        // Set the new value
        this.cache[key] = {
            value: value,
            timeout: setTimeout(() => {
                delete this.cache[key];
            }, this.expirationTime)
        };
    }

    get(key: string): T | null {
        const entry = this.cache[key];
        if (entry) {
            console.log("HIT")
            return entry.value;
        }
        return null;
    }
}

class AutocompleteServer {
    LATENCY: number;
    timeout: null | ReturnType<typeof setTimeout> = null;
    cache = new ExpiringCache<string>(15000)
    constructor(latency: number = 3000) {
        this.LATENCY = latency;
    }
    query = (searchText: string): SearchPromise => {
        if (this.timeout !== null) {
            clearTimeout(this.timeout);
            this.timeout = null;
        }
        const dismiss = () => {
            if (this.timeout !== null) {
                console.log("DISMISSED")
                clearTimeout(this.timeout);
                this.timeout = null;
            }
        };
        const promise: Promise<null | string> = new Promise((resolve, reject) => {
            this.timeout = setTimeout(async () => {
                const searchTextLength = searchText.length;
                if (searchText === '' || searchTextLength < 4) {
                    return resolve(null);
                }
                const data
                    = this.cache.get(searchText) || await fetch(`/api/chat`, {method: 'POST', body: JSON.stringify({input: searchText})}).then(res => res.json())
                    .then(data => data.text).catch(e => reject(e));
                this.cache.set(searchText, data);
                return resolve(data);
            }, this.LATENCY);
        });

        return {
            dismiss,
            promise,
        };
    };
}


// use this instead of the original useQuery

function useQuery(server: AutocompleteServer): (searchText: string) => SearchPromise {
    return useCallback((searchText: string) => {
        console.time('query');
        const response = server.query(searchText);
        console.timeEnd('query');
        return response;
    }, [server]);
}

You might want to tweak the following for the ASCII entries.

function $search(
    selection: null | RangeSelection | NodeSelection | GridSelection,
): [boolean, string] {
    if (!$isRangeSelection(selection) || !selection.isCollapsed()) {
        return [false, ''];
    }
    const node = selection.getNodes()[0];
    const anchor = selection.anchor;
    // Check siblings?
    if (!$isTextNode(node) || !node.isSimpleText() || !$isAtNodeEnd(anchor)) {
        return [false, ''];
    }
    const word = [];
    const text = node.getTextContent();
  
    let i = node.getTextContentSize();
    let c;
    while (i-- && i >= 0 && (c = text[i]) !== ' ') { // <---- if using ASCII, this will trim the sentence to the last word.
        word.push(c);
    }
    if (word.length === 0) {
        return [false, ''];
    }
    return [true, word.reverse().join('')];
}

For ChatGPT you want to send all the texts belong to the node so fix it as below:

function $search(
    selection: null | RangeSelection | NodeSelection | GridSelection,
): [boolean, string] {
    if (!$isRangeSelection(selection) || !selection.isCollapsed()) {
        return [false, ''];
    }
    const node = selection.getNodes()[0];
    const anchor = selection.anchor;
    // Check siblings?
    if (!$isTextNode(node) || !node.isSimpleText() || !$isAtNodeEnd(anchor)) {
        return [false, ''];
    }
    return [true, node.getTextContent()];
}

You also want to remove query from dependencies.

        return mergeRegister(
            editor.registerNodeTransform(
                AutocompleteNode,
                handleAutocompleteNodeTransform,
            ),
            editor.registerUpdateListener(handleUpdate),
            editor.registerCommand(
                KEY_TAB_COMMAND,
                $handleKeypressCommand,
                COMMAND_PRIORITY_LOW,
            ),
            editor.registerCommand(
                KEY_ARROW_RIGHT_COMMAND,
                $handleKeypressCommand,
                COMMAND_PRIORITY_LOW,
            ),
            unmountSuggestion,
        );
    }, [editor, setSuggestion]); // <-- remove query!

Screen.Recording.2023-08-22.at.21.29.12.mov

AI_writer2.mov