From 5893e177d4adeeb74cee6ce3fd35a1431eb855a7 Mon Sep 17 00:00:00 2001
From: Matthew Lipski <matthewlipski@gmail.com>
Date: Thu, 4 Dec 2025 18:23:42 +0100
Subject: [PATCH] Fixed AI menu not updating position on new line

---
 packages/xl-ai/src/components/AIMenu/AIMenuController.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx b/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx
index f3615bd70c..6072d1b911 100644
--- a/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx
+++ b/packages/xl-ai/src/components/AIMenu/AIMenuController.tsx
@@ -5,7 +5,7 @@ import {
   useExtension,
   useExtensionState,
 } from "@blocknote/react";
-import { offset, size } from "@floating-ui/react";
+import { autoUpdate, offset, size } from "@floating-ui/react";
 import { FC, useMemo } from "react";
 
 import { AIExtension } from "../../AIExtension.js";
@@ -55,6 +55,11 @@ export const AIMenuController = (props: {
             ai.rejectChanges();
           }
         },
+        whileElementsMounted(reference, floating, update) {
+          return autoUpdate(reference, floating, update, {
+            animationFrame: true,
+          });
+        },
       },
       useDismissProps: {
         enabled: