fixup! ✨(frontend) integrate new Blocknote AI feature

Author: AntoLC

src/frontend/apps/e2e/__tests__/app-impress/doc-editor.spec.ts

@@ -391,6 +391,10 @@ test.describe('Doc Editor', () => {
     // Check Suggestion menu
     await page.locator('.bn-block-outer').last().fill('/');
     await expect(page.getByText('Write with AI')).toBeVisible();
+
+    // Reload the page to check that the AI change is still there
+    await page.goto(page.url());
+    await expect(editor.getByText('Bonjour le monde')).toBeVisible();
   });
 
   test(`it checks ai_proxy ability`, async ({ page, browserName }) => {

src/frontend/apps/impress/package.json

@@ -16,7 +16,7 @@
   },
   "dependencies": {
     "@ag-media/react-pdf-table": "2.0.3",
-    "@ai-sdk/openai": "1.3.22",
+    "@ai-sdk/openai-compatible": "0.2.14",
     "@blocknote/code-block": "0.32.0",
     "@blocknote/core": "0.32.0",
     "@blocknote/mantine": "0.32.0",

src/frontend/apps/impress/src/features/docs/doc-editor/components/AI/useAI.tsx

@@ -1,8 +1,8 @@
-import { createOpenAI } from '@ai-sdk/openai';
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
 import { createAIExtension, llmFormats } from '@blocknote/xl-ai';
 import { useMemo } from 'react';
 
-import { fetchAPI } from '@/api';
+import { baseApiUrl, fetchAPI } from '@/api';
 import { useConfig } from '@/core';
 import { Doc } from '@/docs/doc-management';
 
@@ -17,8 +17,9 @@ export const useAI = (docId: Doc['id'], aiAllowed: boolean) => {
       return;
     }
 
-    const openai = createOpenAI({
-      apiKey: '', // The API key will be set by the AI proxy
+    const openai = createOpenAICompatible({
+      name: 'AI Proxy',
+      baseURL: `${baseApiUrl('1.0')}documents/${docId}/ai-proxy/`, // Necessary for initialization..
       fetch: (input, init) => {
         // Create a new headers object without the Authorization header
         const headers = new Headers(init?.headers);
@@ -30,7 +31,8 @@ export const useAI = (docId: Doc['id'], aiAllowed: boolean) => {
         });
       },
     });
-    const model = openai.chat(conf.AI_MODEL);
+
+    const model = openai.chatModel(conf.AI_MODEL);
 
     const extension = createAIExtension({
       stream: conf.AI_STREAM,

src/frontend/apps/impress/src/features/docs/doc-editor/hook/useSaveDoc.tsx

@@ -33,11 +33,18 @@ const useSaveDoc = (docId: string, yDoc: Y.Doc, canSave: boolean) => {
     ) => {
       /**
        * When the AI edit the doc transaction.local is false,
-       * so we check if the origin is null to know if the change
-       * is local or not.
+       * so we check if the origin constructor to know where
+       * the transaction comes from.
+       *
        * TODO: see if we can get the local changes from the AI
        */
-      setIsLocalChange(transaction.local || transaction.origin === null);
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-member-access
+      const transactionOrigin = transaction?.origin?.constructor?.name;
+      const AI_ORIGIN_CONSTRUCTOR = 'ao';
+
+      setIsLocalChange(
+        transaction.local || transactionOrigin === AI_ORIGIN_CONSTRUCTOR,
+      );
     };
 
     yDoc.on('update', onUpdate);

src/frontend/yarn.lock

@@ -12,10 +12,10 @@
   resolved "https://registry.yarnpkg.com/@ag-media/react-pdf-table/-/react-pdf-table-2.0.3.tgz#113554b583b46e41a098cf64fecb5decd59ba004"
   integrity sha512-IscjfAOKwsyQok9YmzvuToe6GojN7J8hF0kb8C+K8qZX1DvhheGO+hRSAPxbv2nKMbSpvk7CIhSqJEkw++XVWg==
 
-"@ai-sdk/openai@1.3.22":
-  version "1.3.22"
-  resolved "https://registry.yarnpkg.com/@ai-sdk/openai/-/openai-1.3.22.tgz#ed52af8f8fb3909d108e945d12789397cb188b9b"
-  integrity sha512-QwA+2EkG0QyjVR+7h6FE7iOu2ivNqAVMm9UJZkVxxTk5OIq5fFJDTEI/zICEMuHImTTXR2JjsL6EirJ28Jc4cw==
+"@ai-sdk/openai[email protected]":
+  version "0.2.14"
+  resolved "https://registry.yarnpkg.com/@ai-sdk/openai-compatible/-/openai-compatible-0.2.14.tgz#f31e3dd1d767f3a44efaef2a0f0b2389d5c2b8a1"
+  integrity sha512-icjObfMCHKSIbywijaoLdZ1nSnuRnWgMEMLgwoxPJgxsUHMx0aVORnsLUid4SPtdhHI3X2masrt6iaEQLvOSFw==
   dependencies:
     "@ai-sdk/provider" "1.1.3"
     "@ai-sdk/provider-utils" "2.2.8"