Add public command to close the AI toolbar (#2380)

marcbouchenoire · web-flow · commit 3ac49222a49b · 2025-04-24T09:34:52.000+02:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,8 +2,9 @@
 
 ### `@liveblocks/tiptap`
 
-- Fix `FloatingToolbar` focus behavior in Safari.
+- Add `closeAi` Tiptap command to manually close the AI toolbar.
 - Fix `AiToolbar` focus behavior in Safari.
+- Fix `FloatingToolbar` focus behavior in Safari.
 
 ### `@liveblocks/lexical`
 
diff --git a/packages/liveblocks-react-tiptap/src/ai/AiExtension.ts b/packages/liveblocks-react-tiptap/src/ai/AiExtension.ts
@@ -169,6 +169,12 @@ export const AiExtension = Extension.create<
         return true;
       },
 
+      closeAi: () => () => {
+        (this.editor.commands as unknown as AiCommands).$closeAiToolbar();
+
+        return true;
+      },
+
       $acceptAiToolbarResponse:
         () =>
         ({ tr, view }: CommandProps) => {
diff --git a/packages/liveblocks-react-tiptap/src/types.ts b/packages/liveblocks-react-tiptap/src/types.ts
@@ -265,8 +265,16 @@ export type CommentsCommands<ReturnType = boolean> = {
 };
 
 export type AiCommands<ReturnType = boolean> = {
+  /**
+   * Open the AI toolbar, with an optional prompt.
+   */
   askAi: (prompt?: string) => ReturnType;
 
+  /**
+   * Close the AI toolbar.
+   */
+  closeAi: () => ReturnType;
+
   // Transitions (see AiToolbarState)
 
   /**
