From b5dfdc1c3f4cb760685b61e3925e82ff894cf4ab Mon Sep 17 00:00:00 2001
From: zbeyens <zbeyens@udecode.dev>
Date: Thu, 24 Oct 2024 01:01:05 +0200
Subject: [PATCH] fix

---
 .changeset/stale-mails-rescue.md         |  5 +++++
 packages/ai/src/lib/transforms/undoAI.ts | 12 ++++++++++--
 2 files changed, 15 insertions(+), 2 deletions(-)
 create mode 100644 .changeset/stale-mails-rescue.md

diff --git a/.changeset/stale-mails-rescue.md b/.changeset/stale-mails-rescue.md
new file mode 100644
index 0000000000..acee783163
--- /dev/null
+++ 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
index 8b379e000b..e4861bce5c 100644
--- 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();
   }