@blocknote/xl-ai 0.32.0 → 0.34.0

package/dist/blocknote-xl-ai.js

@@ -1,33 +1,33 @@
 var et = Object.defineProperty;
 var tt = (e, t, o) => t in e ? et(e, t, { enumerable: !0, configurable: !0, writable: !0, value: o }) : e[t] = o;
-var j = (e, t, o) => tt(e, typeof t != "symbol" ? t + "" : t, o);
+var D = (e, t, o) => tt(e, typeof t != "symbol" ? t + "" : t, o);
 import { getPmSchema as ot, UnreachableCaseError as Pe, getNodeById as rt, updateBlockTr as nt, insertBlocks as st, trackPosition as se, removeAndInsertBlocks as it, getBlock as Oe, defaultProps as Me, isStyledTextInlineContent as at, isLinkInlineContent as ct, BlockNoteExtension as lt, filterSuggestionItems as ut, mergeCSSClasses as dt } from "@blocknote/core";
-import { applySuggestions as Be, suggestChanges as pt, revertSuggestions as ie } from "@blocknote/prosemirror-suggest-changes";
-import { jsonSchema as xe, streamObject as mt, generateObject as ft, APICallError as ht, RetryError as yt } from "ai";
+import { applySuggestions as Be, suggestChanges as mt, revertSuggestions as ie } from "@blocknote/prosemirror-suggest-changes";
+import { jsonSchema as xe, streamObject as pt, generateObject as ft, APICallError as ht, RetryError as yt } from "ai";
 import { TextSelection as gt, Plugin as Le, PluginKey as _e } from "prosemirror-state";
 import { fixTablesKey as kt } from "prosemirror-tables";
 import bt from "lodash.isequal";
 import { Slice as Y, Fragment as ae } from "prosemirror-model";
-import { Transform as N, Mapping as Ee, ReplaceStep as L, ReplaceAroundStep as St } from "prosemirror-transform";
+import { Transform as A, Mapping as Ie, ReplaceStep as _, ReplaceAroundStep as St } from "prosemirror-transform";
 import { ChangeSet as ce, simplifyChanges as wt } from "prosemirror-changeset";
 import { DecorationSet as le, Decoration as ue } from "prosemirror-view";
 import { defaultSelectionBuilder as Tt } from "y-prosemirror";
 import { jsxs as vt, jsx as y } from "react/jsx-runtime";
 import { useBlockNoteContext as Ct, useComponentsContext as G, useSuggestionMenuKeyboardHandler as Pt, useBlockNoteEditor as K, useUIElementPositioning as Ot } from "@blocknote/react";
-import B, { useState as Ie, useCallback as D, useMemo as E, useEffect as Ne } from "react";
+import B, { useState as Ee, useCallback as $, useMemo as E, useEffect as Ne } from "react";
 import { useStore as Ae } from "zustand";
 import { offset as Mt, size as Bt, autoUpdate as xt } from "@floating-ui/react";
 const de = (e) => {
   let t;
-  const o = /* @__PURE__ */ new Set(), r = (u, m) => {
+  const o = /* @__PURE__ */ new Set(), r = (u, p) => {
     const l = typeof u == "function" ? u(t) : u;
     if (!Object.is(l, t)) {
       const f = t;
-      t = m ?? (typeof l != "object" || l === null) ? l : Object.assign({}, t, l), o.forEach((p) => p(t, f));
+      t = p ?? (typeof l != "object" || l === null) ? l : Object.assign({}, t, l), o.forEach((m) => m(t, f));
     }
-  }, n = () => t, s = { setState: r, getState: n, getInitialState: () => c, subscribe: (u) => (o.add(u), () => o.delete(u)) }, c = t = e(r, n, s);
-  return s;
-}, pe = (e) => e ? de(e) : de;
+  }, n = () => t, i = { setState: r, getState: n, getInitialState: () => c, subscribe: (u) => (o.add(u), () => o.delete(u)) }, c = t = e(r, n, i);
+  return i;
+}, me = (e) => e ? de(e) : de;
 function Lt(e) {
   const { properties: t, required: o, $defs: r, ...n } = e.parameters;
   return {
@@ -84,7 +84,7 @@ function _t(e) {
     }
   });
 }
-function I(e) {
+function N(e) {
   if (e.locked)
     throw new Error(
       "Stream (source) is already locked and cannot be iterated."
@@ -103,22 +103,22 @@ function I(e) {
   }, t;
 }
 function De(e) {
-  return I(_t(e));
+  return N(_t(e));
 }
-async function* Et(e) {
+async function* It(e) {
   var n;
   let t = 0, o = !0, r;
-  for await (const i of e)
-    if ((n = i.operations) != null && n.length) {
-      for (let a = t; a < i.operations.length; a++) {
-        const s = i.operations[a];
-        r = s, yield {
-          partialOperation: s,
+  for await (const s of e)
+    if ((n = s.operations) != null && n.length) {
+      for (let a = t; a < s.operations.length; a++) {
+        const i = s.operations[a];
+        r = i, yield {
+          partialOperation: i,
           isUpdateToPreviousOperation: a === t && !o,
-          isPossiblyPartial: a === i.operations.length - 1
+          isPossiblyPartial: a === s.operations.length - 1
         }, o = !1;
       }
-      t = i.operations.length - 1;
+      t = s.operations.length - 1;
     }
   if (!r)
     throw new Error("No operations seen");
@@ -145,7 +145,7 @@ async function* $e(e, t) {
 async function* Re(e, t) {
   for await (const o of e) {
     const r = t.find(
-      (i) => i.name === o.partialOperation.type
+      (s) => s.name === o.partialOperation.type
     );
     if (!r) {
       yield {
@@ -165,7 +165,7 @@ async function* Re(e, t) {
     };
   }
 }
-async function* It(e, t) {
+async function* Et(e, t) {
   const o = Re(
     e,
     t
@@ -196,30 +196,31 @@ async function At(e, t) {
     throw new Error(
       "Cannot provide output or schema in _generateObjectOptions"
     );
-  const i = {
+  const s = {
     // non-overridable options for streamObject
     output: "object",
     schema: xe(je(e)),
     // configurable options for streamObject
     // - optional, with defaults
     // mistral somehow needs "auto", while groq/llama needs "tool"
+    // google needs "auto" because https://github.com/vercel/ai/issues/6959
     // TODO: further research this and / or make configurable
     // for now stick to "tool" by default as this has been tested mostly
-    mode: r.model.provider === "mistral.chat" ? "auto" : "tool",
+    mode: r.model.provider === "mistral.chat" || r.model.provider === "google.generative-ai" ? "auto" : "tool",
     //  - mandatory ones:
     ...r,
     // extra options for streamObject
     ...o ?? {}
-  }, a = await ft(i), s = jt(a.object);
-  if (!s.ok)
-    throw new Error(s.error);
+  }, a = await ft(s), i = jt(a.object);
+  if (!i.ok)
+    throw new Error(i.error);
   let c;
   return {
     streamObjectResult: void 0,
     generateObjectResult: a,
     get operationsSource() {
       return c || (c = De(
-        Nt(s.value, e)
+        Nt(i.value, e)
       )), c;
     },
     async getGeneratedOperations() {
@@ -259,36 +260,37 @@ async function Dt(e, t, o = () => {
     // configurable options for streamObject
     // - optional, with defaults
     // mistral somehow needs "auto", while groq/llama needs "tool"
+    // google needs "auto" because https://github.com/vercel/ai/issues/6959
     // TODO: further research this and / or make configurable
     // for now stick to "tool" by default as this has been tested mostly
-    mode: n.model.provider === "mistral.chat" ? "auto" : "tool",
+    mode: n.model.provider === "mistral.chat" || n.model.provider === "google.generative-ai" ? "auto" : "tool",
     //  - mandatory ones:
     ...n,
     // extra options for streamObject
     ...t._streamObjectOptions ?? {}
-  }, s = mt(a);
+  }, i = pt(a);
   let c;
-  const [u, m] = s.fullStream.tee(), l = (async () => {
+  const [u, p] = i.fullStream.tee(), l = (async () => {
     let f = {
       operations: []
     };
-    const p = I(
-      Ht(m)
+    const m = N(
+      Ht(p)
     );
-    for await (const h of p)
+    for await (const h of m)
       h && typeof h == "object" && "operations" in h && (f = h);
     return f;
   })();
   return {
-    streamObjectResult: s,
+    streamObjectResult: i,
     generateObjectResult: void 0,
     get operationsSource() {
       return c || (c = De(
-        It(
-          Et(
+        Et(
+          It(
             $t(
               Rt(
-                I(u)
+                N(u)
               ),
               o
             )
@@ -308,7 +310,7 @@ async function* $t(e, t) {
     o && (t(), o = !1), yield r;
 }
 function Rt(e) {
-  return I(
+  return N(
     e.pipeThrough(
       new TransformStream({
         transform(t, o) {
@@ -333,7 +335,7 @@ function Rt(e) {
   );
 }
 function Ht(e) {
-  return I(
+  return N(
     e.pipeThrough(
       new TransformStream({
         transform(t, o) {
@@ -394,14 +396,32 @@ class qt {
       console.log(JSON.stringify(t, null, 2));
   }
 }
-function z(e, t) {
+function Ut(e, t, o = !0, r = !0) {
+  let n = 0, s = e.length;
+  if (o)
+    for (; n < s && t(e[n]); )
+      n++;
+  if (r)
+    for (; s > n && t(e[s - 1]); )
+      s--;
+  return e.slice(n, s);
+}
+function x(e, t) {
+  return Ut(
+    e,
+    (r) => X(r) && (t == null ? void 0 : t.cursorBlockId) !== r.id,
+    (t == null ? void 0 : t.trimStart) ?? !1,
+    (t == null ? void 0 : t.trimEnd) ?? !0
+  );
+}
+function F(e, t) {
   const o = e.getTextCursorPosition(), r = [];
   for (const n of t) {
-    const i = n.id === o.block.id;
+    const s = n.id === o.block.id;
     r.push({
       id: n.id,
       block: n.block
-    }), i && r.push({
+    }), s && r.push({
       cursor: !0
     });
   }
@@ -422,39 +442,23 @@ function T(e) {
     ...T(t.children)
   ]);
 }
-function x(e) {
+function L(e) {
   return e.map((t) => typeof t == "object" && t && "id" in t ? {
     ...t,
     id: `${t.id}$`
   } : t);
 }
-function Ut(e, t, o = !0, r = !0) {
-  let n = 0, i = e.length;
-  if (o)
-    for (; n < i && t(e[n]); )
-      n++;
-  if (r)
-    for (; i > n && t(e[i - 1]); )
-      i--;
-  return e.slice(n, i);
-}
-function F(e, t) {
-  return e.length === 1 ? e : Ut(
-    e,
-    (r) => X(r),
-    (t == null ? void 0 : t.trimStart) ?? !1,
-    (t == null ? void 0 : t.trimEnd) ?? !0
-  );
-}
 async function He(e, t) {
-  const o = F(e.document), r = await C(
-    T(o),
-    async (s) => e.blocksToHTMLLossy([s])
-  ), i = z(e, r).filter(
-    (s) => "cursor" in s || !(t.excludeBlockIds || []).includes(s.id)
+  const o = e.getTextCursorPosition().block.id, r = x(e.document, {
+    cursorBlockId: o
+  }), n = await C(
+    T(r),
+    async (c) => e.blocksToHTMLLossy([c])
+  ), a = F(e, n).filter(
+    (c) => "cursor" in c || !(t.excludeBlockIds || []).includes(c.id)
   );
   return {
-    htmlBlocks: x(i)
+    htmlBlocks: L(a)
   };
 }
 async function qe(e, t) {
@@ -463,7 +467,7 @@ async function qe(e, t) {
     async (n) => e.blocksToHTMLLossy([n])
   );
   return {
-    htmlSelectedBlocks: x(o),
+    htmlSelectedBlocks: L(o),
     htmlDocument: (await C(T(e.document), async (n) => e.blocksToHTMLLossy([n]))).map(({ block: n }) => ({ block: n }))
     // strip ids so LLM can't accidentally issue updates to ids not in selection
   };
@@ -521,8 +525,9 @@ function Ft(e) {
   ];
 }
 const Jt = async (e, t) => {
+  const o = x(e.document).length === 0;
   if (t.selectedBlocks) {
-    const o = await qe(e, {
+    const r = await qe(e, {
       selectedBlocks: t.selectedBlocks
     });
     return t.previousMessages ? [
@@ -534,7 +539,7 @@ const Jt = async (e, t) => {
       },
       {
         role: "system",
-        content: JSON.stringify(o.htmlSelectedBlocks)
+        content: JSON.stringify(r.htmlSelectedBlocks)
       },
       {
         role: "system",
@@ -542,7 +547,7 @@ const Jt = async (e, t) => {
       },
       {
         role: "system",
-        content: JSON.stringify(o.htmlDocument)
+        content: JSON.stringify(r.htmlDocument)
       },
       {
         role: "system",
@@ -556,12 +561,11 @@ const Jt = async (e, t) => {
         content: t.userPrompt
       }
     ] : zt({
-      ...o,
-      userPrompt: t.userPrompt,
-      isEmptyDocument: e.isEmpty
+      ...r,
+      userPrompt: t.userPrompt
     });
   } else {
-    const o = await He(e, t);
+    const r = await He(e, t);
     return t.previousMessages ? [
       ...t.previousMessages,
       {
@@ -571,7 +575,7 @@ const Jt = async (e, t) => {
       },
       {
         role: "system",
-        content: JSON.stringify(o.htmlBlocks)
+        content: JSON.stringify(r.htmlBlocks)
       },
       {
         role: "system",
@@ -585,9 +589,9 @@ const Jt = async (e, t) => {
         content: t.userPrompt
       }
     ] : Ft({
-      ...o,
+      ...r,
       userPrompt: t.userPrompt,
-      isEmptyDocument: e.isEmpty
+      isEmptyDocument: o
     });
   }
 };
@@ -595,36 +599,36 @@ function Vt(e) {
   let t, o = !1;
   return e.descendants((r, n) => o ? !1 : (r.isText && (o = !0, t = n), !0)), t;
 }
-function $(e) {
-  var i;
-  const t = ot(e), { modification: o } = t.marks, r = [], n = new N(e.before);
+function R(e) {
+  var s;
+  const t = ot(e), { modification: o } = t.marks, r = [], n = new A(e.before);
   for (let a = 0; a < e.steps.length; a++) {
-    const s = e.steps[a], c = new Ee(e.mapping.maps.slice(0, a)).invert();
-    if (s.structure) {
-      if (s instanceof L) {
-        if (s.to !== s.from + 1 || s.slice.openStart !== 0 || s.slice.openEnd !== 1 || s.slice.content.size !== 2)
+    const i = e.steps[a], c = new Ie(e.mapping.maps.slice(0, a)).invert();
+    if (i.structure) {
+      if (i instanceof _) {
+        if (i.to !== i.from + 1 || i.slice.openStart !== 0 || i.slice.openEnd !== 1 || i.slice.content.size !== 2)
           throw new Error(
             "Structure change is not in expected format (ReplaceStep)"
           );
-      } else if (s instanceof St) {
-        if (s.insert !== 1 || s.slice.size !== 2 || s.gapTo !== s.to - 1 || s.gapFrom !== s.from + 1)
+      } else if (i instanceof St) {
+        if (i.insert !== 1 || i.slice.size !== 2 || i.gapTo !== i.to - 1 || i.gapFrom !== i.from + 1)
           throw new Error(
             "Structure change is not in expected format (ReplaceAroundStep)"
           );
       } else
         throw new Error("Step is not a ReplaceStep or ReplaceAroundStep");
-      const b = c.map(s.from), g = s.slice.content.firstChild, k = n.doc.resolve(n.mapping.map(b)).nodeAfter;
+      const b = c.map(i.from), g = i.slice.content.firstChild, k = n.doc.resolve(n.mapping.map(b)).nodeAfter;
       let w = g.marks || [];
       g.type !== k.type && (w = o.create({
         type: "nodeType",
         previousValue: k.type.name,
         newValue: g.type.name
       }).addToSet(w));
-      const A = /* @__PURE__ */ new Set([
+      const j = /* @__PURE__ */ new Set([
         ...Object.keys(g.attrs),
         ...Object.keys(k.attrs)
       ]);
-      for (const P of A)
+      for (const P of j)
         g.attrs[P] !== k.attrs[P] && (w = o.create({
           type: "attr",
           attrName: P,
@@ -644,10 +648,10 @@ function $(e) {
       });
       continue;
     }
-    if (!(s instanceof L))
+    if (!(i instanceof _))
       throw new Error("Step is not a ReplaceStep");
-    const u = c.map(s.from), m = c.map(s.to);
-    if (s.slice.openStart > 0 || s.slice.openEnd > 0)
+    const u = c.map(i.from), p = c.map(i.to);
+    if (i.slice.openStart > 0 || i.slice.openEnd > 0)
       throw new Error(
         "Slice has openStart or openEnd > 0, but structure=false"
       );
@@ -655,32 +659,32 @@ function $(e) {
       prosemirrorSteps: [],
       selection: {
         anchor: n.mapping.map(u),
-        head: n.mapping.map(m)
+        head: n.mapping.map(p)
       },
       type: "select"
     });
-    const l = s.slice.content.textBetween(0, s.slice.size), f = l === n.doc.textBetween(n.mapping.map(u), n.mapping.map(m));
-    let p;
+    const l = i.slice.content.textBetween(0, i.slice.size), f = l === n.doc.textBetween(n.mapping.map(u), n.mapping.map(p));
+    let m;
     if (f)
-      p = s.slice.content.size;
+      m = i.slice.content.size;
     else if (l.length === 0)
-      p = s.slice.content.size;
+      m = i.slice.content.size;
     else {
-      const b = Vt(s.slice.content);
+      const b = Vt(i.slice.content);
       if (b === void 0)
         throw new Error("unexpected: no first character found");
-      p = b + 1;
+      m = b + 1;
     }
-    let h = n.mapping.map(m);
-    const d = n.mapping.map(m);
+    let h = n.mapping.map(p);
+    const d = n.mapping.map(p);
     let S = !0;
-    for (let b = p; b <= s.slice.content.size; b++) {
-      const g = S && u !== m, k = n.steps.length;
+    for (let b = m; b <= i.slice.content.size; b++) {
+      const g = S && u !== p, k = n.steps.length;
       if (g) {
         const M = n.doc.resolve(n.mapping.map(u));
-        (i = M.nodeAfter) != null && i.isBlock && n.addNodeMark(M.pos, t.mark("deletion", {})), n.addMark(M.pos, h, t.mark("deletion", {})), h = n.mapping.map(m);
+        (s = M.nodeAfter) != null && s.isBlock && n.addNodeMark(M.pos, t.mark("deletion", {})), n.addMark(M.pos, h, t.mark("deletion", {})), h = n.mapping.map(p);
       }
-      const w = new Y(s.slice.content.cut(0, b), 0, 0);
+      const w = new Y(i.slice.content.cut(0, b), 0, 0);
       n.replace(d, h, w).addMark(
         d,
         d + w.content.size,
@@ -690,15 +694,15 @@ function $(e) {
         d + w.content.size,
         (M, P) => P < d || P > d + w.content.size ? !0 : (M.isBlock && n.addNodeMark(P, t.mark("insertion", {})), !1)
       ), h = n.mapping.slice(k).map(h);
-      const A = gt.near(
+      const j = gt.near(
         n.doc.resolve(d + w.content.size),
         -1
       );
       r.push({
         prosemirrorSteps: n.steps.slice(k),
         selection: {
-          anchor: A.from,
-          head: A.from
+          anchor: j.from,
+          head: j.from
         },
         type: g ? "replace" : "insert"
         // 3. Insert the replacement character by character
@@ -731,52 +735,52 @@ function ee(e, t) {
   return e;
 }
 function Zt(e, t, o) {
-  const r = new N(t);
+  const r = new A(t);
   for (const a of e) {
-    const s = new L(
+    const i = new _(
       r.mapping.map(a.fromA),
       r.mapping.map(a.toA),
       o.slice(a.fromB, a.toB)
     );
-    r.step(s);
+    r.step(i);
   }
   const n = r.mapping.invert();
-  let i = r.doc.content.findDiffStart(o.content);
-  for (; i !== null; ) {
-    const a = o.resolve(i).nodeAfter, s = r.doc.resolve(i).nodeAfter;
-    if (!a || !s)
+  let s = r.doc.content.findDiffStart(o.content);
+  for (; s !== null; ) {
+    const a = o.resolve(s).nodeAfter, i = r.doc.resolve(s).nodeAfter;
+    if (!a || !i)
       throw new Error("diffNode not found");
-    const c = !a.isLeaf && a.content.eq(s.content), u = c ? 1 : Math.min(a.nodeSize, s.nodeSize), m = i + u, l = n.map(i), f = n.map(m);
-    let p = e.length;
+    const c = !a.isLeaf && a.content.eq(i.content), u = c ? 1 : Math.min(a.nodeSize, i.nodeSize), p = s + u, l = n.map(s), f = n.map(p);
+    let m = e.length;
     for (let d = 0; d < e.length; d++)
       if (e[d].fromA >= f) {
-        p = d;
+        m = d;
         break;
       }
-    e.splice(p, 0, {
+    e.splice(m, 0, {
       fromA: l,
       toA: f,
-      fromB: i,
-      toB: m,
+      fromB: s,
+      toB: p,
       deleted: [],
       inserted: [],
       type: c ? "node-type-or-attr-update" : "mark-update"
     }), r.step(
-      new L(
-        i,
-        m,
-        o.slice(i, m),
+      new _(
+        s,
+        p,
+        o.slice(s, p),
         c
       )
     );
     const h = r.doc.content.findDiffStart(o.content);
-    if (h === i)
+    if (h === s)
       throw new Error("diffStart not moving");
-    i = h;
+    s = h;
   }
   return e;
 }
-const me = (e, t) => {
+const pe = (e, t) => {
   const o = /* @__PURE__ */ new Set(), r = /* @__PURE__ */ new Set();
   e.descendants((a) => {
     a.type.name === "tableCell" && o.add(JSON.stringify(a.toJSON()));
@@ -790,58 +794,58 @@ const me = (e, t) => {
     encodeCharacter: (a) => a,
     encodeNodeStart: (a) => {
       if (a.type.name === "tableCell") {
-        const s = JSON.stringify(a.toJSON());
-        return n.has(s) ? s : a.type.name;
+        const i = JSON.stringify(a.toJSON());
+        return n.has(i) ? i : a.type.name;
       }
       return a.type.name;
     },
     encodeNodeEnd: (a) => {
       if (a.type.name === "tableCell") {
-        const s = JSON.stringify(a.toJSON());
-        return n.has(s) ? s : -1;
+        const i = JSON.stringify(a.toJSON());
+        return n.has(i) ? i : -1;
       }
       return -1;
     },
-    compareTokens: (a, s) => a === s
+    compareTokens: (a, i) => a === i
   };
 };
 function J(e, t, o = !1, r, n) {
-  const i = rt(e.id, t), a = new N(t);
+  const s = rt(e.id, t), a = new A(t);
   nt(
     a,
-    i.posBeforeNode,
+    s.posBeforeNode,
     e.block,
     r,
     n
   );
-  let s = a.doc, c = ce.create(
+  let i = a.doc, c = ce.create(
     t,
     void 0,
-    me(t, s)
+    pe(t, i)
   );
-  if (c = c.addSteps(s, a.mapping.maps, 0), o && c.changes.length > 0) {
-    const l = c.changes[c.changes.length - 1], f = l.toA - l.fromA, p = l.toB - l.fromB;
-    if (f > p) {
+  if (c = c.addSteps(i, a.mapping.maps, 0), o && c.changes.length > 0) {
+    const l = c.changes[c.changes.length - 1], f = l.toA - l.fromA, m = l.toB - l.fromB;
+    if (f > m) {
       const h = t.slice(
-        l.fromA + p,
+        l.fromA + m,
         l.toA
       );
       a.step(
-        new L(l.toB, l.toB, h)
-      ), s = a.doc, c = ce.create(
+        new _(l.toB, l.toB, h)
+      ), i = a.doc, c = ce.create(
         c.startDoc,
         void 0,
-        me(c.startDoc, s)
-      ), c = c.addSteps(s, a.mapping.maps, 0);
+        pe(c.startDoc, i)
+      ), c = c.addSteps(i, a.mapping.maps, 0);
     }
   }
-  const u = [], m = wt(
+  const u = [], p = wt(
     c.changes,
-    s
+    i
   );
-  for (let l = 0; l < m.length; l++) {
-    const f = m[l], p = s.slice(f.fromB, f.toB);
-    if (p.openEnd === 1 && p.openStart === 0 && (f.type = "node-type-or-attr-update", p.size > 2)) {
+  for (let l = 0; l < p.length; l++) {
+    const f = p[l], m = i.slice(f.fromB, f.toB);
+    if (m.openEnd === 1 && m.openStart === 0 && (f.type = "node-type-or-attr-update", m.size > 2)) {
       const h = {
         fromA: f.fromA,
         toA: f.fromA + 1,
@@ -858,21 +862,21 @@ function J(e, t, o = !1, r, n) {
         deleted: [],
         inserted: []
       };
-      m.splice(l, 1, h, d), l++;
+      p.splice(l, 1, h, d), l++;
     }
   }
-  Zt(m, t, s);
-  for (let l = 0; l < m.length; l++) {
-    const f = m[l], p = s.slice(f.fromB, f.toB);
-    if (p.openEnd > 0 && p.size > 1)
+  Zt(p, t, i);
+  for (let l = 0; l < p.length; l++) {
+    const f = p[l], m = i.slice(f.fromB, f.toB);
+    if (m.openEnd > 0 && m.size > 1)
       throw new Error(
         "unexpected, openEnd > 0 and size > 1, this should have been split into two steps"
       );
-    l === m.length - 1 && o && f.type === "mark-update" || u.push(
-      new L(
+    l === p.length - 1 && o && f.type === "mark-update" || u.push(
+      new _(
         f.fromA,
         f.toA,
-        p,
+        m,
         f.type === "node-type-or-attr-update"
       )
     );
@@ -942,83 +946,83 @@ function te(e) {
             ok: !1,
             error: "referenceId and blocks are required"
           };
-        let i = n.referenceId;
+        let s = n.referenceId;
         if (o.idsSuffixed) {
-          if (!(i != null && i.endsWith("$")))
+          if (!(s != null && s.endsWith("$")))
             return {
               ok: !1,
               error: "referenceId must end with $"
             };
-          i = i.slice(0, -1);
+          s = s.slice(0, -1);
         }
-        if (!t.getBlock(i))
+        if (!t.getBlock(s))
           return {
             ok: !1,
             error: "referenceId not found"
           };
-        const s = Wt(
+        const i = Wt(
           n.blocks,
           (c) => e.validateBlock(c, t)
         );
-        return s.ok ? {
+        return i.ok ? {
           ok: !0,
           value: {
             type: n.type,
-            referenceId: i,
+            referenceId: s,
             position: n.position,
-            blocks: s.value
+            blocks: i.value
           }
-        } : s;
+        } : i;
       },
       // Note: functionality mostly tested in jsontools.test.ts
       // would be nicer to add a direct unit test
       execute: async function* (n) {
-        var s;
-        let i = [];
+        var i;
+        let s = [];
         const a = {};
         for await (const c of n) {
-          if (c.isUpdateToPreviousOperation || (i = []), c.operation.type !== "add") {
+          if (c.isUpdateToPreviousOperation || (s = []), c.operation.type !== "add") {
             yield c;
             continue;
           }
-          const u = c.operation, m = await e.toJSONToolCall(t, {
+          const u = c.operation, p = await e.toJSONToolCall(t, {
             ...c,
             operation: u
           });
-          if (m && !(c.isPossiblyPartial && X(
-            m.blocks[m.blocks.length - 1]
+          if (p && !(c.isPossiblyPartial && X(
+            p.blocks[p.blocks.length - 1]
           ))) {
-            for (let l = 0; l < m.blocks.length; l++) {
-              const f = m.blocks[l], p = t.prosemirrorState.tr;
+            for (let l = 0; l < p.blocks.length; l++) {
+              const f = p.blocks[l], m = t.prosemirrorState.tr;
               let h = [];
-              if (l < i.length) {
-                const d = await e.rebaseTool(i[l], t), b = J(
+              if (l < s.length) {
+                const d = await e.rebaseTool(s[l], t), b = J(
                   {
-                    id: i[l],
+                    id: s[l],
                     block: f
                   },
                   d.doc,
                   !1
                 ).map((g) => g.map(d.invertMap));
                 for (const g of b)
-                  p.step(g.map(p.mapping));
-                h = $(p), h = h.filter((g) => g.type !== "select");
+                  m.step(g.map(m.mapping));
+                h = R(m), h = h.filter((g) => g.type !== "select");
               } else {
                 const d = u.position === "after" ? a[u.referenceId] : void 0, S = st(
-                  p,
+                  m,
                   [f],
-                  l > 0 ? i[l - 1] : d || u.referenceId,
+                  l > 0 ? s[l - 1] : d || u.referenceId,
                   l > 0 ? "after" : u.position
                 );
-                i.push(...S.map((b) => b.id)), h = $(p);
+                s.push(...S.map((b) => b.id)), h = R(m);
               }
               h.find((d) => d.type === "replace");
               for (const d of h)
                 o.withDelays && await Q(d), t.transact((S) => {
                   ee(S, d);
-                }), (s = o.onBlockUpdate) == null || s.call(o, i[l]);
+                }), (i = o.onBlockUpdate) == null || i.call(o, s[l]);
             }
-            c.isPossiblyPartial || u.position === "after" && (a[u.referenceId] = i[i.length - 1]);
+            c.isPossiblyPartial || u.position === "after" && (a[u.referenceId] = s[s.length - 1]);
           }
         }
       }
@@ -1054,42 +1058,42 @@ function oe(e) {
             ok: !1,
             error: "id is required"
           };
-        let i = n.id;
+        let s = n.id;
         if (o.idsSuffixed) {
-          if (!(i != null && i.endsWith("$")))
+          if (!(s != null && s.endsWith("$")))
             return {
               ok: !1,
               error: "id must end with $"
             };
-          i = i.slice(0, -1);
+          s = s.slice(0, -1);
         }
         if (!n.block)
           return {
             ok: !1,
             error: "block is required"
           };
-        const a = t.getBlock(i);
+        const a = t.getBlock(s);
         if (!a)
-          return console.error("BLOCK NOT FOUND", i), {
+          return console.error("BLOCK NOT FOUND", s), {
             ok: !1,
             error: "block not found"
           };
-        const s = e.validateBlock(n.block, t, a.type);
-        return s.ok ? {
+        const i = e.validateBlock(n.block, t, a.type);
+        return i.ok ? {
           ok: !0,
           value: {
             type: n.type,
-            id: i,
-            block: s.value
+            id: s,
+            block: i.value
           }
-        } : s;
+        } : i;
       },
       // Note: functionality mostly tested in jsontools.test.ts
       // would be nicer to add a direct unit test
       execute: async function* (n) {
         var c;
         let a = 50;
-        const s = o.updateSelection ? {
+        const i = o.updateSelection ? {
           from: se(t, o.updateSelection.from),
           to: se(t, o.updateSelection.to)
         } : void 0;
@@ -1098,15 +1102,15 @@ function oe(e) {
             yield u;
             continue;
           }
-          const m = u.operation;
+          const p = u.operation;
           if (u.isPossiblyPartial) {
-            const k = JSON.stringify(m.block).length;
+            const k = JSON.stringify(p.block).length;
             if (k < a)
               continue;
             a = k + 50;
           } else
             a = 50;
-          const l = await e.rebaseTool(m.id, t), f = s ? l.invertMap.invert().map(s.from()) : void 0, p = s ? l.invertMap.invert().map(s.to()) : void 0, h = await e.toJSONToolCall(t, u);
+          const l = await e.rebaseTool(p.id, t), f = i ? l.invertMap.invert().map(i.from()) : void 0, m = i ? l.invertMap.invert().map(i.to()) : void 0, h = await e.toJSONToolCall(t, u);
           if (!h)
             continue;
           const d = J(
@@ -1114,18 +1118,18 @@ function oe(e) {
             l.doc,
             u.isPossiblyPartial,
             f,
-            p
+            m
           );
           if (d.length === 1 && u.isPossiblyPartial)
             continue;
-          const S = d.map((k) => k.map(l.invertMap)), b = new N(t.prosemirrorState.doc);
+          const S = d.map((k) => k.map(l.invertMap)), b = new A(t.prosemirrorState.doc);
           for (const k of S)
             b.step(k.map(b.mapping));
-          const g = $(b);
+          const g = R(b);
           for (const k of g)
             o.withDelays && await Q(k), t.transact((w) => {
               ee(w, k);
-            }), (c = o.onBlockUpdate) == null || c.call(o, m.id);
+            }), (c = o.onBlockUpdate) == null || c.call(o, p.id);
         }
       }
     };
@@ -1184,13 +1188,13 @@ const re = (e, t) => ({
         yield n;
         continue;
       }
-      const i = n.operation, a = e.prosemirrorState.tr;
-      it(a, [i.id], []);
-      const s = $(a);
-      for (const c of s)
+      const s = n.operation, a = e.prosemirrorState.tr;
+      it(a, [s.id], []);
+      const i = R(a);
+      for (const c of i)
         t.withDelays && await Q(c), e.transact((u) => {
           ee(u, c);
-        }), (r = t.onBlockUpdate) == null || r.call(t, i.id);
+        }), (r = t.onBlockUpdate) == null || r.call(t, s.id);
     }
   }
 });
@@ -1204,13 +1208,13 @@ function fe(e) {
     0,
     r.length - n[0].length
   ));
-  const s = new DOMParser().parseFromString(
+  const i = new DOMParser().parseFromString(
     `<div>${r}</div>`,
     "text/html"
   ).body.firstChild;
-  return s ? s.innerHTML : "";
+  return i ? i.innerHTML : "";
 }
-function R(e) {
+function H(e) {
   let t;
   if (Be(e.prosemirrorState, (o) => {
     t = o;
@@ -1218,7 +1222,7 @@ function R(e) {
     throw new Error("applySuggestionsTr is not set");
   return t;
 }
-function H(e, t) {
+function q(e, t) {
   const o = t.mapping.invert();
   return {
     doc: t.doc,
@@ -1226,7 +1230,7 @@ function H(e, t) {
      * Return a new transform that has the projection applied.
      * You can add new operations to this transform and later rebase those on the original document with `rebaseTr`
      */
-    tr: () => new N(t.doc),
+    tr: () => new A(t.doc),
     /**
      * Invert map created by the projection.
      * You can use this to map positions on the "clean" document (the projection) to positions on the original document
@@ -1240,8 +1244,8 @@ function H(e, t) {
       if (r.steps.length === 0)
         throw new Error("No steps to apply");
       let n = e.prosemirrorState.tr;
-      for (const i of r.steps) {
-        const a = i.map(o);
+      for (const s of r.steps) {
+        const a = s.map(o);
         if (!a)
           throw new Error("Step is not mapped");
         n = n.step(a);
@@ -1252,7 +1256,7 @@ function H(e, t) {
 }
 async function he(e, t) {
   var u;
-  const o = R(t), r = Oe(o.doc, e);
+  const o = H(t), r = Oe(o.doc, e);
   if (!r)
     throw new Error("block not found");
   const n = await t.blocksToHTMLLossy([
@@ -1260,19 +1264,19 @@ async function he(e, t) {
       ...r,
       children: []
     }
-  ]), i = (u = window.__TEST_OPTIONS) == null ? void 0 : u.mockID, a = await t.tryParseHTMLToBlocks(n);
-  if (window.__TEST_OPTIONS && (window.__TEST_OPTIONS.mockID = i), a.length !== 1)
+  ]), s = (u = window.__TEST_OPTIONS) == null ? void 0 : u.mockID, a = await t.tryParseHTMLToBlocks(n);
+  if (window.__TEST_OPTIONS && (window.__TEST_OPTIONS.mockID = s), a.length !== 1)
     throw new Error("html diff invalid block count");
-  const s = a[0];
-  if (s.id = e, J(
+  const i = a[0];
+  if (i.id = e, J(
     {
       id: e,
-      block: s
+      block: i
     },
     o.doc
   ).length)
     throw new Error("html diff");
-  return H(t, o);
+  return q(t, o);
 }
 function ye(e) {
   return typeof e != "string" ? {
@@ -1288,13 +1292,8 @@ const V = {
     description: "Insert new blocks",
     schema: {
       block: {
-        $ref: "#/$defs/block"
-      },
-      $defs: {
-        block: {
-          type: "string",
-          description: "html of block (MUST be a single HTML element)"
-        }
+        type: "string",
+        description: "html of block (MUST be a single HTML element)"
       }
     },
     validateBlock: ye,
@@ -1302,10 +1301,10 @@ const V = {
     toJSONToolCall: async (e, t) => {
       var n;
       const o = (n = window.__TEST_OPTIONS) == null ? void 0 : n.mockID, r = (await Promise.all(
-        t.operation.blocks.map(async (i) => {
-          const a = t.isPossiblyPartial ? fe(i) : i;
+        t.operation.blocks.map(async (s) => {
+          const a = t.isPossiblyPartial ? fe(s) : s;
           return a ? (await e.tryParseHTMLToBlocks(a)).map(
-            (s) => (delete s.id, s)
+            (i) => (delete i.id, i)
           ) : [];
         })
       )).flat();
@@ -1320,13 +1319,8 @@ const V = {
     description: "Update a block",
     schema: {
       block: {
-        $ref: "#/$defs/block"
-      },
-      $defs: {
-        block: {
-          type: "string",
-          description: "html of block (MUST be a single HTML element)"
-        }
+        type: "string",
+        description: "html of block (MUST be a single HTML element)"
       }
     },
     validateBlock: ye,
@@ -1345,14 +1339,14 @@ const V = {
   delete: re
 };
 function Yt(e, t, o, r, n) {
-  const i = {
+  const s = {
     add: !0,
     update: !0,
     delete: !0,
     ...o
   };
   return [
-    ...i.update ? [
+    ...s.update ? [
       V.update(e, {
         idsSuffixed: !0,
         withDelays: t,
@@ -1360,8 +1354,8 @@ function Yt(e, t, o, r, n) {
         onBlockUpdate: n
       })
     ] : [],
-    ...i.add ? [V.add(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : [],
-    ...i.delete ? [V.delete(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : []
+    ...s.add ? [V.add(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : [],
+    ...s.delete ? [V.delete(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : []
   ];
 }
 const Ue = {
@@ -1387,12 +1381,12 @@ async function Gt(e, t) {
     userPrompt: o,
     useSelection: r,
     deleteEmptyCursorBlock: n,
-    stream: i,
+    stream: s,
     onStart: a,
-    withDelays: s,
+    withDelays: i,
     dataFormat: c,
     previousResponse: u,
-    ...m
+    ...p
   } = {
     maxRetries: 2,
     deleteEmptyCursorBlock: !0,
@@ -1400,14 +1394,15 @@ async function Gt(e, t) {
     withDelays: !0,
     dataFormat: Ue,
     ...t
-  }, l = t.promptBuilder ?? c.defaultPromptBuilder, f = c.getStreamTools, p = r ? void 0 : e.getTextCursorPosition().block, h = p && n && X(p) && e.document.length > 1 ? p.id : void 0, d = r ? e.getSelectionCutBlocks() : void 0;
+  }, l = t.promptBuilder ?? c.defaultPromptBuilder, f = c.getStreamTools, m = r ? void 0 : e.getTextCursorPosition().block, h = m && n && X(m) && x(e.document).length > 0 ? m.id : void 0, d = r ? e.getSelectionCutBlocks() : void 0;
   let S;
   u && (S = u.messages.map((w) => w.role === "user" && typeof w.content == "string" ? {
     role: "system",
     content: `USER_MESSAGE: ${w.content}`
   } : w), S.push({
-    role: "assistant",
-    content: `These are the operations returned by a previous LLM call: 
+    role: "system",
+    // using "assistant" here doesn't work with gemini because we can't mix system / assistant messages
+    content: `ASSISTANT_MESSAGE: These are the operations returned by a previous LLM call: 
 ` + JSON.stringify(
       await u.llmResult.getGeneratedOperations()
     )
@@ -1419,38 +1414,40 @@ async function Gt(e, t) {
     previousMessages: S
   }), g = f(
     e,
-    s,
+    i,
     t.defaultStreamTools,
     d ? { from: d._meta.startPos, to: d._meta.endPos } : void 0,
     t.onBlockUpdate
   );
   let k;
-  return i ? k = await Dt(
+  return s ? k = await Dt(
     g,
     {
       messages: b,
-      ...m
+      ...p
     },
     () => {
       h && e.removeBlocks([h]), a == null || a();
     }
   ) : (k = await At(g, {
     messages: b,
-    ...m
+    ...p
   }), h && e.removeBlocks([h]), a == null || a()), new qt(b, k, g);
 }
 async function ze(e, t) {
-  const o = F(e.document), r = await C(
-    T(o),
-    async (s) => ({
-      ...s,
+  const o = e.getTextCursorPosition().block.id, r = x(e.document, {
+    cursorBlockId: o
+  }), n = await C(
+    T(r),
+    async (c) => ({
+      ...c,
       children: void 0
     })
-  ), i = z(e, r).filter(
-    (s) => "cursor" in s || !(t.excludeBlockIds || []).includes(s.id)
+  ), a = F(e, n).filter(
+    (c) => "cursor" in c || !(t.excludeBlockIds || []).includes(c.id)
   );
   return {
-    jsonBlocks: x(i)
+    jsonBlocks: L(a)
   };
 }
 async function Fe(e, t) {
@@ -1459,7 +1456,7 @@ async function Fe(e, t) {
     async (n) => n
   );
   return {
-    jsonSelectedBlocks: x(o),
+    jsonSelectedBlocks: L(o),
     jsonDocument: (await C(T(e.document), async (n) => ({
       ...n,
       id: void 0,
@@ -1558,37 +1555,37 @@ function Xt(e) {
   ];
 }
 const Qt = async (e, t) => {
+  const o = x(e.document).length === 0;
   if (t.selectedBlocks) {
-    const o = await Fe(e, {
+    const r = await Fe(e, {
       selectedBlocks: t.selectedBlocks
     });
     return Kt({
-      ...o,
-      userPrompt: t.userPrompt,
-      isEmptyDocument: e.isEmpty
+      ...r,
+      userPrompt: t.userPrompt
     });
   } else {
-    const o = await ze(e, t);
+    const r = await ze(e, t);
     return Xt({
-      ...o,
+      ...r,
       userPrompt: t.userPrompt,
-      isEmptyDocument: e.isEmpty
+      isEmptyDocument: o
     });
   }
 };
 function eo(e) {
   const t = {}, o = {};
   return e.forEach((n) => {
-    const { type: i, ...a } = n.properties, s = JSON.stringify(a);
-    t[s] ? t[s].push(i.enum[0]) : (t[s] = [i.enum[0]], o[s] = n);
+    const { type: s, ...a } = n.properties, i = JSON.stringify(a);
+    t[i] ? t[i].push(s.enum[0]) : (t[i] = [s.enum[0]], o[i] = n);
   }), Object.keys(
     t
   ).map((n) => {
-    const i = o[n];
+    const s = o[n];
     return {
-      ...i,
+      ...s,
       properties: {
-        ...i.properties,
+        ...s.properties,
         type: {
           type: "string",
           enum: t[n]
@@ -1798,7 +1795,7 @@ function ke(e, t, o) {
         ok: !0,
         value: e
       };
-    if (!e.content.every((i) => Ve(i, t)))
+    if (!e.content.every((s) => Ve(s, t)))
       return {
         ok: !1,
         error: "block content must be an array of inline content"
@@ -1819,7 +1816,7 @@ const Z = {
       ...ge(e.schema)
     }),
     validateBlock: ke,
-    rebaseTool: async (e, t) => H(t, R(t)),
+    rebaseTool: async (e, t) => q(t, H(t)),
     toJSONToolCall: async (e, t) => t.operation
   }),
   update: oe({
@@ -1831,7 +1828,7 @@ const Z = {
       ...ge(e.schema)
     }),
     validateBlock: ke,
-    rebaseTool: async (e, t) => H(t, R(t)),
+    rebaseTool: async (e, t) => q(t, H(t)),
     toJSONToolCall: async (e, t) => {
       const o = Object.fromEntries(
         Object.entries(Me).map(([r, n]) => [r, n.default])
@@ -1888,14 +1885,16 @@ const ao = {
   }
 };
 async function Ze(e, t) {
-  const o = F(e.document), r = await C(
-    T(o),
-    async (s) => e.blocksToMarkdownLossy([s])
-  ), i = z(e, r).filter(
-    (s) => "cursor" in s || !(t.excludeBlockIds || []).includes(s.id)
+  const o = e.getTextCursorPosition().block.id, r = x(e.document, {
+    cursorBlockId: o
+  }), n = await C(
+    T(r),
+    async (c) => e.blocksToMarkdownLossy([c])
+  ), a = F(e, n).filter(
+    (c) => "cursor" in c || !(t.excludeBlockIds || []).includes(c.id)
   );
   return {
-    markdownBlocks: x(i)
+    markdownBlocks: L(a)
   };
 }
 async function We(e, t) {
@@ -1904,7 +1903,7 @@ async function We(e, t) {
     async (n) => e.blocksToMarkdownLossy([n])
   );
   return {
-    markdownSelectedBlocks: x(o),
+    markdownSelectedBlocks: L(o),
     markdownDocument: (await C(T(e.document), async (n) => e.blocksToMarkdownLossy([n]))).map(({ block: n }) => ({ block: n }))
     // strip ids so LLM can't accidentally issue updates to ids not in selection
   };
@@ -1983,20 +1982,20 @@ const uo = async (e, t) => {
   }
 };
 async function be(e, t) {
-  const o = R(t), r = await t.blocksToMarkdownLossy([Oe(o.doc, e)]), n = await t.tryParseMarkdownToBlocks(r), i = J(
+  const o = H(t), r = await t.blocksToMarkdownLossy([Oe(o.doc, e)]), n = await t.tryParseMarkdownToBlocks(r), s = J(
     {
       id: e,
       block: n[0]
     },
     o.doc
-  ), a = new Ee();
-  for (const s of i) {
-    const c = s.map(a);
+  ), a = new Ie();
+  for (const i of s) {
+    const c = i.map(a);
     if (!c)
       throw new Error("Failed to map step");
     o.step(c), a.appendMap(c.getMap());
   }
-  return H(t, o);
+  return q(t, o);
 }
 function Se(e) {
   return typeof e != "string" ? {
@@ -2055,7 +2054,7 @@ const W = {
   }),
   delete: re
 };
-function po(e, t, o, r) {
+function mo(e, t, o, r) {
   const n = {
     add: !0,
     update: !0,
@@ -2074,11 +2073,11 @@ function po(e, t, o, r) {
     ...n.delete ? [W.delete(e, { idsSuffixed: !0, withDelays: t })] : []
   ];
 }
-const mo = {
+const po = {
   /**
    * Function to get the stream tools that can apply BlockNote Markdown block updates to the editor
    */
-  getStreamTools: po,
+  getStreamTools: mo,
   /**
    * The default PromptBuilder that determines how a userPrompt is converted to an array of
    * LLM Messages (CoreMessage[])
@@ -2092,14 +2091,14 @@ const mo = {
     getDataForPromptWithSelection: We
   }
 }, ir = {
-  addCursorPosition: z,
+  addCursorPosition: F,
   flattenBlocks: T,
-  suffixIDs: x,
-  trimEmptyBlocks: F,
+  suffixIDs: L,
+  trimEmptyBlocks: x,
   convertBlocks: C
 }, fo = {
   _experimental_json: ao,
-  _experimental_markdown: mo,
+  _experimental_markdown: po,
   html: Ue
 }, we = new _e("blocknote-agent-cursor");
 function ho(e) {
@@ -2130,9 +2129,9 @@ function ho(e) {
             side: 10
           })
         );
-        const i = Math.min(r.anchor, r.head), a = Math.max(r.anchor, r.head);
+        const s = Math.min(r.anchor, r.head), a = Math.max(r.anchor, r.head);
         return n.push(
-          ue.inline(i, a, Tt(e), {
+          ue.inline(s, a, Tt(e), {
             inclusiveEnd: !0,
             inclusiveStart: !1
           })
@@ -2155,17 +2154,17 @@ class Ye extends lt {
    */
   constructor(o, r) {
     super();
-    j(this, "previousRequestOptions");
+    D(this, "previousRequestOptions");
     // internal store including setters
-    j(this, "_store", pe()((o) => ({
+    D(this, "_store", me()((o) => ({
       aiMenuState: "closed"
     })));
     /**
      * Returns a zustand store with the global configuration of the AI Extension.
      * These options are used by default across all LLM calls when calling {@link doLLMRequest}
      */
-    j(this, "options");
-    this.editor = o, this.options = pe()((n) => ({
+    D(this, "options");
+    this.editor = o, this.options = me()((n) => ({
       dataFormat: fo.html,
       stream: !0,
       ...r
@@ -2174,11 +2173,11 @@ class Ye extends lt {
         key: go,
         filterTransaction: (n) => {
           var a;
-          const i = this.store.getState().aiMenuState;
-          return !(i !== "closed" && i.status === "ai-writing" && (a = n.getMeta(kt)) != null && a.fixTables);
+          const s = this.store.getState().aiMenuState;
+          return !(s !== "closed" && s.status === "ai-writing" && (a = n.getMeta(kt)) != null && a.fixTables);
         }
       })
-    ), this.addProsemirrorPlugin(pt()), this.addProsemirrorPlugin(
+    ), this.addProsemirrorPlugin(mt()), this.addProsemirrorPlugin(
       ho(
         r.agentCursor || { name: "AI", color: "#8bc6ff" }
       )
@@ -2219,18 +2218,18 @@ class Ye extends lt {
   acceptChanges() {
     var r;
     const o = this.editor.prosemirrorState.doc;
-    this.editor.exec((n, i) => ie(n, (a) => {
-      i == null || i(a.setMeta("addToHistory", !1));
-    })), this.editor.exec((n, i) => {
+    this.editor.exec((n, s) => ie(n, (a) => {
+      s == null || s(a.setMeta("addToHistory", !1));
+    })), this.editor.exec((n, s) => {
       const a = n.tr;
       a.replace(
         0,
         a.doc.content.size,
         new Y(ae.from(o), 0, 0)
       );
-      const s = n.apply(a);
-      return Be(s, (c) => {
-        i == null || i(
+      const i = n.apply(a);
+      return Be(i, (c) => {
+        s == null || s(
           a.replace(
             0,
             a.doc.content.size,
@@ -2245,8 +2244,8 @@ class Ye extends lt {
    */
   rejectChanges() {
     var o;
-    this.editor.exec((r, n) => ie(r, (i) => {
-      n == null || n(i.setMeta("addToHistory", !1));
+    this.editor.exec((r, n) => ie(r, (s) => {
+      n == null || n(s.setMeta("addToHistory", !1));
     })), (o = this.editor.forkYDocPlugin) == null || o.merge({ keepChanges: !1 }), this.closeAIMenu();
   }
   /**
@@ -2304,34 +2303,34 @@ class Ye extends lt {
     this.setAIResponseStatus("thinking"), (n = this.editor.forkYDocPlugin) == null || n.fork();
     let r;
     try {
-      const i = {
+      const s = {
         ...this.options.getState(),
         ...o,
         previousResponse: this.store.getState().llmResponse
       };
-      this.previousRequestOptions = i, r = await Gt(this.editor, {
-        ...i,
+      this.previousRequestOptions = s, r = await Gt(this.editor, {
+        ...s,
         onStart: () => {
           var a;
           this.setAIResponseStatus("ai-writing"), (a = o.onStart) == null || a.call(o);
         },
         onBlockUpdate: (a) => {
-          var s;
+          var i;
           this._store.setState({
             aiMenuState: {
               blockId: a,
               status: "ai-writing"
             }
-          }), (s = o.onBlockUpdate) == null || s.call(o, a);
+          }), (i = o.onBlockUpdate) == null || i.call(o, a);
         }
       }), this._store.setState({
         llmResponse: r
       }), await r.execute(), this.setAIResponseStatus("user-reviewing");
-    } catch (i) {
+    } catch (s) {
       this.setAIResponseStatus({
         status: "error",
-        error: i
-      }), console.warn("Error calling LLM", i);
+        error: s
+      }), console.warn("Error calling LLM", s);
     }
     return r;
   }
@@ -2343,7 +2342,7 @@ function O(e) {
   return e.extension(Ye);
 }
 const ko = (e, t) => async (o, r) => {
-  const n = new Request(o, r), i = new Request(
+  const n = new Request(o, r), s = new Request(
     `${e}?provider=${encodeURIComponent(
       t
     )}&url=${encodeURIComponent(n.url)}`,
@@ -2357,7 +2356,7 @@ const ko = (e, t) => async (o, r) => {
     }
   );
   try {
-    return await fetch(i);
+    return await fetch(s);
   } catch (a) {
     throw new TypeError("fetch failed", {
       cause: a
@@ -2396,9 +2395,9 @@ function So(e, t) {
   if (e == null) return {};
   var o = wo(e, t), r, n;
   if (Object.getOwnPropertySymbols) {
-    var i = Object.getOwnPropertySymbols(e);
-    for (n = 0; n < i.length; n++)
-      r = i[n], !(t.indexOf(r) >= 0) && Object.prototype.propertyIsEnumerable.call(e, r) && (o[r] = e[r]);
+    var s = Object.getOwnPropertySymbols(e);
+    for (n = 0; n < s.length; n++)
+      r = s[n], !(t.indexOf(r) >= 0) && Object.prototype.propertyIsEnumerable.call(e, r) && (o[r] = e[r]);
   }
   return o;
 }
@@ -2412,15 +2411,15 @@ function wo(e, t) {
     }
   return o;
 }
-function q() {
-  return q = Object.assign ? Object.assign.bind() : function(e) {
+function U() {
+  return U = Object.assign ? Object.assign.bind() : function(e) {
     for (var t = 1; t < arguments.length; t++) {
       var o = arguments[t];
       for (var r in o)
         Object.prototype.hasOwnProperty.call(o, r) && (e[r] = o[r]);
     }
     return e;
-  }, q.apply(this, arguments);
+  }, U.apply(this, arguments);
 }
 function ve(e, t) {
   var o = Object.keys(e);
@@ -2432,7 +2431,7 @@ function ve(e, t) {
   }
   return o;
 }
-function U(e) {
+function z(e) {
   for (var t = 1; t < arguments.length; t++) {
     var o = arguments[t] != null ? arguments[t] : {};
     t % 2 ? ve(Object(o), !0).forEach(function(r) {
@@ -2461,13 +2460,13 @@ function Co(e, t) {
   return (t === "string" ? String : Number)(e);
 }
 function Ke(e) {
-  return e && e.map((t, o) => /* @__PURE__ */ B.createElement(t.tag, U({
+  return e && e.map((t, o) => /* @__PURE__ */ B.createElement(t.tag, z({
     key: o
   }, t.attr), Ke(t.child)));
 }
 function v(e) {
-  return (t) => /* @__PURE__ */ B.createElement(Po, q({
-    attr: U({}, e.attr)
+  return (t) => /* @__PURE__ */ B.createElement(Po, U({
+    attr: z({}, e.attr)
   }, t), Ke(e.child));
 }
 function Po(e) {
@@ -2475,21 +2474,21 @@ function Po(e) {
     var {
       attr: r,
       size: n,
-      title: i
-    } = e, a = So(e, bo), s = n || o.size || "1em", c;
-    return o.className && (c = o.className), e.className && (c = (c ? c + " " : "") + e.className), /* @__PURE__ */ B.createElement("svg", q({
+      title: s
+    } = e, a = So(e, bo), i = n || o.size || "1em", c;
+    return o.className && (c = o.className), e.className && (c = (c ? c + " " : "") + e.className), /* @__PURE__ */ B.createElement("svg", U({
       stroke: "currentColor",
       fill: "currentColor",
       strokeWidth: "0"
     }, o.attr, r, a, {
       className: c,
-      style: U(U({
+      style: z(z({
         color: e.color || o.color
       }, o.style), e.style),
-      height: s,
-      width: s,
+      height: i,
+      width: i,
       xmlns: "http://www.w3.org/2000/svg"
-    }), i && /* @__PURE__ */ B.createElement("title", null, i), e.children);
+    }), s && /* @__PURE__ */ B.createElement("title", null, s), e.children);
   };
   return Te !== void 0 ? /* @__PURE__ */ B.createElement(Te.Consumer, null, (o) => t(o)) : t(Ge);
 }
@@ -2517,45 +2516,45 @@ function Lo(e) {
 function _o(e) {
   return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M9.9997 15.1709L19.1921 5.97852L20.6063 7.39273L9.9997 17.9993L3.63574 11.6354L5.04996 10.2212L9.9997 15.1709Z" }, child: [] }] })(e);
 }
-function Eo(e) {
+function Io(e) {
   return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M9.9997 15.1709L19.1921 5.97852L20.6063 7.39273L9.9997 17.9993L3.63574 11.6354L5.04996 10.2212L9.9997 15.1709Z" }, child: [] }] })(e);
 }
-function Io(e) {
+function Eo(e) {
   return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M12 4C9.4095 4 7.10606 5.23053 5.64274 7.14274L8 9.5H2V3.5L4.21863 5.71863C6.05061 3.452 8.85558 2 12 2 17.5228 2 22 6.47715 22 12H20C20 7.58172 16.4183 4 12 4ZM4 12C4 16.4183 7.58172 20 12 20 14.5905 20 16.894 18.7695 18.3573 16.8573L16 14.5 22 14.5V20.5L19.7814 18.2814C17.9494 20.548 15.1444 22 12 22 6.47715 22 2 17.5228 2 12H4Z" }, child: [] }] })(e);
 }
 function ne(e) {
   return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M17.0007 1.20825 18.3195 3.68108 20.7923 4.99992 18.3195 6.31876 17.0007 8.79159 15.6818 6.31876 13.209 4.99992 15.6818 3.68108 17.0007 1.20825ZM8.00065 4.33325 10.6673 9.33325 15.6673 11.9999 10.6673 14.6666 8.00065 19.6666 5.33398 14.6666.333984 11.9999 5.33398 9.33325 8.00065 4.33325ZM19.6673 16.3333 18.0007 13.2083 16.334 16.3333 13.209 17.9999 16.334 19.6666 18.0007 22.7916 19.6673 19.6666 22.7923 17.9999 19.6673 16.3333Z" }, child: [] }] })(e);
 }
-function _(e) {
+function I(e) {
   if (!e.dictionary.ai)
     throw new Error("AI dictionary not found");
   return e.dictionary.ai;
 }
 function Qe() {
   const e = Ct();
-  return _(e.editor);
+  return I(e.editor);
 }
 const No = (e) => {
-  const t = G(), { onManualPromptSubmit: o, promptText: r, onPromptTextChange: n } = e, [i, a] = Ie(""), s = r || i, c = D(
+  const t = G(), { onManualPromptSubmit: o, promptText: r, onPromptTextChange: n } = e, [s, a] = Ee(""), i = r || s, c = $(
     async (d) => {
-      d.key === "Enter" && o(s);
+      d.key === "Enter" && o(i);
     },
-    [s, o]
-  ), u = D(
+    [i, o]
+  ), u = $(
     (d) => {
       const S = d.currentTarget.value;
       n && n(S), r === void 0 && a(S);
     },
     [n, a, r]
-  ), m = E(() => ut(e.items, s), [s, e.items]), { selectedIndex: l, setSelectedIndex: f, handler: p } = Pt(m, (d) => d.onItemClick()), h = D(
+  ), p = E(() => ut(e.items, i), [i, e.items]), { selectedIndex: l, setSelectedIndex: f, handler: m } = Pt(p, (d) => d.onItemClick()), h = $(
     (d) => {
-      d.key === "Enter" ? m.length > 0 ? p(d) : c(d) : p(d);
+      d.key === "Enter" ? p.length > 0 ? m(d) : c(d) : m(d);
     },
-    [c, p, m.length]
+    [c, m, p.length]
   );
   return Ne(() => {
     f(0);
-  }, [s, f]), /* @__PURE__ */ vt("div", { className: "bn-combobox", children: [
+  }, [i, f]), /* @__PURE__ */ vt("div", { className: "bn-combobox", children: [
     /* @__PURE__ */ y(t.Generic.Form.Root, { children: /* @__PURE__ */ y(
       t.Generic.Form.TextInput,
       {
@@ -2563,7 +2562,7 @@ const No = (e) => {
         name: "ai-prompt",
         variant: "large",
         icon: e.icon,
-        value: s || "",
+        value: i || "",
         autoFocus: !0,
         placeholder: e.placeholder,
         disabled: e.disabled,
@@ -2579,7 +2578,7 @@ const No = (e) => {
       {
         className: "bn-combobox-items",
         id: "ai-suggestion-menu",
-        children: m.map((d, S) => /* @__PURE__ */ y(
+        children: p.map((d, S) => /* @__PURE__ */ y(
           t.SuggestionMenu.Item,
           {
             className: dt(
@@ -2598,7 +2597,7 @@ const No = (e) => {
   ] });
 };
 function Ao(e) {
-  const t = _(e), o = O(e);
+  const t = I(e), o = O(e);
   return [
     {
       key: "continue_writing",
@@ -2607,7 +2606,7 @@ function Ao(e) {
       icon: /* @__PURE__ */ y(Ce, { size: 18 }),
       onItemClick: async () => {
         await o.callLLM({
-          userPrompt: "Continue writing: write more text at the current cursor position related to the previous text",
+          userPrompt: "Continue writing at the current cursor position related to the previous text. Add multiple blocks if needed. If the document looks like a template / draft, follow the template. Be extensive if needed.",
           // By default, LLM will be able to add / update / delete blocks. For "continue writing", we only want to allow adding new blocks.
           defaultStreamTools: {
             add: !0,
@@ -2667,7 +2666,7 @@ function Ao(e) {
   ];
 }
 function jo(e) {
-  const t = _(e), o = O(e);
+  const t = I(e), o = O(e);
   return [
     {
       key: "improve_writing",
@@ -2692,7 +2691,7 @@ function jo(e) {
       key: "fix_spelling",
       title: t.ai_default_commands.fix_spelling.title,
       aliases: t.ai_default_commands.fix_spelling.aliases,
-      icon: /* @__PURE__ */ y(Eo, { size: 18 }),
+      icon: /* @__PURE__ */ y(Io, { size: 18 }),
       onItemClick: async () => {
         await o.callLLM({
           useSelection: !0,
@@ -2739,7 +2738,7 @@ function jo(e) {
   ];
 }
 function Do(e) {
-  const t = _(e), o = O(e);
+  const t = I(e), o = O(e);
   return [
     {
       key: "accept",
@@ -2764,13 +2763,13 @@ function Do(e) {
   ];
 }
 function $o(e) {
-  const t = _(e), o = O(e);
+  const t = I(e), o = O(e);
   return [
     {
       key: "retry",
       title: t.ai_menu.actions.retry.title,
       aliases: t.ai_menu.actions.retry.aliases,
-      icon: /* @__PURE__ */ y(Io, { size: 18 }),
+      icon: /* @__PURE__ */ y(Eo, { size: 18 }),
       onItemClick: async () => {
         await o.retry();
       },
@@ -2792,38 +2791,38 @@ function Ro(e, t) {
   return t === "user-input" ? e.getSelection() ? jo(e) : Ao(e) : t === "user-reviewing" ? Do(e) : t === "error" ? $o(e) : [];
 }
 const Ho = (e) => {
-  const t = K(), [o, r] = Ie(""), n = Qe(), i = G(), a = O(t), s = Ae(
+  const t = K(), [o, r] = Ee(""), n = Qe(), s = G(), a = O(t), i = Ae(
     a.store,
-    (p) => p.aiMenuState !== "closed" ? p.aiMenuState.status : "closed"
+    (m) => m.aiMenuState !== "closed" ? m.aiMenuState.status : "closed"
   ), { items: c } = e, u = E(() => {
-    let p = [];
-    return c ? p = c(t, s) : p = Ro(t, s), p.map((h) => ({
+    let m = [];
+    return c ? m = c(t, i) : m = Ro(t, i), m.map((h) => ({
       ...h,
       onItemClick: () => {
         h.onItemClick(r);
       }
     }));
-  }, [c, s, t]), m = D(
-    async (p) => {
+  }, [c, i, t]), p = $(
+    async (m) => {
       await a.callLLM({
-        userPrompt: p,
+        userPrompt: m,
         useSelection: t.getSelection() !== void 0
       });
     },
     [a, t]
   );
   Ne(() => {
-    (s === "user-reviewing" || s === "error") && r("");
-  }, [s]);
-  const l = E(() => s === "thinking" ? n.ai_menu.status.thinking : s === "ai-writing" ? n.ai_menu.status.editing : s === "error" ? n.ai_menu.status.error : n.ai_menu.input_placeholder, [s, n]), f = E(() => {
-    if (s === "thinking" || s === "ai-writing")
+    (i === "user-reviewing" || i === "error") && r("");
+  }, [i]);
+  const l = E(() => i === "thinking" ? n.ai_menu.status.thinking : i === "ai-writing" ? n.ai_menu.status.editing : i === "error" ? n.ai_menu.status.error : n.ai_menu.input_placeholder, [i, n]), f = E(() => {
+    if (i === "thinking" || i === "ai-writing")
       return /* @__PURE__ */ y(
-        i.SuggestionMenu.Loader,
+        s.SuggestionMenu.Loader,
         {
           className: "bn-suggestion-menu-loader bn-combobox-right-section"
         }
       );
-    if (s === "error")
+    if (i === "error")
       return /* @__PURE__ */ y("div", { className: "bn-combobox-right-section bn-combobox-error", children: /* @__PURE__ */ y(
         "svg",
         {
@@ -2835,16 +2834,16 @@ const Ho = (e) => {
           children: /* @__PURE__ */ y("path", { d: "M480-280q17 0 28.5-11.5T520-320q0-17-11.5-28.5T480-360q-17 0-28.5 11.5T440-320q0 17 11.5 28.5T480-280Zm0-160q17 0 28.5-11.5T520-480v-160q0-17-11.5-28.5T480-680q-17 0-28.5 11.5T440-640v160q0 17 11.5 28.5T480-440Zm0 360q-83 0-156-31.5T197-197q-54-54-85.5-127T80-480q0-83 31.5-156T197-763q54-54 127-85.5T480-880q83 0 156 31.5T763-763q54 54 85.5 127T880-480q0 83-31.5 156T763-197q-54 54-127 85.5T480-80Zm0-80q134 0 227-93t93-227q0-134-93-227t-227-93q-134 0-227 93t-93 227q0 134 93 227t227 93Zm0-320Z" })
         }
       ) });
-  }, [i, s]);
+  }, [s, i]);
   return /* @__PURE__ */ y(
     No,
     {
-      onManualPromptSubmit: e.onManualPromptSubmit || m,
+      onManualPromptSubmit: e.onManualPromptSubmit || p,
       items: u,
       promptText: o,
       onPromptTextChange: r,
       placeholder: l,
-      disabled: s === "thinking" || s === "ai-writing",
+      disabled: i === "thinking" || i === "ai-writing",
       icon: /* @__PURE__ */ y("div", { className: "bn-combobox-icon", children: /* @__PURE__ */ y(ne, {}) }),
       rightSection: f
     }
@@ -2853,7 +2852,7 @@ const Ho = (e) => {
   const t = e.blockID ? document.querySelector(`[data-id="${e.blockID}"]`) : void 0, o = E(() => t ? {
     getBoundingClientRect: () => t.getBoundingClientRect(),
     contextElement: t
-  } : null, [t]), { isMounted: r, ref: n, style: i, getFloatingProps: a, isPositioned: s } = Ot(!!t, o, 3e3, {
+  } : null, [t]), { isMounted: r, ref: n, style: s, getFloatingProps: a, isPositioned: i } = Ot(!!t, o, 3e3, {
     canDismiss: {
       enabled: !0,
       escapeKey: !0,
@@ -2872,23 +2871,24 @@ const Ho = (e) => {
       })
     ],
     onOpenChange: e.onOpenChange,
-    whileElementsMounted: (c, u, m) => xt(c, u, m, {
+    whileElementsMounted: (c, u, p) => xt(c, u, p, {
       animationFrame: !0
-    })
+    }),
+    ...e.floatingOptions
   });
   return r ? /* @__PURE__ */ y(
     "div",
     {
       ref: n,
       style: {
-        ...i
+        ...s
       },
       ...a(),
-      children: s && e.children
+      children: i && e.children
     }
   ) : null;
 }, lr = (e) => {
-  const t = K(), o = O(t), r = Ae(o.store, (a) => a.aiMenuState), n = r === "closed" ? void 0 : r.blockId, i = e.aiMenu || Ho;
+  const t = K(), o = O(t), r = Ae(o.store, (a) => a.aiMenuState), n = r === "closed" ? void 0 : r.blockId, s = e.aiMenu || Ho;
   return /* @__PURE__ */ y(
     qo,
     {
@@ -2897,16 +2897,16 @@ const Ho = (e) => {
       onOpenChange: (a) => {
         a || r === "closed" || (r.status === "user-input" ? o.closeAIMenu() : (r.status === "user-reviewing" || r.status === "error") && o.rejectChanges());
       },
-      children: /* @__PURE__ */ y(i, {})
+      children: /* @__PURE__ */ y(s, {})
     }
   );
 }, ur = () => {
   const e = Qe(), t = G(), o = K(), r = O(o), n = () => {
     o.formattingToolbar.closeMenu();
-    const i = o.getSelection();
-    if (!i)
+    const s = o.getSelection();
+    if (!s)
       throw new Error("No selection");
-    const a = i.blocks[i.blocks.length - 1].id;
+    const a = s.blocks[s.blocks.length - 1].id;
     r.openAIMenuAtBlock(a);
   };
   return o.isEditable ? /* @__PURE__ */ y(
@@ -2932,7 +2932,7 @@ function dr(e) {
         r.block.content && Array.isArray(r.block.content) && // isarray check not ideal
         r.block.content.length === 0 && r.prevBlock ? t.openAIMenuAtBlock(r.prevBlock.id) : t.openAIMenuAtBlock(r.block.id);
       },
-      ..._(e).slash_menu.ai,
+      ...I(e).slash_menu.ai,
       icon: /* @__PURE__ */ y(Uo.AI, {})
     }
   ];
@@ -2947,7 +2947,7 @@ export {
   Gt as callLLM,
   ar as createAIExtension,
   cr as createBlockNoteAIClient,
-  _ as getAIDictionary,
+  I as getAIDictionary,
   O as getAIExtension,
   dr as getAISlashMenuItems,
   Ro as getDefaultAIMenuItems,