Function submitAssistantAsk

explorer/src/js/assistant.js:229–286

()

function submitAssistantAsk() {

    const data = {
        sql: window.editor?.state.doc.toString() ?? null,
        connection_id: document.getElementById("id_database_connection")?.value ?? null,
        assistant_request: document.getElementById("id_assistant_input")?.value ?? null,
        selected_tables: assistantChoices.getValue(true),
        db_error: getErrorMessage()
    };

    document.getElementById("assistant_response").innerHTML = '';
    document.getElementById("response_block").classList.remove('d-none');
    document.getElementById("assistant_spinner").classList.remove('d-none');

    fetch(`${window.baseUrlPath}assistant/`, {
        method: 'POST',
        headers: {
            'Content-Type': 'application/json',
            'X-CSRFToken': getCsrfToken()
        },
        body: JSON.stringify(data)
    })
    .then(response => {
        if (!response.ok) {
            throw new Error('Network response was not ok');
        }
        return response.json();
    })
    .then(data => {
        const output = DOMPurify.sanitize(marked.parse(data.message));
        document.getElementById("assistant_response").innerHTML = output;
        document.getElementById("assistant_spinner").classList.add('d-none');

        // If there is exactly one code block in the response and the SQL editor is empty
        // then copy the code directly into the editor
        const preElements = document.querySelectorAll('#assistant_response pre');
        if (preElements.length === 1 && window.editor?.state.doc.toString().trim() === "") {
            window.editor.dispatch({
                changes: {
                    from: 0,
                    insert: preElements[0].textContent
                }
            });
        }

        // Similarly, if there is no description, copy the prompt into the description
        const prompt = document.getElementById("id_assistant_input")?.value;
        const description = document.getElementById("id_description");
        if (description?.value === "") {
            description.value = prompt;
        }

        setUpCopyButtons();
    })
        .catch(error => {
        console.error('Error:', error);
    });
}