fix

wax-prosemirror-services/src/AiService/AskAiContentService.js

@@ -15,19 +15,15 @@ class AskAiContentService extends Service {
     );
 
     const createOverlay = this.container.get('CreateOverlay');
-    const { config } = this;
+    const config = this.config;
 
     // Create the overlay
-    createOverlay(
-      AskAIOverlay,
-      { config },
-      {
-        nodeType: '',
-        markType: '',
-        followCursor: false,
-        selection: true,
-      },
-    );
+    createOverlay(AskAIOverlay, config, {
+      nodeType: '',
+      markType: '',
+      followCursor: false,
+      selection: true,
+    });
   }
 
   register() {

wax-prosemirror-services/src/AiService/ReplaceSelectedText.js

@@ -8,6 +8,7 @@ const elementFromString = string => {
 };
 
 const replaceSelectedText = (view, responseText, replace = false) => {
+  if (!view) return;
   let { state } = view;
   let { tr } = state;
   const { from, to } = tr.selection;

wax-prosemirror-services/src/AiService/components/AskAIOverlay.js

@@ -192,7 +192,7 @@ const AskAIOverlay = ({ setPosition, position, config }) => {
     }
   };
 
-  return options.AiOn ? (
+  return options?.AiOn ? (
     <Wrapper id="ai-overlay">
       <AskAIForm>
         <AskAIFormInput

wax-prosemirror-services/src/AiService/plugins/AskAiSelectionPlugin.js

@@ -14,10 +14,14 @@ export default () => {
         let createDecoration;
         const askAiInput = document.getElementById('askAiInput');
         if (askAiInput) {
-          // const selectionWhenBlured = tr.getMeta(key);
+          const selectionWhenBlured = tr.getMeta(key);
 
-          const from = newState.selection.from;
-          const to = newState.selection.to;
+          const from = selectionWhenBlured
+            ? selectionWhenBlured.from
+            : newState.selection.from;
+          const to = selectionWhenBlured
+            ? selectionWhenBlured.to
+            : newState.selection.to;
 
           createDecoration = DecorationSet.create(newState.doc, [
             Decoration.inline(from, to, {
@@ -38,7 +42,8 @@ export default () => {
       },
       handleDOMEvents: {
         blur(view) {
-          view.dispatch(view.state.tr.setMeta(key, view.state.selection));
+          if (view)
+            view.dispatch(view.state.tr.setMeta(key, view.state.selection));
         },
         //   focus(view) {
         //     view.dispatch(view.state.tr.setMeta(key, true));