fix

udecode · Oct 23, 2024 · b5dfdc1 · b5dfdc1
1 parent d19d103
commit b5dfdc1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/.changeset/stale-mails-rescue.md b/.changeset/stale-mails-rescue.md
@@ -0,0 +1,5 @@
+---
+'@udecode/plate-ai': patch
+---
+
+fix accept + undo
diff --git a/packages/ai/src/lib/transforms/undoAI.ts b/packages/ai/src/lib/transforms/undoAI.ts
@@ -1,7 +1,15 @@
-import type { SlateEditor } from '@udecode/plate-common';
+import { type SlateEditor, someNode } from '@udecode/plate-common';
+
+import { AIPlugin } from '../../react';
 
 export const undoAI = (editor: SlateEditor) => {
-  if ((editor.history.undos.at(-1) as any)?.ai) {
+  if (
+    (editor.history.undos.at(-1) as any)?.ai &&
+    someNode(editor, {
+      at: [],
+      match: (n) => !!n[AIPlugin.key],
+    })
+  ) {
     editor.undo();
     editor.history.redos.pop();
   }