@blocknote/xl-ai 0.51.4 → 0.52.1

package/package.json

@@ -11,7 +11,7 @@
     "directory": "packages/xl-ai"
   },
   "license": "GPL-3.0 OR PROPRIETARY",
-  "version": "0.51.4",
+  "version": "0.52.1",
   "files": [
     "dist",
     "types",
@@ -28,7 +28,6 @@
     "wysiwyg",
     "rich-text-editor",
     "notion",
-    "yjs",
     "block-based",
     "tiptap"
   ],
@@ -63,9 +62,6 @@
   "dependencies": {
     "@ai-sdk/provider-utils": "^4.0.2",
     "@ai-sdk/react": "^3.0.5",
-    "@blocknote/core": "0.51.4",
-    "@blocknote/mantine": "0.51.4",
-    "@blocknote/react": "0.51.4",
     "@floating-ui/react": "^0.27.18",
     "@handlewithcare/prosemirror-suggest-changes": "^0.1.8",
     "@tiptap/core": "^3.13.0",
@@ -78,30 +74,27 @@
     "prosemirror-tables": "^1.8.3",
     "prosemirror-transform": "^1.11.0",
     "prosemirror-view": "^1.41.4",
-    "react": "^19.2.5",
-    "react-dom": "^19.2.5",
-    "react-icons": "^5.5.0",
-    "y-prosemirror": "^1.3.7"
+    "@blocknote/core": "^0.52.1",
+    "@blocknote/react": "^0.52.1"
   },
   "devDependencies": {
+    "react-icons": "^5.5.0",
     "@ai-sdk/anthropic": "^3.0.2",
     "@ai-sdk/google": "^3.0.2",
     "@ai-sdk/groq": "^3.0.2",
     "@ai-sdk/mistral": "^3.0.2",
     "@ai-sdk/openai": "^3.0.2",
     "@ai-sdk/openai-compatible": "^2.0.2",
-    "@blocknote/xl-multi-column": "0.51.4",
     "@mswjs/interceptors": "^0.37.6",
     "@types/diff": "^6.0.0",
     "@types/json-diff": "^1.0.3",
     "@types/json-schema": "^7.0.15",
     "@types/lodash.isequal": "^4.5.8",
     "@types/lodash.merge": "^4.6.9",
+    "@types/node": "22.13.13",
     "@types/react": "^19.2.3",
     "@types/react-dom": "^19.2.3",
     "@vitejs/plugin-react": "^6.0.1",
-    "@vitest/runner": "^4.1.2",
-    "eslint": "^8.57.1",
     "glob": "^10.5.0",
     "headers-polyfill": "^4.0.3",
     "msw": "^2.11.5",
@@ -110,27 +103,21 @@
     "rollup-plugin-webpack-stats": "^0.2.6",
     "typescript": "^5.9.3",
     "undici": "^6.22.0",
-    "vite": "^8.0.8",
-    "vite-plugin-eslint": "^1.8.1",
     "vite-plugin-externalize-deps": "^0.10.0",
-    "vitest": "^4.1.2"
+    "vite-plus": "^0.1.24",
+    "@blocknote/shared": "^0.30.0",
+    "@blocknote/xl-multi-column": "^0.52.1"
   },
   "peerDependencies": {
     "react": "^18.0 || ^19.0 || >= 19.0.0-rc",
     "react-dom": "^18.0 || ^19.0 || >= 19.0.0-rc"
   },
-  "eslintConfig": {
-    "extends": [
-      "../../.eslintrc.json"
-    ]
-  },
   "scripts": {
-    "dev": "vite",
-    "build": "tsc && vite build",
-    "lint": "eslint src --max-warnings 0",
-    "test": "NODE_EXTRA_CA_CERTS=\"$(mkcert -CAROOT)/rootCA.pem\" vitest --run",
-    "test-watch": "NODE_EXTRA_CA_CERTS=\"$(mkcert -CAROOT)/rootCA.pem\" vitest watch",
+    "dev": "vp dev",
+    "lint": "vp lint src",
+    "test-watch": "NODE_EXTRA_CA_CERTS=\"$(mkcert -CAROOT)/rootCA.pem\" vp test watch",
     "rename-msw-snapshots": "node scripts/rename-msw-snapshots.mjs",
+    "clean": "rimraf dist && rimraf types",
     "email": "email dev"
   }
 }

package/src/AIExtension.ts

@@ -6,10 +6,8 @@ import {
   getNodeById,
   UnreachableCaseError,
 } from "@blocknote/core";
-import {
-  ForkYDocExtension,
-  ShowSelectionExtension,
-} from "@blocknote/core/extensions";
+import { ShowSelectionExtension } from "@blocknote/core/extensions";
+import type { ForkYDocExtension } from "@blocknote/core/yjs";
 import {
   applySuggestions,
   revertSuggestions,
@@ -22,6 +20,7 @@ import { fixTablesKey } from "prosemirror-tables";
 import { buildAIRequest, sendMessageWithAIRequest } from "./api/index.js";
 import { createAgentCursorPlugin } from "./plugins/AgentCursorPlugin.js";
 import { AIRequestHelpers, InvokeAIOptions } from "./types.js";
+import { AttributionMarksExtension } from "./prosemirror/AttributionMarks.js";
 
 type AIPluginState = {
   aiMenuState:
@@ -83,6 +82,7 @@ export const AIExtension = createExtension(
       key: "ai",
       options,
       store,
+      blockNoteExtensions: [AttributionMarksExtension()],
       mount({ signal }: { signal: AbortSignal }) {
         let scrollInProgress = false;
         // Listens for `scroll` and `scrollend` events to see if a new scroll was
@@ -220,7 +220,9 @@ export const AIExtension = createExtension(
         });
 
         // If in collaboration mode, merge the changes back into the original yDoc
-        editor.getExtension(ForkYDocExtension)?.merge({ keepChanges: true });
+        editor
+          .getExtension<typeof ForkYDocExtension>("yForkDoc")
+          ?.merge({ keepChanges: true });
 
         this.closeAIMenu();
       },
@@ -238,7 +240,9 @@ export const AIExtension = createExtension(
         });
 
         // If in collaboration mode, discard the changes and revert to the original yDoc
-        editor.getExtension(ForkYDocExtension)?.merge({ keepChanges: false });
+        editor
+          .getExtension<typeof ForkYDocExtension>("yForkDoc")
+          ?.merge({ keepChanges: false });
         this.closeAIMenu();
       },
 
@@ -379,7 +383,8 @@ export const AIExtension = createExtension(
        */
       async invokeAI(opts: InvokeAIOptions) {
         this.setAIResponseStatus("thinking");
-        editor.getExtension(ForkYDocExtension)?.fork();
+        // If in collaboration mode, fork the yDoc to allow modifications without affecting the remote
+        editor.getExtension<typeof ForkYDocExtension>("yForkDoc")?.fork();
 
         try {
           // Create a new AbortController for this request

package/src/api/formats/base-tools/createAddBlocksTool.ts

@@ -54,7 +54,7 @@ export function createAddBlocksTool<T>(config: {
   rebaseTool: (
     id: string,
     editor: BlockNoteEditor<any, any, any>,
-  ) => Promise<RebaseTool>;
+  ) => RebaseTool;
   /**
    * Converts the operation from `AddBlocksToolCall<T>` to `AddBlocksToolCall<PartialBlock<any, any, any>>`
    *
@@ -68,7 +68,7 @@ export function createAddBlocksTool<T>(config: {
       isUpdateToPreviousOperation: boolean;
       isPossiblyPartial: boolean;
     },
-  ) => Promise<AddBlocksToolCall<PartialBlock<any, any, any>> | undefined>;
+  ) => AddBlocksToolCall<PartialBlock<any, any, any>> | undefined;
 }) {
   return (
     editor: BlockNoteEditor<any, any, any>,
@@ -196,7 +196,7 @@ export function createAddBlocksTool<T>(config: {
 
             const operation = chunk.operation as AddBlocksToolCall<T>;
 
-            const jsonToolCall = await config.toJSONToolCall(editor, {
+            const jsonToolCall = config.toJSONToolCall(editor, {
               ...chunk,
               operation,
             });
@@ -223,7 +223,7 @@ export function createAddBlocksTool<T>(config: {
               let agentSteps: AgentStep[] = [];
               if (i < addedBlockIds.length) {
                 // we have already added this block, so we need to update it
-                const tool = await config.rebaseTool(addedBlockIds[i], editor);
+                const tool = config.rebaseTool(addedBlockIds[i], editor);
                 const steps = updateToReplaceSteps(
                   {
                     id: addedBlockIds[i],

package/src/api/formats/base-tools/createUpdateBlockTool.ts

@@ -59,7 +59,7 @@ export function createUpdateBlockTool<T>(config: {
   rebaseTool: (
     id: string,
     editor: BlockNoteEditor<any, any, any>,
-  ) => Promise<RebaseTool>;
+  ) => RebaseTool;
   /**
    * Converts the operation from `AddBlocksToolCall<T>` to `AddBlocksToolCall<PartialBlock<any, any, any>>`
    *
@@ -73,7 +73,7 @@ export function createUpdateBlockTool<T>(config: {
       isUpdateToPreviousOperation: boolean;
       isPossiblyPartial: boolean;
     },
-  ) => Promise<UpdateBlockToolCall<PartialBlock<any, any, any>> | undefined>;
+  ) => UpdateBlockToolCall<PartialBlock<any, any, any>> | undefined;
 }) {
   return (
     editor: BlockNoteEditor<any, any, any>,
@@ -202,7 +202,7 @@ export function createUpdateBlockTool<T>(config: {
 
             // REC: we could investigate whether we can use a single rebasetool across operations instead of
             // creating a new one every time (possibly expensive)
-            const tool = await config.rebaseTool(operation.id, editor);
+            const tool = config.rebaseTool(operation.id, editor);
 
             const fromPos = selectionPositions
               ? tool.invertMap.invert().map(selectionPositions.from())
@@ -212,7 +212,7 @@ export function createUpdateBlockTool<T>(config: {
               ? tool.invertMap.invert().map(selectionPositions.to())
               : undefined;
 
-            const jsonToolCall = await config.toJSONToolCall(editor, {
+            const jsonToolCall = config.toJSONToolCall(editor, {
               ...chunk,
               operation,
             });

package/src/api/formats/html-blocks/columnContainerDocumentState.test.ts

@@ -2,7 +2,7 @@
 
 import { BlockNoteEditor, BlockNoteSchema } from "@blocknote/core";
 import { withMultiColumn } from "@blocknote/xl-multi-column";
-import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import { afterEach, beforeEach, describe, expect, it } from "vite-plus/test";
 
 import { htmlBlockLLMFormat } from "./htmlBlocks.js";
 

package/src/api/formats/html-blocks/htmlBlocks.test.ts

@@ -1,7 +1,7 @@
 import { snapshot } from "msw-snapshot";
 import { setupServer } from "msw/node";
 import path from "path";
-import { afterAll, afterEach, beforeAll, describe, it } from "vitest";
+import { afterAll, afterEach, beforeAll, describe, it } from "vite-plus/test";
 import { testAIModels } from "../../../testUtil/testAIModels.js";
 
 import { BlockNoteEditor } from "@blocknote/core";
@@ -102,11 +102,8 @@ describe("Models", () => {
 });
 
 describe("streamToolsProvider", () => {
-  it("should return the correct stream tools", () => {
+  it.skip("should return the correct stream tools", () => {
     // test skipped, this is only to validate type inference
-    return;
-
-    // eslint-disable-next-line no-unreachable
     const editor = BlockNoteEditor.create();
     const streamTools = htmlBlockLLMFormat
       .getStreamToolsProvider({
@@ -118,14 +115,14 @@ describe("streamToolsProvider", () => {
 
     const executor = new StreamToolExecutor(streamTools);
 
-    executor.executeOne({
+    void executor.executeOne({
       type: "add",
       blocks: ["<p>test</p>"],
       referenceId: "1",
       position: "after",
     });
 
-    executor.executeOne({
+    void executor.executeOne({
       // @ts-expect-error
       type: "update",
       blocks: ["<p>test</p>"],
@@ -133,12 +130,13 @@ describe("streamToolsProvider", () => {
       position: "after",
     });
 
-    executor.executeOne({
+    void executor.executeOne({
       type: "add",
       // @ts-expect-error
       blocks: [{ type: "paragraph", content: "test" }],
       referenceId: "1",
       position: "after",
     });
+    /* eslint-enable no-unreachable */
   });
 });

package/src/api/formats/html-blocks/tools/getPartialHTML.test.ts

@@ -1,4 +1,4 @@
-import { describe, expect, it } from "vitest";
+import { describe, expect, it } from "vite-plus/test";
 import { getPartialHTML } from "./getPartialHTML.js";
 
 describe("getPartialHTML", () => {

package/src/api/formats/html-blocks/tools/index.ts

@@ -23,27 +23,21 @@ export const tools = {
     },
     validateBlock: validateBlockFunction,
     rebaseTool: createHTMLRebaseTool,
-    toJSONToolCall: async (editor, chunk) => {
+    toJSONToolCall: (editor, chunk) => {
       const initialMockID = (window as any).__TEST_OPTIONS?.mockID;
 
-      const blocks = (
-        await Promise.all(
-          chunk.operation.blocks.map(async (html) => {
-            const parsedHtml = chunk.isPossiblyPartial
-              ? getPartialHTML(html)
-              : html;
-            if (!parsedHtml) {
-              return [];
-            }
-            return (await editor.tryParseHTMLToBlocks(parsedHtml)).map(
-              (block) => {
-                delete (block as any).id;
-                return block;
-              },
-            );
-          }),
-        )
-      ).flat();
+      const blocks = chunk.operation.blocks.flatMap((html) => {
+        const parsedHtml = chunk.isPossiblyPartial
+          ? getPartialHTML(html)
+          : html;
+        if (!parsedHtml) {
+          return [];
+        }
+        return editor.tryParseHTMLToBlocks(parsedHtml).map((block) => {
+          delete (block as any).id;
+          return block;
+        });
+      });
 
       // hacky
       if ((window as any).__TEST_OPTIONS) {
@@ -71,7 +65,7 @@ export const tools = {
     },
     validateBlock: validateBlockFunction,
     rebaseTool: createHTMLRebaseTool,
-    toJSONToolCall: async (editor, chunk) => {
+    toJSONToolCall: (editor, chunk) => {
       const html = chunk.isPossiblyPartial
         ? getPartialHTML(chunk.operation.block)
         : chunk.operation.block;
@@ -80,7 +74,7 @@ export const tools = {
         return undefined;
       }
 
-      const block = (await editor.tryParseHTMLToBlocks(html))[0];
+      const block = editor.tryParseHTMLToBlocks(html)[0];
 
       // console.log("update", operation.block);
       // console.log("html", html);

package/src/api/formats/html-blocks/tools/rebaseTool.ts

@@ -5,7 +5,7 @@ import {
   rebaseTool,
 } from "../../../../prosemirror/rebaseTool.js";
 
-export async function createHTMLRebaseTool(
+export function createHTMLRebaseTool(
   id: string,
   editor: BlockNoteEditor<any, any, any>,
 ) {
@@ -15,7 +15,7 @@ export async function createHTMLRebaseTool(
     throw new Error("block not found");
   }
 
-  const html = await editor.blocksToHTMLLossy([
+  const html = editor.blocksToHTMLLossy([
     {
       ...block,
       children: [],
@@ -24,7 +24,7 @@ export async function createHTMLRebaseTool(
 
   const initialMockID = (window as any).__TEST_OPTIONS?.mockID;
 
-  const blocks = await editor.tryParseHTMLToBlocks(html);
+  const blocks = editor.tryParseHTMLToBlocks(html);
 
   // hacky
   if ((window as any).__TEST_OPTIONS) {

package/src/api/formats/json/errorHandling.test.ts

@@ -1,4 +1,11 @@
-import { afterAll, afterEach, beforeAll, describe, expect, it } from "vitest";
+import {
+  afterAll,
+  afterEach,
+  beforeAll,
+  describe,
+  expect,
+  it,
+} from "vite-plus/test";
 
 import { BlockNoteEditor } from "@blocknote/core";
 import { HttpResponse, http } from "msw";

package/src/api/formats/json/json.test.ts

@@ -1,4 +1,4 @@
-import { afterAll, afterEach, beforeAll, describe } from "vitest";
+import { afterAll, afterEach, beforeAll, describe } from "vite-plus/test";
 
 import { snapshot } from "msw-snapshot";
 import { setupServer } from "msw/node";

package/src/api/formats/json/tools/index.ts

@@ -19,9 +19,9 @@ export const tools = {
       ...(blockNoteSchemaToJSONSchema(editor.schema) as any),
     }),
     validateBlock: validateBlockFunction,
-    rebaseTool: async (_id, editor) =>
+    rebaseTool: (_id, editor) =>
       rebaseTool(editor, getApplySuggestionsTr(editor)),
-    toJSONToolCall: async (_editor, chunk) => {
+    toJSONToolCall: (_editor, chunk) => {
       return chunk.operation;
     },
   }),
@@ -35,9 +35,9 @@ export const tools = {
       ...(blockNoteSchemaToJSONSchema(editor.schema) as any),
     }),
     validateBlock: validateBlockFunction,
-    rebaseTool: async (_id, editor) =>
+    rebaseTool: (_id, editor) =>
       rebaseTool(editor, getApplySuggestionsTr(editor)),
-    toJSONToolCall: async (_editor, chunk) => {
+    toJSONToolCall: (_editor, chunk) => {
       const defaultPropsVals = Object.fromEntries(
         Object.entries(defaultProps).map(([key, val]) => {
           return [key, val.default];

package/src/api/formats/json/tools/jsontools.test.ts

@@ -1,6 +1,6 @@
 /* eslint-disable jest/valid-title */
 import { BlockNoteEditor, expandPMRangeToWords } from "@blocknote/core";
-import { describe, expect, it } from "vitest";
+import { describe, expect, it } from "vite-plus/test";
 import { addOperationTestCases } from "../../../../testUtil/cases/addOperationTestCases.js";
 import { combinedOperationsTestCases } from "../../../../testUtil/cases/combinedOperationsTestCases.js";
 import { deleteOperationTestCases } from "../../../../testUtil/cases/deleteOperationTestCases.js";

package/src/api/formats/markdown-blocks/markdownBlocks.test.ts

@@ -1,4 +1,4 @@
-import { afterAll, afterEach, beforeAll, describe } from "vitest";
+import { afterAll, afterEach, beforeAll, describe } from "vite-plus/test";
 
 import { snapshot } from "msw-snapshot";
 import { setupServer } from "msw/node";

package/src/api/formats/markdown-blocks/tools/index.ts

@@ -24,14 +24,12 @@ export const tools = {
     },
     validateBlock: validateBlockFunction,
     rebaseTool: createMDRebaseTool,
-    toJSONToolCall: async (editor, chunk) => {
-      const blocks = await Promise.all(
-        chunk.operation.blocks.map(async (md) => {
-          const block = (await editor.tryParseMarkdownToBlocks(md.trim()))[0]; // TODO: trim
-          delete (block as any).id;
-          return block;
-        }),
-      );
+    toJSONToolCall: (editor, chunk) => {
+      const blocks = chunk.operation.blocks.map((md) => {
+        const block = editor.tryParseMarkdownToBlocks(md.trim())[0]; // TODO: trim
+        delete (block as any).id;
+        return block;
+      });
 
       // hacky
       if ((window as any).__TEST_OPTIONS) {
@@ -58,9 +56,9 @@ export const tools = {
     },
     validateBlock: validateBlockFunction,
     rebaseTool: createMDRebaseTool,
-    toJSONToolCall: async (editor, chunk) => {
-      const block = (
-        await editor.tryParseMarkdownToBlocks(chunk.operation.block.trim())
+    toJSONToolCall: (editor, chunk) => {
+      const block = editor.tryParseMarkdownToBlocks(
+        chunk.operation.block.trim(),
       )[0];
 
       delete (block as any).id;

package/src/api/formats/markdown-blocks/tools/rebaseTool.ts

@@ -6,13 +6,13 @@ import {
   rebaseTool,
 } from "../../../../prosemirror/rebaseTool.js";
 
-export async function createMDRebaseTool(
+export function createMDRebaseTool(
   id: string,
   editor: BlockNoteEditor<any, any, any>,
 ) {
   const tr = getApplySuggestionsTr(editor);
-  const md = await editor.blocksToMarkdownLossy([getBlock(tr.doc, id)!]);
-  const blocks = await editor.tryParseMarkdownToBlocks(md);
+  const md = editor.blocksToMarkdownLossy([getBlock(tr.doc, id)!]);
+  const blocks = editor.tryParseMarkdownToBlocks(md);
 
   const steps = updateToReplaceSteps(
     {

package/src/api/formats/tests/sharedTestCases.ts

@@ -1,9 +1,8 @@
 import { Chat, UIMessage } from "@ai-sdk/react";
 import { BlockNoteEditor } from "@blocknote/core";
-import { getCurrentTest, TestContext } from "@vitest/runner";
 import path from "path";
 import { TextSelection } from "prosemirror-state";
-import { describe, expect, it } from "vitest";
+import { describe, expect, it, TestContext, TestRunner } from "vite-plus/test";
 import { AIExtension } from "../../../AIExtension.js";
 import { sendMessageWithAIRequest } from "../../../index.js";
 import { addOperationTestCases } from "../../../testUtil/cases/addOperationTestCases.js";
@@ -23,7 +22,7 @@ const BASE_FILE_PATH = path.resolve(__dirname, "__snapshots__");
 // @ts-expect-error - unused helper kept for debugging snapshots
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
 async function matchFileSnapshot(data: any, postFix = "") {
-  const t = getCurrentTest()!;
+  const t = TestRunner.getCurrentTest()!;
   // this uses the same snapshot path, regardless of the model / streaming params
   await expect(data).toMatchFileSnapshot(
     path.resolve(

package/src/api/formats/tests/snapshotPath.ts

@@ -1,6 +1,6 @@
-import { getCurrentTest } from "@vitest/runner";
 import { getSortedEntries, toHashString } from "msw-snapshot";
 import path from "node:path";
+import { TestRunner } from "vite-plus/test";
 
 async function createRequestHash(req: Request): Promise<string> {
   const url = new URL(req.url);
@@ -18,7 +18,7 @@ export function createSnapshotPathFn(_basePath: string) {
   const fetchCountMap: Record<string, number> = {};
 
   return async (info: { request: Request }): Promise<string> => {
-    const t = getCurrentTest()!;
+    const t = TestRunner.getCurrentTest()!;
     const mswPath = path.join(
       t.suite!.name,
       "__msw_snapshots__",

package/src/api/formats/tests/validateTestEnvironment.test.ts

@@ -1,6 +1,6 @@
 import * as glob from "glob";
 import path from "path";
-import { describe, expect, it } from "vitest";
+import { describe, expect, it } from "vite-plus/test";
 
 /**
  * We should only have one snapshot file per test, this test ensures that

package/src/api/promptHelpers/suffixIds.ts

@@ -3,7 +3,7 @@ export function suffixIDs<T>(source: Array<T>): Array<T> {
     if (typeof el === "object" && el && "id" in el) {
       return {
         ...el,
-        id: `${el.id}$`,
+        id: `${String(el.id)}$`,
       };
     }
     return el;

package/src/api/schema/__snapshots__/schemaToJSONSchema.test.ts.snap

@@ -64,7 +64,7 @@ exports[`creates json schema 1`] = `
           "additionalProperties": false,
           "properties": {
             "content": {
-              "$ref": "#/$defs/inlinecontent",
+              "type": "string",
             },
             "props": {
               "additionalProperties": false,

package/src/api/schema/schemaToJSONSchema.test.ts

@@ -1,12 +1,12 @@
 import { BlockNoteEditor } from "@blocknote/core";
-import { expect, it } from "vitest";
+import { expect, it } from "vite-plus/test";
 import { blockNoteSchemaToJSONSchema } from "./schemaToJSONSchema.js";
 
-it("creates json schema", async () => {
+it("creates json schema", () => {
   // eslint-disable-next-line no-console
   const editor = BlockNoteEditor.create();
 
   const jsonSchema = blockNoteSchemaToJSONSchema(editor.schema);
-  await expect(jsonSchema).toMatchSnapshot();
+  expect(jsonSchema).toMatchSnapshot();
   // console.log(JSON.stringify(blockNoteSchemaToJSONSchema(editor.schema), undefined, 2));
 });

package/src/api/schema/schemaToJSONSchema.ts

@@ -217,9 +217,11 @@ function blockSchemaToJSONSchema(schema: BlockSchema) {
             content:
               val.content === "inline"
                 ? { $ref: "#/$defs/inlinecontent" }
-                : val.content === "table"
-                  ? { type: "object", properties: {} } // TODO
-                  : undefined,
+                : val.content === "plain"
+                  ? { type: "string" }
+                  : val.content === "table"
+                    ? { type: "object", properties: {} } // TODO
+                    : undefined,
             // filter out default props (TODO: make option)
             props: propSchemaToJSONSchema(val.propSchema),
             // Object.fromEntries(

package/src/components/AIMenu/PromptSuggestionMenu.tsx

@@ -83,7 +83,7 @@ export const PromptSuggestionMenu = (props: PromptSuggestionMenuProps) => {
           handler(event);
         } else {
           // TODO: check focus?
-          handleEnter(event);
+          void handleEnter(event);
         }
       } else {
         handler(event);

package/src/plugins/AgentCursorPlugin.ts

@@ -1,6 +1,13 @@
 import { Plugin, PluginKey } from "prosemirror-state";
 import { Decoration, DecorationSet } from "prosemirror-view";
-import { defaultSelectionBuilder } from "y-prosemirror";
+
+// Pulled from `y-prosemirror`https://github.com/yjs/y-prosemirror/blob/v1.3.7/src/plugins/cursor-plugin.js
+const defaultSelectionBuilder = (user: { name: string; color: string }) => {
+  return {
+    style: `background-color: ${user.color}70`,
+    class: "ProseMirror-yjs-selection",
+  };
+};
 
 type AgentCursorState = {
   selection: { anchor: number; head: number } | undefined;