@blocknote/xl-ai 0.39.1 → 0.41.0

package/dist/blocknote-xl-ai.js

@@ -1,35 +1,37 @@
-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 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 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";
+var ot = Object.defineProperty;
+var rt = (e, t, o) => t in e ? ot(e, t, { enumerable: !0, configurable: !0, writable: !0, value: o }) : e[t] = o;
+var M = (e, t, o) => rt(e, typeof t != "symbol" ? t + "" : t, o);
+import { Chat as nt } from "@ai-sdk/react";
+import { getPmSchema as st, UnreachableCaseError as Be, getNodeById as Me, updateBlockTr as it, insertBlocks as at, trackPosition as ae, removeAndInsertBlocks as lt, getBlock as Ee, defaultProps as _e, isStyledTextInlineContent as ct, isLinkInlineContent as ut, BlockNoteExtension as dt, filterSuggestionItems as pt, mergeCSSClasses as mt } from "@blocknote/core";
+import { applySuggestions as Ne, suggestChanges as ft, revertSuggestions as le } from "@blocknote/prosemirror-suggest-changes";
+import { Slice as G, Fragment as ce } from "prosemirror-model";
+import { TextSelection as ht, Plugin as Le, PluginKey as Ae } from "prosemirror-state";
+import { fixTablesKey as yt } from "prosemirror-tables";
+import { DecorationSet as ue, Decoration as de } from "prosemirror-view";
+import { defaultSelectionBuilder as gt } from "y-prosemirror";
 import bt from "lodash.isequal";
-import { Slice as Y, Fragment as ae } from "prosemirror-model";
-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 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) => {
+import { getErrorMessage as K } from "@ai-sdk/provider-utils";
+import { parsePartialJson as De, isToolUIPart as kt, tool as St, jsonSchema as pe, generateObject as wt, convertToModelMessages as Y, streamObject as Tt, streamText as xt } from "ai";
+import { Transform as D, Mapping as je, ReplaceStep as N, ReplaceAroundStep as vt } from "prosemirror-transform";
+import { ChangeSet as me, simplifyChanges as Pt } from "prosemirror-changeset";
+import { jsxs as Ct, jsx as y } from "react/jsx-runtime";
+import { useBlockNoteContext as Ot, useComponentsContext as X, useSuggestionMenuKeyboardHandler as It, useBlockNoteEditor as Q, useUIElementPositioning as Bt } from "@blocknote/react";
+import E, { useState as $e, useCallback as $, useMemo as A, useEffect as He } from "react";
+import { useStore as qe } from "zustand";
+import { offset as Mt, size as Et, autoUpdate as _t } from "@floating-ui/react";
+const fe = (e) => {
   let t;
-  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 = p ?? (typeof l != "object" || l === null) ? l : Object.assign({}, t, l), o.forEach((m) => m(t, f));
+  const o = /* @__PURE__ */ new Set(), r = (c, u) => {
+    const m = typeof c == "function" ? c(t) : c;
+    if (!Object.is(m, t)) {
+      const p = t;
+      t = u ?? (typeof m != "object" || m === null) ? m : Object.assign({}, t, m), o.forEach((d) => d(t, p));
     }
-  }, n = () => t, i = { setState: r, getState: n, getInitialState: () => c, subscribe: (u) => (o.add(u), () => o.delete(u)) }, c = t = e(r, n, i);
+  }, n = () => t, i = { setState: r, getState: n, getInitialState: () => l, subscribe: (c) => (o.add(c), () => o.delete(c)) }, l = 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;
+}, he = (e) => e ? fe(e) : fe;
+function Nt(e) {
+  const { properties: t, required: o, $defs: r, ...n } = e.inputSchema;
   return {
     schema: {
       type: "object",
@@ -48,8 +50,8 @@ function Lt(e) {
     $defs: r
   };
 }
-function je(e) {
-  const t = e.map((r) => Lt(r)), o = {};
+function Lt(e) {
+  const t = e.map((r) => Nt(r)), o = {};
   for (const r of t)
     for (const n in r.$defs) {
       if (o[n] && !bt(o[n], r.$defs[n]))
@@ -60,6 +62,8 @@ function je(e) {
     type: "object",
     properties: {
       operations: {
+        //description:
+        // "Operations to apply to the document. Put all operations in this array in ONE tool call / function call. DO NOT use multiple operation arrays with parallel tool calls.",
         type: "array",
         items: {
           anyOf: t.map((r) => r.schema)
@@ -71,7 +75,170 @@ function je(e) {
     $defs: Object.keys(o).length > 0 ? o : void 0
   };
 }
-function _t(e) {
+function At(e, t) {
+  return {
+    async sendAIRequest(o, r) {
+      const n = await t(o);
+      return await e(o.chat.messages, n), o.chat.sendMessage(void 0, {
+        ...r,
+        body: {
+          ...(r == null ? void 0 : r.body) ?? {},
+          toolDefinitions: {
+            applyDocumentOperations: {
+              name: "applyDocumentOperations",
+              inputSchema: Lt(o.streamTools),
+              outputSchema: { type: "object" }
+            }
+          }
+        },
+        // we pass the promptData as metadata
+        // so the transport can decide whether or not to submit this to the server
+        // (DefaultChatTransport will not)
+        metadata: { promptData: n }
+      });
+    }
+  };
+}
+class ze extends Error {
+  constructor(t, o, r) {
+    super(t, r), this.chunk = o, this.name = "ChunkExecutionError";
+  }
+}
+class Dt {
+  /**
+   * @param streamTools - The StreamTools to use to apply the StreamToolCalls
+   */
+  constructor(t) {
+    M(this, "stream");
+    this.streamTools = t, this.stream = this.createStream();
+  }
+  createStream() {
+    let t;
+    const o = new TransformStream({
+      transform: async (i, l) => {
+        const c = typeof i == "string" ? await jt(
+          i,
+          (t == null ? void 0 : t.isPossiblyPartial) ?? !1,
+          this.streamTools
+        ) : i;
+        c && (t = c, l.enqueue(c));
+      },
+      flush: (i) => {
+        t != null && t.isPossiblyPartial && i.error(new Error("stream ended with a partial operation"));
+      }
+    }), r = o.readable.pipeThrough(this.createExecutor()), [n, s] = r.tee(), a = n.pipeTo(new WritableStream());
+    return {
+      writable: o.writable,
+      // expose externalReadable to the consumer
+      readable: s,
+      finishPromise: a
+    };
+  }
+  createExecutor() {
+    const t = this.streamTools.map((o) => o.executor());
+    return new TransformStream({
+      transform: async (o, r) => {
+        var s;
+        let n = !1;
+        for (const a of t)
+          try {
+            if (await a.execute(o)) {
+              r.enqueue({ status: "ok", chunk: o }), n = !0;
+              break;
+            }
+          } catch (i) {
+            throw new ze(
+              `Tool execution failed: ${K(i)}`,
+              o,
+              {
+                cause: i
+              }
+            );
+          }
+        if (!n) {
+          const a = ((s = o.operation) == null ? void 0 : s.type) || "unknown";
+          throw new Error(
+            `No tool could handle operation of type: ${a}`
+          );
+        }
+      }
+    });
+  }
+  /**
+   * Returns a WritableStream that can be used to write StreamToolCalls to the executor.
+   *
+   * The WriteableStream accepts JSON strings or Operation objects.
+   *
+   * Make sure to call `close` on the StreamToolExecutor instead of on the writable returned here!
+   */
+  get writable() {
+    return this.stream.writable;
+  }
+  /**
+   * Returns a ReadableStream that can be used to read the results of the executor.
+   */
+  get readable() {
+    return this.stream.readable;
+  }
+  async finish() {
+    await this.stream.finishPromise;
+  }
+  async executeOperationsArray(t) {
+    const o = this.writable.getWriter();
+    for await (const r of t) {
+      const n = await De(r);
+      if (n.state === "undefined-input" || n.state === "failed-parse" || !n)
+        return;
+      await o.write(r);
+    }
+    await o.close(), await this.finish();
+  }
+  /**
+   * Accepts an async iterable and writes each chunk to the internal stream.
+   *
+   * (alternative to writing to the writable stream using {@link writable})
+   */
+  async execute(t) {
+    const o = this.writable.getWriter();
+    for await (const r of t)
+      await o.write(r);
+    await o.close(), await this.finish();
+  }
+  /**
+   * Accepts a single chunk and processes it using the same logic.
+   *
+   * (alternative to writing to the writable stream using {@link writable})
+   */
+  async executeOne(t) {
+    await this.execute(
+      async function* () {
+        yield {
+          operation: t,
+          isUpdateToPreviousOperation: !1,
+          isPossiblyPartial: !1,
+          metadata: {}
+        };
+      }()
+    );
+  }
+}
+async function jt(e, t, o) {
+  const r = await De(e);
+  if (r.state === "undefined-input" || r.state === "failed-parse" || !r)
+    return;
+  const n = o.find((a) => {
+    var i;
+    return a.name === ((i = r.value) == null ? void 0 : i.type);
+  }), s = n && n.validate(r.value);
+  if (s != null && s.ok)
+    return {
+      operation: s.value,
+      isPossiblyPartial: r.state === "repaired-parse",
+      isUpdateToPreviousOperation: t,
+      metadata: void 0
+    };
+}
+function $t(e) {
   return new ReadableStream({
     async start(t) {
       try {
@@ -84,7 +251,7 @@ function _t(e) {
     }
   });
 }
-function N(e) {
+function Je(e) {
   if (e.locked)
     throw new Error(
       "Stream (source) is already locked and cannot be iterated."
@@ -102,47 +269,50 @@ function N(e) {
     };
   }, t;
 }
-function De(e) {
-  return N(_t(e));
-}
-async function* It(e) {
-  var n;
-  let t = 0, o = !0, r;
-  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 === s.operations.length - 1
-        }, o = !1;
+function Ht(e) {
+  return Je($t(e));
+}
+async function* qt(e, t) {
+  var s;
+  let o = 0, r = !0, n;
+  for await (const a of e)
+    if ((s = a.operations) != null && s.length) {
+      for (let i = o; i < a.operations.length; i++) {
+        const l = a.operations[i];
+        n = l, yield {
+          partialOperation: l,
+          isUpdateToPreviousOperation: i === o && !r,
+          isPossiblyPartial: i === a.operations.length - 1,
+          metadata: t
+        }, r = !1;
       }
-      t = s.operations.length - 1;
+      o = a.operations.length - 1;
     }
-  if (!r)
+  if (!n)
     throw new Error("No operations seen");
   yield {
-    partialOperation: r,
+    partialOperation: n,
     isUpdateToPreviousOperation: !0,
-    isPossiblyPartial: !1
+    isPossiblyPartial: !1,
+    metadata: t
   };
 }
-async function* $e(e, t) {
+async function* zt(e, t) {
   let o = !1;
   for await (const r of e) {
     const n = r.operation;
     n.ok ? (yield {
       operation: n.value,
       isUpdateToPreviousOperation: o ? !1 : r.isUpdateToPreviousOperation,
-      isPossiblyPartial: r.isPossiblyPartial
+      isPossiblyPartial: r.isPossiblyPartial,
+      metadata: r.metadata
     }, o = !1) : (o = o || !r.isUpdateToPreviousOperation, t == null || t({
       ...r,
       operation: n
     }));
   }
 }
-async function* Re(e, t) {
+async function* Jt(e, t) {
   for await (const o of e) {
     const r = t.find(
       (s) => s.name === o.partialOperation.type
@@ -154,23 +324,25 @@ async function* Re(e, t) {
           error: `No matching function for ${o.partialOperation.type}`
         },
         isUpdateToPreviousOperation: o.isUpdateToPreviousOperation,
-        isPossiblyPartial: o.isPossiblyPartial
+        isPossiblyPartial: o.isPossiblyPartial,
+        metadata: o.metadata
       };
       continue;
     }
     yield {
       operation: r.validate(o.partialOperation),
       isUpdateToPreviousOperation: o.isUpdateToPreviousOperation,
-      isPossiblyPartial: o.isPossiblyPartial
+      isPossiblyPartial: o.isPossiblyPartial,
+      metadata: o.metadata
     };
   }
 }
-async function* Et(e, t) {
-  const o = Re(
+async function* Ut(e, t) {
+  const o = Jt(
     e,
     t
   );
-  yield* $e(
+  yield* zt(
     o,
     (n) => {
       if (!n.isPossiblyPartial)
@@ -178,225 +350,147 @@ async function* Et(e, t) {
     }
   );
 }
-async function* Nt(e, t) {
-  const o = Re(
-    e,
-    t
-  );
-  yield* $e(
-    o,
-    (n) => {
-      throw new Error("invalid operation: " + n.operation.error);
-    }
+function Vt(e, t, o) {
+  return Ht(
+    Ut(
+      qt(
+        Je(e),
+        o
+      ),
+      t
+    )
   );
 }
-async function At(e, t) {
-  const { _generateObjectOptions: o, ...r } = t;
-  if (o && ("output" in o || "schema" in o))
-    throw new Error(
-      "Cannot provide output or schema in _generateObjectOptions"
-    );
-  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" || r.model.provider === "google.generative-ai" ? "auto" : "tool",
-    //  - mandatory ones:
-    ...r,
-    // extra options for streamObject
-    ...o ?? {}
-  }, 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(i.value, e)
-      )), c;
+async function Ft(e, t, o) {
+  const r = new Dt(e), n = Rt();
+  n.output.pipeTo(r.writable);
+  const s = /* @__PURE__ */ new Map();
+  let a = !0;
+  const i = t["~registerMessagesCallback"](() => {
+    Zt(t, (d) => {
+      if (!s.has(d.toolCallId)) {
+        const h = Wt(
+          e,
+          d.toolName,
+          d.toolCallId
+        );
+        n.append(h.operationsStream), s.set(d.toolCallId, h), a && (a = !1, o == null || o());
+      }
+      return s.get(d.toolCallId);
+    });
+  });
+  await new Promise((d) => {
+    const h = t["~registerStatusCallback"](() => {
+      (t.status === "ready" || t.status === "error") && (i(), h(), t.status !== "error" && f(), d());
+    }), f = t["~registerErrorCallback"](() => {
+      if (t.error) {
+        f();
+        for (const k of s.values())
+          k.complete || k.writer.abort(t.error);
+      }
+    });
+  }), await n.finalize();
+  const c = (await Promise.allSettled([r.finish()]))[0];
+  let u;
+  if (c.status === "rejected") {
+    if (c.reason instanceof ze)
+      u = c.reason;
+    else if (!t.error)
+      throw new Error(
+        "Unexpected: no ChunkExecutionError but also no chat.error (network error?)"
+      );
+  }
+  let m = !1;
+  const p = Array.from(
+    s.values().filter((d) => d.complete)
+  );
+  if (p.forEach((d, h) => {
+    const f = d.toolCallId === (u == null ? void 0 : u.chunk.metadata.toolCallId);
+    f && (m = !0), t.addToolResult({
+      tool: p[h].toolName,
+      toolCallId: p[h].toolCallId,
+      output: m === !1 ? { status: "ok" } : f ? { status: "error", error: K(u) } : { status: "not-executed-previous-tool-errored" }
+    });
+  }), u)
+    throw u;
+  if (t.error)
+    throw t.error;
+}
+function Rt() {
+  let e, t = Promise.resolve(), o = !1;
+  const r = new ReadableStream({
+    start(a) {
+      e = a;
     },
-    async getGeneratedOperations() {
-      return a.object;
+    cancel(a) {
+      o = !0, e.error(a);
     }
-  };
+  });
+  async function n(a) {
+    if (o)
+      throw new Error("Appendable stream canceled, can't append");
+    const i = a.getReader();
+    return t = t.then(async () => {
+      for (; ; )
+        try {
+          const { done: l, value: c } = await i.read();
+          if (l || o)
+            break;
+          e.enqueue(c);
+        } catch (l) {
+          o = !0, e.error(l);
+          break;
+        }
+    }), t;
+  }
+  async function s() {
+    await t, o || e.close();
+  }
+  return { output: r, append: n, finalize: s };
 }
-function jt(e) {
-  if (!e || typeof e != "object" || !("operations" in e) || !Array.isArray(e.operations))
-    return {
-      ok: !1,
-      error: "No operations returned"
-    };
-  const t = e.operations;
-  async function* o() {
-    for (const r of t)
-      yield {
-        partialOperation: r,
-        isUpdateToPreviousOperation: !1,
-        isPossiblyPartial: !1
-      };
+function Zt(e, t) {
+  var o;
+  for (const r of ((o = e.lastMessage) == null ? void 0 : o.parts) ?? []) {
+    if (!kt(r) || r.type.replace("tool-", "") !== "applyDocumentOperations")
+      continue;
+    const s = r.toolCallId, a = t({
+      toolName: r.type.replace("tool-", ""),
+      toolCallId: s
+    });
+    Yt(r, a);
   }
-  return {
-    ok: !0,
-    value: o()
-  };
 }
-async function Dt(e, t, o = () => {
-}) {
-  const { _streamObjectOptions: r, ...n } = t;
-  if (r && ("output" in r || "schema" in r))
-    throw new Error("Cannot provide output or schema in _streamObjectOptions");
-  const a = {
-    // 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: n.model.provider === "mistral.chat" || n.model.provider === "google.generative-ai" ? "auto" : "tool",
-    //  - mandatory ones:
-    ...n,
-    // extra options for streamObject
-    ...t._streamObjectOptions ?? {}
-  }, i = pt(a);
-  let c;
-  const [u, p] = i.fullStream.tee(), l = (async () => {
-    let f = {
-      operations: []
-    };
-    const m = N(
-      Ht(p)
-    );
-    for await (const h of m)
-      h && typeof h == "object" && "operations" in h && (f = h);
-    return f;
-  })();
+function Wt(e, t, o) {
+  const r = new TransformStream(), n = Vt(
+    r.readable,
+    e,
+    { toolCallId: o }
+  );
   return {
-    streamObjectResult: i,
-    generateObjectResult: void 0,
-    get operationsSource() {
-      return c || (c = De(
-        Et(
-          It(
-            $t(
-              Rt(
-                N(u)
-              ),
-              o
-            )
-          ),
-          e
-        )
-      )), c;
-    },
-    async getGeneratedOperations() {
-      return l;
-    }
+    // stream,
+    writer: r.writable.getWriter(),
+    complete: !1,
+    // executor,
+    operationsStream: n,
+    toolName: t,
+    toolCallId: o
   };
 }
-async function* $t(e, t) {
-  let o = !0;
-  for await (const r of e)
-    o && (t(), o = !1), yield r;
-}
-function Rt(e) {
-  return N(
-    e.pipeThrough(
-      new TransformStream({
-        transform(t, o) {
-          switch (t.type) {
-            case "object":
-              o.enqueue(t.object);
-              break;
-            case "text-delta":
-            case "finish":
-              break;
-            case "error":
-              o.error(t.error);
-              break;
-            default: {
-              const r = t;
-              throw new Error(`Unsupported chunk type: ${r}`);
-            }
-          }
-        }
-      })
-    )
-  );
-}
-function Ht(e) {
-  return N(
-    e.pipeThrough(
-      new TransformStream({
-        transform(t, o) {
-          switch (t.type) {
-            case "object":
-              o.enqueue(t.object);
-              break;
-            case "text-delta":
-            case "finish":
-              break;
-            case "error":
-              break;
-            default: {
-              const r = t;
-              throw new Error(`Unsupported chunk type: ${r}`);
-            }
-          }
-        }
-      })
-    )
-  );
+function Yt(e, t) {
+  if (e.state === "input-streaming") {
+    const o = e.input;
+    o !== void 0 && t.writer.write(o);
+  } else if (e.state === "input-available") {
+    const o = e.input;
+    if (o === void 0)
+      throw new Error("input is undefined");
+    t.complete || (t.complete = !0, t.writer.write(o), t.writer.close());
+  }
 }
-function X(e) {
+function ee(e) {
   return (e.type === "paragraph" || !e.type) && !e.content || Array.isArray(e.content) && e.content.length === 0;
 }
-class qt {
-  /**
-   * @internal
-   */
-  constructor(t, o, r) {
-    this.messages = t, this.llmResult = o, this.streamTools = r;
-  }
-  /**
-   * Apply the operations to the editor and return a stream of results.
-   *
-   * (this method consumes underlying streams in `llmResult`)
-   */
-  async *applyToolCalls() {
-    let t = this.llmResult.operationsSource;
-    for (const o of this.streamTools)
-      t = o.execute(t);
-    yield* t;
-  }
-  /**
-   * Helper method to apply all operations to the editor if you're not interested in intermediate operations and results.
-   *
-   * (this method consumes underlying streams in `llmResult`)
-   */
-  async execute() {
-    for await (const t of this.applyToolCalls())
-      ;
-  }
-  /**
-   * @internal
-   */
-  async _logToolCalls() {
-    for await (const t of this.llmResult.operationsSource)
-      console.log(JSON.stringify(t, null, 2));
-  }
-}
-function Ut(e, t, o = !0, r = !0) {
+function Gt(e, t, o = !0, r = !0) {
   let n = 0, s = e.length;
   if (o)
     for (; n < s && t(e[n]); )
@@ -406,15 +500,208 @@ function Ut(e, t, o = !0, r = !0) {
       s--;
   return e.slice(n, s);
 }
-function x(e, t) {
-  return Ut(
+function w(e, t) {
+  return Gt(
     e,
-    (r) => X(r) && (t == null ? void 0 : t.cursorBlockId) !== r.id,
+    (r) => ee(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) {
+function Kt(e) {
+  const { useSelection: t, deleteEmptyCursorBlock: o, streamToolsProvider: r } = {
+    useSelection: e.useSelection ?? !1,
+    deleteEmptyCursorBlock: e.deleteEmptyCursorBlock ?? !0,
+    streamToolsProvider: e.streamToolsProvider ?? O.html.getStreamToolsProvider()
+  }, n = t ? void 0 : e.editor.getTextCursorPosition().block, s = n && o && ee(n) && w(e.editor.document).length > 0 ? n.id : void 0, a = t ? e.editor.getSelectionCutBlocks() : void 0, i = r.getStreamTools(
+    e.editor,
+    a ? {
+      from: a._meta.startPos,
+      to: a._meta.endPos
+    } : void 0,
+    e.onBlockUpdated
+  );
+  return {
+    editor: e.editor,
+    chat: e.chat,
+    userPrompt: e.userPrompt,
+    selectedBlocks: a == null ? void 0 : a.blocks,
+    streamTools: i,
+    emptyCursorBlockToDelete: s
+  };
+}
+async function Xt(e) {
+  const { aiRequest: t, sender: o, chatRequestOptions: r, onStart: n } = e, s = Ft(
+    t.streamTools,
+    t.chat,
+    () => {
+      n == null || n(), t.emptyCursorBlockToDelete && t.editor.getBlock(t.emptyCursorBlockToDelete) && t.editor.removeBlocks([t.emptyCursorBlockToDelete]);
+    }
+  );
+  await o.sendAIRequest(t, r), await s;
+}
+function Qt(e, t) {
+  e.length > 0 && e.push(
+    {
+      role: "assistant",
+      id: "document-state-" + e.length,
+      parts: [
+        {
+          type: "text",
+          text: "This is the latest state of the selection (ignore previous selections, you MUST issue operations against this latest version of the selection):"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.htmlSelectedBlocks)
+        },
+        {
+          type: "text",
+          text: "This is the latest state of the document (INCLUDING the selected text), find the selected text in there to understand the context:"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.htmlDocument)
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt-" + e.length,
+      parts: [
+        {
+          type: "text",
+          text: "The user asks you to do the following:"
+        },
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  ), e.push(
+    {
+      role: "system",
+      id: "document-state-intro",
+      parts: [
+        {
+          type: "text",
+          text: "You're manipulating a selected part of a text document using HTML blocks. \n      Make sure to follow the json schema provided and always include the trailing $ in ids.\n      List items are 1 block with 1 list item each, so block content `<ul><li>item1</li></ul>` is valid, but `<ul><li>item1</li><li>item2</li></ul>` is invalid. We'll merge them automatically.\n      This is the selection as an array of html blocks:"
+        }
+      ]
+    },
+    {
+      role: "system",
+      id: "document-state-selection",
+      parts: [
+        {
+          type: "text",
+          text: JSON.stringify(t.htmlSelectedBlocks)
+        }
+      ]
+    },
+    {
+      role: "system",
+      id: "document-state-context",
+      parts: [
+        {
+          type: "text",
+          text: "This is the entire document (INCLUDING the selected text), find the selected text in there to understand the context:"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.htmlDocument)
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt",
+      parts: [
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  );
+}
+function eo(e, t) {
+  if (e.length > 0) {
+    e.push(
+      {
+        role: "assistant",
+        id: "document-state-" + e.length,
+        parts: [
+          {
+            type: "text",
+            text: "This is the latest state of the document (ignore previous documents, you MUST issue operations against this latest version of the document):"
+          },
+          {
+            type: "text",
+            text: JSON.stringify(t.htmlBlocks)
+          }
+        ]
+      },
+      {
+        role: "user",
+        id: "user-prompt-" + e.length,
+        parts: [
+          {
+            type: "text",
+            text: t.userPrompt
+          }
+        ]
+      }
+    );
+    return;
+  }
+  e.push(
+    {
+      role: "system",
+      id: "document-state-intro",
+      parts: [
+        {
+          type: "text",
+          text: "You're manipulating a text document using HTML blocks. \n        Make sure to follow the json schema provided. When referencing ids they MUST be EXACTLY the same (including the trailing $). \n        List items are 1 block with 1 list item each, so block content `<ul><li>item1</li></ul>` is valid, but `<ul><li>item1</li><li>item2</li></ul>` is invalid. We'll merge them automatically.\n        For code blocks, you can use the `data-language` attribute on a <code> block (wrapped with <pre>) to specify the language.\n        This is the initial document as an array of html blocks (the cursor is BETWEEN two blocks as indicated by cursor: true):"
+        }
+      ]
+    },
+    {
+      role: "system",
+      id: "document-state-data",
+      parts: [
+        {
+          type: "text",
+          text: JSON.stringify(t.htmlBlocks)
+        }
+      ]
+    },
+    {
+      role: "system",
+      id: "extended-instructions",
+      parts: [
+        {
+          type: "text",
+          text: 'First, determine what part of the document the user is talking about. You SHOULD probably take cursor info into account if needed.\n       EXAMPLE: if user says "below" (without pointing to a specific part of the document) he / she probably indicates the block(s) after the cursor. \n       EXAMPLE: If you want to insert content AT the cursor position (UNLESS indicated otherwise by the user), \n       then you need `referenceId` to point to the block before the cursor with position `after` (or block below and `before`).\n      \n      ' + (t.isEmptyDocument ? "Because the document is empty, first update the empty block before adding new blocks." : "Prefer updating existing blocks over removing and adding (but this also depends on the user's question).")
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt",
+      parts: [
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  );
+}
+const to = async (e, t) => {
+  t.selection ? Qt(e, t) : eo(e, t);
+};
+function Z(e, t) {
   const o = e.getTextCursorPosition(), r = [];
   for (const n of t) {
     const s = n.id === o.block.id;
@@ -427,7 +714,7 @@ function F(e, t) {
   }
   return r;
 }
-async function C(e, t) {
+async function P(e, t) {
   return await Promise.all(e.map(async (o) => ({
     id: o.id,
     block: await t(o)
@@ -442,215 +729,109 @@ function T(e) {
     ...T(t.children)
   ]);
 }
-function L(e) {
+function _(e) {
   return e.map((t) => typeof t == "object" && t && "id" in t ? {
     ...t,
     id: `${t.id}$`
   } : t);
 }
-async function He(e, t) {
-  const o = e.getTextCursorPosition().block.id, r = x(e.document, {
-    cursorBlockId: o
-  }), n = await C(
-    T(r),
+async function oo(e) {
+  return e.selectedBlocks ? {
+    ...await Ve(e.editor, {
+      selectedBlocks: e.selectedBlocks
+    }),
+    userPrompt: e.userPrompt
+  } : {
+    ...await Ue(e.editor, {
+      excludeBlockIds: e.emptyCursorBlockToDelete ? [e.emptyCursorBlockToDelete] : void 0
+    }),
+    userPrompt: e.userPrompt
+  };
+}
+async function Ue(e, t) {
+  const o = w(e.document).length === 0, r = e.getTextCursorPosition().block.id, n = w(e.document, {
+    cursorBlockId: r
+  }), s = await P(
+    T(n),
     async (c) => e.blocksToHTMLLossy([c])
-  ), a = F(e, n).filter(
+  ), i = Z(e, s).filter(
     (c) => "cursor" in c || !(t.excludeBlockIds || []).includes(c.id)
   );
   return {
-    htmlBlocks: L(a)
+    selection: !1,
+    htmlBlocks: _(i),
+    isEmptyDocument: o
   };
 }
-async function qe(e, t) {
-  const o = await C(
+async function Ve(e, t) {
+  const o = w(e.document).length === 0, r = await P(
     T(t.selectedBlocks),
-    async (n) => e.blocksToHTMLLossy([n])
-  );
+    async (s) => e.blocksToHTMLLossy([s])
+  ), n = _(r);
   return {
-    htmlSelectedBlocks: L(o),
-    htmlDocument: (await C(T(e.document), async (n) => e.blocksToHTMLLossy([n]))).map(({ block: n }) => ({ block: n }))
+    isEmptyDocument: o,
+    selection: !0,
+    htmlSelectedBlocks: n,
+    htmlDocument: (await P(T(e.document), async (s) => e.blocksToHTMLLossy([s]))).map(({ block: s }) => ({ block: s }))
     // strip ids so LLM can't accidentally issue updates to ids not in selection
   };
 }
-function zt(e) {
-  return [
-    {
-      role: "system",
-      content: "You're manipulating a selected part of a text document using HTML blocks. \n      Make sure to follow the json schema provided and always include the trailing $ in ids. \n      List items are 1 block with 1 list item each, so block content `<ul><li>item1</li></ul>` is valid, but `<ul><li>item1</li><li>item2</li></ul>` is invalid. We'll merge them automatically.\n      This is the selection as an array of html blocks:"
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.htmlSelectedBlocks)
-    },
-    {
-      role: "system",
-      content: "This is the entire document (INCLUDING the selected text), find the selected text in there to understand the context:"
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.htmlDocument)
-    },
-    {
-      role: "system",
-      content: "The user asks you to do the following:"
-    },
-    {
-      role: "user",
-      content: e.userPrompt
-    }
-  ];
-}
-function Ft(e) {
-  return [
-    {
-      role: "system",
-      content: "You're manipulating a text document using HTML blocks. \n        Make sure to follow the json schema provided. When referencing ids they MUST be EXACTLY the same (including the trailing $). \n        List items are 1 block with 1 list item each, so block content `<ul><li>item1</li></ul>` is valid, but `<ul><li>item1</li><li>item2</li></ul>` is invalid. We'll merge them automatically.\n        For code blocks, you can use the `data-language` attribute on a code block to specify the language.\n        This is the document as an array of html blocks (the cursor is BETWEEN two blocks as indicated by cursor: true):"
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.htmlBlocks)
-    },
-    {
-      role: "system",
-      content: 'First, determine what part of the document the user is talking about. You SHOULD probably take cursor info into account if needed.\n       EXAMPLE: if user says "below" (without pointing to a specific part of the document) he / she probably indicates the block(s) after the cursor. \n       EXAMPLE: If you want to insert content AT the cursor position (UNLESS indicated otherwise by the user), \n       then you need `referenceId` to point to the block before the cursor with position `after` (or block below and `before`).\n      \n      ' + (e.isEmptyDocument ? "Because the document is empty, first update the empty block before adding new blocks." : "Prefer updating existing blocks over removing and adding (but this also depends on the user's question).")
-    },
-    {
-      role: "system",
-      content: "The user asks you to do the following:"
-    },
-    {
-      role: "user",
-      content: e.userPrompt
-    }
-  ];
-}
-const Jt = async (e, t) => {
-  const o = x(e.document).length === 0;
-  if (t.selectedBlocks) {
-    const r = await qe(e, {
-      selectedBlocks: t.selectedBlocks
-    });
-    return t.previousMessages ? [
-      ...t.previousMessages,
-      {
-        role: "system",
-        content: `After processing the previous response, this is the updated selection.
-            Ignore previous documents, you MUST issue operations against this latest version of the document:`
-      },
-      {
-        role: "system",
-        content: JSON.stringify(r.htmlSelectedBlocks)
-      },
-      {
-        role: "system",
-        content: "This is the updated entire document:"
-      },
-      {
-        role: "system",
-        content: JSON.stringify(r.htmlDocument)
-      },
-      {
-        role: "system",
-        content: `You SHOULD use "update" operations to update blocks you added / edited previously 
-          (unless the user explicitly asks you otherwise to add or delete other blocks).
-          
-          The user now asks you to do the following:`
-      },
-      {
-        role: "user",
-        content: t.userPrompt
-      }
-    ] : zt({
-      ...r,
-      userPrompt: t.userPrompt
-    });
-  } else {
-    const r = await He(e, t);
-    return t.previousMessages ? [
-      ...t.previousMessages,
-      {
-        role: "system",
-        content: `After processing the previous response, this is the updated document.
-            Ignore previous documents, you MUST issue operations against this latest version of the document:`
-      },
-      {
-        role: "system",
-        content: JSON.stringify(r.htmlBlocks)
-      },
-      {
-        role: "system",
-        content: `You SHOULD use "update" operations to update blocks you added / edited previously 
-          (unless the user explicitly asks you otherwise to add or delete other blocks).
-          
-          The user now asks you to do the following:`
-      },
-      {
-        role: "user",
-        content: t.userPrompt
-      }
-    ] : Ft({
-      ...r,
-      userPrompt: t.userPrompt,
-      isEmptyDocument: o
-    });
-  }
-};
-function Vt(e) {
+function ro(e) {
   let t, o = !1;
   return e.descendants((r, n) => o ? !1 : (r.isText && (o = !0, t = n), !0)), t;
 }
-function R(e) {
+function J(e) {
   var s;
-  const t = ot(e), { modification: o } = t.marks, r = [], n = new A(e.before);
+  const t = st(e), { modification: o } = t.marks, r = [], n = new D(e.before);
   for (let a = 0; a < e.steps.length; a++) {
-    const i = e.steps[a], c = new Ie(e.mapping.maps.slice(0, a)).invert();
+    const i = e.steps[a], l = new je(e.mapping.maps.slice(0, a)).invert();
     if (i.structure) {
-      if (i instanceof _) {
+      if (i instanceof N) {
         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 (i instanceof St) {
+      } else if (i instanceof vt) {
         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(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({
+      const b = l.map(i.from), g = i.slice.content.firstChild, v = n.doc.resolve(n.mapping.map(b)).nodeAfter;
+      let S = g.marks || [];
+      g.type !== v.type && (S = o.create({
         type: "nodeType",
-        previousValue: k.type.name,
+        previousValue: v.type.name,
         newValue: g.type.name
-      }).addToSet(w));
+      }).addToSet(S));
       const j = /* @__PURE__ */ new Set([
         ...Object.keys(g.attrs),
-        ...Object.keys(k.attrs)
+        ...Object.keys(v.attrs)
       ]);
-      for (const P of j)
-        g.attrs[P] !== k.attrs[P] && (w = o.create({
+      for (const C of j)
+        g.attrs[C] !== v.attrs[C] && (S = o.create({
           type: "attr",
-          attrName: P,
-          previousValue: k.attrs[P],
-          newValue: g.attrs[P]
-        }).addToSet(w));
-      const M = n.steps.length;
+          attrName: C,
+          previousValue: v.attrs[C],
+          newValue: g.attrs[C]
+        }).addToSet(S));
+      const B = n.steps.length;
       n.setNodeMarkup(
         n.mapping.map(b),
         g.type,
         g.attrs,
-        w
+        S
       ), r.push({
-        prosemirrorSteps: n.steps.slice(M),
+        prosemirrorSteps: n.steps.slice(B),
         selection: void 0,
         type: "replace"
       });
       continue;
     }
-    if (!(i instanceof _))
+    if (!(i instanceof N))
       throw new Error("Step is not a ReplaceStep");
-    const u = c.map(i.from), p = c.map(i.to);
+    const c = l.map(i.from), u = l.map(i.to);
     if (i.slice.openStart > 0 || i.slice.openEnd > 0)
       throw new Error(
         "Slice has openStart or openEnd > 0, but structure=false"
@@ -658,60 +839,60 @@ function R(e) {
     r.push({
       prosemirrorSteps: [],
       selection: {
-        anchor: n.mapping.map(u),
-        head: n.mapping.map(p)
+        anchor: n.mapping.map(c),
+        head: n.mapping.map(u)
       },
       type: "select"
     });
-    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)
-      m = i.slice.content.size;
-    else if (l.length === 0)
-      m = i.slice.content.size;
+    const m = i.slice.content.textBetween(0, i.slice.size), p = m === n.doc.textBetween(n.mapping.map(c), n.mapping.map(u));
+    let d;
+    if (p)
+      d = i.slice.content.size;
+    else if (m.length === 0)
+      d = i.slice.content.size;
     else {
-      const b = Vt(i.slice.content);
+      const b = ro(i.slice.content);
       if (b === void 0)
         throw new Error("unexpected: no first character found");
-      m = b + 1;
+      d = b + 1;
     }
-    let h = n.mapping.map(p);
-    const d = n.mapping.map(p);
-    let S = !0;
-    for (let b = m; b <= i.slice.content.size; b++) {
-      const g = S && u !== p, k = n.steps.length;
+    let h = n.mapping.map(u);
+    const f = n.mapping.map(u);
+    let k = !0;
+    for (let b = d; b <= i.slice.content.size; b++) {
+      const g = k && c !== u, v = n.steps.length;
       if (g) {
-        const M = n.doc.resolve(n.mapping.map(u));
-        (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 B = n.doc.resolve(n.mapping.map(c));
+        (s = B.nodeAfter) != null && s.isBlock && n.addNodeMark(B.pos, t.mark("deletion", {})), n.addMark(B.pos, h, t.mark("deletion", {})), h = n.mapping.map(u);
       }
-      const w = new Y(i.slice.content.cut(0, b), 0, 0);
-      n.replace(d, h, w).addMark(
-        d,
-        d + w.content.size,
+      const S = new G(i.slice.content.cut(0, b), 0, 0);
+      n.replace(f, h, S).addMark(
+        f,
+        f + S.content.size,
         t.mark("insertion", {})
       ), n.doc.nodesBetween(
-        d,
-        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 j = gt.near(
-        n.doc.resolve(d + w.content.size),
+        f,
+        f + S.content.size,
+        (B, C) => C < f || C > f + S.content.size ? !0 : (B.isBlock && n.addNodeMark(C, t.mark("insertion", {})), !1)
+      ), h = n.mapping.slice(v).map(h);
+      const j = ht.near(
+        n.doc.resolve(f + S.content.size),
         -1
       );
       r.push({
-        prosemirrorSteps: n.steps.slice(k),
+        prosemirrorSteps: n.steps.slice(v),
         selection: {
           anchor: j.from,
           head: j.from
         },
         type: g ? "replace" : "insert"
         // 3. Insert the replacement character by character
-      }), S = !1;
+      }), k = !1;
     }
   }
   return r;
 }
-async function Q(e) {
+async function te(e) {
   const t = Math.random() * 0.3 + 0.85;
   if (e.type === "select")
     await new Promise((o) => setTimeout(o, 100 * t));
@@ -720,9 +901,9 @@ async function Q(e) {
   else if (e.type === "replace")
     await new Promise((o) => setTimeout(o, 200 * t));
   else
-    throw new Pe(e.type);
+    throw new Be(e.type);
 }
-function ee(e, t) {
+function oe(e, t) {
   e.setMeta("addToHistory", !1), t.selection && e.setMeta("aiAgent", {
     selection: {
       anchor: t.selection.anchor,
@@ -734,10 +915,10 @@ function ee(e, t) {
       throw new Error("failed to apply step");
   return e;
 }
-function Zt(e, t, o) {
-  const r = new A(t);
+function no(e, t, o) {
+  const r = new D(t);
   for (const a of e) {
-    const i = new _(
+    const i = new N(
       r.mapping.map(a.fromA),
       r.mapping.map(a.toA),
       o.slice(a.fromB, a.toB)
@@ -750,27 +931,27 @@ function Zt(e, t, o) {
     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(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) {
-        m = d;
+    const l = !a.isLeaf && a.content.eq(i.content), c = l ? 1 : Math.min(a.nodeSize, i.nodeSize), u = s + c, m = n.map(s), p = n.map(u);
+    let d = e.length;
+    for (let f = 0; f < e.length; f++)
+      if (e[f].fromA >= p) {
+        d = f;
         break;
       }
-    e.splice(m, 0, {
-      fromA: l,
-      toA: f,
+    e.splice(d, 0, {
+      fromA: m,
+      toA: p,
       fromB: s,
-      toB: p,
+      toB: u,
       deleted: [],
       inserted: [],
-      type: c ? "node-type-or-attr-update" : "mark-update"
+      type: l ? "node-type-or-attr-update" : "mark-update"
     }), r.step(
-      new _(
+      new N(
         s,
-        p,
-        o.slice(s, p),
-        c
+        u,
+        o.slice(s, u),
+        l
       )
     );
     const h = r.doc.content.findDiffStart(o.content);
@@ -780,7 +961,7 @@ function Zt(e, t, o) {
   }
   return e;
 }
-const pe = (e, t) => {
+const ye = (e, t) => {
   const o = /* @__PURE__ */ new Set(), r = /* @__PURE__ */ new Set();
   e.descendants((a) => {
     a.type.name === "tableCell" && o.add(JSON.stringify(a.toJSON()));
@@ -809,81 +990,81 @@ const pe = (e, t) => {
     compareTokens: (a, i) => a === i
   };
 };
-function J(e, t, o = !1, r, n) {
-  const s = rt(e.id, t), a = new A(t);
-  nt(
+function W(e, t, o = !1, r, n) {
+  const s = Me(e.id, t), a = new D(t);
+  it(
     a,
     s.posBeforeNode,
     e.block,
     r,
     n
   );
-  let i = a.doc, c = ce.create(
+  let i = a.doc, l = me.create(
     t,
     void 0,
-    pe(t, i)
+    ye(t, i)
   );
-  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) {
+  if (l = l.addSteps(i, a.mapping.maps, 0), o && l.changes.length > 0) {
+    const m = l.changes[l.changes.length - 1], p = m.toA - m.fromA, d = m.toB - m.fromB;
+    if (p > d) {
       const h = t.slice(
-        l.fromA + m,
-        l.toA
+        m.fromA + d,
+        m.toA
       );
       a.step(
-        new _(l.toB, l.toB, h)
-      ), i = a.doc, c = ce.create(
-        c.startDoc,
+        new N(m.toB, m.toB, h)
+      ), i = a.doc, l = me.create(
+        l.startDoc,
         void 0,
-        pe(c.startDoc, i)
-      ), c = c.addSteps(i, a.mapping.maps, 0);
+        ye(l.startDoc, i)
+      ), l = l.addSteps(i, a.mapping.maps, 0);
     }
   }
-  const u = [], p = wt(
-    c.changes,
+  const c = [], u = Pt(
+    l.changes,
     i
   );
-  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)) {
+  for (let m = 0; m < u.length; m++) {
+    const p = u[m], d = i.slice(p.fromB, p.toB);
+    if (d.openEnd === 1 && d.openStart === 0 && (p.type = "node-type-or-attr-update", d.size > 2)) {
       const h = {
-        fromA: f.fromA,
-        toA: f.fromA + 1,
-        fromB: f.fromB,
-        toB: f.fromB + 1,
+        fromA: p.fromA,
+        toA: p.fromA + 1,
+        fromB: p.fromB,
+        toB: p.fromB + 1,
         deleted: [],
         inserted: [],
         type: "node-type-or-attr-update"
-      }, d = {
-        fromA: f.fromA + 1,
-        toA: f.toA,
-        fromB: f.fromB + 1,
-        toB: f.toB,
+      }, f = {
+        fromA: p.fromA + 1,
+        toA: p.toA,
+        fromB: p.fromB + 1,
+        toB: p.toB,
         deleted: [],
         inserted: []
       };
-      p.splice(l, 1, h, d), l++;
+      u.splice(m, 1, h, f), m++;
     }
   }
-  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)
+  no(u, t, i);
+  for (let m = 0; m < u.length; m++) {
+    const p = u[m], d = i.slice(p.fromB, p.toB);
+    if (d.openEnd > 0 && d.size > 1)
       throw new Error(
         "unexpected, openEnd > 0 and size > 1, this should have been split into two steps"
       );
-    l === p.length - 1 && o && f.type === "mark-update" || u.push(
-      new _(
-        f.fromA,
-        f.toA,
-        m,
-        f.type === "node-type-or-attr-update"
+    m === u.length - 1 && o && p.type === "mark-update" || c.push(
+      new N(
+        p.fromA,
+        p.toA,
+        d,
+        p.type === "node-type-or-attr-update"
       )
     );
   }
-  return u;
+  return c;
 }
-function Wt(e, t) {
+function so(e, t) {
   if (!e || !Array.isArray(e) || e.length === 0)
     return {
       ok: !1,
@@ -904,13 +1085,13 @@ function Wt(e, t) {
     value: o
   };
 }
-function te(e) {
+function re(e) {
   return (t, o) => {
     const r = typeof e.schema == "function" ? e.schema(t) : e.schema;
     return {
       name: "add",
       description: e.description,
-      parameters: {
+      inputSchema: {
         type: "object",
         properties: {
           referenceId: {
@@ -960,9 +1141,9 @@ function te(e) {
             ok: !1,
             error: "referenceId not found"
           };
-        const i = Wt(
+        const i = so(
           n.blocks,
-          (c) => e.validateBlock(c, t)
+          (l) => e.validateBlock(l, t)
         );
         return i.ok ? {
           ok: !0,
@@ -976,66 +1157,67 @@ function te(e) {
       },
       // Note: functionality mostly tested in jsontools.test.ts
       // would be nicer to add a direct unit test
-      execute: async function* (n) {
-        var i;
-        let s = [];
-        const a = {};
-        for await (const c of n) {
-          if (c.isUpdateToPreviousOperation || (s = []), c.operation.type !== "add") {
-            yield c;
-            continue;
-          }
-          const u = c.operation, p = await e.toJSONToolCall(t, {
-            ...c,
-            operation: u
-          });
-          if (p && !(c.isPossiblyPartial && X(
-            p.blocks[p.blocks.length - 1]
-          ))) {
-            for (let l = 0; l < p.blocks.length; l++) {
-              const f = p.blocks[l], m = t.prosemirrorState.tr;
-              let h = [];
-              if (l < s.length) {
-                const d = await e.rebaseTool(s[l], t), b = J(
+      executor: () => {
+        let n = [];
+        const s = {};
+        return {
+          execute: async (a) => {
+            var c;
+            if (a.isUpdateToPreviousOperation || (n = []), a.operation.type !== "add")
+              return !1;
+            const i = a.operation, l = await e.toJSONToolCall(t, {
+              ...a,
+              operation: i
+            });
+            if (!l || a.isPossiblyPartial && ee(
+              l.blocks[l.blocks.length - 1]
+            ))
+              return !0;
+            for (let u = 0; u < l.blocks.length; u++) {
+              const m = l.blocks[u], p = t.prosemirrorState.tr;
+              let d = [];
+              if (u < n.length) {
+                const h = await e.rebaseTool(n[u], t), k = W(
                   {
-                    id: s[l],
-                    block: f
+                    id: n[u],
+                    block: m
                   },
-                  d.doc,
+                  h.doc,
                   !1
-                ).map((g) => g.map(d.invertMap));
-                for (const g of b)
-                  m.step(g.map(m.mapping));
-                h = R(m), h = h.filter((g) => g.type !== "select");
+                ).map((b) => b.map(h.invertMap));
+                for (const b of k)
+                  p.step(b.map(p.mapping));
+                d = J(p), d = d.filter(
+                  (b) => b.type !== "select"
+                );
               } else {
-                const d = u.position === "after" ? a[u.referenceId] : void 0, S = st(
-                  m,
-                  [f],
-                  l > 0 ? s[l - 1] : d || u.referenceId,
-                  l > 0 ? "after" : u.position
+                const h = i.position === "after" ? s[i.referenceId] : void 0, f = at(
+                  p,
+                  [m],
+                  u > 0 ? n[u - 1] : h || i.referenceId,
+                  u > 0 ? "after" : i.position
                 );
-                s.push(...S.map((b) => b.id)), h = R(m);
+                n.push(...f.map((k) => k.id)), d = J(p);
               }
-              h.find((d) => d.type === "replace");
-              for (const d of h)
-                o.withDelays && await Q(d), t.transact((S) => {
-                  ee(S, d);
-                }), (i = o.onBlockUpdate) == null || i.call(o, s[l]);
+              for (const h of d)
+                o.withDelays && await te(h), t.transact((f) => {
+                  oe(f, h);
+                }), (c = o.onBlockUpdate) == null || c.call(o, n[u]);
             }
-            c.isPossiblyPartial || u.position === "after" && (a[u.referenceId] = s[s.length - 1]);
+            return a.isPossiblyPartial || i.position === "after" && (s[i.referenceId] = n[n.length - 1]), !0;
           }
-        }
+        };
       }
     };
   };
 }
-function oe(e) {
+function ne(e) {
   return (t, o) => {
     const r = typeof e.schema == "function" ? e.schema(t) : e.schema;
     return {
       name: "update",
       description: e.description,
-      parameters: {
+      inputSchema: {
         type: "object",
         properties: {
           id: {
@@ -1090,55 +1272,59 @@ function oe(e) {
       },
       // 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 i = o.updateSelection ? {
-          from: se(t, o.updateSelection.from),
-          to: se(t, o.updateSelection.to)
+      executor: () => {
+        let s = 50;
+        const a = o.updateSelection ? {
+          from: ae(t, o.updateSelection.from),
+          to: ae(t, o.updateSelection.to)
         } : void 0;
-        for await (const u of n) {
-          if (u.operation.type !== "update") {
-            yield u;
-            continue;
+        return {
+          execute: async (i) => {
+            var b;
+            if (i.operation.type !== "update")
+              return !1;
+            const l = i.operation;
+            if (i.isPossiblyPartial) {
+              const g = JSON.stringify(l.block).length;
+              if (g < s)
+                return !0;
+              s = g + 50;
+            } else
+              s = 50;
+            const c = await e.rebaseTool(l.id, t), u = a ? c.invertMap.invert().map(a.from()) : void 0, m = a ? c.invertMap.invert().map(a.to()) : void 0, p = await e.toJSONToolCall(t, {
+              ...i,
+              operation: l
+            });
+            if (!p)
+              return !0;
+            const d = W(
+              p,
+              c.doc,
+              i.isPossiblyPartial,
+              u,
+              m
+            );
+            if (d.length === 1 && i.isPossiblyPartial)
+              return !0;
+            const h = d.map((g) => g.map(c.invertMap)), f = new D(t.prosemirrorState.doc);
+            for (const g of h)
+              f.step(g.map(f.mapping));
+            const k = J(f);
+            for (const g of k)
+              o.withDelays && await te(g), t.transact((v) => {
+                oe(v, g);
+              }), (b = o.onBlockUpdate) == null || b.call(o, l.id);
+            return !0;
           }
-          const p = u.operation;
-          if (u.isPossiblyPartial) {
-            const k = JSON.stringify(p.block).length;
-            if (k < a)
-              continue;
-            a = k + 50;
-          } else
-            a = 50;
-          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(
-            h,
-            l.doc,
-            u.isPossiblyPartial,
-            f,
-            m
-          );
-          if (d.length === 1 && u.isPossiblyPartial)
-            continue;
-          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 = 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, p.id);
-        }
+        };
       }
     };
   };
 }
-const re = (e, t) => ({
+const se = (e, t) => ({
   name: "delete",
   description: "Delete a block",
-  parameters: {
+  inputSchema: {
     type: "object",
     properties: {
       id: {
@@ -1181,24 +1367,23 @@ const re = (e, t) => ({
   },
   // Note: functionality mostly tested in jsontools.test.ts
   // would be nicer to add a direct unit test
-  execute: async function* (o) {
-    var r;
-    for await (const n of o) {
-      if (n.operation.type !== "delete") {
-        yield n;
-        continue;
-      }
-      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, s.id);
+  executor: () => ({
+    execute: async (o) => {
+      var a;
+      if (o.operation.type !== "delete")
+        return !1;
+      const r = o.operation, n = e.prosemirrorState.tr;
+      lt(n, [r.id], []);
+      const s = J(n);
+      for (const i of s)
+        t.withDelays && await te(i), e.transact((l) => {
+          oe(l, i);
+        }), (a = t.onBlockUpdate) == null || a.call(t, r.id);
+      return !0;
     }
-  }
+  })
 });
-function fe(e) {
+function ge(e) {
   const t = e.lastIndexOf("<"), o = e.lastIndexOf(">");
   let r = e;
   if (t > o && (r = e.substring(0, t), !r.trim()))
@@ -1214,15 +1399,15 @@ function fe(e) {
   ).body.firstChild;
   return i ? i.innerHTML : "";
 }
-function H(e) {
+function U(e) {
   let t;
-  if (Be(e.prosemirrorState, (o) => {
+  if (Ne(e.prosemirrorState, (o) => {
     t = o;
   }), !t)
     throw new Error("applySuggestionsTr is not set");
   return t;
 }
-function q(e, t) {
+function V(e, t) {
   const o = t.mapping.invert();
   return {
     doc: t.doc,
@@ -1230,7 +1415,7 @@ function q(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 A(t.doc),
+    tr: () => new D(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
@@ -1254,9 +1439,9 @@ function q(e, t) {
     }
   };
 }
-async function he(e, t) {
-  var u;
-  const o = H(t), r = Oe(o.doc, e);
+async function be(e, t) {
+  var c;
+  const o = U(t), r = Ee(o.doc, e);
   if (!r)
     throw new Error("block not found");
   const n = await t.blocksToHTMLLossy([
@@ -1264,11 +1449,11 @@ async function he(e, t) {
       ...r,
       children: []
     }
-  ]), s = (u = window.__TEST_OPTIONS) == null ? void 0 : u.mockID, a = await t.tryParseHTMLToBlocks(n);
+  ]), s = (c = window.__TEST_OPTIONS) == null ? void 0 : c.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 i = a[0];
-  if (i.id = e, J(
+  if (i.id = e, W(
     {
       id: e,
       block: i
@@ -1276,9 +1461,9 @@ async function he(e, t) {
     o.doc
   ).length)
     throw new Error("html diff");
-  return q(t, o);
+  return V(t, o);
 }
-function ye(e) {
+function ke(e) {
   return typeof e != "string" ? {
     ok: !1,
     error: "block must be a string"
@@ -1287,8 +1472,8 @@ function ye(e) {
     value: e
   };
 }
-const V = {
-  add: te({
+const H = {
+  add: re({
     description: "Insert new blocks",
     schema: {
       block: {
@@ -1296,13 +1481,13 @@ const V = {
         description: "html of block (MUST be a single, VALID HTML element)"
       }
     },
-    validateBlock: ye,
-    rebaseTool: he,
+    validateBlock: ke,
+    rebaseTool: be,
     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 (s) => {
-          const a = t.isPossiblyPartial ? fe(s) : s;
+          const a = t.isPossiblyPartial ? ge(s) : s;
           return a ? (await e.tryParseHTMLToBlocks(a)).map(
             (i) => (delete i.id, i)
           ) : [];
@@ -1315,7 +1500,7 @@ const V = {
         };
     }
   }),
-  update: oe({
+  update: ne({
     description: "Update a block",
     schema: {
       block: {
@@ -1323,10 +1508,10 @@ const V = {
         description: "html of block (MUST be a single HTML element)"
       }
     },
-    validateBlock: ye,
-    rebaseTool: he,
+    validateBlock: ke,
+    rebaseTool: be,
     toJSONToolCall: async (e, t) => {
-      const o = t.isPossiblyPartial ? fe(t.operation.block) : t.operation.block;
+      const o = t.isPossiblyPartial ? ge(t.operation.block) : t.operation.block;
       if (!o)
         return;
       const r = (await e.tryParseHTMLToBlocks(o))[0];
@@ -1336,244 +1521,115 @@ const V = {
       };
     }
   }),
-  delete: re
+  delete: se
 };
-function Yt(e, t, o, r, n) {
-  const s = {
+function io(e, t, o, r, n) {
+  if (typeof r == "boolean") {
+    const i = r ? e.getSelectionCutBlocks() : void 0;
+    r = i ? {
+      from: i._meta.startPos,
+      to: i._meta.endPos
+    } : void 0;
+  }
+  const s = o ?? {
     add: !0,
     update: !0,
-    delete: !0,
-    ...o
+    delete: !0
   };
   return [
     ...s.update ? [
-      V.update(e, {
+      H.update(e, {
         idsSuffixed: !0,
         withDelays: t,
         updateSelection: r,
         onBlockUpdate: n
       })
     ] : [],
-    ...s.add ? [V.add(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : [],
-    ...s.delete ? [V.delete(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : []
+    ...s.add ? [H.add(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : [],
+    ...s.delete ? [H.delete(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : []
   ];
 }
-const Ue = {
+const ao = {
   /**
    * Function to get the stream tools that can apply HTML block updates to the editor
    */
-  getStreamTools: Yt,
+  getStreamToolsProvider: (e = {}) => ({
+    getStreamTools: (t, o, r) => io(
+      t,
+      e.withDelays ?? !0,
+      e.defaultStreamTools,
+      o,
+      r
+    )
+  }),
+  tools: H,
   /**
    * The default PromptBuilder that determines how a userPrompt is converted to an array of
    * LLM Messages (CoreMessage[])
    */
-  defaultPromptBuilder: Jt,
+  defaultPromptBuilder: to,
+  /**
+   * The default PromptInputDataBuilder that can take an editor and user request and convert it to the input required for the PromptBuilder
+   */
+  defaultPromptInputDataBuilder: oo,
   /**
    * Helper functions which can be used when implementing a custom PromptBuilder
    */
   promptHelpers: {
-    getDataForPromptNoSelection: He,
-    getDataForPromptWithSelection: qe
+    getDataForPromptNoSelection: Ue,
+    getDataForPromptWithSelection: Ve
   }
 };
-async function Gt(e, t) {
-  const {
-    userPrompt: o,
-    useSelection: r,
-    deleteEmptyCursorBlock: n,
-    stream: s,
-    onStart: a,
-    withDelays: i,
-    dataFormat: c,
-    previousResponse: u,
-    ...p
-  } = {
-    maxRetries: 2,
-    deleteEmptyCursorBlock: !0,
-    stream: !0,
-    withDelays: !0,
-    dataFormat: Ue,
-    ...t
-  }, 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: "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()
-    )
-  }));
-  const b = await l(e, {
-    selectedBlocks: d == null ? void 0 : d.blocks,
-    userPrompt: o,
-    excludeBlockIds: h ? [h] : void 0,
-    previousMessages: S
-  }), g = f(
-    e,
-    i,
-    t.defaultStreamTools,
-    d ? { from: d._meta.startPos, to: d._meta.endPos } : void 0,
-    t.onBlockUpdate
-  );
-  let k;
-  return s ? k = await Dt(
-    g,
-    {
-      messages: b,
-      ...p
-    },
-    () => {
-      h && e.getBlock(h) && e.removeBlocks([h]), a == null || a();
-    }
-  ) : (k = await At(g, {
-    messages: b,
-    ...p
-  }), h && e.getBlock(h) && e.removeBlocks([h]), a == null || a()), new qt(b, k, g);
-}
-async function ze(e, t) {
-  const o = e.getTextCursorPosition().block.id, r = x(e.document, {
-    cursorBlockId: o
-  }), n = await C(
-    T(r),
+async function lo(e) {
+  return e.selectedBlocks ? {
+    ...await Re(e.editor, {
+      selectedBlocks: e.selectedBlocks
+    }),
+    userPrompt: e.userPrompt
+  } : {
+    ...await Fe(e.editor, {
+      excludeBlockIds: e.emptyCursorBlockToDelete ? [e.emptyCursorBlockToDelete] : void 0
+    }),
+    userPrompt: e.userPrompt
+  };
+}
+async function Fe(e, t) {
+  const o = w(e.document).length === 0, r = e.getTextCursorPosition().block.id, n = w(e.document, {
+    cursorBlockId: r
+  }), s = await P(
+    T(n),
     async (c) => ({
       ...c,
       children: void 0
     })
-  ), a = F(e, n).filter(
+  ), i = Z(e, s).filter(
     (c) => "cursor" in c || !(t.excludeBlockIds || []).includes(c.id)
   );
   return {
-    jsonBlocks: L(a)
+    selection: !1,
+    jsonBlocks: _(i),
+    isEmptyDocument: o
   };
 }
-async function Fe(e, t) {
-  const o = await C(
+async function Re(e, t) {
+  const o = w(e.document).length === 0, r = await P(
     T(t.selectedBlocks),
-    async (n) => n
-  );
+    async (s) => s
+  ), n = _(r);
   return {
-    jsonSelectedBlocks: L(o),
-    jsonDocument: (await C(T(e.document), async (n) => ({
-      ...n,
+    isEmptyDocument: o,
+    selection: !0,
+    jsonSelectedBlocks: n,
+    jsonDocument: (await P(T(e.document), async (s) => ({
+      ...s,
       id: void 0,
       // don't pass id, because LLM should use `jsonSelectedBlocks` for this
       children: void 0
-    }))).map(({ block: n }) => ({ block: n }))
+    }))).map(({ block: s }) => ({ block: s }))
     // strip ids so LLM can't accidentally issue updates to ids not in selection
   };
 }
-function Kt(e) {
-  return [
-    {
-      role: "system",
-      content: `You're manipulating a selected part of a text document using JSON blocks. 
-      Make sure to follow the json schema provided and always include the trailing $ in ids. 
-      This is the selection as an array of JSON blocks:`
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.jsonSelectedBlocks)
-    },
-    {
-      role: "system",
-      content: "This is the entire document (INCLUDING the selected text), find the selected text in there to understand the context:"
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.jsonDocument)
-    },
-    {
-      role: "system",
-      content: "The user asks you to do the following:"
-    },
-    {
-      role: "user",
-      content: e.userPrompt
-    }
-  ];
-}
-function Xt(e) {
-  return [
-    {
-      role: "system",
-      content: `You're manipulating a text document using JSON blocks. 
-        Make sure to follow the json schema provided. When referencing ids they MUST be EXACTLY the same (including the trailing $). 
-        This is the document as an array of JSON blocks (the cursor is BETWEEN two blocks as indicated by cursor: true):`
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.jsonBlocks)
-    },
-    ...e.isEmptyDocument ? [
-      {
-        role: "system",
-        content: "Because the actual document is empty, this is an example document to understand the schema:"
-      },
-      {
-        role: "system",
-        content: JSON.stringify({
-          id: "ref3",
-          type: "paragraph",
-          content: [
-            {
-              type: "text",
-              text: "Hello, world! ",
-              styles: {}
-            },
-            {
-              type: "text",
-              text: "Bold text. ",
-              styles: {
-                bold: !0
-              }
-            },
-            {
-              type: "link",
-              href: "https://www.w3.org",
-              content: "Link."
-            }
-          ]
-        })
-      }
-    ] : [],
-    {
-      role: "system",
-      content: "The user asks you to do the following:"
-    },
-    {
-      role: "system",
-      content: 'First, determine what part of the document the user is talking about. You SHOULD probably take cursor info into account if needed.\n       EXAMPLE: if user says "below" (without pointing to a specific part of the document) he / she probably indicates the block(s) after the cursor. \n       EXAMPLE: If you want to insert content AT the cursor position (UNLESS indicated otherwise by the user), then you need `referenceId` to point to the block before the cursor with position `after` (or block below and `before`).\n      \n      ' + (e.isEmptyDocument ? "Because the document is empty, first update the empty block before adding new blocks." : "Prefer updating existing blocks over removing and adding (but this also depends on the user's question).")
-    },
-    {
-      role: "user",
-      content: e.userPrompt
-    }
-  ];
-}
-const Qt = async (e, t) => {
-  const o = x(e.document).length === 0;
-  if (t.selectedBlocks) {
-    const r = await Fe(e, {
-      selectedBlocks: t.selectedBlocks
-    });
-    return Kt({
-      ...r,
-      userPrompt: t.userPrompt
-    });
-  } else {
-    const r = await ze(e, t);
-    return Xt({
-      ...r,
-      userPrompt: t.userPrompt,
-      isEmptyDocument: o
-    });
-  }
-};
-function eo(e) {
+function co(e) {
   const t = {}, o = {};
   return e.forEach((n) => {
     const { type: s, ...a } = n.properties, i = JSON.stringify(a);
@@ -1594,7 +1650,7 @@ function eo(e) {
     };
   });
 }
-function to(e) {
+function uo(e) {
   return {
     type: "object",
     properties: Object.fromEntries(
@@ -1608,7 +1664,7 @@ function to(e) {
     additionalProperties: !1
   };
 }
-function oo() {
+function po() {
   return {
     type: "object",
     properties: {
@@ -1627,7 +1683,7 @@ function oo() {
     required: ["type", "text"]
   };
 }
-function Je(e) {
+function Ze(e) {
   return {
     type: "object",
     properties: Object.fromEntries(
@@ -1642,7 +1698,7 @@ function Je(e) {
     additionalProperties: !1
   };
 }
-function ro(e) {
+function mo(e) {
   return {
     type: "array",
     items: {
@@ -1680,7 +1736,7 @@ function ro(e) {
               $ref: "#/$defs/styledtext"
             }
           } : void 0,
-          props: Je(o.propSchema)
+          props: Ze(o.propSchema)
         },
         additionalProperties: !1,
         required: ["type", ...o.content === "styled" ? ["content"] : []]
@@ -1688,9 +1744,9 @@ function ro(e) {
     }
   };
 }
-function no(e) {
+function fo(e) {
   return {
-    anyOf: eo(
+    anyOf: co(
       Object.entries(e).map(([t, o]) => ({
         type: "object",
         properties: {
@@ -1700,7 +1756,7 @@ function no(e) {
           },
           content: o.content === "inline" ? { $ref: "#/$defs/inlinecontent" } : o.content === "table" ? { type: "object", properties: {} } : void 0,
           // filter out default props (TODO: make option)
-          props: Je(o.propSchema)
+          props: Ze(o.propSchema)
           // Object.fromEntries(
           //   Object.entries(val.propSchema).filter(
           //     (key) => typeof (defaultProps as any)[key[0]] === "undefined"
@@ -1715,7 +1771,7 @@ function no(e) {
     )
   };
 }
-function so(e) {
+function ho(e) {
   const t = JSON.parse(
     JSON.stringify({
       blockSchema: e.blockSchema,
@@ -1724,6 +1780,7 @@ function so(e) {
     })
   );
   return {
+    // TODO
     removeFileBlocks() {
       return t.blockSchema = Object.fromEntries(
         Object.entries(t.blockSchema).filter(
@@ -1742,7 +1799,7 @@ function so(e) {
             ...r,
             propSchema: Object.fromEntries(
               Object.entries(r.propSchema).filter(
-                (n) => typeof Me[n[0]] > "u"
+                (n) => typeof _e[n[0]] > "u"
               )
             )
           }
@@ -1754,22 +1811,22 @@ function so(e) {
     }
   };
 }
-function ge(e) {
-  return e = so(e).removeFileBlocks().removeDefaultProps().get(), {
+function Se(e) {
+  return e = ho(e).removeFileBlocks().removeDefaultProps().get(), {
     $defs: {
-      styles: to(e.styleSchema),
-      styledtext: oo(),
-      inlinecontent: ro(
+      styles: uo(e.styleSchema),
+      styledtext: po(),
+      inlinecontent: mo(
         e.inlineContentSchema
       ),
-      block: no(e.blockSchema)
+      block: fo(e.blockSchema)
     }
   };
 }
-function Ve(e, t) {
-  return !t.schema.inlineContentSchema[e.type] || at(e) && !("text" in e) ? !1 : ct(e) ? !("content" in e) || !("href" in e) ? !1 : Ve(e.content, t) : !0;
+function We(e, t) {
+  return !t.schema.inlineContentSchema[e.type] || ct(e) && !("text" in e) ? !1 : ut(e) ? !("content" in e) || !("href" in e) ? !1 : We(e.content, t) : !0;
 }
-function ke(e, t, o) {
+function we(e, t, o) {
   const r = e.type || o, n = t.schema.blockSchema[r];
   if (!n)
     return {
@@ -1798,7 +1855,7 @@ function ke(e, t, o) {
         ok: !0,
         value: e
       };
-    if (!e.content.every((s) => Ve(s, t)))
+    if (!e.content.every((s) => We(s, t)))
       return {
         ok: !1,
         error: "block content must be an array of inline content"
@@ -1809,32 +1866,32 @@ function ke(e, t, o) {
     value: e
   };
 }
-const Z = {
-  add: te({
+const q = {
+  add: re({
     description: "Insert new blocks",
     schema: (e) => ({
       block: {
         $ref: "#/$defs/block"
       },
-      ...ge(e.schema)
+      ...Se(e.schema)
     }),
-    validateBlock: ke,
-    rebaseTool: async (e, t) => q(t, H(t)),
+    validateBlock: we,
+    rebaseTool: async (e, t) => V(t, U(t)),
     toJSONToolCall: async (e, t) => t.operation
   }),
-  update: oe({
+  update: ne({
     description: "Update a block, the new block will replace the existing block.",
     schema: (e) => ({
       block: {
         $ref: "#/$defs/block"
       },
-      ...ge(e.schema)
+      ...Se(e.schema)
     }),
-    validateBlock: ke,
-    rebaseTool: async (e, t) => q(t, H(t)),
+    validateBlock: we,
+    rebaseTool: async (e, t) => V(t, U(t)),
     toJSONToolCall: async (e, t) => {
       const o = Object.fromEntries(
-        Object.entries(Me).map(([r, n]) => [r, n.default])
+        Object.entries(_e).map(([r, n]) => [r, n.default])
       );
       return {
         ...t.operation,
@@ -1848,159 +1905,264 @@ const Z = {
       };
     }
   }),
-  delete: re
+  delete: se
+};
+function yo(e, t) {
+  e.length > 0 && e.push(
+    {
+      role: "assistant",
+      id: "document-state-" + e.length,
+      parts: [
+        {
+          type: "text",
+          text: "This is the latest state of the selection (ignore previous selections, you MUST issue operations against this latest version of the selection):"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.jsonSelectedBlocks)
+        },
+        {
+          type: "text",
+          text: "This is the latest state of the document (INCLUDING the selected text), find the selected text in there to understand the context:"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.jsonDocument)
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt-" + e.length,
+      parts: [
+        {
+          type: "text",
+          text: "The user asks you to do the following:"
+        },
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  ), e.push(
+    {
+      role: "system",
+      id: "document-state-intro",
+      parts: [
+        {
+          type: "text",
+          text: `You're manipulating a selected part of a text document using JSON blocks. 
+      Make sure to follow the json schema provided and always include the trailing $ in ids.
+      This is the selection as an array of JSON blocks:`
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.jsonSelectedBlocks)
+        },
+        {
+          type: "text",
+          text: "This is the entire document (INCLUDING the selected text), find the selected text in there to understand the context:"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.jsonDocument)
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt",
+      parts: [
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  );
+}
+function go(e, t) {
+  if (e.length > 0) {
+    e.push(
+      {
+        role: "assistant",
+        id: "document-state-" + e.length,
+        parts: [
+          {
+            type: "text",
+            text: "This is the latest state of the document (ignore previous documents, you MUST issue operations against this latest version of the document):"
+          },
+          {
+            type: "text",
+            text: JSON.stringify(t.jsonBlocks)
+          }
+        ]
+      },
+      {
+        role: "user",
+        id: "user-prompt-" + e.length,
+        parts: [
+          {
+            type: "text",
+            text: t.userPrompt
+          }
+        ]
+      }
+    );
+    return;
+  }
+  e.push(
+    {
+      role: "system",
+      id: "document-state",
+      parts: [
+        {
+          type: "text",
+          text: `You're manipulating a text document using JSON blocks. 
+        Make sure to follow the json schema provided. When referencing ids they MUST be EXACTLY the same (including the trailing $). 
+        This is the initial document as an array of JSON blocks (the cursor is BETWEEN two blocks as indicated by cursor: true):`
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.jsonBlocks)
+        },
+        {
+          type: "text",
+          text: 'First, determine what part of the document the user is talking about. You SHOULD probably take cursor info into account if needed.\n       EXAMPLE: if user says "below" (without pointing to a specific part of the document) he / she probably indicates the block(s) after the cursor. \n       EXAMPLE: If you want to insert content AT the cursor position (UNLESS indicated otherwise by the user), \n       then you need `referenceId` to point to the block before the cursor with position `after` (or block below and `before`).\n      \n      ' + (t.isEmptyDocument ? "Because the document is empty, first update the empty block before adding new blocks." : "Prefer updating existing blocks over removing and adding (but this also depends on the user's question).")
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt",
+      parts: [
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  );
+}
+const bo = async (e, t) => {
+  t.selection ? yo(e, t) : go(e, t);
 };
-function io(e, t, o, r) {
-  const n = {
+function ko(e, t, o, r, n) {
+  if (typeof r == "boolean") {
+    const i = r ? e.getSelectionCutBlocks() : void 0;
+    r = i ? {
+      from: i._meta.startPos,
+      to: i._meta.endPos
+    } : void 0;
+  }
+  const s = o ?? {
     add: !0,
     update: !0,
-    delete: !0,
-    ...o
+    delete: !0
   };
   return [
-    ...n.update ? [
-      Z.update(e, {
+    ...s.update ? [
+      q.update(e, {
         idsSuffixed: !0,
         withDelays: t,
-        updateSelection: r
+        updateSelection: r,
+        onBlockUpdate: n
       })
     ] : [],
-    ...n.add ? [Z.add(e, { idsSuffixed: !0, withDelays: t })] : [],
-    ...n.delete ? [Z.delete(e, { idsSuffixed: !0, withDelays: t })] : []
+    ...s.add ? [q.add(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : [],
+    ...s.delete ? [q.delete(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : []
   ];
 }
-const ao = {
+const So = {
   /**
-   * Function to get the stream tools that can apply BlockNote JSON block updates to the editor
+   * Function to get the stream tools that can apply JSON block updates to the editor
    */
-  getStreamTools: io,
+  getStreamToolsProvider: (e = {}) => ({
+    getStreamTools: (t, o, r) => ko(
+      t,
+      e.withDelays ?? !0,
+      e.defaultStreamTools,
+      o,
+      r
+    )
+  }),
+  tools: q,
   /**
    * The default PromptBuilder that determines how a userPrompt is converted to an array of
    * LLM Messages (CoreMessage[])
    */
-  defaultPromptBuilder: Qt,
+  defaultPromptBuilder: bo,
+  /**
+   * The default PromptInputDataBuilder that can take an editor and user request and convert it to the input required for the PromptBuilder
+   */
+  defaultPromptInputDataBuilder: lo,
   /**
    * Helper functions which can be used when implementing a custom PromptBuilder
    */
   promptHelpers: {
-    getDataForPromptNoSelection: ze,
-    getDataForPromptWithSelection: Fe
+    getDataForPromptNoSelection: Fe,
+    getDataForPromptWithSelection: Re
   }
 };
-async function Ze(e, t) {
-  const o = e.getTextCursorPosition().block.id, r = x(e.document, {
-    cursorBlockId: o
-  }), n = await C(
-    T(r),
+async function wo(e) {
+  return e.selectedBlocks ? {
+    ...await Ge(e.editor, {
+      selectedBlocks: e.selectedBlocks
+    }),
+    userPrompt: e.userPrompt
+  } : {
+    ...await Ye(e.editor, {
+      excludeBlockIds: e.emptyCursorBlockToDelete ? [e.emptyCursorBlockToDelete] : void 0
+    }),
+    userPrompt: e.userPrompt
+  };
+}
+async function Ye(e, t) {
+  const o = w(e.document).length === 0, r = e.getTextCursorPosition().block.id, n = w(e.document, {
+    cursorBlockId: r
+  }), s = await P(
+    T(n),
     async (c) => e.blocksToMarkdownLossy([c])
-  ), a = F(e, n).filter(
+  ), i = Z(e, s).filter(
     (c) => "cursor" in c || !(t.excludeBlockIds || []).includes(c.id)
   );
   return {
-    markdownBlocks: L(a)
+    selection: !1,
+    markdownBlocks: _(i),
+    isEmptyDocument: o
   };
 }
-async function We(e, t) {
-  const o = await C(
+async function Ge(e, t) {
+  const o = w(e.document).length === 0, r = await P(
     T(t.selectedBlocks),
-    async (n) => e.blocksToMarkdownLossy([n])
-  );
+    async (s) => e.blocksToMarkdownLossy([s])
+  ), n = _(r);
   return {
-    markdownSelectedBlocks: L(o),
-    markdownDocument: (await C(T(e.document), async (n) => e.blocksToMarkdownLossy([n]))).map(({ block: n }) => ({ block: n }))
+    isEmptyDocument: o,
+    selection: !0,
+    markdownSelectedBlocks: n,
+    markdownDocument: (await P(T(e.document), async (s) => e.blocksToMarkdownLossy([s]))).map(({ block: s }) => ({ block: s }))
     // strip ids so LLM can't accidentally issue updates to ids not in selection
   };
 }
-function co(e) {
-  return [
-    {
-      role: "system",
-      content: `You're manipulating a selected part of a text document using markdown blocks. 
-      Make sure to follow the json schema provided and always include the trailing $ in ids. 
-      This is the selection as an array of markdown blocks:`
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.markdownSelectedBlocks)
-    },
-    {
-      role: "system",
-      content: "This is the entire document (INCLUDING the selected text), find the selected text in there to understand the context:"
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.markdownDocument)
-    },
-    {
-      role: "system",
-      content: "The user asks you to do the following:"
-    },
-    {
-      role: "user",
-      content: e.userPrompt
-    }
-  ];
-}
-function lo(e) {
-  return [
-    {
-      role: "system",
-      content: `You're manipulating a text document using markdown blocks. 
-        Make sure to follow the json schema provided. When referencing ids they MUST be EXACTLY the same (including the trailing $). 
-        This is the document as an array of markdown blocks (the cursor is BETWEEN two blocks as indicated by cursor: true):`
-    },
-    {
-      role: "system",
-      content: JSON.stringify(e.markdownBlocks)
-    },
-    {
-      role: "system",
-      content: "The user asks you to do the following:"
-    },
-    {
-      role: "user",
-      content: e.userPrompt
-    },
-    {
-      role: "system",
-      content: 'First, determine what part of the document the user is talking about. You SHOULD probably take cursor info into account if needed.\n       EXAMPLE: if user says "below" (without pointing to a specific part of the document) he / she probably indicates the block(s) after the cursor. \n       EXAMPLE: If you want to insert content AT the cursor position (UNLESS indicated otherwise by the user), then you need `referenceId` to point to the block before the cursor with position `after` (or block below and `before`).\n      \n      Prefer updating blocks over adding or removing (but this also depends on the user\'s question).'
-    }
-  ];
-}
-const uo = async (e, t) => {
-  if (t.selectedBlocks) {
-    const o = await We(e, {
-      selectedBlocks: t.selectedBlocks
-    });
-    return co({
-      ...o,
-      userPrompt: t.userPrompt
-    });
-  } else {
-    const o = await Ze(e, t);
-    return lo({
-      ...o,
-      userPrompt: t.userPrompt
-    });
-  }
-};
-async function be(e, t) {
-  const o = H(t), r = await t.blocksToMarkdownLossy([Oe(o.doc, e)]), n = await t.tryParseMarkdownToBlocks(r), s = J(
+async function Te(e, t) {
+  const o = U(t), r = await t.blocksToMarkdownLossy([Ee(o.doc, e)]), n = await t.tryParseMarkdownToBlocks(r), s = W(
     {
       id: e,
       block: n[0]
     },
     o.doc
-  ), a = new Ie();
+  ), a = new je();
   for (const i of s) {
-    const c = i.map(a);
-    if (!c)
+    const l = i.map(a);
+    if (!l)
       throw new Error("Failed to map step");
-    o.step(c), a.appendMap(c.getMap());
+    o.step(l), a.appendMap(l.getMap());
   }
-  return q(t, o);
+  return V(t, o);
 }
-function Se(e) {
+function xe(e) {
   return typeof e != "string" ? {
     ok: !1,
     error: "block must be a string"
@@ -2009,8 +2171,8 @@ function Se(e) {
     value: e
   };
 }
-const W = {
-  add: te({
+const z = {
+  add: re({
     description: "Insert new blocks",
     schema: {
       block: {
@@ -2020,8 +2182,8 @@ const W = {
         block: { type: "string", description: "markdown of block" }
       }
     },
-    validateBlock: Se,
-    rebaseTool: be,
+    validateBlock: xe,
+    rebaseTool: Te,
     toJSONToolCall: async (e, t) => {
       const o = await Promise.all(
         t.operation.blocks.map(async (r) => {
@@ -2035,7 +2197,7 @@ const W = {
       };
     }
   }),
-  update: oe({
+  update: ne({
     description: "Update a block, the new block will replace the existing block.",
     schema: {
       block: {
@@ -2045,8 +2207,8 @@ const W = {
         block: { type: "string", description: "markdown of block" }
       }
     },
-    validateBlock: Se,
-    rebaseTool: be,
+    validateBlock: xe,
+    rebaseTool: Te,
     toJSONToolCall: async (e, t) => {
       const o = (await e.tryParseMarkdownToBlocks(t.operation.block.trim()))[0];
       return delete o.id, window.__TEST_OPTIONS && (window.__TEST_OPTIONS.mockID = void 0), {
@@ -2055,58 +2217,219 @@ const W = {
       };
     }
   }),
-  delete: re
+  delete: se
+};
+function To(e, t) {
+  e.length > 0 && e.push(
+    {
+      role: "assistant",
+      id: "document-state-" + e.length,
+      parts: [
+        {
+          type: "text",
+          text: "This is the latest state of the selection (ignore previous selections, you MUST issue operations against this latest version of the selection):"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.markdownSelectedBlocks)
+        },
+        {
+          type: "text",
+          text: "This is the latest state of the document (INCLUDING the selected text), find the selected text in there to understand the context:"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.markdownDocument)
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt-" + e.length,
+      parts: [
+        {
+          type: "text",
+          text: "The user asks you to do the following:"
+        },
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  ), e.push(
+    {
+      role: "system",
+      id: "document-state-intro",
+      parts: [
+        {
+          type: "text",
+          text: `You're manipulating a selected part of a text document using Markdown blocks. 
+      Make sure to follow the json schema provided and always include the trailing $ in ids.
+      This is the selection as an array of Markdown blocks:`
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.markdownSelectedBlocks)
+        },
+        {
+          type: "text",
+          text: "This is the entire document (INCLUDING the selected text), find the selected text in there to understand the context:"
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.markdownDocument)
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt",
+      parts: [
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  );
+}
+function xo(e, t) {
+  if (e.length > 0) {
+    e.push(
+      {
+        role: "assistant",
+        id: "document-state-" + e.length,
+        parts: [
+          {
+            type: "text",
+            text: "This is the latest state of the document (ignore previous documents, you MUST issue operations against this latest version of the document):"
+          },
+          {
+            type: "text",
+            text: JSON.stringify(t.markdownBlocks)
+          }
+        ]
+      },
+      {
+        role: "user",
+        id: "user-prompt-" + e.length,
+        parts: [
+          {
+            type: "text",
+            text: t.userPrompt
+          }
+        ]
+      }
+    );
+    return;
+  }
+  e.push(
+    {
+      role: "system",
+      id: "document-state",
+      parts: [
+        {
+          type: "text",
+          text: `You're manipulating a text document using Markdown blocks. 
+        Make sure to follow the json schema provided. When referencing ids they MUST be EXACTLY the same (including the trailing $). 
+        This is the initial document as an array of Markdown blocks (the cursor is BETWEEN two blocks as indicated by cursor: true):`
+        },
+        {
+          type: "text",
+          text: JSON.stringify(t.markdownBlocks)
+        },
+        {
+          type: "text",
+          text: 'First, determine what part of the document the user is talking about. You SHOULD probably take cursor info into account if needed.\n       EXAMPLE: if user says "below" (without pointing to a specific part of the document) he / she probably indicates the block(s) after the cursor. \n       EXAMPLE: If you want to insert content AT the cursor position (UNLESS indicated otherwise by the user), \n       then you need `referenceId` to point to the block before the cursor with position `after` (or block below and `before`).\n      \n      ' + (t.isEmptyDocument ? "Because the document is empty, first update the empty block before adding new blocks." : "Prefer updating existing blocks over removing and adding (but this also depends on the user's question).")
+        }
+      ]
+    },
+    {
+      role: "user",
+      id: "user-prompt",
+      parts: [
+        {
+          type: "text",
+          text: t.userPrompt
+        }
+      ]
+    }
+  );
+}
+const vo = async (e, t) => {
+  t.selection ? To(e, t) : xo(e, t);
 };
-function mo(e, t, o, r) {
-  const n = {
+function Po(e, t, o, r, n) {
+  if (typeof r == "boolean") {
+    const i = r ? e.getSelectionCutBlocks() : void 0;
+    r = i ? {
+      from: i._meta.startPos,
+      to: i._meta.endPos
+    } : void 0;
+  }
+  const s = o ?? {
     add: !0,
     update: !0,
-    delete: !0,
-    ...o
+    delete: !0
   };
   return [
-    ...n.update ? [
-      W.update(e, {
+    ...s.update ? [
+      z.update(e, {
         idsSuffixed: !0,
         withDelays: t,
-        updateSelection: r
+        updateSelection: r,
+        onBlockUpdate: n
       })
     ] : [],
-    ...n.add ? [W.add(e, { idsSuffixed: !0, withDelays: t })] : [],
-    ...n.delete ? [W.delete(e, { idsSuffixed: !0, withDelays: t })] : []
+    ...s.add ? [z.add(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : [],
+    ...s.delete ? [z.delete(e, { idsSuffixed: !0, withDelays: t, onBlockUpdate: n })] : []
   ];
 }
-const po = {
+const Co = {
   /**
-   * Function to get the stream tools that can apply BlockNote Markdown block updates to the editor
+   * Function to get the stream tools that can apply Markdown block updates to the editor
    */
-  getStreamTools: mo,
+  getStreamToolsProvider: (e = {}) => ({
+    getStreamTools: (t, o, r) => Po(
+      t,
+      e.withDelays ?? !0,
+      e.defaultStreamTools,
+      o,
+      r
+    )
+  }),
+  tools: z,
   /**
    * The default PromptBuilder that determines how a userPrompt is converted to an array of
    * LLM Messages (CoreMessage[])
    */
-  defaultPromptBuilder: uo,
+  defaultPromptBuilder: vo,
+  /**
+   * The default PromptInputDataBuilder that can take an editor and user request and convert it to the input required for the PromptBuilder
+   */
+  defaultPromptInputDataBuilder: wo,
   /**
    * Helper functions which can be used when implementing a custom PromptBuilder
    */
   promptHelpers: {
-    getDataForPromptNoSelection: Ze,
-    getDataForPromptWithSelection: We
+    getDataForPromptNoSelection: Ye,
+    getDataForPromptWithSelection: Ge
   }
-}, ir = {
-  addCursorPosition: F,
+}, O = {
+  _experimental_json: So,
+  _experimental_markdown: Co,
+  html: ao
+}, xr = {
+  addCursorPosition: Z,
   flattenBlocks: T,
-  suffixIDs: L,
-  trimEmptyBlocks: x,
-  convertBlocks: C
-}, fo = {
-  _experimental_json: ao,
-  _experimental_markdown: po,
-  html: Ue
-}, we = new _e("blocknote-agent-cursor");
-function ho(e) {
+  suffixIDs: _,
+  trimEmptyBlocks: w,
+  convertBlocks: P
+}, ve = new Ae("blocknote-agent-cursor");
+function Oo(e) {
   return new Le({
-    key: we,
+    key: ve,
     view: (t) => ({}),
     state: {
       init: () => ({
@@ -2123,67 +2446,79 @@ function ho(e) {
     },
     props: {
       decorations: (t) => {
-        const { doc: o } = t, { selection: r } = we.getState(t), n = [];
+        const { doc: o } = t, { selection: r } = ve.getState(t), n = [];
         if (!r)
-          return le.create(o, []);
+          return ue.create(o, []);
         n.push(
-          ue.widget(r.head, () => yo(e), {
+          de.widget(r.head, () => Io(e), {
             key: "agent-cursor",
             side: 10
           })
         );
         const s = Math.min(r.anchor, r.head), a = Math.max(r.anchor, r.head);
         return n.push(
-          ue.inline(s, a, Tt(e), {
+          de.inline(s, a, gt(e), {
             inclusiveEnd: !0,
             inclusiveStart: !1
           })
-        ), le.create(o, n);
+        ), ue.create(o, n);
       }
     }
   });
 }
-const yo = (e) => {
+const Io = (e) => {
   const t = document.createElement("span");
   t.classList.add("bn-collaboration-cursor__base"), t.setAttribute("data-active", "true");
   const o = document.createElement("span");
   o.setAttribute("contentedEditable", "false"), o.classList.add("bn-collaboration-cursor__caret"), o.setAttribute("style", `background-color: ${e.color}`);
   const r = document.createElement("span");
   return r.classList.add("bn-collaboration-cursor__label"), r.setAttribute("style", `background-color: ${e.color}`), r.insertBefore(document.createTextNode(e.name), null), o.insertBefore(r, null), t.insertBefore(document.createTextNode("⁠"), null), t.insertBefore(o, null), t.insertBefore(document.createTextNode("⁠"), null), t;
-}, go = new _e("blocknote-ai-plugin");
-class Ye extends lt {
+}, Bo = new Ae("blocknote-ai-plugin");
+class Ke extends dt {
   /**
    * @internal use `createAIExtension` instead
    */
   constructor(o, r) {
     super();
-    D(this, "previousRequestOptions");
+    M(this, "chatSession");
+    M(this, "scrollInProgress", !1);
+    M(this, "autoScroll", !1);
     // internal store including setters
-    D(this, "_store", me()((o) => ({
+    M(this, "_store", he()((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}
+     * These options are used by default across all LLM calls when calling {@link executeLLMRequest}
      */
-    D(this, "options");
-    this.editor = o, this.options = me()((n) => ({
-      dataFormat: fo.html,
-      stream: !0,
+    M(this, "options");
+    this.editor = o, this.options = he()((n) => ({
       ...r
     })), this.addProsemirrorPlugin(
       new Le({
-        key: go,
+        key: Bo,
         filterTransaction: (n) => {
           var a;
           const s = this.store.getState().aiMenuState;
-          return !(s !== "closed" && s.status === "ai-writing" && (a = n.getMeta(kt)) != null && a.fixTables);
+          return !(s !== "closed" && s.status === "ai-writing" && (a = n.getMeta(yt)) != null && a.fixTables);
         }
       })
-    ), this.addProsemirrorPlugin(mt()), this.addProsemirrorPlugin(
-      ho(
+    ), this.addProsemirrorPlugin(ft()), this.addProsemirrorPlugin(
+      Oo(
         r.agentCursor || { name: "AI", color: "#8bc6ff" }
       )
+    ), document.addEventListener(
+      "scroll",
+      () => {
+        this.scrollInProgress && (this.autoScroll = !1), this.scrollInProgress = !0;
+      },
+      !0
+    ), document.addEventListener(
+      "scrollend",
+      () => {
+        this.scrollInProgress = !1;
+      },
+      !0
     );
   }
   static key() {
@@ -2199,21 +2534,25 @@ class Ye extends lt {
    * Open the AI menu at a specific block
    */
   openAIMenuAtBlock(o) {
+    var n;
     this.editor.setForceSelectionVisible(!0), this.editor.isEditable = !1, this._store.setState({
       aiMenuState: {
         blockId: o,
         status: "user-input"
       }
     });
+    const r = (n = this.editor.domElement) == null ? void 0 : n.querySelector(
+      `[data-node-type="blockContainer"][data-id="${o}"]`
+    );
+    r == null || r.scrollIntoView({ block: "center" });
   }
   /**
    * Close the AI menu
    */
   closeAIMenu() {
-    this.previousRequestOptions = void 0, this._store.setState({
-      aiMenuState: "closed",
-      llmResponse: void 0
-    }), this.editor.setForceSelectionVisible(!1), this.editor.isEditable = !0, this.editor.focus();
+    this._store.setState({
+      aiMenuState: "closed"
+    }), this.chatSession = void 0, this.editor.setForceSelectionVisible(!1), this.editor.isEditable = !0, this.editor.focus();
   }
   /**
    * Accept the changes made by the LLM
@@ -2221,22 +2560,22 @@ class Ye extends lt {
   acceptChanges() {
     var r;
     const o = this.editor.prosemirrorState.doc;
-    this.editor.exec((n, s) => ie(n, (a) => {
+    this.editor.exec((n, s) => le(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)
+        new G(ce.from(o), 0, 0)
       );
       const i = n.apply(a);
-      return Be(i, (c) => {
+      return Ne(i, (l) => {
         s == null || s(
           a.replace(
             0,
             a.doc.content.size,
-            new Y(ae.from(c.doc), 0, 0)
+            new G(ce.from(l.doc), 0, 0)
           )
         );
       });
@@ -2247,7 +2586,7 @@ class Ye extends lt {
    */
   rejectChanges() {
     var o;
-    this.editor.exec((r, n) => ie(r, (s) => {
+    this.editor.exec((r, n) => le(r, (s) => {
       n == null || n(s.setMeta("addToHistory", !1));
     })), (o = this.editor.forkYDocPlugin) == null || o.merge({ keepChanges: !1 }), this.closeAIMenu();
   }
@@ -2257,32 +2596,31 @@ class Ye extends lt {
    * Only valid if the current status is "error"
    */
   async retry() {
-    const o = this.store.getState().aiMenuState;
-    if (o === "closed" || o.status !== "error" || !this.previousRequestOptions)
+    var r;
+    const { aiMenuState: o } = this.store.getState();
+    if (o === "closed" || o.status !== "error" || !this.chatSession)
       throw new Error("retry() is only valid when a previous response failed");
-    if (o.error instanceof ht || o.error instanceof yt)
-      return this.callLLM(this.previousRequestOptions);
-    {
-      const r = o.error instanceof Error ? o.error.message : String(o.error);
-      return this.callLLM({
-        userPrompt: `An error occured: ${r}
-            Please retry the previous user request.`
-      });
-    }
+    return ((r = this.chatSession) == null ? void 0 : r.chat.status) === "error" ? this.invokeAI({
+      ...this.chatSession.previousRequestOptions,
+      userPrompt: "An error occured in the previous request. Please retry to accomplish the last user prompt."
+    }) : this.invokeAI({
+      ...this.chatSession.previousRequestOptions,
+      userPrompt: "An error occured while executing the previous tool call. Please retry to accomplish the last user prompt."
+    });
   }
   /**
    * Update the status of a call to an LLM
    *
    * @warning This method should usually only be used for advanced use-cases
    * if you want to implement how an LLM call is executed. Usually, you should
-   * use {@link doLLMRequest} instead which will handle the status updates for you.
+   * use {@link executeLLMRequest} instead which will handle the status updates for you.
    */
   setAIResponseStatus(o) {
     const r = this.store.getState().aiMenuState;
     if (r !== "closed")
       if (o === "ai-writing" && this.editor.setForceSelectionVisible(!1), typeof o == "object") {
         if (o.status !== "error")
-          throw new Pe(o.status);
+          throw new Be(o.status);
         this._store.setState({
           aiMenuState: {
             status: o.status,
@@ -2299,104 +2637,107 @@ class Ye extends lt {
         });
   }
   /**
-   * Execute a call to an LLM and apply the result to the editor
+   * @deprecated Use {@link invokeAI} instead
    */
   async callLLM(o) {
-    var n;
-    this.setAIResponseStatus("thinking"), (n = this.editor.forkYDocPlugin) == null || n.fork();
-    let r;
+    return this.invokeAI(o);
+  }
+  /**
+   * Execute a call to an LLM and apply the result to the editor
+   */
+  async invokeAI(o) {
+    var r, n;
+    this.setAIResponseStatus("thinking"), (r = this.editor.forkYDocPlugin) == null || r.fork();
     try {
-      const s = {
+      this.chatSession ? this.chatSession.previousRequestOptions = o : this.chatSession = {
+        previousRequestOptions: o,
+        chat: new nt({
+          sendAutomaticallyWhen: () => !1,
+          transport: o.transport || this.options.getState().transport
+        })
+      };
+      const s = this.chatSession.chat;
+      o = {
         ...this.options.getState(),
-        ...o,
-        previousResponse: this.store.getState().llmResponse
+        ...o
       };
-      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 i;
-          this._store.setState({
+      const i = o.aiRequestSender ?? At(
+        O.html.defaultPromptBuilder,
+        O.html.defaultPromptInputDataBuilder
+      ), l = Kt({
+        editor: this.editor,
+        chat: s,
+        userPrompt: o.userPrompt,
+        useSelection: o.useSelection,
+        deleteEmptyCursorBlock: o.deleteEmptyCursorBlock,
+        streamToolsProvider: o.streamToolsProvider,
+        onBlockUpdated: (c) => {
+          if (this._store.setState({
             aiMenuState: {
-              blockId: a,
+              blockId: c,
               status: "ai-writing"
             }
-          }), (i = o.onBlockUpdate) == null || i.call(o, a);
+          }), !this.autoScroll)
+            return;
+          const u = this._store.getState().aiMenuState, m = u === "closed" ? void 0 : u;
+          if (!m || m.status !== "ai-writing")
+            return;
+          const p = Me(
+            m.blockId,
+            this.editor.prosemirrorState.doc
+          );
+          if (!p)
+            return;
+          this.editor.prosemirrorView.domAtPos(
+            p.posBeforeNode + 1
+          ).node.scrollIntoView({
+            block: "center"
+          });
+        }
+      });
+      await Xt({
+        aiRequest: l,
+        sender: i,
+        chatRequestOptions: o.chatRequestOptions,
+        onStart: () => {
+          this.autoScroll = !0, this.setAIResponseStatus("ai-writing");
         }
-      }), this._store.setState({
-        llmResponse: r
-      }), await r.execute(), this.setAIResponseStatus("user-reviewing");
+      }), this.setAIResponseStatus("user-reviewing");
     } catch (s) {
       this.setAIResponseStatus({
         status: "error",
         error: s
-      }), console.warn("Error calling LLM", s);
+      }), console.warn("Error calling LLM", s, (n = this.chatSession) == null ? void 0 : n.chat.messages);
     }
-    return r;
   }
 }
-function ar(e) {
-  return (t) => new Ye(t, e);
-}
-function O(e) {
-  return e.extension(Ye);
+function vr(e) {
+  return (t) => new Ke(t, e);
 }
-const ko = (e, t) => async (o, r) => {
-  const n = new Request(o, r), s = new Request(
-    `${e}?provider=${encodeURIComponent(
-      t
-    )}&url=${encodeURIComponent(n.url)}`,
-    {
-      headers: n.headers,
-      // if we just pass request.body, it's a readablestream which is not visible in chrome inspector,
-      // so use init?.body instead if it's available to make debugging easier
-      body: (r == null ? void 0 : r.body) || n.body,
-      method: n.method,
-      duplex: "half"
-    }
-  );
-  try {
-    return await fetch(s);
-  } catch (a) {
-    throw new TypeError("fetch failed", {
-      cause: a
-    });
-  }
+function I(e) {
+  return e.extension(Ke);
+}
+const Pr = (e) => async (t, o) => {
+  const r = new Request(t, o), n = e(r.url), s = new Request(n, {
+    headers: r.headers,
+    // if we just pass request.body, it's a readablestream which is not visible in chrome inspector,
+    // so use init?.body instead if it's available to make debugging easier
+    body: (o == null ? void 0 : o.body) || r.body,
+    method: r.method,
+    duplex: "half"
+  });
+  return await fetch(s);
 };
-function cr(e) {
-  return {
-    /**
-     * Get settings for AI SDK providers. Pass the returned objects when creating the AI SDK provider, e.g.:
-     *
-     * createOpenAI({
-     *   ...client.getProviderSettings("openai"),
-     * })("gpt-4o-2024-08-06", {});
-     *
-     * Explanation: we override the `fetch` and `apiKey` parameters of the AI SDK provider to instead
-     * use the BlockNote AI server to proxy requests to the provider.
-     *
-     * Note: the `apiKey` is the API key for the @blocknote/xl-ai-server AI server, not the model provider.
-     * The correct API key for the model provider will be added by the BlockNote AI server.
-     */
-    getProviderSettings: (t) => ({
-      apiKey: e.apiKey,
-      fetch: ko(e.baseURL, t)
-    })
-  };
-}
-var Ge = {
+var Xe = {
   color: void 0,
   size: void 0,
   className: void 0,
   style: void 0,
   attr: void 0
-}, Te = B.createContext && /* @__PURE__ */ B.createContext(Ge), bo = ["attr", "size", "title"];
-function So(e, t) {
+}, Pe = E.createContext && /* @__PURE__ */ E.createContext(Xe), Mo = ["attr", "size", "title"];
+function Eo(e, t) {
   if (e == null) return {};
-  var o = wo(e, t), r, n;
+  var o = _o(e, t), r, n;
   if (Object.getOwnPropertySymbols) {
     var s = Object.getOwnPropertySymbols(e);
     for (n = 0; n < s.length; n++)
@@ -2404,7 +2745,7 @@ function So(e, t) {
   }
   return o;
 }
-function wo(e, t) {
+function _o(e, t) {
   if (e == null) return {};
   var o = {};
   for (var r in e)
@@ -2414,17 +2755,17 @@ function wo(e, t) {
     }
   return o;
 }
-function U() {
-  return U = Object.assign ? Object.assign.bind() : function(e) {
+function F() {
+  return F = 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;
-  }, U.apply(this, arguments);
+  }, F.apply(this, arguments);
 }
-function ve(e, t) {
+function Ce(e, t) {
   var o = Object.keys(e);
   if (Object.getOwnPropertySymbols) {
     var r = Object.getOwnPropertySymbols(e);
@@ -2434,25 +2775,25 @@ function ve(e, t) {
   }
   return o;
 }
-function z(e) {
+function R(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) {
-      To(e, r, o[r]);
-    }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(o)) : ve(Object(o)).forEach(function(r) {
+    t % 2 ? Ce(Object(o), !0).forEach(function(r) {
+      No(e, r, o[r]);
+    }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(o)) : Ce(Object(o)).forEach(function(r) {
       Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(o, r));
     });
   }
   return e;
 }
-function To(e, t, o) {
-  return t = vo(t), t in e ? Object.defineProperty(e, t, { value: o, enumerable: !0, configurable: !0, writable: !0 }) : e[t] = o, e;
+function No(e, t, o) {
+  return t = Lo(t), t in e ? Object.defineProperty(e, t, { value: o, enumerable: !0, configurable: !0, writable: !0 }) : e[t] = o, e;
 }
-function vo(e) {
-  var t = Co(e, "string");
+function Lo(e) {
+  var t = Ao(e, "string");
   return typeof t == "symbol" ? t : t + "";
 }
-function Co(e, t) {
+function Ao(e, t) {
   if (typeof e != "object" || !e) return e;
   var o = e[Symbol.toPrimitive];
   if (o !== void 0) {
@@ -2462,102 +2803,102 @@ function Co(e, t) {
   }
   return (t === "string" ? String : Number)(e);
 }
-function Ke(e) {
-  return e && e.map((t, o) => /* @__PURE__ */ B.createElement(t.tag, z({
+function Qe(e) {
+  return e && e.map((t, o) => /* @__PURE__ */ E.createElement(t.tag, R({
     key: o
-  }, t.attr), Ke(t.child)));
+  }, t.attr), Qe(t.child)));
 }
-function v(e) {
-  return (t) => /* @__PURE__ */ B.createElement(Po, U({
-    attr: z({}, e.attr)
-  }, t), Ke(e.child));
+function x(e) {
+  return (t) => /* @__PURE__ */ E.createElement(Do, F({
+    attr: R({}, e.attr)
+  }, t), Qe(e.child));
 }
-function Po(e) {
+function Do(e) {
   var t = (o) => {
     var {
       attr: r,
       size: n,
       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({
+    } = e, a = Eo(e, Mo), i = n || o.size || "1em", l;
+    return o.className && (l = o.className), e.className && (l = (l ? l + " " : "") + e.className), /* @__PURE__ */ E.createElement("svg", F({
       stroke: "currentColor",
       fill: "currentColor",
       strokeWidth: "0"
     }, o.attr, r, a, {
-      className: c,
-      style: z(z({
+      className: l,
+      style: R(R({
         color: e.color || o.color
       }, o.style), e.style),
       height: i,
       width: i,
       xmlns: "http://www.w3.org/2000/svg"
-    }), s && /* @__PURE__ */ B.createElement("title", null, s), e.children);
+    }), s && /* @__PURE__ */ E.createElement("title", null, s), e.children);
   };
-  return Te !== void 0 ? /* @__PURE__ */ B.createElement(Te.Consumer, null, (o) => t(o)) : t(Ge);
+  return Pe !== void 0 ? /* @__PURE__ */ E.createElement(Pe.Consumer, null, (o) => t(o)) : t(Xe);
 }
-function Xe(e) {
-  return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M8 7V11L2 6L8 1V5H13C17.4183 5 21 8.58172 21 13C21 17.4183 17.4183 21 13 21H4V19H13C16.3137 19 19 16.3137 19 13C19 9.68629 16.3137 7 13 7H8Z" }, child: [] }] })(e);
+function et(e) {
+  return x({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M8 7V11L2 6L8 1V5H13C17.4183 5 21 8.58172 21 13C21 17.4183 17.4183 21 13 21H4V19H13C16.3137 19 19 16.3137 19 13C19 9.68629 16.3137 7 13 7H8Z" }, child: [] }] })(e);
 }
-function Ce(e) {
-  return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M17.8492 11.6983L17.1421 10.9912L7.24264 20.8907H3V16.648L14.3137 5.33432L19.9706 10.9912C20.3611 11.3817 20.3611 12.0149 19.9706 12.4054L12.8995 19.4765L11.4853 18.0622L17.8492 11.6983ZM15.7279 9.57696L14.3137 8.16274L5 17.4765V18.8907H6.41421L15.7279 9.57696ZM18.5563 2.50589L21.3848 5.33432C21.7753 5.72484 21.7753 6.35801 21.3848 6.74853L19.9706 8.16274L15.7279 3.9201L17.1421 2.50589C17.5327 2.11537 18.1658 2.11537 18.5563 2.50589Z" }, child: [] }] })(e);
+function Oe(e) {
+  return x({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M17.8492 11.6983L17.1421 10.9912L7.24264 20.8907H3V16.648L14.3137 5.33432L19.9706 10.9912C20.3611 11.3817 20.3611 12.0149 19.9706 12.4054L12.8995 19.4765L11.4853 18.0622L17.8492 11.6983ZM15.7279 9.57696L14.3137 8.16274L5 17.4765V18.8907H6.41421L15.7279 9.57696ZM18.5563 2.50589L21.3848 5.33432C21.7753 5.72484 21.7753 6.35801 21.3848 6.74853L19.9706 8.16274L15.7279 3.9201L17.1421 2.50589C17.5327 2.11537 18.1658 2.11537 18.5563 2.50589Z" }, child: [] }] })(e);
 }
-function Oo(e) {
-  return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M15.1986 9.94447C14.7649 9.5337 14.4859 8.98613 14.4085 8.39384L14.0056 5.31138L11.275 6.79724C10.7503 7.08274 10.1433 7.17888 9.55608 7.06948L6.49998 6.50015L7.06931 9.55625C7.17871 10.1435 7.08257 10.7505 6.79707 11.2751L5.31121 14.0057L8.39367 14.4086C8.98596 14.4861 9.53353 14.7651 9.94431 15.1987L12.0821 17.4557L13.4178 14.6486C13.6745 14.1092 14.109 13.6747 14.6484 13.418L17.4555 12.0823L15.1986 9.94447ZM15.2238 15.5079L13.0111 20.1581C12.8687 20.4573 12.5107 20.5844 12.2115 20.442C12.1448 20.4103 12.0845 20.3665 12.0337 20.3129L8.49229 16.5741C8.39749 16.474 8.27113 16.4096 8.13445 16.3918L3.02816 15.7243C2.69958 15.6814 2.46804 15.3802 2.51099 15.0516C2.52056 14.9784 2.54359 14.9075 2.5789 14.8426L5.04031 10.3192C5.1062 10.1981 5.12839 10.058 5.10314 9.92253L4.16 4.85991C4.09931 4.53414 4.3142 4.22086 4.63997 4.16017C4.7126 4.14664 4.78711 4.14664 4.85974 4.16017L9.92237 5.10331C10.0579 5.12855 10.198 5.10637 10.319 5.04048L14.8424 2.57907C15.1335 2.42068 15.4979 2.52825 15.6562 2.81931C15.6916 2.88421 15.7146 2.95507 15.7241 3.02833L16.3916 8.13462C16.4095 8.2713 16.4739 8.39766 16.5739 8.49245L20.3127 12.0338C20.5533 12.2617 20.5636 12.6415 20.3357 12.8821C20.2849 12.9357 20.2246 12.9795 20.1579 13.0112L15.5078 15.224C15.3833 15.2832 15.283 15.3835 15.2238 15.5079ZM16.0206 17.435L17.4348 16.0208L21.6775 20.2634L20.2633 21.6776L16.0206 17.435Z" }, child: [] }] })(e);
+function jo(e) {
+  return x({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M15.1986 9.94447C14.7649 9.5337 14.4859 8.98613 14.4085 8.39384L14.0056 5.31138L11.275 6.79724C10.7503 7.08274 10.1433 7.17888 9.55608 7.06948L6.49998 6.50015L7.06931 9.55625C7.17871 10.1435 7.08257 10.7505 6.79707 11.2751L5.31121 14.0057L8.39367 14.4086C8.98596 14.4861 9.53353 14.7651 9.94431 15.1987L12.0821 17.4557L13.4178 14.6486C13.6745 14.1092 14.109 13.6747 14.6484 13.418L17.4555 12.0823L15.1986 9.94447ZM15.2238 15.5079L13.0111 20.1581C12.8687 20.4573 12.5107 20.5844 12.2115 20.442C12.1448 20.4103 12.0845 20.3665 12.0337 20.3129L8.49229 16.5741C8.39749 16.474 8.27113 16.4096 8.13445 16.3918L3.02816 15.7243C2.69958 15.6814 2.46804 15.3802 2.51099 15.0516C2.52056 14.9784 2.54359 14.9075 2.5789 14.8426L5.04031 10.3192C5.1062 10.1981 5.12839 10.058 5.10314 9.92253L4.16 4.85991C4.09931 4.53414 4.3142 4.22086 4.63997 4.16017C4.7126 4.14664 4.78711 4.14664 4.85974 4.16017L9.92237 5.10331C10.0579 5.12855 10.198 5.10637 10.319 5.04048L14.8424 2.57907C15.1335 2.42068 15.4979 2.52825 15.6562 2.81931C15.6916 2.88421 15.7146 2.95507 15.7241 3.02833L16.3916 8.13462C16.4095 8.2713 16.4739 8.39766 16.5739 8.49245L20.3127 12.0338C20.5533 12.2617 20.5636 12.6415 20.3357 12.8821C20.2849 12.9357 20.2246 12.9795 20.1579 13.0112L15.5078 15.224C15.3833 15.2832 15.283 15.3835 15.2238 15.5079ZM16.0206 17.435L17.4348 16.0208L21.6775 20.2634L20.2633 21.6776L16.0206 17.435Z" }, child: [] }] })(e);
 }
-function Mo(e) {
-  return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M8.00008 6V9H5.00008V6H8.00008ZM3.00008 4V11H10.0001V4H3.00008ZM13.0001 4H21.0001V6H13.0001V4ZM13.0001 11H21.0001V13H13.0001V11ZM13.0001 18H21.0001V20H13.0001V18ZM10.7072 16.2071L9.29297 14.7929L6.00008 18.0858L4.20718 16.2929L2.79297 17.7071L6.00008 20.9142L10.7072 16.2071Z" }, child: [] }] })(e);
+function $o(e) {
+  return x({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M8.00008 6V9H5.00008V6H8.00008ZM3.00008 4V11H10.0001V4H3.00008ZM13.0001 4H21.0001V6H13.0001V4ZM13.0001 11H21.0001V13H13.0001V11ZM13.0001 18H21.0001V20H13.0001V18ZM10.7072 16.2071L9.29297 14.7929L6.00008 18.0858L4.20718 16.2929L2.79297 17.7071L6.00008 20.9142L10.7072 16.2071Z" }, child: [] }] })(e);
 }
-function Bo(e) {
-  return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M15 18H16.5C17.8807 18 19 16.8807 19 15.5C19 14.1193 17.8807 13 16.5 13H3V11H16.5C18.9853 11 21 13.0147 21 15.5C21 17.9853 18.9853 20 16.5 20H15V22L11 19L15 16V18ZM3 4H21V6H3V4ZM9 18V20H3V18H9Z" }, child: [] }] })(e);
+function Ho(e) {
+  return x({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M15 18H16.5C17.8807 18 19 16.8807 19 15.5C19 14.1193 17.8807 13 16.5 13H3V11H16.5C18.9853 11 21 13.0147 21 15.5C21 17.9853 18.9853 20 16.5 20H15V22L11 19L15 16V18ZM3 4H21V6H3V4ZM9 18V20H3V18H9Z" }, child: [] }] })(e);
 }
-function xo(e) {
-  return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M13 6V21H11V6H5V4H19V6H13Z" }, child: [] }] })(e);
+function qo(e) {
+  return x({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M13 6V21H11V6H5V4H19V6H13Z" }, child: [] }] })(e);
 }
-function Lo(e) {
-  return v({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M6.23509 6.45329C4.85101 7.89148 4 9.84636 4 12C4 16.4183 7.58172 20 12 20C13.0808 20 14.1116 19.7857 15.0521 19.3972C15.1671 18.6467 14.9148 17.9266 14.8116 17.6746C14.582 17.115 13.8241 16.1582 12.5589 14.8308C12.2212 14.4758 12.2429 14.2035 12.3636 13.3943L12.3775 13.3029C12.4595 12.7486 12.5971 12.4209 14.4622 12.1248C15.4097 11.9746 15.6589 12.3533 16.0043 12.8777C16.0425 12.9358 16.0807 12.9928 16.1198 13.0499C16.4479 13.5297 16.691 13.6394 17.0582 13.8064C17.2227 13.881 17.428 13.9751 17.7031 14.1314C18.3551 14.504 18.3551 14.9247 18.3551 15.8472V15.9518C18.3551 16.3434 18.3168 16.6872 18.2566 16.9859C19.3478 15.6185 20 13.8854 20 12C20 8.70089 18.003 5.8682 15.1519 4.64482C14.5987 5.01813 13.8398 5.54726 13.575 5.91C13.4396 6.09538 13.2482 7.04166 12.6257 7.11976C12.4626 7.14023 12.2438 7.12589 12.012 7.11097C11.3905 7.07058 10.5402 7.01606 10.268 7.75495C10.0952 8.2232 10.0648 9.49445 10.6239 10.1543C10.7134 10.2597 10.7307 10.4547 10.6699 10.6735C10.59 10.9608 10.4286 11.1356 10.3783 11.1717C10.2819 11.1163 10.0896 10.8931 9.95938 10.7412C9.64554 10.3765 9.25405 9.92233 8.74797 9.78176C8.56395 9.73083 8.36166 9.68867 8.16548 9.64736C7.6164 9.53227 6.99443 9.40134 6.84992 9.09302C6.74442 8.8672 6.74488 8.55621 6.74529 8.22764C6.74529 7.8112 6.74529 7.34029 6.54129 6.88256C6.46246 6.70541 6.35689 6.56446 6.23509 6.45329ZM12 22C6.47715 22 2 17.5228 2 12C2 6.47715 6.47715 2 12 2C17.5228 2 22 6.47715 22 12C22 17.5228 17.5228 22 12 22Z" }, child: [] }] })(e);
+function zo(e) {
+  return x({ attr: { viewBox: "0 0 24 24", fill: "currentColor" }, child: [{ tag: "path", attr: { d: "M6.23509 6.45329C4.85101 7.89148 4 9.84636 4 12C4 16.4183 7.58172 20 12 20C13.0808 20 14.1116 19.7857 15.0521 19.3972C15.1671 18.6467 14.9148 17.9266 14.8116 17.6746C14.582 17.115 13.8241 16.1582 12.5589 14.8308C12.2212 14.4758 12.2429 14.2035 12.3636 13.3943L12.3775 13.3029C12.4595 12.7486 12.5971 12.4209 14.4622 12.1248C15.4097 11.9746 15.6589 12.3533 16.0043 12.8777C16.0425 12.9358 16.0807 12.9928 16.1198 13.0499C16.4479 13.5297 16.691 13.6394 17.0582 13.8064C17.2227 13.881 17.428 13.9751 17.7031 14.1314C18.3551 14.504 18.3551 14.9247 18.3551 15.8472V15.9518C18.3551 16.3434 18.3168 16.6872 18.2566 16.9859C19.3478 15.6185 20 13.8854 20 12C20 8.70089 18.003 5.8682 15.1519 4.64482C14.5987 5.01813 13.8398 5.54726 13.575 5.91C13.4396 6.09538 13.2482 7.04166 12.6257 7.11976C12.4626 7.14023 12.2438 7.12589 12.012 7.11097C11.3905 7.07058 10.5402 7.01606 10.268 7.75495C10.0952 8.2232 10.0648 9.49445 10.6239 10.1543C10.7134 10.2597 10.7307 10.4547 10.6699 10.6735C10.59 10.9608 10.4286 11.1356 10.3783 11.1717C10.2819 11.1163 10.0896 10.8931 9.95938 10.7412C9.64554 10.3765 9.25405 9.92233 8.74797 9.78176C8.56395 9.73083 8.36166 9.68867 8.16548 9.64736C7.6164 9.53227 6.99443 9.40134 6.84992 9.09302C6.74442 8.8672 6.74488 8.55621 6.74529 8.22764C6.74529 7.8112 6.74529 7.34029 6.54129 6.88256C6.46246 6.70541 6.35689 6.56446 6.23509 6.45329ZM12 22C6.47715 22 2 17.5228 2 12C2 6.47715 6.47715 2 12 2C17.5228 2 22 6.47715 22 12C22 17.5228 17.5228 22 12 22Z" }, child: [] }] })(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 Jo(e) {
+  return x({ 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) {
-  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 Uo(e) {
+  return x({ 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) {
-  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 Vo(e) {
+  return x({ 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 ie(e) {
+  return x({ 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 I(e) {
+function L(e) {
   if (!e.dictionary.ai)
     throw new Error("AI dictionary not found");
   return e.dictionary.ai;
 }
-function Qe() {
-  const e = Ct();
-  return I(e.editor);
+function tt() {
+  const e = Ot();
+  return L(e.editor);
 }
-const No = (e) => {
-  const t = G(), { onManualPromptSubmit: o, promptText: r, onPromptTextChange: n } = e, [s, a] = Ee(""), i = r || s, c = $(
-    async (d) => {
-      d.key === "Enter" && o(i);
+const Fo = (e) => {
+  const t = X(), { onManualPromptSubmit: o, promptText: r, onPromptTextChange: n } = e, [s, a] = $e(""), i = r || s, l = $(
+    async (f) => {
+      f.key === "Enter" && o(i);
     },
     [i, o]
-  ), u = $(
-    (d) => {
-      const S = d.currentTarget.value;
-      n && n(S), r === void 0 && a(S);
+  ), c = $(
+    (f) => {
+      const k = f.currentTarget.value;
+      n && n(k), r === void 0 && a(k);
     },
     [n, a, r]
-  ), 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" ? p.length > 0 ? m(d) : c(d) : m(d);
+  ), u = A(() => pt(e.items, i), [i, e.items]), { selectedIndex: m, setSelectedIndex: p, handler: d } = It(u, (f) => f.onItemClick()), h = $(
+    (f) => {
+      f.key === "Enter" ? u.length > 0 ? d(f) : l(f) : d(f);
     },
-    [c, m, p.length]
+    [l, d, u.length]
   );
-  return Ne(() => {
-    f(0);
-  }, [i, f]), /* @__PURE__ */ vt("div", { className: "bn-combobox", children: [
+  return He(() => {
+    p(0);
+  }, [i, p]), /* @__PURE__ */ Ct("div", { className: "bn-combobox", children: [
     /* @__PURE__ */ y(t.Generic.Form.Root, { children: /* @__PURE__ */ y(
       t.Generic.Form.TextInput,
       {
@@ -2570,7 +2911,7 @@ const No = (e) => {
         placeholder: e.placeholder,
         disabled: e.disabled,
         onKeyDown: h,
-        onChange: u,
+        onChange: c,
         autoComplete: "off",
         rightSection: e.rightSection
       },
@@ -2581,41 +2922,43 @@ const No = (e) => {
       {
         className: "bn-combobox-items",
         id: "ai-suggestion-menu",
-        children: p.map((d, S) => /* @__PURE__ */ y(
+        children: u.map((f, k) => /* @__PURE__ */ y(
           t.SuggestionMenu.Item,
           {
-            className: dt(
+            className: mt(
               "bn-suggestion-menu-item",
-              d.size === "small" ? "bn-suggestion-menu-item-small" : ""
+              f.size === "small" ? "bn-suggestion-menu-item-small" : ""
             ),
-            id: `bn-suggestion-menu-item-${S}`,
-            isSelected: S === l,
-            onClick: d.onItemClick,
-            item: d
+            id: `bn-suggestion-menu-item-${k}`,
+            isSelected: k === m,
+            onClick: f.onItemClick,
+            item: f
           },
-          d.title
+          f.title
         ))
       }
     )
   ] });
 };
-function Ao(e) {
-  const t = I(e), o = O(e);
+function Ro(e) {
+  const t = L(e), o = I(e);
   return [
     {
       key: "continue_writing",
       title: t.ai_default_commands.continue_writing.title,
       aliases: t.ai_default_commands.continue_writing.aliases,
-      icon: /* @__PURE__ */ y(Ce, { size: 18 }),
+      icon: /* @__PURE__ */ y(Oe, { size: 18 }),
       onItemClick: async () => {
-        await o.callLLM({
+        await o.invokeAI({
           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,
-            delete: !1,
-            update: !1
-          }
+          streamToolsProvider: O.html.getStreamToolsProvider({
+            defaultStreamTools: {
+              add: !0,
+              delete: !1,
+              update: !1
+            }
+          })
         });
       },
       size: "small"
@@ -2624,16 +2967,18 @@ function Ao(e) {
       key: "summarize",
       title: t.ai_default_commands.summarize.title,
       aliases: t.ai_default_commands.summarize.aliases,
-      icon: /* @__PURE__ */ y(Bo, { size: 18 }),
+      icon: /* @__PURE__ */ y(Ho, { size: 18 }),
       onItemClick: async () => {
-        await o.callLLM({
+        await o.invokeAI({
           userPrompt: "Summarize",
           // By default, LLM will be able to add / update / delete blocks. For "summarize", we only want to allow adding new blocks.
-          defaultStreamTools: {
-            add: !0,
-            delete: !1,
-            update: !1
-          }
+          streamToolsProvider: O.html.getStreamToolsProvider({
+            defaultStreamTools: {
+              add: !0,
+              delete: !1,
+              update: !1
+            }
+          })
         });
       },
       size: "small"
@@ -2642,16 +2987,18 @@ function Ao(e) {
       key: "action_items",
       title: t.ai_default_commands.add_action_items.title,
       aliases: t.ai_default_commands.add_action_items.aliases,
-      icon: /* @__PURE__ */ y(Mo, { size: 18 }),
+      icon: /* @__PURE__ */ y($o, { size: 18 }),
       onItemClick: async () => {
-        await o.callLLM({
+        await o.invokeAI({
           userPrompt: "Add action items",
           // By default, LLM will be able to add / update / delete blocks. For "summarize", we only want to allow adding new blocks.
-          defaultStreamTools: {
-            add: !0,
-            delete: !1,
-            update: !1
-          }
+          streamToolsProvider: O.html.getStreamToolsProvider({
+            defaultStreamTools: {
+              add: !0,
+              delete: !1,
+              update: !1
+            }
+          })
         });
       },
       size: "small"
@@ -2660,7 +3007,7 @@ function Ao(e) {
       key: "write_anything",
       title: t.ai_default_commands.write_anything.title,
       aliases: t.ai_default_commands.write_anything.aliases,
-      icon: /* @__PURE__ */ y(Ce, { size: 18 }),
+      icon: /* @__PURE__ */ y(Oe, { size: 18 }),
       onItemClick: (r) => {
         r(t.ai_default_commands.write_anything.prompt_placeholder);
       },
@@ -2668,24 +3015,26 @@ function Ao(e) {
     }
   ];
 }
-function jo(e) {
-  const t = I(e), o = O(e);
+function Zo(e) {
+  const t = L(e), o = I(e);
   return [
     {
       key: "improve_writing",
       title: t.ai_default_commands.improve_writing.title,
       aliases: t.ai_default_commands.improve_writing.aliases,
-      icon: /* @__PURE__ */ y(xo, { size: 18 }),
+      icon: /* @__PURE__ */ y(qo, { size: 18 }),
       onItemClick: async () => {
-        await o.callLLM({
+        await o.invokeAI({
           useSelection: !0,
           userPrompt: "Improve writing",
           // By default, LLM will be able to add / update / delete blocks. For "summarize", we only want to allow adding new blocks.
-          defaultStreamTools: {
-            add: !1,
-            delete: !1,
-            update: !0
-          }
+          streamToolsProvider: O.html.getStreamToolsProvider({
+            defaultStreamTools: {
+              add: !1,
+              delete: !1,
+              update: !0
+            }
+          })
         });
       },
       size: "small"
@@ -2694,17 +3043,19 @@ 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(Io, { size: 18 }),
+      icon: /* @__PURE__ */ y(Uo, { size: 18 }),
       onItemClick: async () => {
-        await o.callLLM({
+        await o.invokeAI({
           useSelection: !0,
           userPrompt: "Fix spelling",
           // By default, LLM will be able to add / update / delete blocks. For "summarize", we only want to allow adding new blocks.
-          defaultStreamTools: {
-            add: !1,
-            delete: !1,
-            update: !0
-          }
+          streamToolsProvider: O.html.getStreamToolsProvider({
+            defaultStreamTools: {
+              add: !1,
+              delete: !1,
+              update: !0
+            }
+          })
         });
       },
       size: "small"
@@ -2713,7 +3064,7 @@ function jo(e) {
       key: "translate",
       title: t.ai_default_commands.translate.title,
       aliases: t.ai_default_commands.translate.aliases,
-      icon: /* @__PURE__ */ y(Lo, { size: 18 }),
+      icon: /* @__PURE__ */ y(zo, { size: 18 }),
       onItemClick: (r) => {
         r(t.ai_default_commands.translate.prompt_placeholder);
       },
@@ -2723,31 +3074,33 @@ function jo(e) {
       key: "simplify",
       title: t.ai_default_commands.simplify.title,
       aliases: t.ai_default_commands.simplify.aliases,
-      icon: /* @__PURE__ */ y(Oo, { size: 18 }),
+      icon: /* @__PURE__ */ y(jo, { size: 18 }),
       onItemClick: async () => {
-        await o.callLLM({
+        await o.invokeAI({
           useSelection: !0,
           userPrompt: "Simplify",
           // By default, LLM will be able to add / update / delete blocks. For "summarize", we only want to allow adding new blocks.
-          defaultStreamTools: {
-            add: !1,
-            delete: !1,
-            update: !0
-          }
+          streamToolsProvider: O.html.getStreamToolsProvider({
+            defaultStreamTools: {
+              add: !1,
+              delete: !1,
+              update: !0
+            }
+          })
         });
       },
       size: "small"
     }
   ];
 }
-function Do(e) {
-  const t = I(e), o = O(e);
+function Wo(e) {
+  const t = L(e), o = I(e);
   return [
     {
       key: "accept",
       title: t.ai_menu.actions.accept.title,
       aliases: t.ai_menu.actions.accept.aliases,
-      icon: /* @__PURE__ */ y(_o, { size: 18 }),
+      icon: /* @__PURE__ */ y(Jo, { size: 18 }),
       onItemClick: () => {
         o.acceptChanges();
       },
@@ -2757,7 +3110,7 @@ function Do(e) {
       key: "revert",
       title: t.ai_menu.actions.revert.title,
       aliases: t.ai_menu.actions.revert.aliases,
-      icon: /* @__PURE__ */ y(Xe, { size: 18 }),
+      icon: /* @__PURE__ */ y(et, { size: 18 }),
       onItemClick: () => {
         o.rejectChanges();
       },
@@ -2765,14 +3118,14 @@ function Do(e) {
     }
   ];
 }
-function $o(e) {
-  const t = I(e), o = O(e);
+function Yo(e) {
+  const t = L(e), o = I(e);
   return [
     {
       key: "retry",
       title: t.ai_menu.actions.retry.title,
       aliases: t.ai_menu.actions.retry.aliases,
-      icon: /* @__PURE__ */ y(Eo, { size: 18 }),
+      icon: /* @__PURE__ */ y(Vo, { size: 18 }),
       onItemClick: async () => {
         await o.retry();
       },
@@ -2782,7 +3135,7 @@ function $o(e) {
       key: "cancel",
       title: t.ai_menu.actions.cancel.title,
       aliases: t.ai_menu.actions.cancel.aliases,
-      icon: /* @__PURE__ */ y(Xe, { size: 18 }),
+      icon: /* @__PURE__ */ y(et, { size: 18 }),
       onItemClick: () => {
         o.rejectChanges();
       },
@@ -2790,34 +3143,34 @@ function $o(e) {
     }
   ];
 }
-function Ro(e, t) {
-  return t === "user-input" ? e.getSelection() ? jo(e) : Ao(e) : t === "user-reviewing" ? Do(e) : t === "error" ? $o(e) : [];
+function Go(e, t) {
+  return t === "user-input" ? e.getSelection() ? Zo(e) : Ro(e) : t === "user-reviewing" ? Wo(e) : t === "error" ? Yo(e) : [];
 }
-const Ho = (e) => {
-  const t = K(), [o, r] = Ee(""), n = Qe(), s = G(), a = O(t), i = Ae(
+const Ko = (e) => {
+  const t = Q(), [o, r] = $e(""), n = tt(), s = X(), a = I(t), i = qe(
     a.store,
-    (m) => m.aiMenuState !== "closed" ? m.aiMenuState.status : "closed"
-  ), { items: c } = e, u = E(() => {
-    let m = [];
-    return c ? m = c(t, i) : m = Ro(t, i), m.map((h) => ({
+    (d) => d.aiMenuState !== "closed" ? d.aiMenuState.status : "closed"
+  ), { items: l } = e, c = A(() => {
+    let d = [];
+    return l ? d = l(t, i) : d = Go(t, i), d.map((h) => ({
       ...h,
       onItemClick: () => {
         h.onItemClick(r);
       }
     }));
-  }, [c, i, t]), p = $(
-    async (m) => {
-      await a.callLLM({
-        userPrompt: m,
+  }, [l, i, t]), u = $(
+    async (d) => {
+      await a.invokeAI({
+        userPrompt: d,
         useSelection: t.getSelection() !== void 0
       });
     },
     [a, t]
   );
-  Ne(() => {
+  He(() => {
     (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(() => {
+  const m = A(() => 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]), p = A(() => {
     if (i === "thinking" || i === "ai-writing")
       return /* @__PURE__ */ y(
         s.SuggestionMenu.Loader,
@@ -2839,23 +3192,23 @@ const Ho = (e) => {
       ) });
   }, [s, i]);
   return /* @__PURE__ */ y(
-    No,
+    Fo,
     {
-      onManualPromptSubmit: e.onManualPromptSubmit || p,
-      items: u,
+      onManualPromptSubmit: e.onManualPromptSubmit || u,
+      items: c,
       promptText: o,
       onPromptTextChange: r,
-      placeholder: l,
+      placeholder: m,
       disabled: i === "thinking" || i === "ai-writing",
-      icon: /* @__PURE__ */ y("div", { className: "bn-combobox-icon", children: /* @__PURE__ */ y(ne, {}) }),
-      rightSection: f
+      icon: /* @__PURE__ */ y("div", { className: "bn-combobox-icon", children: /* @__PURE__ */ y(ie, {}) }),
+      rightSection: p
     }
   );
-}, qo = (e) => {
-  const t = e.blockID ? document.querySelector(`[data-id="${e.blockID}"]`) : void 0, o = E(() => t ? {
+}, Xo = (e) => {
+  const t = e.blockID ? document.querySelector(`[data-id="${e.blockID}"]`) : void 0, o = A(() => t ? {
     getBoundingClientRect: () => t.getBoundingClientRect(),
     contextElement: t
-  } : null, [t]), { isMounted: r, ref: n, style: s, getFloatingProps: a, isPositioned: i } = Ot(!!t, o, 3e3, {
+  } : null, [t]), { isMounted: r, ref: n, style: s, getFloatingProps: a, isPositioned: i } = Bt(!!t, o, 3e3, {
     canDismiss: {
       enabled: !0,
       escapeKey: !0,
@@ -2865,16 +3218,16 @@ const Ho = (e) => {
     middleware: [
       Mt(10),
       // flip(),
-      Bt({
-        apply({ rects: c, elements: u }) {
-          Object.assign(u.floating.style, {
-            width: `${c.reference.width}px`
+      Et({
+        apply({ rects: l, elements: c }) {
+          Object.assign(c.floating.style, {
+            width: `${l.reference.width}px`
           });
         }
       })
     ],
     onOpenChange: e.onOpenChange,
-    whileElementsMounted: (c, u, p) => xt(c, u, p, {
+    whileElementsMounted: (l, c, u) => _t(l, c, u, {
       animationFrame: !0
     }),
     ...e.floatingOptions
@@ -2890,10 +3243,10 @@ const Ho = (e) => {
       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, s = e.aiMenu || Ho;
+}, Cr = (e) => {
+  const t = Q(), o = I(t), r = qe(o.store, (a) => a.aiMenuState), n = r === "closed" ? void 0 : r.blockId, s = e.aiMenu || Ko;
   return /* @__PURE__ */ y(
-    qo,
+    Xo,
     {
       canDismissViaOutsidePress: r === "closed" || r.status === "user-input",
       blockID: n,
@@ -2903,8 +3256,8 @@ const Ho = (e) => {
       children: /* @__PURE__ */ y(s, {})
     }
   );
-}, ur = () => {
-  const e = Qe(), t = G(), o = K(), r = O(o), n = () => {
+}, Or = () => {
+  const e = tt(), t = X(), o = Q(), r = I(o), n = () => {
     o.formattingToolbar.closeMenu();
     const s = o.getSelection();
     if (!s)
@@ -2918,15 +3271,15 @@ const Ho = (e) => {
       className: "bn-button",
       label: e.formatting_toolbar.ai.tooltip,
       mainTooltip: e.formatting_toolbar.ai.tooltip,
-      icon: /* @__PURE__ */ y(ne, {}),
+      icon: /* @__PURE__ */ y(ie, {}),
       onClick: n
     }
   ) : null;
-}, Uo = {
-  AI: ne
+}, Qo = {
+  AI: ie
 };
-function dr(e) {
-  const t = O(e);
+function Ir(e) {
+  const t = I(e);
   return [
     {
       key: "ai",
@@ -2935,26 +3288,214 @@ 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);
       },
-      ...I(e).slash_menu.ai,
-      icon: /* @__PURE__ */ y(Uo.AI, {})
+      ...L(e).slash_menu.ai,
+      icon: /* @__PURE__ */ y(Qo.AI, {})
     }
   ];
 }
+function er(e) {
+  return Object.fromEntries(
+    Object.entries(e).map(([t, o]) => [
+      t,
+      St({
+        ...o,
+        inputSchema: pe(o.inputSchema),
+        outputSchema: pe(o.outputSchema)
+      })
+    ])
+  );
+}
+function tr(e, t) {
+  let o = "";
+  return e.pipeThrough(
+    new TransformStream({
+      start(r) {
+        r.enqueue({ type: "start" }), r.enqueue({ type: "start-step" }), r.enqueue({
+          type: "tool-input-start",
+          toolCallId: "call_object_1",
+          toolName: t
+        });
+      },
+      transform(r, n) {
+        switch (r.type) {
+          case "text-delta":
+            o += r.textDelta, n.enqueue({
+              type: "tool-input-delta",
+              toolCallId: "call_object_1",
+              inputTextDelta: r.textDelta
+            });
+            break;
+          case "object":
+          case "finish":
+            break;
+          case "error":
+            n.enqueue({
+              type: "error",
+              errorText: K(r.error)
+            });
+            break;
+          default: {
+            const s = r;
+            throw new Error(`Unsupported chunk type: ${s}`);
+          }
+        }
+      },
+      async flush(r) {
+        r.enqueue({
+          type: "tool-input-available",
+          toolCallId: "call_object_1",
+          toolName: t,
+          input: JSON.parse(o)
+        }), r.enqueue({ type: "finish-step" }), r.enqueue({ type: "finish" });
+      }
+    })
+  );
+}
+function or(e, t) {
+  return new ReadableStream({
+    start(r) {
+      r.enqueue({ type: "start" }), r.enqueue({ type: "start-step" }), r.enqueue({
+        type: "tool-input-start",
+        toolCallId: "call_object_1",
+        toolName: t
+      }), r.enqueue({
+        type: "tool-input-delta",
+        toolCallId: "call_object_1",
+        inputTextDelta: JSON.stringify(e)
+      }), r.enqueue({
+        type: "tool-input-available",
+        toolCallId: "call_object_1",
+        toolName: t,
+        input: e
+      }), r.enqueue({ type: "finish-step" }), r.enqueue({ type: "finish" }), r.close();
+    }
+  });
+}
+const rr = {
+  "mistral.chat": {
+    mode: "auto"
+  },
+  "google.generative-ai": {
+    mode: "auto"
+  },
+  "groq.chat": {
+    providerOptions: {
+      groq: {
+        structuredOutputs: !1
+      }
+    }
+  }
+};
+function Ie(e) {
+  return rr[e.provider] || {};
+}
+class Br {
+  constructor(t) {
+    this.opts = t;
+  }
+  /**
+   * Calls an LLM with StreamTools, using the `generateObject` of the AI SDK.
+   *
+   * This is the non-streaming version.
+   */
+  async generateObject(t, o) {
+    const { model: r, _additionalOptions: n } = this.opts;
+    if (typeof r == "string")
+      throw new Error("model must be a LanguageModelV2");
+    const s = Object.keys(o)[0], a = o[s].inputSchema, i = await wt({
+      output: "object",
+      schema: a,
+      model: r,
+      mode: "tool",
+      messages: Y(t),
+      ...Ie(r),
+      ...n ?? {}
+    });
+    return or(i.object, s);
+  }
+  /**
+   * Calls an LLM with StreamTools, using the `streamObject` of the AI SDK.
+   *
+   * This is the streaming version.
+   */
+  async streamObject(t, o) {
+    const { model: r, _additionalOptions: n } = this.opts;
+    if (typeof r == "string")
+      throw new Error("model must be a LanguageModelV2");
+    const s = Object.keys(o)[0], a = o[s].inputSchema, i = Tt({
+      output: "object",
+      schema: a,
+      model: r,
+      mode: "tool",
+      messages: Y(t),
+      ...Ie(r),
+      ...n ?? {}
+    });
+    return tr(
+      i.fullStream,
+      s
+    );
+  }
+  /**
+   * Calls an LLM with StreamTools, using the `streamText` of the AI SDK.
+   *
+   * This is the streaming version.
+   */
+  async streamText(t, o) {
+    const { model: r, _additionalOptions: n } = this.opts;
+    return xt({
+      model: r,
+      messages: Y(t),
+      tools: o,
+      toolChoice: "required",
+      // extra options for streamObject
+      ...n ?? {}
+      // activeTools: ["applyDocumentOperations"],
+    }).toUIMessageStream();
+  }
+  async sendMessages({
+    messages: t,
+    body: o
+    // metadata,
+  }) {
+    const r = this.opts.stream ?? !0, n = o.toolDefinitions, s = er(n);
+    if (this.opts.objectGeneration)
+      return r ? this.streamObject(t, s) : this.generateObject(t, s);
+    if (r)
+      return await this.streamText(t, s);
+    throw new Error("Not implemented (generateText)");
+  }
+  reconnectToStream() {
+    throw new Error("Not implemented");
+  }
+}
 export {
-  Ye as AIExtension,
-  Ho as AIMenu,
-  lr as AIMenuController,
-  ur as AIToolbarButton,
-  qo as BlockPositioner,
-  No as PromptSuggestionMenu,
-  Gt as callLLM,
-  ar as createAIExtension,
-  cr as createBlockNoteAIClient,
-  I as getAIDictionary,
-  O as getAIExtension,
-  dr as getAISlashMenuItems,
-  Ro as getDefaultAIMenuItems,
-  fo as llmFormats,
-  ir as promptHelpers
+  Ke as AIExtension,
+  Ko as AIMenu,
+  Cr as AIMenuController,
+  Or as AIToolbarButton,
+  Xo as BlockPositioner,
+  ze as ChunkExecutionError,
+  Br as ClientSideTransport,
+  rr as PROVIDER_OVERRIDES,
+  Fo as PromptSuggestionMenu,
+  Dt as StreamToolExecutor,
+  O as aiDocumentFormats,
+  Kt as buildAIRequest,
+  vr as createAIExtension,
+  Lt as createStreamToolsArraySchema,
+  At as defaultAIRequestSender,
+  Xt as executeAIRequest,
+  Pr as fetchViaProxy,
+  L as getAIDictionary,
+  I as getAIExtension,
+  Ir as getAISlashMenuItems,
+  Go as getDefaultAIMenuItems,
+  Ie as getProviderOverrides,
+  or as objectAsToolCallInUIMessageStream,
+  Vt as objectStreamToOperationsResult,
+  tr as partialObjectStreamAsToolCallInUIMessageStream,
+  xr as promptHelpers,
+  er as toolDefinitionsToToolSet
 };
 //# sourceMappingURL=blocknote-xl-ai.js.map