From 323da6a2e3782197490cb6039441bf6ea9e1f628 Mon Sep 17 00:00:00 2001
From: Victor Mutai <victor.mutai@dillieduck.com>
Date: Tue, 13 Feb 2024 11:57:07 +0300
Subject: [PATCH] chore: aiOn from config

---
 wax-prosemirror-core/src/WaxContext.js                       | 2 +-
 .../src/AiService/components/AskAIOverlay.js                 | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/wax-prosemirror-core/src/WaxContext.js b/wax-prosemirror-core/src/WaxContext.js
index d97ec379c..4d3ee58f9 100644
--- a/wax-prosemirror-core/src/WaxContext.js
+++ b/wax-prosemirror-core/src/WaxContext.js
@@ -18,7 +18,7 @@ export default props => {
     pmViews: props.view || {},
     activeView: props.activeView || {},
     activeViewId: props.activeViewId || {},
-    options: { fullScreen: false },
+    options: { fullScreen: false, AiOn: false },
     transaction: {},
     setTransaction: tr => {
       Object.assign(context.transaction, tr);
diff --git a/wax-prosemirror-services/src/AiService/components/AskAIOverlay.js b/wax-prosemirror-services/src/AiService/components/AskAIOverlay.js
index 8fabe7623..5e5464b60 100644
--- a/wax-prosemirror-services/src/AiService/components/AskAIOverlay.js
+++ b/wax-prosemirror-services/src/AiService/components/AskAIOverlay.js
@@ -105,6 +105,7 @@ const AskAIOverlay = ({ setPosition, position, config }) => {
   const { t, i18n } = useTranslation();
   const {
     pmViews: { main },
+    options,
   } = useContext(WaxContext);
   const [result, setResult] = useState('');
   const [isSubmitted, setIsSubmitted] = useState(false);
@@ -134,7 +135,7 @@ const AskAIOverlay = ({ setPosition, position, config }) => {
     }
 
     setPosition({ ...position, left, top });
-  }, [position.left, AiOn]);
+  }, [position.left, options.AiOn, AiOn]);
 
   const tryAgain = () => {
     // Reset the state to initial values
@@ -197,7 +198,7 @@ const AskAIOverlay = ({ setPosition, position, config }) => {
     }
   };
 
-  return AiOn ? (
+  return options?.AiOn && AiOn ? (
     <Wrapper id="ai-overlay">
       <AskAIForm>
         <AskAIFormInput
-- 
GitLab