@blocknote/xl-ai 0.49.0 → 0.50.0

package/dist/webpack-stats.json

@@ -1 +1 @@
-{"builtAt":1777035767190,"assets":[{"name":"blocknote-xl-ai.cjs","size":16875},{"name":"locales.cjs","size":23775},{"name":"server.cjs","size":760},{"name":"server-D-Xyj9BS.cjs","size":38286},{"name":"blocknote-xl-ai.cjs.map","size":63114},{"name":"locales.cjs.map","size":56423},{"name":"server-D-Xyj9BS.cjs.map","size":186169}],"chunks":[{"id":"a1ee98a","entry":true,"initial":true,"files":["blocknote-xl-ai.cjs"],"names":["blocknote-xl-ai"]},{"id":"842df4c","entry":true,"initial":true,"files":["locales.cjs"],"names":["locales"]},{"id":"791e108","entry":true,"initial":true,"files":["server.cjs"],"names":["server"]},{"id":"91fd206","entry":false,"initial":true,"files":["server-D-Xyj9BS.cjs"],"names":["server"]}],"modules":[{"name":"./src/style.css","size":0,"chunks":["a1ee98a"]},{"name":"./src/plugins/AgentCursorPlugin.ts","size":2201,"chunks":["a1ee98a"]},{"name":"./src/AIExtension.ts","size":7732,"chunks":["a1ee98a"]},{"name":"./src/hooks/useAIDictionary.ts","size":161,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/PromptSuggestionMenu.tsx","size":3233,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/getDefaultAIMenuItems.tsx","size":6452,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/AIMenu.tsx","size":3485,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/AIMenuController.tsx","size":2391,"chunks":["a1ee98a"]},{"name":"./src/components/FormattingToolbar/AIToolbarButton.tsx","size":1068,"chunks":["a1ee98a"]},{"name":"./src/components/SuggestionMenu/getAISlashMenuItems.tsx","size":670,"chunks":["a1ee98a"]},{"name":"./src/index.ts","size":0,"chunks":["a1ee98a"]},{"name":"./src/i18n/locales/ar.ts","size":1427,"chunks":["842df4c"]},{"name":"./src/i18n/locales/de.ts","size":1481,"chunks":["842df4c"]},{"name":"./src/i18n/locales/en.ts","size":1379,"chunks":["842df4c"]},{"name":"./src/i18n/locales/es.ts","size":1449,"chunks":["842df4c"]},{"name":"./src/i18n/locales/fr.ts","size":1465,"chunks":["842df4c"]},{"name":"./src/i18n/locales/he.ts","size":1297,"chunks":["842df4c"]},{"name":"./src/i18n/locales/hr.ts","size":1390,"chunks":["842df4c"]},{"name":"./src/i18n/locales/is.ts","size":1458,"chunks":["842df4c"]},{"name":"./src/i18n/locales/it.ts","size":1451,"chunks":["842df4c"]},{"name":"./src/i18n/locales/ja.ts","size":1268,"chunks":["842df4c"]},{"name":"./src/i18n/locales/ko.ts","size":1290,"chunks":["842df4c"]},{"name":"./src/i18n/locales/nl.ts","size":1432,"chunks":["842df4c"]},{"name":"./src/i18n/locales/no.ts","size":1405,"chunks":["842df4c"]},{"name":"./src/i18n/locales/pl.ts","size":1406,"chunks":["842df4c"]},{"name":"./src/i18n/locales/pt.ts","size":1444,"chunks":["842df4c"]},{"name":"./src/i18n/locales/ru.ts","size":1445,"chunks":["842df4c"]},{"name":"./src/i18n/locales/sk.ts","size":1407,"chunks":["842df4c"]},{"name":"./src/i18n/locales/uk.ts","size":1452,"chunks":["842df4c"]},{"name":"./src/i18n/locales/vi.ts","size":1402,"chunks":["842df4c"]},{"name":"./src/i18n/locales/zh-tw.ts","size":1244,"chunks":["842df4c"]},{"name":"./src/i18n/locales/zh.ts","size":1236,"chunks":["842df4c"]},{"name":"./src/i18n/locales/index.ts","size":0,"chunks":["842df4c"]},{"name":"./rolldown/runtime.js","size":1008,"chunks":["91fd206"]},{"name":"./src/util/emptyBlock.ts","size":216,"chunks":["91fd206"]},{"name":"./src/util/trimArray.ts","size":320,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/trimEmptyBlocks.ts","size":270,"chunks":["91fd206"]},{"name":"./src/api/aiRequest/builder.ts","size":1312,"chunks":["91fd206"]},{"name":"./src/streamTool/jsonSchema.ts","size":1326,"chunks":["91fd206"]},{"name":"./src/streamTool/ChunkExecutionError.ts","size":295,"chunks":["91fd206"]},{"name":"./src/streamTool/StreamToolExecutor.ts","size":3431,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/injectDocumentStateMessages.ts","size":1755,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/toolDefinitions.ts","size":745,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/clientside/ClientSideTransport.ts","size":1172,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/appendableStream.ts","size":1069,"chunks":["91fd206"]},{"name":"./src/util/stream.ts","size":1003,"chunks":["91fd206"]},{"name":"./src/streamTool/filterNewOrUpdatedOperations.ts","size":925,"chunks":["91fd206"]},{"name":"./src/streamTool/filterValidOperations.ts","size":692,"chunks":["91fd206"]},{"name":"./src/streamTool/toValidatedOperations.ts","size":1106,"chunks":["91fd206"]},{"name":"./src/streamTool/preprocess.ts","size":431,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/UIMessageStreamToOperationsResult.ts","size":348,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/chatHandlers.ts","size":4033,"chunks":["91fd206"]},{"name":"./src/api/aiRequest/sendMessageWithAIRequest.ts","size":786,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/addCursorPosition.ts","size":410,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/convertBlocks.ts","size":234,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/flattenBlocks.ts","size":201,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/suffixIds.ts","size":229,"chunks":["91fd206"]},{"name":"./src/api/formats/DocumentStateBuilder.ts","size":1530,"chunks":["91fd206"]},{"name":"./src/prosemirror/fragmentUtil.ts","size":285,"chunks":["91fd206"]},{"name":"./src/prosemirror/agent.ts","size":5303,"chunks":["91fd206"]},{"name":"./src/prosemirror/changeset.ts","size":5344,"chunks":["91fd206"]},{"name":"./src/streamTool/streamTool.ts","size":100,"chunks":["91fd206"]},{"name":"./src/util/AbortError.ts","size":179,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/util/validateBlockArray.ts","size":587,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/createAddBlocksTool.ts","size":4117,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/createUpdateBlockTool.ts","size":3314,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/delete.ts","size":1440,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/getPartialHTML.ts","size":657,"chunks":["91fd206"]},{"name":"./src/prosemirror/rebaseTool.ts","size":940,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/rebaseTool.ts","size":837,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/validate.ts","size":246,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/index.ts","size":1618,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/htmlBlocks.ts","size":2552,"chunks":["91fd206"]},{"name":"./src/api/schema/mergeSchema.ts","size":767,"chunks":["91fd206"]},{"name":"./src/api/schema/schemaToJSONSchema.ts","size":3581,"chunks":["91fd206"]},{"name":"./src/api/formats/json/tools/validate.ts","size":1409,"chunks":["91fd206"]},{"name":"./src/api/formats/json/tools/index.ts","size":1242,"chunks":["91fd206"]},{"name":"./src/api/formats/json/json.ts","size":2244,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/tools/rebaseTool.ts","size":651,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/tools/validate.ts","size":248,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/tools/index.ts","size":1377,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/markdownBlocks.ts","size":2373,"chunks":["91fd206"]},{"name":"./src/api/formats/formats.ts","size":195,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/index.ts","size":160,"chunks":["91fd206"]},{"name":"./src/i18n/dictionary.ts","size":185,"chunks":["91fd206"]},{"name":"./src/blocknoteAIClient/client.ts","size":376,"chunks":["91fd206"]},{"name":"./src/server.ts","size":0,"chunks":["91fd206"]}]}
+{"builtAt":1777910869037,"assets":[{"name":"blocknote-xl-ai.cjs","size":16875},{"name":"locales.cjs","size":23775},{"name":"server.cjs","size":760},{"name":"server-D-Xyj9BS.cjs","size":38286},{"name":"blocknote-xl-ai.cjs.map","size":63114},{"name":"locales.cjs.map","size":56423},{"name":"server-D-Xyj9BS.cjs.map","size":186169}],"chunks":[{"id":"a1ee98a","entry":true,"initial":true,"files":["blocknote-xl-ai.cjs"],"names":["blocknote-xl-ai"]},{"id":"842df4c","entry":true,"initial":true,"files":["locales.cjs"],"names":["locales"]},{"id":"791e108","entry":true,"initial":true,"files":["server.cjs"],"names":["server"]},{"id":"91fd206","entry":false,"initial":true,"files":["server-D-Xyj9BS.cjs"],"names":["server"]}],"modules":[{"name":"./src/style.css","size":0,"chunks":["a1ee98a"]},{"name":"./src/plugins/AgentCursorPlugin.ts","size":2201,"chunks":["a1ee98a"]},{"name":"./src/AIExtension.ts","size":7732,"chunks":["a1ee98a"]},{"name":"./src/hooks/useAIDictionary.ts","size":161,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/PromptSuggestionMenu.tsx","size":3233,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/getDefaultAIMenuItems.tsx","size":6452,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/AIMenu.tsx","size":3485,"chunks":["a1ee98a"]},{"name":"./src/components/AIMenu/AIMenuController.tsx","size":2391,"chunks":["a1ee98a"]},{"name":"./src/components/FormattingToolbar/AIToolbarButton.tsx","size":1068,"chunks":["a1ee98a"]},{"name":"./src/components/SuggestionMenu/getAISlashMenuItems.tsx","size":670,"chunks":["a1ee98a"]},{"name":"./src/index.ts","size":0,"chunks":["a1ee98a"]},{"name":"./src/i18n/locales/ar.ts","size":1427,"chunks":["842df4c"]},{"name":"./src/i18n/locales/de.ts","size":1481,"chunks":["842df4c"]},{"name":"./src/i18n/locales/en.ts","size":1379,"chunks":["842df4c"]},{"name":"./src/i18n/locales/es.ts","size":1449,"chunks":["842df4c"]},{"name":"./src/i18n/locales/fr.ts","size":1465,"chunks":["842df4c"]},{"name":"./src/i18n/locales/he.ts","size":1297,"chunks":["842df4c"]},{"name":"./src/i18n/locales/hr.ts","size":1390,"chunks":["842df4c"]},{"name":"./src/i18n/locales/is.ts","size":1458,"chunks":["842df4c"]},{"name":"./src/i18n/locales/it.ts","size":1451,"chunks":["842df4c"]},{"name":"./src/i18n/locales/ja.ts","size":1268,"chunks":["842df4c"]},{"name":"./src/i18n/locales/ko.ts","size":1290,"chunks":["842df4c"]},{"name":"./src/i18n/locales/nl.ts","size":1432,"chunks":["842df4c"]},{"name":"./src/i18n/locales/no.ts","size":1405,"chunks":["842df4c"]},{"name":"./src/i18n/locales/pl.ts","size":1406,"chunks":["842df4c"]},{"name":"./src/i18n/locales/pt.ts","size":1444,"chunks":["842df4c"]},{"name":"./src/i18n/locales/ru.ts","size":1445,"chunks":["842df4c"]},{"name":"./src/i18n/locales/sk.ts","size":1407,"chunks":["842df4c"]},{"name":"./src/i18n/locales/uk.ts","size":1452,"chunks":["842df4c"]},{"name":"./src/i18n/locales/vi.ts","size":1402,"chunks":["842df4c"]},{"name":"./src/i18n/locales/zh-tw.ts","size":1244,"chunks":["842df4c"]},{"name":"./src/i18n/locales/zh.ts","size":1236,"chunks":["842df4c"]},{"name":"./src/i18n/locales/index.ts","size":0,"chunks":["842df4c"]},{"name":"./rolldown/runtime.js","size":1008,"chunks":["91fd206"]},{"name":"./src/util/emptyBlock.ts","size":216,"chunks":["91fd206"]},{"name":"./src/util/trimArray.ts","size":320,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/trimEmptyBlocks.ts","size":270,"chunks":["91fd206"]},{"name":"./src/api/aiRequest/builder.ts","size":1312,"chunks":["91fd206"]},{"name":"./src/streamTool/jsonSchema.ts","size":1326,"chunks":["91fd206"]},{"name":"./src/streamTool/ChunkExecutionError.ts","size":295,"chunks":["91fd206"]},{"name":"./src/streamTool/StreamToolExecutor.ts","size":3431,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/injectDocumentStateMessages.ts","size":1755,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/toolDefinitions.ts","size":745,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/clientside/ClientSideTransport.ts","size":1172,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/appendableStream.ts","size":1069,"chunks":["91fd206"]},{"name":"./src/util/stream.ts","size":1003,"chunks":["91fd206"]},{"name":"./src/streamTool/filterNewOrUpdatedOperations.ts","size":925,"chunks":["91fd206"]},{"name":"./src/streamTool/filterValidOperations.ts","size":692,"chunks":["91fd206"]},{"name":"./src/streamTool/toValidatedOperations.ts","size":1106,"chunks":["91fd206"]},{"name":"./src/streamTool/preprocess.ts","size":431,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/UIMessageStreamToOperationsResult.ts","size":348,"chunks":["91fd206"]},{"name":"./src/streamTool/vercelAiSdk/util/chatHandlers.ts","size":4033,"chunks":["91fd206"]},{"name":"./src/api/aiRequest/sendMessageWithAIRequest.ts","size":786,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/addCursorPosition.ts","size":410,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/convertBlocks.ts","size":234,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/flattenBlocks.ts","size":201,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/suffixIds.ts","size":229,"chunks":["91fd206"]},{"name":"./src/api/formats/DocumentStateBuilder.ts","size":1530,"chunks":["91fd206"]},{"name":"./src/prosemirror/fragmentUtil.ts","size":285,"chunks":["91fd206"]},{"name":"./src/prosemirror/agent.ts","size":5303,"chunks":["91fd206"]},{"name":"./src/prosemirror/changeset.ts","size":5344,"chunks":["91fd206"]},{"name":"./src/streamTool/streamTool.ts","size":100,"chunks":["91fd206"]},{"name":"./src/util/AbortError.ts","size":179,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/util/validateBlockArray.ts","size":587,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/createAddBlocksTool.ts","size":4117,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/createUpdateBlockTool.ts","size":3314,"chunks":["91fd206"]},{"name":"./src/api/formats/base-tools/delete.ts","size":1440,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/getPartialHTML.ts","size":657,"chunks":["91fd206"]},{"name":"./src/prosemirror/rebaseTool.ts","size":940,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/rebaseTool.ts","size":837,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/validate.ts","size":246,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/tools/index.ts","size":1618,"chunks":["91fd206"]},{"name":"./src/api/formats/html-blocks/htmlBlocks.ts","size":2552,"chunks":["91fd206"]},{"name":"./src/api/schema/mergeSchema.ts","size":767,"chunks":["91fd206"]},{"name":"./src/api/schema/schemaToJSONSchema.ts","size":3581,"chunks":["91fd206"]},{"name":"./src/api/formats/json/tools/validate.ts","size":1409,"chunks":["91fd206"]},{"name":"./src/api/formats/json/tools/index.ts","size":1242,"chunks":["91fd206"]},{"name":"./src/api/formats/json/json.ts","size":2244,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/tools/rebaseTool.ts","size":651,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/tools/validate.ts","size":248,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/tools/index.ts","size":1377,"chunks":["91fd206"]},{"name":"./src/api/formats/markdown-blocks/markdownBlocks.ts","size":2373,"chunks":["91fd206"]},{"name":"./src/api/formats/formats.ts","size":195,"chunks":["91fd206"]},{"name":"./src/api/promptHelpers/index.ts","size":160,"chunks":["91fd206"]},{"name":"./src/i18n/dictionary.ts","size":185,"chunks":["91fd206"]},{"name":"./src/blocknoteAIClient/client.ts","size":376,"chunks":["91fd206"]},{"name":"./src/server.ts","size":0,"chunks":["91fd206"]}]}

package/package.json

@@ -11,7 +11,7 @@
     "directory": "packages/xl-ai"
   },
   "license": "GPL-3.0 OR PROPRIETARY",
-  "version": "0.49.0",
+  "version": "0.50.0",
   "files": [
     "dist",
     "types",
@@ -63,9 +63,9 @@
   "dependencies": {
     "@ai-sdk/provider-utils": "^4.0.2",
     "@ai-sdk/react": "^3.0.5",
-    "@blocknote/core": "0.49.0",
-    "@blocknote/mantine": "0.49.0",
-    "@blocknote/react": "0.49.0",
+    "@blocknote/core": "0.50.0",
+    "@blocknote/mantine": "0.50.0",
+    "@blocknote/react": "0.50.0",
     "@floating-ui/react": "^0.27.18",
     "@handlewithcare/prosemirror-suggest-changes": "^0.1.8",
     "@tiptap/core": "^3.13.0",
@@ -132,6 +132,7 @@
     "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",
+    "rename-msw-snapshots": "node scripts/rename-msw-snapshots.mjs",
     "email": "email dev"
   }
 }

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

@@ -1,5 +1,4 @@
-import { getCurrentTest } from "@vitest/runner";
-import { getSortedEntries, snapshot, toHashString } from "msw-snapshot";
+import { snapshot } from "msw-snapshot";
 import { setupServer } from "msw/node";
 import path from "path";
 import { afterAll, afterEach, beforeAll, describe, it } from "vitest";
@@ -8,6 +7,7 @@ import { testAIModels } from "../../../testUtil/testAIModels.js";
 import { BlockNoteEditor } from "@blocknote/core";
 import { StreamToolExecutor } from "../../../streamTool/StreamToolExecutor.js";
 import { ClientSideTransport } from "../../../streamTool/vercelAiSdk/clientside/ClientSideTransport.js";
+import { createSnapshotPathFn } from "../tests/snapshotPath.js";
 import { generateSharedTestCases } from "../tests/sharedTestCases.js";
 import { htmlBlockLLMFormat } from "./htmlBlocks.js";
 
@@ -17,54 +17,14 @@ const BASE_FILE_PATH = path.resolve(
   path.basename(__filename),
 );
 
-const fetchCountMap: Record<string, number> = {};
-
-async function createRequestHash(req: Request) {
-  const url = new URL(req.url);
-  return [
-    // url.host,
-    // url.pathname,
-    toHashString([
-      req.method,
-      url.origin,
-      url.pathname,
-      getSortedEntries(url.searchParams),
-      getSortedEntries(req.headers),
-      // getSortedEntries(req.cookies),
-      new TextDecoder("utf-8").decode(await req.arrayBuffer()),
-    ]),
-  ].join("/");
-}
-
 // Main test suite with snapshot middleware
 describe("Models", () => {
   // Define server with snapshot middleware for the main tests
   const server = setupServer(
     snapshot({
       updateSnapshots: "missing",
-      // onSnapshotUpdated: "all",
-      // ignoreSnapshots: true,
-      async createSnapshotPath(info) {
-        // use a unique path for each model
-        const t = getCurrentTest()!;
-        const mswPath = path.join(
-          t.suite!.name, // same directory as the test snapshot
-          "__msw_snapshots__",
-          t.suite!.suite!.name, // model / streaming params
-          t.name,
-        );
-        // in case there are multiple requests in a test, we need to use a separate snapshot for each request
-        fetchCountMap[mswPath] = (fetchCountMap[mswPath] || 0) + 1;
-        const hash = await createRequestHash(info.request);
-        return mswPath + `_${fetchCountMap[mswPath]}_${hash}.json`;
-      },
+      createSnapshotPath: createSnapshotPathFn(BASE_FILE_PATH),
       basePath: BASE_FILE_PATH,
-      // onFetchFromSnapshot(info, snapshot) {
-      // console.log("onFetchFromSnapshot", info, snapshot);
-      // },
-      // onFetchFromServer(info, snapshot) {
-      //   console.log("onFetchFromServer", info, snapshot);
-      // },
     }),
   );
 

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

@@ -1,9 +1,9 @@
 import { afterAll, afterEach, beforeAll, describe } from "vitest";
 
-import { getCurrentTest } from "@vitest/runner";
-import { getSortedEntries, snapshot, toHashString } from "msw-snapshot";
+import { snapshot } from "msw-snapshot";
 import { setupServer } from "msw/node";
 import path from "path";
+import { createSnapshotPathFn } from "../tests/snapshotPath.js";
 import { generateSharedTestCases } from "../tests/sharedTestCases.js";
 
 import { ClientSideTransport } from "../../../streamTool/vercelAiSdk/clientside/ClientSideTransport.js";
@@ -16,54 +16,14 @@ const BASE_FILE_PATH = path.resolve(
   path.basename(__filename),
 );
 
-const fetchCountMap: Record<string, number> = {};
-
-async function createRequestHash(req: Request) {
-  const url = new URL(req.url);
-  return [
-    // url.host,
-    // url.pathname,
-    toHashString([
-      req.method,
-      url.origin,
-      url.pathname,
-      getSortedEntries(url.searchParams),
-      getSortedEntries(req.headers),
-      // getSortedEntries(req.cookies),
-      new TextDecoder("utf-8").decode(await req.arrayBuffer()),
-    ]),
-  ].join("/");
-}
-
 // Main test suite with snapshot middleware
 describe.skip("Models", () => {
   // Define server with snapshot middleware for the main tests
   const server = setupServer(
     snapshot({
       updateSnapshots: "missing",
-      // updateSnapshots: "all",
-      // ignoreSnapshots: true,
-      async createSnapshotPath(info) {
-        // use a unique path for each model
-        const t = getCurrentTest()!;
-        const mswPath = path.join(
-          t.suite!.name, // same directory as the test snapshot
-          "__msw_snapshots__",
-          t.suite!.suite!.name, // model / streaming params
-          t.name,
-        );
-        // in case there are multiple requests in a test, we need to use a separate snapshot for each request
-        fetchCountMap[mswPath] = (fetchCountMap[mswPath] || 0) + 1;
-        const hash = await createRequestHash(info.request);
-        return mswPath + `_${fetchCountMap[mswPath]}_${hash}.json`;
-      },
+      createSnapshotPath: createSnapshotPathFn(BASE_FILE_PATH),
       basePath: BASE_FILE_PATH,
-      // onFetchFromSnapshot(info, snapshot) {
-      // console.log("onFetchFromSnapshot", info, snapshot);
-      // },
-      // onFetchFromServer(info, snapshot) {
-      //   console.log("onFetchFromServer", info, snapshot);
-      // },
     }),
   );
 

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

@@ -1,7 +1,6 @@
 import { afterAll, afterEach, beforeAll, describe } from "vitest";
 
-import { getCurrentTest } from "@vitest/runner";
-import { getSortedEntries, snapshot, toHashString } from "msw-snapshot";
+import { snapshot } from "msw-snapshot";
 import { setupServer } from "msw/node";
 import path from "path";
 
@@ -9,6 +8,7 @@ import { ClientSideTransport } from "../../../streamTool/vercelAiSdk/clientside/
 import { testAIModels } from "../../../testUtil/testAIModels.js";
 import { aiDocumentFormats } from "../../index.js";
 import { generateSharedTestCases } from "../tests/sharedTestCases.js";
+import { createSnapshotPathFn } from "../tests/snapshotPath.js";
 
 const BASE_FILE_PATH = path.resolve(
   __dirname,
@@ -16,54 +16,14 @@ const BASE_FILE_PATH = path.resolve(
   path.basename(__filename),
 );
 
-const fetchCountMap: Record<string, number> = {};
-
-async function createRequestHash(req: Request) {
-  const url = new URL(req.url);
-  return [
-    // url.host,
-    // url.pathname,
-    toHashString([
-      req.method,
-      url.origin,
-      url.pathname,
-      getSortedEntries(url.searchParams),
-      getSortedEntries(req.headers),
-      // getSortedEntries(req.cookies),
-      new TextDecoder("utf-8").decode(await req.arrayBuffer()),
-    ]),
-  ].join("/");
-}
-
 // Main test suite with snapshot middleware
 describe("Models", () => {
   // Define server with snapshot middleware for the main tests
   const server = setupServer(
     snapshot({
       updateSnapshots: "missing",
-      // onSnapshotUpdated: "all",
-      // ignoreSnapshots: true,
-      async createSnapshotPath(info) {
-        // use a unique path for each model
-        const t = getCurrentTest()!;
-        const mswPath = path.join(
-          t.suite!.name, // same directory as the test snapshot
-          "__msw_snapshots__",
-          t.suite!.suite!.name, // model / streaming params
-          t.name,
-        );
-        // in case there are multiple requests in a test, we need to use a separate snapshot for each request
-        fetchCountMap[mswPath] = (fetchCountMap[mswPath] || 0) + 1;
-        const hash = await createRequestHash(info.request);
-        return mswPath + `_${fetchCountMap[mswPath]}_${hash}.json`;
-      },
+      createSnapshotPath: createSnapshotPathFn(BASE_FILE_PATH),
       basePath: BASE_FILE_PATH,
-      // onFetchFromSnapshot(info, snapshot) {
-      // console.log("onFetchFromSnapshot", info, snapshot);
-      // },
-      // onFetchFromServer(info, snapshot) {
-      //   console.log("onFetchFromServer", info, snapshot);
-      // },
     }),
   );
 

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

@@ -0,0 +1,33 @@
+import { getCurrentTest } from "@vitest/runner";
+import { getSortedEntries, toHashString } from "msw-snapshot";
+import path from "node:path";
+
+async function createRequestHash(req: Request): Promise<string> {
+  const url = new URL(req.url);
+  return toHashString([
+    req.method,
+    url.origin,
+    url.pathname,
+    getSortedEntries(url.searchParams),
+    getSortedEntries(req.headers),
+    new TextDecoder("utf-8").decode(await req.arrayBuffer()),
+  ]);
+}
+
+export function createSnapshotPathFn(_basePath: string) {
+  const fetchCountMap: Record<string, number> = {};
+
+  return async (info: { request: Request }): Promise<string> => {
+    const t = getCurrentTest()!;
+    const mswPath = path.join(
+      t.suite!.name,
+      "__msw_snapshots__",
+      t.suite!.suite!.name,
+      t.name,
+    );
+    fetchCountMap[mswPath] = (fetchCountMap[mswPath] || 0) + 1;
+    const counter = fetchCountMap[mswPath];
+    const hash = await createRequestHash(info.request);
+    return `${mswPath}_${counter}_${hash}.json`;
+  };
+}