diff --git a/editors/demo/src/Editoria/config/config.js b/editors/demo/src/Editoria/config/config.js
index ba03fd62792a602fc1a5726076ca33a4ea11c153..4ad83143d299f6a9ae648b85ca997df11c45b037 100644
--- a/editors/demo/src/Editoria/config/config.js
+++ b/editors/demo/src/Editoria/config/config.js
@@ -138,6 +138,10 @@ export default {
       templateArea: 'BottomRightInfo',
       toolGroups: ['InfoToolGroup'],
     },
+    {
+      templateArea: 'ToggleAiToolBar',
+      toolGroups: ['ToggleAi'],
+    },
   ],
 
   // CommentsService: { readOnly: true },
diff --git a/editors/demo/src/Editoria/layout/EditoriaLayout.js b/editors/demo/src/Editoria/layout/EditoriaLayout.js
index 68b7e97be73b9c4c2427a3a91b907ccc4a2774b7..29fae5cb73c7b55bf663e9a6dc3c4f4cfcc53206 100644
--- a/editors/demo/src/Editoria/layout/EditoriaLayout.js
+++ b/editors/demo/src/Editoria/layout/EditoriaLayout.js
@@ -59,6 +59,15 @@ const Main = styled.div`
   height: 100%;
 `;
 
+const ToggleAiMenu = styled.div`
+  display: flex;
+  min-height: 20px;
+  user-select: none;
+  margin-left: auto;
+  margin-right: 30px;
+  margin-bottom: 10px;
+`;
+
 const TopMenu = styled.div`
   display: flex;
   min-height: 40px;
@@ -208,6 +217,7 @@ const getNotes = main => {
   return notes;
 };
 
+const ToggleAiToolBar = ComponentPlugin('ToggleAiToolBar');
 const LeftSideBar = ComponentPlugin('leftSideBar');
 const MainMenuToolBar = ComponentPlugin('mainMenuToolBar');
 const NotesArea = ComponentPlugin('notesArea');
@@ -260,6 +270,9 @@ const EditoriaLayout = ({ editor }) => {
   return (
     <ThemeProvider theme={cokoTheme}>
       <Wrapper style={fullScreenStyles} id="wax-container">
+        <ToggleAiMenu>
+          <ToggleAiToolBar />
+        </ToggleAiMenu>
         <TopMenu>
           <MainMenuToolBar />
         </TopMenu>
diff --git a/wax-prosemirror-services/package.json b/wax-prosemirror-services/package.json
index 24c94788faa01ca3f6bfe14105406cc77540507e..7341b65c7afd5ec00386558e03188d1ac74bab80 100644
--- a/wax-prosemirror-services/package.json
+++ b/wax-prosemirror-services/package.json
@@ -31,6 +31,7 @@
     "prosemirror-state": "1.4.2",
     "prosemirror-transform": "1.7.1",
     "prosemirror-view": "1.30.2",
+    "rc-switch": "^3.2.2",
     "react-dropdown": "^1.6.2",
     "use-deep-compare-effect": "^1.3.1",
     "uuid": "^7.0.3",