This commit is contained in:
Vanessa 2023-04-06 09:29:16 +08:00
parent 0d706baa1d
commit a674f8fbec

View file

@ -14,6 +14,9 @@ import {hasClosestByClassName} from "../protyle/util/hasClosest";
import {escapeAttr, escapeHtml} from "../util/escape"; import {escapeAttr, escapeHtml} from "../util/escape";
export const fillContent = (protyle: IProtyle, data: string, elements: Element[]) => { export const fillContent = (protyle: IProtyle, data: string, elements: Element[]) => {
if (!data) {
return;
}
setLastNodeRange(getContenteditableElement(elements[elements.length - 1]), protyle.toolbar.range); setLastNodeRange(getContenteditableElement(elements[elements.length - 1]), protyle.toolbar.range);
protyle.toolbar.range.collapse(true); protyle.toolbar.range.collapse(true);
insertHTML(data, protyle, true, true); insertHTML(data, protyle, true, true);
@ -55,7 +58,7 @@ export const AIActions = (elements: Element[], protyle: IProtyle) => {
action: inputElement.value, action: inputElement.value,
}, (response) => { }, (response) => {
dialog.destroy(); dialog.destroy();
fillContent(protyle, response.data || "", elements); fillContent(protyle, response.data, elements);
}); });
}); });
} }
@ -119,11 +122,7 @@ export const AIActions = (elements: Element[], protyle: IProtyle) => {
ids, ids,
action: item.memo, action: item.memo,
}, (response) => { }, (response) => {
let respContent = ""; fillContent(protyle, response.data, elements);
if (response.data && "" !== response.data) {
respContent = "\n\n" + response.data;
}
fillContent(protyle, `${item.memo}${respContent}`, elements);
}); });
window.siyuan.menus.menu.remove(); window.siyuan.menus.menu.remove();
} }