From 9797cc95bb04659b121ef4ed67f587c79ca38e13 Mon Sep 17 00:00:00 2001
From: Ilia Znamenskii <ilia@appsmith.com>
Date: Mon, 7 Oct 2024 13:35:04 +0200
Subject: [PATCH 1/6] remove fake message

---
 .../ui/wds/WDSAIChatWidget/widget/index.tsx   | 19 +------------------
 1 file changed, 1 insertion(+), 18 deletions(-)

diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
index f94fb5935f7..94f920ecfd3 100644
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
@@ -43,24 +43,7 @@ class WDSAIChatWidget extends BaseWidget<WDSAIChatWidgetProps, State> {
   static type = "WDS_AI_CHAT_WIDGET";
 
   state = {
-    messages: [
-      {
-        id: "1",
-        content: "Hello! How can I help you?",
-        role: "assistant" as const,
-      },
-      {
-        id: "2",
-        content: "Find stuck support requests",
-        role: "user" as const,
-      },
-      {
-        id: "3",
-        content:
-          "I'm finding these customer support requests that have been waiting for a response for over a day:",
-        role: "assistant" as const,
-      },
-    ],
+    messages: [],
     prompt: "",
     isWaitingForResponse: false,
   };

From 54509b9ab49f6cbb4ca77a847b7959314782d629 Mon Sep 17 00:00:00 2001
From: Ilia Znamenskii <ilia@appsmith.com>
Date: Mon, 7 Oct 2024 13:36:57 +0200
Subject: [PATCH 2/6] move AIChat component back to widgets

---
 app/client/packages/design-system/widgets/src/index.ts          | 1 -
 .../modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/index.ts  | 0
 .../ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/AIChat.tsx    | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/ChatTitle/ChatTitle.tsx  | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/ChatTitle/index.ts       | 0
 .../wds/WDSAIChatWidget}/AIChat/src/ChatTitle/styles.module.css | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/ChatTitle/types.ts       | 0
 .../WDSAIChatWidget}/AIChat/src/ThreadMessage/ThreadMessage.tsx | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/ThreadMessage/index.ts   | 0
 .../WDSAIChatWidget}/AIChat/src/ThreadMessage/styles.module.css | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/ThreadMessage/types.ts   | 0
 .../wds/WDSAIChatWidget}/AIChat/src/UserAvatar/UserAvatar.tsx   | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/UserAvatar/index.ts      | 0
 .../WDSAIChatWidget}/AIChat/src/UserAvatar/styles.module.css    | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/UserAvatar/types.ts      | 0
 .../ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/index.ts      | 0
 .../ui/wds/WDSAIChatWidget}/AIChat/src/styles.module.css        | 0
 .../ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/types.ts      | 0
 .../modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx  | 2 +-
 19 files changed, 1 insertion(+), 2 deletions(-)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/index.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/AIChat.tsx (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ChatTitle/ChatTitle.tsx (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ChatTitle/index.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ChatTitle/styles.module.css (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ChatTitle/types.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ThreadMessage/ThreadMessage.tsx (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ThreadMessage/index.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ThreadMessage/styles.module.css (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/ThreadMessage/types.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/UserAvatar/UserAvatar.tsx (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/UserAvatar/index.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/UserAvatar/styles.module.css (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/UserAvatar/types.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/index.ts (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/styles.module.css (100%)
 rename app/client/{packages/design-system/widgets/src/components => src/modules/ui-builder/ui/wds/WDSAIChatWidget}/AIChat/src/types.ts (100%)

diff --git a/app/client/packages/design-system/widgets/src/index.ts b/app/client/packages/design-system/widgets/src/index.ts
index 06aafcc9422..5f308407904 100644
--- a/app/client/packages/design-system/widgets/src/index.ts
+++ b/app/client/packages/design-system/widgets/src/index.ts
@@ -1,4 +1,3 @@
-export * from "./components/AIChat";
 export * from "./components/Icon";
 export * from "./components/Button";
 export * from "./components/IconButton";
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/index.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/index.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/index.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/AIChat.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/AIChat.tsx
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/AIChat.tsx
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/AIChat.tsx
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/ChatTitle.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/ChatTitle.tsx
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/ChatTitle.tsx
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/ChatTitle.tsx
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/index.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/index.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/index.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/styles.module.css
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/styles.module.css
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/styles.module.css
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/types.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ChatTitle/types.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/types.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/ThreadMessage.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/ThreadMessage.tsx
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/ThreadMessage.tsx
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/ThreadMessage.tsx
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/index.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/index.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/index.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/styles.module.css
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/styles.module.css
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/styles.module.css
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/types.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/ThreadMessage/types.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/types.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/UserAvatar.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/UserAvatar.tsx
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/UserAvatar.tsx
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/UserAvatar.tsx
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/index.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/index.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/index.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/styles.module.css
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/styles.module.css
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/styles.module.css
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/types.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/UserAvatar/types.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/types.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/index.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/index.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/index.ts
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/styles.module.css
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/styles.module.css
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/styles.module.css
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/src/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/types.ts
similarity index 100%
rename from app/client/packages/design-system/widgets/src/components/AIChat/src/types.ts
rename to app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/types.ts
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
index 94f920ecfd3..b79b883408d 100644
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
@@ -1,4 +1,3 @@
-import { AIChat, type ChatMessage } from "@appsmith/wds";
 import {
   EventType,
   type ExecutionResult,
@@ -15,6 +14,7 @@ import type { DerivedPropertiesMap } from "WidgetProvider/factory";
 import type { WidgetProps, WidgetState } from "widgets/BaseWidget";
 import BaseWidget from "widgets/BaseWidget";
 import type { ContainerWidgetProps } from "widgets/ContainerWidget/widget";
+import { AIChat, type ChatMessage } from "../AIChat";
 import {
   anvilConfig,
   autocompleteConfig,

From d19b3726a635ea59dc438dd71857b02039c03cd6 Mon Sep 17 00:00:00 2001
From: Ilia Znamenskii <ilia@appsmith.com>
Date: Mon, 7 Oct 2024 15:45:35 +0200
Subject: [PATCH 3/6] remove AIChat and move widget index to ce/ee

---
 app/client/src/{ => ce}/widgets/index.ts      | 126 ++++++-----
 app/client/src/ee/widgets/index.ts            |   1 +
 .../ui/wds/WDSAIChatWidget/AIChat/index.ts    |   1 -
 .../wds/WDSAIChatWidget/AIChat/src/AIChat.tsx |  81 --------
 .../AIChat/src/ChatTitle/ChatTitle.tsx        |  17 --
 .../AIChat/src/ChatTitle/index.ts             |   2 -
 .../AIChat/src/ChatTitle/styles.module.css    |  22 --
 .../AIChat/src/ChatTitle/types.ts             |   5 -
 .../src/ThreadMessage/ThreadMessage.tsx       |  64 ------
 .../AIChat/src/ThreadMessage/index.ts         |   2 -
 .../src/ThreadMessage/styles.module.css       |  27 ---
 .../AIChat/src/ThreadMessage/types.ts         |   7 -
 .../AIChat/src/UserAvatar/UserAvatar.tsx      |  27 ---
 .../AIChat/src/UserAvatar/index.ts            |   2 -
 .../AIChat/src/UserAvatar/styles.module.css   |  13 --
 .../AIChat/src/UserAvatar/types.ts            |   5 -
 .../wds/WDSAIChatWidget/AIChat/src/index.ts   |   2 -
 .../AIChat/src/styles.module.css              |  48 -----
 .../wds/WDSAIChatWidget/AIChat/src/types.ts   |  18 --
 .../ui/wds/WDSAIChatWidget/index.tsx          |   3 -
 .../widget/config/anvilConfig.ts              |  11 -
 .../widget/config/autocompleteConfig.ts       |   5 -
 .../widget/config/defaultConfig.ts            |  10 -
 .../WDSAIChatWidget/widget/config/index.ts    |  17 --
 .../widget/config/metaConfig.ts               |   9 -
 .../widget/config/methodsConfig.ts            |   6 -
 .../widget/config/propertyPaneContent.ts      | 140 -------------
 .../widget/config/propertyPaneStyle.ts        |   1 -
 .../ui/wds/WDSAIChatWidget/widget/index.tsx   | 196 ------------------
 app/client/src/utils/editor/EditorUtils.ts    |   2 +-
 .../src/utils/testPropertyPaneConfig.test.ts  |   2 +-
 .../workers/common/DataTreeEvaluator/test.ts  |   2 +-
 32 files changed, 66 insertions(+), 808 deletions(-)
 rename app/client/src/{ => ce}/widgets/index.ts (53%)
 create mode 100644 app/client/src/ee/widgets/index.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/index.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/AIChat.tsx
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/ChatTitle.tsx
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/index.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/styles.module.css
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/types.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/ThreadMessage.tsx
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/index.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/styles.module.css
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/types.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/UserAvatar.tsx
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/index.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/styles.module.css
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/types.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/index.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/styles.module.css
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/types.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/index.tsx
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/anvilConfig.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/autocompleteConfig.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/defaultConfig.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/index.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/metaConfig.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/methodsConfig.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneContent.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneStyle.ts
 delete mode 100644 app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx

diff --git a/app/client/src/widgets/index.ts b/app/client/src/ce/widgets/index.ts
similarity index 53%
rename from app/client/src/widgets/index.ts
rename to app/client/src/ce/widgets/index.ts
index 23ca19c6934..098751c5a10 100644
--- a/app/client/src/widgets/index.ts
+++ b/app/client/src/ce/widgets/index.ts
@@ -1,69 +1,68 @@
-import AudioRecorderWidget from "./AudioRecorderWidget";
-import AudioWidget from "./AudioWidget";
-import ButtonGroupWidget from "./ButtonGroupWidget";
-import ButtonWidget from "./ButtonWidget";
-import SelectWidget from "./SelectWidget";
-import CameraWidget from "./CameraWidget";
-import CanvasWidget from "./CanvasWidget";
-import ChartWidget from "./ChartWidget";
-import CheckboxGroupWidget from "./CheckboxGroupWidget";
-import CheckboxWidget from "./CheckboxWidget";
-import CircularProgressWidget from "./CircularProgressWidget";
-import ContainerWidget from "./ContainerWidget";
-import CurrencyInputWidget from "./CurrencyInputWidget";
-import DatePickerWidget from "./DatePickerWidget";
-import DatePickerWidget2 from "./DatePickerWidget2";
-import DividerWidget from "./DividerWidget";
-import MultiSelectWidgetV2 from "./MultiSelectWidgetV2";
-import DocumentViewerWidget from "./DocumentViewerWidget";
-import DropdownWidget from "./DropdownWidget";
-import FilePickerWidget from "./FilepickerWidget";
-import FilePickerWidgetV2 from "./FilePickerWidgetV2";
-import FormButtonWidget from "./FormButtonWidget";
-import FormWidget from "./FormWidget";
-import IconButtonWidget from "./IconButtonWidget";
-import IconWidget from "./IconWidget";
-import IframeWidget from "./IframeWidget";
-import ImageWidget from "./ImageWidget";
-import InputWidget from "./InputWidget";
-import InputWidgetV2 from "./InputWidgetV2";
-import ListWidget from "./ListWidget";
-import MapChartWidget from "./MapChartWidget";
-import MapWidget from "./MapWidget";
-import MenuButtonWidget from "./MenuButtonWidget";
-import ModalWidget from "./ModalWidget";
-import MultiSelectTreeWidget from "./MultiSelectTreeWidget";
-import MultiSelectWidget from "./MultiSelectWidget";
-import PhoneInputWidget from "./PhoneInputWidget";
-import ProgressBarWidget from "./ProgressBarWidget";
-import RadioGroupWidget from "./RadioGroupWidget";
-import RateWidget from "./RateWidget";
-import RichTextEditorWidget from "./RichTextEditorWidget";
-import SingleSelectTreeWidget from "./SingleSelectTreeWidget";
-import SkeletonWidget from "./SkeletonWidget";
-import StatboxWidget from "./StatboxWidget";
-import JSONFormWidget from "./JSONFormWidget";
-import SwitchGroupWidget from "./SwitchGroupWidget";
-import SwitchWidget from "./SwitchWidget";
-import TableWidget from "./TableWidget";
-import TabsMigratorWidget from "./TabsMigrator";
-import TabsWidget from "./TabsWidget";
-import TextWidget from "./TextWidget";
-import VideoWidget from "./VideoWidget";
-import ProgressWidget from "./ProgressWidget";
-import TableWidgetV2 from "./TableWidgetV2";
-import NumberSliderWidget from "./NumberSliderWidget";
-import RangeSliderWidget from "./RangeSliderWidget";
-import CategorySliderWidget from "./CategorySliderWidget";
-import CodeScannerWidget from "./CodeScannerWidget";
-import ListWidgetV2 from "./ListWidgetV2";
-import { WDSAIChatWidget } from "modules/ui-builder/ui/wds/WDSAIChatWidget";
+import AudioRecorderWidget from "../../widgets/AudioRecorderWidget";
+import AudioWidget from "../../widgets/AudioWidget";
+import ButtonGroupWidget from "../../widgets/ButtonGroupWidget";
+import ButtonWidget from "../../widgets/ButtonWidget";
+import SelectWidget from "../../widgets/SelectWidget";
+import CameraWidget from "../../widgets/CameraWidget";
+import CanvasWidget from "../../widgets/CanvasWidget";
+import ChartWidget from "../../widgets/ChartWidget";
+import CheckboxGroupWidget from "../../widgets/CheckboxGroupWidget";
+import CheckboxWidget from "../../widgets/CheckboxWidget";
+import CircularProgressWidget from "../../widgets/CircularProgressWidget";
+import ContainerWidget from "../../widgets/ContainerWidget";
+import CurrencyInputWidget from "../../widgets/CurrencyInputWidget";
+import DatePickerWidget from "../../widgets/DatePickerWidget";
+import DatePickerWidget2 from "../../widgets/DatePickerWidget2";
+import DividerWidget from "../../widgets/DividerWidget";
+import MultiSelectWidgetV2 from "../../widgets/MultiSelectWidgetV2";
+import DocumentViewerWidget from "../../widgets/DocumentViewerWidget";
+import DropdownWidget from "../../widgets/DropdownWidget";
+import FilePickerWidget from "../../widgets/FilepickerWidget";
+import FilePickerWidgetV2 from "../../widgets/FilePickerWidgetV2";
+import FormButtonWidget from "../../widgets/FormButtonWidget";
+import FormWidget from "../../widgets/FormWidget";
+import IconButtonWidget from "../../widgets/IconButtonWidget";
+import IconWidget from "../../widgets/IconWidget";
+import IframeWidget from "../../widgets/IframeWidget";
+import ImageWidget from "../../widgets/ImageWidget";
+import InputWidget from "../../widgets/InputWidget";
+import InputWidgetV2 from "../../widgets/InputWidgetV2";
+import ListWidget from "../../widgets/ListWidget";
+import MapChartWidget from "../../widgets/MapChartWidget";
+import MapWidget from "../../widgets/MapWidget";
+import MenuButtonWidget from "../../widgets/MenuButtonWidget";
+import ModalWidget from "../../widgets/ModalWidget";
+import MultiSelectTreeWidget from "../../widgets/MultiSelectTreeWidget";
+import MultiSelectWidget from "../../widgets/MultiSelectWidget";
+import PhoneInputWidget from "../../widgets/PhoneInputWidget";
+import ProgressBarWidget from "../../widgets/ProgressBarWidget";
+import RadioGroupWidget from "../../widgets/RadioGroupWidget";
+import RateWidget from "../../widgets/RateWidget";
+import RichTextEditorWidget from "../../widgets/RichTextEditorWidget";
+import SingleSelectTreeWidget from "../../widgets/SingleSelectTreeWidget";
+import SkeletonWidget from "../../widgets/SkeletonWidget";
+import StatboxWidget from "../../widgets/StatboxWidget";
+import JSONFormWidget from "../../widgets/JSONFormWidget";
+import SwitchGroupWidget from "../../widgets/SwitchGroupWidget";
+import SwitchWidget from "../../widgets/SwitchWidget";
+import TableWidget from "../../widgets/TableWidget";
+import TabsMigratorWidget from "../../widgets/TabsMigrator";
+import TabsWidget from "../../widgets/TabsWidget";
+import TextWidget from "../../widgets/TextWidget";
+import VideoWidget from "../../widgets/VideoWidget";
+import ProgressWidget from "../../widgets/ProgressWidget";
+import TableWidgetV2 from "../../widgets/TableWidgetV2";
+import NumberSliderWidget from "../../widgets/NumberSliderWidget";
+import RangeSliderWidget from "../../widgets/RangeSliderWidget";
+import CategorySliderWidget from "../../widgets/CategorySliderWidget";
+import CodeScannerWidget from "../../widgets/CodeScannerWidget";
+import ListWidgetV2 from "../../widgets/ListWidgetV2";
 import { WDSButtonWidget } from "modules/ui-builder/ui/wds/WDSButtonWidget";
 import { WDSInputWidget } from "modules/ui-builder/ui/wds/WDSInputWidget";
 import { WDSCheckboxWidget } from "modules/ui-builder/ui/wds/WDSCheckboxWidget";
 import { WDSIconButtonWidget } from "modules/ui-builder/ui/wds/WDSIconButtonWidget";
-import type BaseWidget from "./BaseWidget";
-import ExternalWidget from "./ExternalWidget";
+import type BaseWidget from "../../widgets/BaseWidget";
+import ExternalWidget from "../../widgets/ExternalWidget";
 import { WDSTableWidget } from "modules/ui-builder/ui/wds/WDSTableWidget";
 import { WDSCurrencyInputWidget } from "modules/ui-builder/ui/wds/WDSCurrencyInputWidget";
 import { WDSToolbarButtonsWidget } from "modules/ui-builder/ui/wds/WDSToolbarButtonsWidget";
@@ -74,7 +73,7 @@ import { WDSSwitchWidget } from "modules/ui-builder/ui/wds/WDSSwitchWidget";
 import { WDSSwitchGroupWidget } from "modules/ui-builder/ui/wds/WDSSwitchGroupWidget";
 import { WDSRadioGroupWidget } from "modules/ui-builder/ui/wds/WDSRadioGroupWidget";
 import { WDSMenuButtonWidget } from "modules/ui-builder/ui/wds/WDSMenuButtonWidget";
-import CustomWidget from "./CustomWidget";
+import CustomWidget from "../../widgets/CustomWidget";
 import { WDSSectionWidget } from "modules/ui-builder/ui/wds/WDSSectionWidget";
 import { WDSZoneWidget } from "modules/ui-builder/ui/wds/WDSZoneWidget";
 import { WDSHeadingWidget } from "modules/ui-builder/ui/wds/WDSHeadingWidget";
@@ -158,7 +157,6 @@ const DeprecatedWidgets = [
 
 const WDSWidgets = [
   // WDS Widgets
-  WDSAIChatWidget,
   WDSButtonWidget,
   WDSInputWidget,
   WDSCheckboxWidget,
diff --git a/app/client/src/ee/widgets/index.ts b/app/client/src/ee/widgets/index.ts
new file mode 100644
index 00000000000..ab94ad1cd24
--- /dev/null
+++ b/app/client/src/ee/widgets/index.ts
@@ -0,0 +1 @@
+export { default } from "ce/widgets";
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/index.ts
deleted file mode 100644
index 3bd16e178a0..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/index.ts
+++ /dev/null
@@ -1 +0,0 @@
-export * from "./src";
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/AIChat.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/AIChat.tsx
deleted file mode 100644
index d6975d9750a..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/AIChat.tsx
+++ /dev/null
@@ -1,81 +0,0 @@
-import { Button, Text, TextArea } from "@appsmith/wds";
-import type { FormEvent, ForwardedRef, KeyboardEvent } from "react";
-import React, { forwardRef, useCallback } from "react";
-import { ChatTitle } from "./ChatTitle";
-import styles from "./styles.module.css";
-import { ThreadMessage } from "./ThreadMessage";
-import type { AIChatProps, ChatMessage } from "./types";
-import { UserAvatar } from "./UserAvatar";
-
-const MIN_PROMPT_LENGTH = 3;
-
-const _AIChat = (props: AIChatProps, ref: ForwardedRef<HTMLDivElement>) => {
-  const {
-    // assistantName,
-    chatTitle,
-    isWaitingForResponse = false,
-    onPromptChange,
-    onSubmit,
-    prompt,
-    promptInputPlaceholder,
-    thread,
-    username,
-    ...rest
-  } = props;
-
-  const handleFormSubmit = useCallback(
-    (event: FormEvent<HTMLFormElement>) => {
-      event.preventDefault();
-      onSubmit?.();
-    },
-    [onSubmit],
-  );
-
-  const handlePromptInputKeyDown = useCallback(
-    (event: KeyboardEvent<HTMLTextAreaElement>) => {
-      if (event.key === "Enter" && event.shiftKey) {
-        event.preventDefault();
-        onSubmit?.();
-      }
-    },
-    [onSubmit],
-  );
-
-  return (
-    <div className={styles.root} ref={ref} {...rest}>
-      <div className={styles.header}>
-        <ChatTitle title={chatTitle} />
-
-        <div className={styles.username}>
-          <UserAvatar username={username} />
-          <Text data-testid="t--aichat-username" size="body">
-            {username}
-          </Text>
-        </div>
-      </div>
-
-      <ul className={styles.thread} data-testid="t--aichat-thread">
-        {thread.map((message: ChatMessage) => (
-          <ThreadMessage {...message} key={message.id} username={username} />
-        ))}
-      </ul>
-
-      <form className={styles.promptForm} onSubmit={handleFormSubmit}>
-        <TextArea
-          // TODO: Handle isWaitingForResponse: true state
-          isDisabled={isWaitingForResponse}
-          name="prompt"
-          onChange={onPromptChange}
-          onKeyDown={handlePromptInputKeyDown}
-          placeholder={promptInputPlaceholder}
-          value={prompt}
-        />
-        <Button isDisabled={prompt.length < MIN_PROMPT_LENGTH} type="submit">
-          Send
-        </Button>
-      </form>
-    </div>
-  );
-};
-
-export const AIChat = forwardRef(_AIChat);
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/ChatTitle.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/ChatTitle.tsx
deleted file mode 100644
index 2819d352314..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/ChatTitle.tsx
+++ /dev/null
@@ -1,17 +0,0 @@
-import { clsx } from "clsx";
-import React from "react";
-import styles from "./styles.module.css";
-import type { ChatTitleProps } from "./types";
-
-export const ChatTitle = ({ className, title, ...rest }: ChatTitleProps) => {
-  return (
-    <div
-      className={clsx(styles.root, className)}
-      {...rest}
-      data-testid="t--aichat-chat-title"
-    >
-      <div className={styles.logo} />
-      {title}
-    </div>
-  );
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/index.ts
deleted file mode 100644
index 78b61712de7..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/index.ts
+++ /dev/null
@@ -1,2 +0,0 @@
-export * from "./ChatTitle";
-export * from "./types";
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/styles.module.css
deleted file mode 100644
index a952c30dbf7..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/styles.module.css
+++ /dev/null
@@ -1,22 +0,0 @@
-.root {
-  display: flex;
-  gap: 12px;
-  align-items: center;
-  /* TODO: --type-title doesn't exists. Define it */
-  font-size: var(--type-title, 22.499px);
-  font-style: normal;
-  font-weight: 500;
-  /* TODO: --type-title-lineheight doesn't exists. Define it */
-  line-height: var(--type-title-lineheight, 31.7px);
-}
-
-.logo {
-  display: inline-block;
-  width: 48px;
-  min-width: 48px;
-  height: 48px;
-  border-radius: 8px;
-  /* TODO: --bd-neutral doesn't exists. Define it */
-  border: 1px solid var(--bd-neutral, #81858b);
-  background: #f8f8f8;
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/types.ts
deleted file mode 100644
index 220bbc4efa7..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ChatTitle/types.ts
+++ /dev/null
@@ -1,5 +0,0 @@
-import type { HTMLProps } from "react";
-
-export interface ChatTitleProps extends HTMLProps<HTMLDivElement> {
-  title?: string;
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/ThreadMessage.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/ThreadMessage.tsx
deleted file mode 100644
index 37ea1f1c792..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/ThreadMessage.tsx
+++ /dev/null
@@ -1,64 +0,0 @@
-import { Text } from "@appsmith/wds";
-import { clsx } from "clsx";
-import React from "react";
-import Markdown from "react-markdown";
-import SyntaxHighlighter from "react-syntax-highlighter";
-import { monokai } from "react-syntax-highlighter/dist/cjs/styles/hljs";
-import { UserAvatar } from "../UserAvatar";
-import styles from "./styles.module.css";
-import type { ThreadMessageProps } from "./types";
-
-export const ThreadMessage = ({
-  className,
-  content,
-  isAssistant,
-  username,
-  ...rest
-}: ThreadMessageProps) => {
-  return (
-    <li
-      className={clsx(styles.root, className)}
-      data-assistant={isAssistant}
-      {...rest}
-    >
-      {isAssistant ? (
-        <div>
-          <Text className={styles.content}>
-            <Markdown
-              // eslint-disable-next-line react-perf/jsx-no-new-object-as-prop
-              components={{
-                code(props) {
-                  const { children, className, ...rest } = props;
-                  const match = /language-(\w+)/.exec(className ?? "");
-
-                  return match ? (
-                    <SyntaxHighlighter
-                      PreTag="div"
-                      language={match[1]}
-                      style={monokai}
-                    >
-                      {String(children).replace(/\n$/, "")}
-                    </SyntaxHighlighter>
-                  ) : (
-                    <code {...rest} className={className}>
-                      {children}
-                    </code>
-                  );
-                },
-              }}
-            >
-              {content}
-            </Markdown>
-          </Text>
-        </div>
-      ) : (
-        <>
-          <UserAvatar className={styles.userAvatar} username={username} />
-          <div>
-            <Text className={styles.content}>{content}</Text>
-          </div>
-        </>
-      )}
-    </li>
-  );
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/index.ts
deleted file mode 100644
index 9916fbd0d56..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/index.ts
+++ /dev/null
@@ -1,2 +0,0 @@
-export * from "./ThreadMessage";
-export * from "./types";
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/styles.module.css
deleted file mode 100644
index e8081ecad00..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/styles.module.css
+++ /dev/null
@@ -1,27 +0,0 @@
-.root {
-  display: flex;
-  gap: 16px;
-  padding: 12px 0;
-}
-
-@container (min-width: 700px) {
-  .root {
-    padding: 24px 0;
-  }
-}
-
-.root[data-assistant="false"] {
-  flex-direction: row-reverse;
-}
-
-.root[data-assistant="false"] .sentTime {
-  text-align: right;
-}
-
-.sentTime {
-  margin: 0 0 8px;
-  /* TODO: --type-caption doesn't exists. Define it */
-  font-size: var(--type-caption, 12.247px);
-  /* TODO: --type-caption-lineheight doesn't exists. Define it */
-  line-height: var(--type-caption-lineheight, 17.25px);
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/types.ts
deleted file mode 100644
index 8935dfe1e5c..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/ThreadMessage/types.ts
+++ /dev/null
@@ -1,7 +0,0 @@
-import type { HTMLProps } from "react";
-
-export interface ThreadMessageProps extends HTMLProps<HTMLLIElement> {
-  content: string;
-  isAssistant: boolean;
-  username: string;
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/UserAvatar.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/UserAvatar.tsx
deleted file mode 100644
index 2c1667a4f22..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/UserAvatar.tsx
+++ /dev/null
@@ -1,27 +0,0 @@
-import { clsx } from "clsx";
-import React from "react";
-import styles from "./styles.module.css";
-import type { UserAvatarProps } from "./types";
-
-export const UserAvatar = ({
-  className,
-  username,
-  ...rest
-}: UserAvatarProps) => {
-  const getNameInitials = (username: string) => {
-    const names = username.split(" ");
-
-    // If there is only one name, return the first character of the name.
-    if (names.length === 1) {
-      return `${names[0].charAt(0)}`;
-    }
-
-    return `${names[0].charAt(0)}${names[1]?.charAt(0)}`;
-  };
-
-  return (
-    <span className={clsx(styles.root, className)} {...rest}>
-      {getNameInitials(username)}
-    </span>
-  );
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/index.ts
deleted file mode 100644
index ee08802f33b..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/index.ts
+++ /dev/null
@@ -1,2 +0,0 @@
-export * from "./UserAvatar";
-export * from "./types";
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/styles.module.css
deleted file mode 100644
index b11f3896efc..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/styles.module.css
+++ /dev/null
@@ -1,13 +0,0 @@
-.root {
-  display: inline-block;
-  width: 28px;
-  min-width: 28px;
-  height: 28px;
-  color: var(--bg-elevation-2, #fff);
-  text-align: center;
-  font-size: 14px;
-  font-weight: 500;
-  line-height: 28px;
-  border-radius: var(--inner-spacing-1, 4px);
-  background: #000;
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/types.ts
deleted file mode 100644
index 493da46824f..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/UserAvatar/types.ts
+++ /dev/null
@@ -1,5 +0,0 @@
-import type { HTMLProps } from "react";
-
-export interface UserAvatarProps extends HTMLProps<HTMLSpanElement> {
-  username: string;
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/index.ts
deleted file mode 100644
index a2b07d66147..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/index.ts
+++ /dev/null
@@ -1,2 +0,0 @@
-export * from "./types";
-export { AIChat } from "./AIChat";
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/styles.module.css b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/styles.module.css
deleted file mode 100644
index 3c0780cf620..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/styles.module.css
+++ /dev/null
@@ -1,48 +0,0 @@
-.root {
-  width: 100%;
-  border-radius: var(--border-radius-elevation-1);
-  border: 1px solid var(--color-bd-elevation-1);
-  /* TODO: --bg-elevation-1 doesn't exists. Define it */
-  background: var(--bg-elevation-1, #fbfcfd);
-}
-
-.header {
-  display: flex;
-  flex-direction: column;
-  gap: 8px;
-  padding: 22px 40px;
-  align-items: flex-start;
-  border-bottom: 1px solid var(--color-bd-elevation-1);
-  background: rgba(255, 255, 255, 0.45);
-}
-
-@container (min-width: 700px) {
-  .header {
-    flex-direction: row;
-    justify-content: space-between;
-    align-items: center;
-  }
-}
-
-.username {
-  display: flex;
-  align-items: center;
-  gap: 8px;
-}
-
-.thread {
-  display: flex;
-  flex-direction: column;
-  gap: 12px;
-  align-self: stretch;
-  padding: 0px 40px var(--inner-spacing-5) 40px;
-}
-
-.promptForm {
-  display: flex;
-  align-items: flex-start;
-  margin: var(--outer-spacing-4) 0 0 0;
-  padding: 0 var(--inner-spacing-5) var(--inner-spacing-5)
-    var(--inner-spacing-5);
-  gap: var(--outer-spacing-3);
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/types.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/types.ts
deleted file mode 100644
index 69572c5d6f8..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/AIChat/src/types.ts
+++ /dev/null
@@ -1,18 +0,0 @@
-export interface ChatMessage {
-  id: string;
-  content: string;
-  isAssistant: boolean;
-}
-
-export interface AIChatProps {
-  thread: ChatMessage[];
-  prompt: string;
-  username: string;
-  promptInputPlaceholder?: string;
-  chatTitle?: string;
-  chatDescription?: string;
-  assistantName?: string;
-  isWaitingForResponse?: boolean;
-  onPromptChange: (prompt: string) => void;
-  onSubmit?: () => void;
-}
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/index.tsx
deleted file mode 100644
index 99a42a9692a..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/index.tsx
+++ /dev/null
@@ -1,3 +0,0 @@
-import WDSAIChatWidget from "./widget";
-
-export { WDSAIChatWidget };
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/anvilConfig.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/anvilConfig.ts
deleted file mode 100644
index dc7fe21e103..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/anvilConfig.ts
+++ /dev/null
@@ -1,11 +0,0 @@
-import type { AnvilConfig } from "WidgetProvider/constants";
-
-export const anvilConfig: AnvilConfig = {
-  isLargeWidget: false,
-  widgetSize: {
-    minWidth: {
-      base: "100%",
-      "180px": "sizing-30",
-    },
-  },
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/autocompleteConfig.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/autocompleteConfig.ts
deleted file mode 100644
index 26579c80b83..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/autocompleteConfig.ts
+++ /dev/null
@@ -1,5 +0,0 @@
-import { DefaultAutocompleteDefinitions } from "widgets/WidgetUtils";
-
-export const autocompleteConfig = {
-  isVisible: DefaultAutocompleteDefinitions.isVisible,
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/defaultConfig.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/defaultConfig.ts
deleted file mode 100644
index 6d618658a41..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/defaultConfig.ts
+++ /dev/null
@@ -1,10 +0,0 @@
-import { ResponsiveBehavior } from "layoutSystems/common/utils/constants";
-import type { WidgetDefaultProps } from "WidgetProvider/constants";
-
-export const defaultsConfig = {
-  isVisible: true,
-  widgetName: "AIChat",
-  widgetType: "AI_CHAT",
-  version: 1,
-  responsiveBehavior: ResponsiveBehavior.Fill,
-} as unknown as WidgetDefaultProps;
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/index.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/index.ts
deleted file mode 100644
index fdf9289d5f4..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/index.ts
+++ /dev/null
@@ -1,17 +0,0 @@
-import { anvilConfig } from "./anvilConfig";
-import { metaConfig } from "./metaConfig";
-import { defaultsConfig } from "./defaultConfig";
-import { propertyPaneContent } from "./propertyPaneContent";
-import { propertyPaneStyle } from "./propertyPaneStyle";
-import { methodsConfig } from "./methodsConfig";
-import { autocompleteConfig } from "./autocompleteConfig";
-
-export {
-  anvilConfig,
-  metaConfig,
-  defaultsConfig,
-  propertyPaneContent,
-  propertyPaneStyle,
-  methodsConfig,
-  autocompleteConfig,
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/metaConfig.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/metaConfig.ts
deleted file mode 100644
index 1b94379578d..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/metaConfig.ts
+++ /dev/null
@@ -1,9 +0,0 @@
-import { WIDGET_TAGS } from "constants/WidgetConstants";
-import type { WidgetBaseConfiguration } from "WidgetProvider/constants";
-
-export const metaConfig: WidgetBaseConfiguration = {
-  name: "AIChat",
-  tags: [WIDGET_TAGS.CONTENT],
-  needsMeta: true,
-  searchTags: ["chat"],
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/methodsConfig.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/methodsConfig.ts
deleted file mode 100644
index 71de829ef47..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/methodsConfig.ts
+++ /dev/null
@@ -1,6 +0,0 @@
-import { AIChatIcon, AIChatThumbnail } from "appsmith-icons";
-
-export const methodsConfig = {
-  IconCmp: AIChatIcon,
-  ThumbnailCmp: AIChatThumbnail,
-};
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneContent.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneContent.ts
deleted file mode 100644
index ac4abc2c8b1..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneContent.ts
+++ /dev/null
@@ -1,140 +0,0 @@
-import { ValidationTypes } from "constants/WidgetValidation";
-
-export const propertyPaneContent = [
-  {
-    sectionName: "General",
-    children: [
-      {
-        helpText: "Select a query to submit when a chat message is sent.",
-        propertyName: "query",
-        label: "Query to trigger",
-        controlType: "INPUT_TEXT",
-        placeholderText: "Value",
-        isJSConvertible: false,
-        isBindProperty: false,
-        isTriggerProperty: false,
-        dependencies: ["queryData", "queryRun"],
-        updateHook: (
-          _props: unknown,
-          propertyPath: string,
-          propertyValue: string,
-        ) => {
-          const propertiesToUpdate = [{ propertyPath, propertyValue }];
-
-          propertiesToUpdate.push({
-            propertyPath: "queryData",
-            propertyValue: `{{${propertyValue}.data}}`,
-          });
-          propertiesToUpdate.push({
-            propertyPath: "queryRun",
-            propertyValue: propertyValue,
-          });
-
-          return propertiesToUpdate;
-        },
-      },
-      {
-        helpText: "Adds a header to the chat",
-        propertyName: "chatTitle",
-        label: "Title",
-        controlType: "INPUT_TEXT",
-        isJSConvertible: false,
-        isBindProperty: false,
-        isTriggerProperty: false,
-        validation: { type: ValidationTypes.TEXT },
-        defaultValue: "",
-      },
-      {
-        helpText:
-          "Adds a description to help users understand how to use the chat",
-        propertyName: "chatDescription",
-        label: "Description",
-        controlType: "INPUT_TEXT",
-        isJSConvertible: false,
-        isBindProperty: false,
-        isTriggerProperty: false,
-        validation: { type: ValidationTypes.TEXT },
-        defaultValue: "",
-      },
-      {
-        helpText: "Adds a placeholder text to the message input box",
-        propertyName: "promptInputPlaceholder",
-        label: "Placeholder",
-        controlType: "INPUT_TEXT",
-        isJSConvertible: false,
-        isBindProperty: false,
-        isTriggerProperty: false,
-        validation: { type: ValidationTypes.TEXT },
-        defaultValue: "",
-      },
-      {
-        helpText: "Gives the open AI Assistant a name to be displayed in chat",
-        propertyName: "assistantName",
-        label: "Assistant Name",
-        controlType: "INPUT_TEXT",
-        isJSConvertible: false,
-        isBindProperty: false,
-        isTriggerProperty: false,
-        validation: { type: ValidationTypes.TEXT },
-        defaultValue: "",
-      },
-      {
-        helpText: "Configures a prompt for the assistant",
-        propertyName: "systemPrompt",
-        label: "Prompt",
-        controlType: "INPUT_TEXT",
-        isJSConvertible: false,
-        isBindProperty: false,
-        isTriggerProperty: false,
-        validation: { type: ValidationTypes.TEXT },
-        defaultValue: "",
-      },
-      {
-        helpText: "Controls the visibility of the widget",
-        propertyName: "isVisible",
-        label: "Visible",
-        controlType: "SWITCH",
-        isJSConvertible: true,
-        isBindProperty: true,
-        isTriggerProperty: false,
-        validation: { type: ValidationTypes.BOOLEAN },
-        defaultValue: true,
-      },
-    ],
-  },
-  {
-    sectionName: "Hidden props",
-    children: [
-      {
-        propertyName: "queryData",
-        label: "",
-        controlType: "INPUT_TEXT",
-        isJSConvertible: true,
-        isBindProperty: true,
-        isTriggerProperty: false,
-        hidden: () => true,
-      },
-      {
-        propertyName: "queryRun",
-        label: "",
-        controlType: "INPUT_TEXT",
-        isJSConvertible: true,
-        isBindProperty: true,
-        isTriggerProperty: false,
-        hidden: () => true,
-      },
-      // Fake hidden prop to pass the username to the widget
-      {
-        propertyName: "username",
-        label: "",
-        controlType: "INPUT_TEXT",
-        defaultValue: "{{appsmith.user.username}}",
-        isJSConvertible: true,
-        isBindProperty: true,
-        isTriggerProperty: false,
-        hidden: () => true,
-        invisible: true,
-      },
-    ],
-  },
-];
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneStyle.ts b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneStyle.ts
deleted file mode 100644
index 090b16597cf..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/config/propertyPaneStyle.ts
+++ /dev/null
@@ -1 +0,0 @@
-export const propertyPaneStyle = [];
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
deleted file mode 100644
index b79b883408d..00000000000
--- a/app/client/src/modules/ui-builder/ui/wds/WDSAIChatWidget/widget/index.tsx
+++ /dev/null
@@ -1,196 +0,0 @@
-import {
-  EventType,
-  type ExecutionResult,
-} from "constants/AppsmithActionConstants/ActionConstants";
-import type { SetterConfig, Stylesheet } from "entities/AppTheming";
-import React, { type FormEvent, type ReactNode } from "react";
-import type {
-  AnvilConfig,
-  AutocompletionDefinitions,
-  WidgetBaseConfiguration,
-  WidgetDefaultProps,
-} from "WidgetProvider/constants";
-import type { DerivedPropertiesMap } from "WidgetProvider/factory";
-import type { WidgetProps, WidgetState } from "widgets/BaseWidget";
-import BaseWidget from "widgets/BaseWidget";
-import type { ContainerWidgetProps } from "widgets/ContainerWidget/widget";
-import { AIChat, type ChatMessage } from "../AIChat";
-import {
-  anvilConfig,
-  autocompleteConfig,
-  defaultsConfig,
-  metaConfig,
-  methodsConfig,
-  propertyPaneContent,
-  propertyPaneStyle,
-} from "./config";
-
-export interface WDSAIChatWidgetProps
-  extends ContainerWidgetProps<WidgetProps> {}
-export interface Message {
-  id: string;
-  content: string;
-  role: "assistant" | "user" | "system";
-}
-
-interface State extends WidgetState {
-  messages: Message[];
-  prompt: string;
-  isWaitingForResponse: boolean;
-}
-
-class WDSAIChatWidget extends BaseWidget<WDSAIChatWidgetProps, State> {
-  static type = "WDS_AI_CHAT_WIDGET";
-
-  state = {
-    messages: [],
-    prompt: "",
-    isWaitingForResponse: false,
-  };
-
-  static getConfig(): WidgetBaseConfiguration {
-    return metaConfig;
-  }
-
-  static getDefaults(): WidgetDefaultProps {
-    return defaultsConfig;
-  }
-
-  static getPropertyPaneConfig() {
-    return [];
-  }
-  static getPropertyPaneContentConfig() {
-    return propertyPaneContent;
-  }
-
-  static getPropertyPaneStyleConfig() {
-    return propertyPaneStyle;
-  }
-
-  static getMethods() {
-    return methodsConfig;
-  }
-
-  static getAutocompleteDefinitions(): AutocompletionDefinitions {
-    return autocompleteConfig;
-  }
-
-  static getSetterConfig(): SetterConfig | null {
-    return {
-      __setters: {
-        setVisibility: {
-          path: "isVisible",
-          type: "boolean",
-        },
-      },
-    };
-  }
-
-  static getDerivedPropertiesMap(): DerivedPropertiesMap {
-    return {};
-  }
-
-  static getDefaultPropertiesMap(): Record<string, string> {
-    return {};
-  }
-
-  static getMetaPropertiesMap(): Record<string, unknown> {
-    return {};
-  }
-
-  static getAnvilConfig(): AnvilConfig | null {
-    return anvilConfig;
-  }
-
-  static getStylesheetConfig(): Stylesheet {
-    return {};
-  }
-
-  adaptMessages(messages: Message[]): ChatMessage[] {
-    return messages.map((message) => ({
-      ...message,
-      isAssistant: message.role === "assistant",
-    }));
-  }
-
-  handleMessageSubmit = (event?: FormEvent<HTMLFormElement>) => {
-    event?.preventDefault();
-
-    this.setState(
-      (state) => ({
-        messages: [
-          ...state.messages,
-          {
-            id: String(Date.now()),
-            content: this.state.prompt,
-            role: "user",
-          },
-        ],
-        prompt: "",
-        isWaitingForResponse: true,
-      }),
-      () => {
-        const messages: Message[] = [...this.state.messages];
-
-        if (this.props.systemPrompt) {
-          messages.unshift({
-            id: String(Date.now()),
-            content: this.props.systemPrompt,
-            role: "system",
-          });
-        }
-
-        const params = {
-          messages,
-        };
-
-        this.executeAction({
-          triggerPropertyName: "onClick",
-          dynamicString: `{{${this.props.queryRun}.run(${JSON.stringify(params)})}}`,
-          event: {
-            type: EventType.ON_CLICK,
-            callback: this.handleActionComplete,
-          },
-        });
-      },
-    );
-  };
-
-  handleActionComplete = (result: ExecutionResult) => {
-    if (result.success) {
-      this.setState((state) => ({
-        messages: [
-          ...state.messages,
-          {
-            id: Math.random().toString(),
-            content: this.props.queryData.choices[0].message.content,
-            role: "assistant",
-          },
-        ],
-        isWaitingForResponse: false,
-      }));
-    }
-  };
-
-  handlePromptChange = (prompt: string) => {
-    this.setState({ prompt });
-  };
-
-  getWidgetView(): ReactNode {
-    return (
-      <AIChat
-        assistantName={this.props.assistantName}
-        chatTitle={this.props.chatTitle}
-        isWaitingForResponse={this.state.isWaitingForResponse}
-        onPromptChange={this.handlePromptChange}
-        onSubmit={this.handleMessageSubmit}
-        prompt={this.state.prompt}
-        promptInputPlaceholder={this.props.promptInputPlaceholder}
-        thread={this.adaptMessages(this.state.messages)}
-        username={this.props.username || ""}
-      />
-    );
-  }
-}
-
-export default WDSAIChatWidget;
diff --git a/app/client/src/utils/editor/EditorUtils.ts b/app/client/src/utils/editor/EditorUtils.ts
index 52ce8faa1ad..fa8cd8568e6 100644
--- a/app/client/src/utils/editor/EditorUtils.ts
+++ b/app/client/src/utils/editor/EditorUtils.ts
@@ -3,7 +3,7 @@ import PropertyControlRegistry from "../PropertyControlRegistry";
 // import Widgets from "widgets";
 import { registerWidgets } from "WidgetProvider/factory/registrationHelper";
 import { registerLayoutComponents } from "layoutSystems/anvil/utils/layouts/layoutUtils";
-import widgets from "widgets";
+import widgets from "ee/widgets";
 
 export const registerEditorWidgets = () => {
   registerWidgets(widgets);
diff --git a/app/client/src/utils/testPropertyPaneConfig.test.ts b/app/client/src/utils/testPropertyPaneConfig.test.ts
index 21cfb9b4627..c9bf0f77924 100644
--- a/app/client/src/utils/testPropertyPaneConfig.test.ts
+++ b/app/client/src/utils/testPropertyPaneConfig.test.ts
@@ -6,7 +6,7 @@ import type {
 } from "constants/PropertyControlConstants";
 import { ValidationTypes } from "constants/WidgetValidation";
 import { isFunction } from "lodash";
-import widgets from "widgets";
+import widgets from "ee/widgets";
 import WidgetFactory from "WidgetProvider/factory";
 import { registerWidgets } from "WidgetProvider/factory/registrationHelper";
 
diff --git a/app/client/src/workers/common/DataTreeEvaluator/test.ts b/app/client/src/workers/common/DataTreeEvaluator/test.ts
index f71e77021ea..33965ba1f25 100644
--- a/app/client/src/workers/common/DataTreeEvaluator/test.ts
+++ b/app/client/src/workers/common/DataTreeEvaluator/test.ts
@@ -14,7 +14,7 @@ import {
 import { updateDependencyMap } from "workers/common/DependencyMap";
 import { replaceThisDotParams } from "./utils";
 import { isDataField } from "./utils";
-import widgets from "widgets";
+import widgets from "ee/widgets";
 import type { WidgetConfiguration } from "WidgetProvider/constants";
 import {
   EvaluationSubstitutionType,

From 41f22a21c55f78abdc38039cdc702ea27ea33711 Mon Sep 17 00:00:00 2001
From: Ilia Znamenskii <ilia@appsmith.com>
Date: Tue, 8 Oct 2024 12:55:46 +0200
Subject: [PATCH 4/6] move wds constants to ee

---
 app/client/cypress/support/Pages/Anvil/Locators/index.ts     | 2 +-
 app/client/src/WidgetProvider/factory/helpers.ts             | 2 +-
 .../src/{ => ce}/modules/ui-builder/ui/wds/constants.ts      | 2 +-
 app/client/src/ee/modules/ui-builder/ui/wds/constants.ts     | 1 +
 .../src/layoutSystems/anvil/common/AnvilFlexComponent.tsx    | 2 +-
 .../anvil/editor/canvasArenas/hooks/useAnvilWidgetDrop.ts    | 2 +-
 .../layoutSystems/anvil/editor/canvasArenas/utils/utils.ts   | 2 +-
 .../layoutSystems/anvil/editor/hooks/useAnvilWidgetStyles.ts | 2 +-
 .../integrations/sagas/anvilWidgetAdditionSagas/index.ts     | 2 +-
 .../anvil/layoutComponents/components/section/index.tsx      | 2 +-
 app/client/src/layoutSystems/anvil/utils/anvilChecksUtils.ts | 2 +-
 app/client/src/layoutSystems/anvil/utils/constants.ts        | 2 +-
 .../anvil/utils/layouts/update/mainCanvasLayoutUtils.ts      | 2 +-
 .../layoutSystems/anvil/utils/layouts/update/sectionUtils.ts | 2 +-
 .../layoutSystems/anvil/utils/layouts/update/zoneUtils.ts    | 2 +-
 .../src/layoutSystems/anvil/utils/paste/sectionPasteUtils.ts | 2 +-
 .../src/layoutSystems/anvil/utils/sectionOperationUtils.ts   | 2 +-
 app/client/src/modules/ui-builder/ui/wds/Container.tsx       | 2 +-
 .../ui-builder/ui/wds/WDSModalWidget/widget/index.tsx        | 4 ++--
 .../ui-builder/ui/wds/WDSSectionWidget/widget/index.tsx      | 5 ++++-
 .../ui-builder/ui/wds/WDSStatsWidget/widget/index.tsx        | 2 +-
 .../modules/ui-builder/ui/wds/WDSZoneWidget/widget/index.tsx | 5 ++++-
 app/client/src/pages/Editor/QueryEditor/BindDataButton.tsx   | 2 +-
 app/client/src/selectors/editorSelectors.tsx                 | 2 +-
 24 files changed, 31 insertions(+), 24 deletions(-)
 rename app/client/src/{ => ce}/modules/ui-builder/ui/wds/constants.ts (97%)
 create mode 100644 app/client/src/ee/modules/ui-builder/ui/wds/constants.ts

diff --git a/app/client/cypress/support/Pages/Anvil/Locators/index.ts b/app/client/cypress/support/Pages/Anvil/Locators/index.ts
index 76f837e73c7..c671083d285 100644
--- a/app/client/cypress/support/Pages/Anvil/Locators/index.ts
+++ b/app/client/cypress/support/Pages/Anvil/Locators/index.ts
@@ -1,6 +1,6 @@
 import { MAIN_CONTAINER_WIDGET_ID } from "../../../../../src/constants/WidgetConstants";
 import { getAnvilCanvasId } from "../../../../../src/layoutSystems/anvil/viewer/canvas/utils";
-import { AnvilDataAttributes } from "../../../../../src/modules/ui-builder/ui/wds/constants";
+import { AnvilDataAttributes } from "../../../../../src/ee/modules/ui-builder/ui/wds/constants";
 
 // anvil widget based selectors
 const anvilWidgetSelector = "[data-testid=t--anvil-widget-wrapper]";
diff --git a/app/client/src/WidgetProvider/factory/helpers.ts b/app/client/src/WidgetProvider/factory/helpers.ts
index 28731d7bc97..8a0b84c0350 100644
--- a/app/client/src/WidgetProvider/factory/helpers.ts
+++ b/app/client/src/WidgetProvider/factory/helpers.ts
@@ -17,7 +17,7 @@ import {
   WidgetFeaturePropertyPaneEnhancements,
 } from "../../utils/WidgetFeatures";
 import { generateReactKey } from "utils/generators";
-import { DEFAULT_WIDGET_ON_CANVAS_UI } from "modules/ui-builder/ui/wds/constants";
+import { DEFAULT_WIDGET_ON_CANVAS_UI } from "ee/modules/ui-builder/ui/wds/constants";
 import type { WidgetDefaultProps } from "WidgetProvider/constants";
 
 export enum PropertyPaneConfigTypes {
diff --git a/app/client/src/modules/ui-builder/ui/wds/constants.ts b/app/client/src/ce/modules/ui-builder/ui/wds/constants.ts
similarity index 97%
rename from app/client/src/modules/ui-builder/ui/wds/constants.ts
rename to app/client/src/ce/modules/ui-builder/ui/wds/constants.ts
index f769115a3c6..45f141e139c 100644
--- a/app/client/src/modules/ui-builder/ui/wds/constants.ts
+++ b/app/client/src/ce/modules/ui-builder/ui/wds/constants.ts
@@ -1,4 +1,4 @@
-import { getCanvasClassName } from "../../../../utils/generators";
+import { getCanvasClassName } from "../../../../../utils/generators";
 
 export const anvilWidgets = {
   SECTION_WIDGET: "SECTION_WIDGET",
diff --git a/app/client/src/ee/modules/ui-builder/ui/wds/constants.ts b/app/client/src/ee/modules/ui-builder/ui/wds/constants.ts
new file mode 100644
index 00000000000..555937febbc
--- /dev/null
+++ b/app/client/src/ee/modules/ui-builder/ui/wds/constants.ts
@@ -0,0 +1 @@
+export * from "ce/modules/ui-builder/ui/wds/constants";
diff --git a/app/client/src/layoutSystems/anvil/common/AnvilFlexComponent.tsx b/app/client/src/layoutSystems/anvil/common/AnvilFlexComponent.tsx
index b59ee2bf8b3..9ee17af37b3 100644
--- a/app/client/src/layoutSystems/anvil/common/AnvilFlexComponent.tsx
+++ b/app/client/src/layoutSystems/anvil/common/AnvilFlexComponent.tsx
@@ -15,7 +15,7 @@ import { Layers } from "constants/Layers";
 import { noop } from "utils/AppsmithUtils";
 import { convertFlexGrowToFlexBasis } from "../sectionSpaceDistributor/utils/spaceDistributionEditorUtils";
 import styles from "./styles.module.css";
-import { AnvilDataAttributes } from "modules/ui-builder/ui/wds/constants";
+import { AnvilDataAttributes } from "ee/modules/ui-builder/ui/wds/constants";
 
 const anvilWidgetStyleProps: CSSProperties = {
   position: "relative",
diff --git a/app/client/src/layoutSystems/anvil/editor/canvasArenas/hooks/useAnvilWidgetDrop.ts b/app/client/src/layoutSystems/anvil/editor/canvasArenas/hooks/useAnvilWidgetDrop.ts
index 9e36ffe8836..6ae68b88863 100644
--- a/app/client/src/layoutSystems/anvil/editor/canvasArenas/hooks/useAnvilWidgetDrop.ts
+++ b/app/client/src/layoutSystems/anvil/editor/canvasArenas/hooks/useAnvilWidgetDrop.ts
@@ -7,7 +7,7 @@ import type { AnvilHighlightInfo } from "layoutSystems/anvil/utils/anvilTypes";
 import { useCallback } from "react";
 import { useDispatch } from "react-redux";
 import type { AnvilDnDListenerStates } from "./useAnvilDnDListenerStates";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 
 export const useAnvilWidgetDrop = (
   canvasId: string,
diff --git a/app/client/src/layoutSystems/anvil/editor/canvasArenas/utils/utils.ts b/app/client/src/layoutSystems/anvil/editor/canvasArenas/utils/utils.ts
index f5e615b6f60..aad08a9e474 100644
--- a/app/client/src/layoutSystems/anvil/editor/canvasArenas/utils/utils.ts
+++ b/app/client/src/layoutSystems/anvil/editor/canvasArenas/utils/utils.ts
@@ -7,7 +7,7 @@ import WidgetFactory from "WidgetProvider/factory";
 import type { CanvasWidgetsReduxState } from "reducers/entityReducers/canvasWidgetsReducer";
 import type { DragDetails } from "reducers/uiReducers/dragResizeReducer";
 import { AnvilDraggedWidgetTypesEnum } from "../types";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import { HIGHLIGHT_SIZE } from "layoutSystems/anvil/utils/constants";
 import { getWidgetHierarchy } from "layoutSystems/anvil/utils/paste/utils";
 
diff --git a/app/client/src/layoutSystems/anvil/editor/hooks/useAnvilWidgetStyles.ts b/app/client/src/layoutSystems/anvil/editor/hooks/useAnvilWidgetStyles.ts
index 80bf9d9e235..4efc175eb4d 100644
--- a/app/client/src/layoutSystems/anvil/editor/hooks/useAnvilWidgetStyles.ts
+++ b/app/client/src/layoutSystems/anvil/editor/hooks/useAnvilWidgetStyles.ts
@@ -4,7 +4,7 @@ import { useSelector } from "react-redux";
 import { useWidgetBorderStyles } from "layoutSystems/anvil/common/hooks/useWidgetBorderStyles";
 import type { AppState } from "ee/reducers";
 import { getIsNewWidgetBeingDragged } from "sagas/selectors";
-import { AnvilDataAttributes } from "modules/ui-builder/ui/wds/constants";
+import { AnvilDataAttributes } from "ee/modules/ui-builder/ui/wds/constants";
 
 export const useAnvilWidgetStyles = (
   widgetId: string,
diff --git a/app/client/src/layoutSystems/anvil/integrations/sagas/anvilWidgetAdditionSagas/index.ts b/app/client/src/layoutSystems/anvil/integrations/sagas/anvilWidgetAdditionSagas/index.ts
index c121a458358..a0e8f829735 100644
--- a/app/client/src/layoutSystems/anvil/integrations/sagas/anvilWidgetAdditionSagas/index.ts
+++ b/app/client/src/layoutSystems/anvil/integrations/sagas/anvilWidgetAdditionSagas/index.ts
@@ -11,7 +11,7 @@ import {
   type ReduxAction,
 } from "ee/constants/ReduxActionConstants";
 import type { WidgetProps } from "widgets/BaseWidget";
-import { WDS_V2_WIDGET_MAP } from "modules/ui-builder/ui/wds/constants";
+import { WDS_V2_WIDGET_MAP } from "ee/modules/ui-builder/ui/wds/constants";
 import { MAIN_CONTAINER_WIDGET_ID } from "constants/WidgetConstants";
 import { getMainCanvasLastRowHighlight } from "../anvilDraggingSagas/helpers";
 import { updateAndSaveAnvilLayout } from "layoutSystems/anvil/utils/anvilChecksUtils";
diff --git a/app/client/src/layoutSystems/anvil/layoutComponents/components/section/index.tsx b/app/client/src/layoutSystems/anvil/layoutComponents/components/section/index.tsx
index 4b30af0c6b3..e65215286ea 100644
--- a/app/client/src/layoutSystems/anvil/layoutComponents/components/section/index.tsx
+++ b/app/client/src/layoutSystems/anvil/layoutComponents/components/section/index.tsx
@@ -7,7 +7,7 @@ import {
 import WidgetRow from "../WidgetRow";
 import { SectionRow } from "./SectionRow";
 import type { FlexLayoutProps } from "../FlexLayout";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import { SectionSpaceDistributor } from "layoutSystems/anvil/sectionSpaceDistributor/SectionSpaceDistributor";
 
 class Section extends WidgetRow {
diff --git a/app/client/src/layoutSystems/anvil/utils/anvilChecksUtils.ts b/app/client/src/layoutSystems/anvil/utils/anvilChecksUtils.ts
index 00bb19299d9..a05f6aa9e41 100644
--- a/app/client/src/layoutSystems/anvil/utils/anvilChecksUtils.ts
+++ b/app/client/src/layoutSystems/anvil/utils/anvilChecksUtils.ts
@@ -4,7 +4,7 @@ import { updateAnvilParentPostWidgetDeletion } from "layoutSystems/anvil/utils/l
 import type { FlattenedWidgetProps } from "WidgetProvider/constants";
 import { MAIN_CONTAINER_WIDGET_ID } from "constants/WidgetConstants";
 import type { CanvasWidgetsReduxState } from "reducers/entityReducers/canvasWidgetsReducer";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import {
   updateSectionWithDefaultSpaceDistribution,
   updateSectionsDistributedSpace,
diff --git a/app/client/src/layoutSystems/anvil/utils/constants.ts b/app/client/src/layoutSystems/anvil/utils/constants.ts
index e00c88b9fd8..7fbde4b4fd1 100644
--- a/app/client/src/layoutSystems/anvil/utils/constants.ts
+++ b/app/client/src/layoutSystems/anvil/utils/constants.ts
@@ -1,6 +1,6 @@
 import { FlexLayerAlignment } from "layoutSystems/common/utils/constants";
 import type { HighlightPayload, HighlightRenderInfo } from "./anvilTypes";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 
 export const MOBILE_BREAKPOINT = 480;
 
diff --git a/app/client/src/layoutSystems/anvil/utils/layouts/update/mainCanvasLayoutUtils.ts b/app/client/src/layoutSystems/anvil/utils/layouts/update/mainCanvasLayoutUtils.ts
index 0b389c9e68f..81e7519321c 100644
--- a/app/client/src/layoutSystems/anvil/utils/layouts/update/mainCanvasLayoutUtils.ts
+++ b/app/client/src/layoutSystems/anvil/utils/layouts/update/mainCanvasLayoutUtils.ts
@@ -11,7 +11,7 @@ import LayoutFactory from "layoutSystems/anvil/layoutComponents/LayoutFactory";
 import { call } from "redux-saga/effects";
 
 import { severTiesFromParents, transformMovedWidgets } from "./moveUtils";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import { MAIN_CONTAINER_WIDGET_ID } from "constants/WidgetConstants";
 import { addNewAnvilWidgetToDSL } from "layoutSystems/anvil/integrations/sagas/anvilWidgetAdditionSagas/helpers";
 
diff --git a/app/client/src/layoutSystems/anvil/utils/layouts/update/sectionUtils.ts b/app/client/src/layoutSystems/anvil/utils/layouts/update/sectionUtils.ts
index 9a4f5df405e..0a13fda47e3 100644
--- a/app/client/src/layoutSystems/anvil/utils/layouts/update/sectionUtils.ts
+++ b/app/client/src/layoutSystems/anvil/utils/layouts/update/sectionUtils.ts
@@ -13,7 +13,7 @@ import type { CanvasWidgetsReduxState } from "reducers/entityReducers/canvasWidg
 import { call } from "redux-saga/effects";
 import { severTiesFromParents, transformMovedWidgets } from "./moveUtils";
 import type { FlattenedWidgetProps } from "WidgetProvider/constants";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import { addNewAnvilWidgetToDSL } from "layoutSystems/anvil/integrations/sagas/anvilWidgetAdditionSagas/helpers";
 
 export function* createSectionAndAddWidget(
diff --git a/app/client/src/layoutSystems/anvil/utils/layouts/update/zoneUtils.ts b/app/client/src/layoutSystems/anvil/utils/layouts/update/zoneUtils.ts
index acffc109491..ca95c472fce 100644
--- a/app/client/src/layoutSystems/anvil/utils/layouts/update/zoneUtils.ts
+++ b/app/client/src/layoutSystems/anvil/utils/layouts/update/zoneUtils.ts
@@ -11,7 +11,7 @@ import { call } from "redux-saga/effects";
 import { addWidgetsToChildTemplate } from "./additionUtils";
 import type { FlattenedWidgetProps } from "WidgetProvider/constants";
 import { isLargeWidget } from "../../widgetUtils";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import {
   moveWidgets,
   severTiesFromParents,
diff --git a/app/client/src/layoutSystems/anvil/utils/paste/sectionPasteUtils.ts b/app/client/src/layoutSystems/anvil/utils/paste/sectionPasteUtils.ts
index f0b7ee22d6c..9411874ca8a 100644
--- a/app/client/src/layoutSystems/anvil/utils/paste/sectionPasteUtils.ts
+++ b/app/client/src/layoutSystems/anvil/utils/paste/sectionPasteUtils.ts
@@ -6,7 +6,7 @@ import type {
 } from "./types";
 import { getDestinedParent } from "./destinationUtils";
 import type { FlattenedWidgetProps } from "WidgetProvider/constants";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import type { LayoutProps } from "../anvilTypes";
 import { all, call } from "redux-saga/effects";
 import { addPastedWidgets } from "./utils";
diff --git a/app/client/src/layoutSystems/anvil/utils/sectionOperationUtils.ts b/app/client/src/layoutSystems/anvil/utils/sectionOperationUtils.ts
index f7febdf4c76..51cf98a9d5f 100644
--- a/app/client/src/layoutSystems/anvil/utils/sectionOperationUtils.ts
+++ b/app/client/src/layoutSystems/anvil/utils/sectionOperationUtils.ts
@@ -13,7 +13,7 @@ import { generateReactKey } from "utils/generators";
 import type BaseLayoutComponent from "../layoutComponents/BaseLayoutComponent";
 import LayoutFactory from "../layoutComponents/LayoutFactory";
 import { defaultHighlightRenderInfo } from "../utils/constants";
-import { anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import { anvilWidgets } from "ee/modules/ui-builder/ui/wds/constants";
 import { getUpdatedListOfWidgetsAfterAddingNewWidget } from "../integrations/sagas/anvilWidgetAdditionSagas";
 
 /**
diff --git a/app/client/src/modules/ui-builder/ui/wds/Container.tsx b/app/client/src/modules/ui-builder/ui/wds/Container.tsx
index 67132176fd4..bba808b1599 100644
--- a/app/client/src/modules/ui-builder/ui/wds/Container.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/Container.tsx
@@ -2,7 +2,7 @@ import type { ReactNode } from "react";
 import React from "react";
 import styled from "styled-components";
 import { generateClassName } from "utils/generators";
-import type { Elevations } from "./constants";
+import type { Elevations } from "../../../../ee/modules/ui-builder/ui/wds/constants";
 import { useAnvilWidgetElevationSetter } from "layoutSystems/anvil/editor/canvas/hooks/useAnvilWidgetElevationSetter";
 
 /**
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSModalWidget/widget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSModalWidget/widget/index.tsx
index 7af7d3e668c..946e5b7a211 100644
--- a/app/client/src/modules/ui-builder/ui/wds/WDSModalWidget/widget/index.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/WDSModalWidget/widget/index.tsx
@@ -8,7 +8,7 @@ import React from "react";
 import { EventType } from "constants/AppsmithActionConstants/ActionConstants";
 import { SelectionRequestType } from "sagas/WidgetSelectUtils";
 import { ModalBody } from "@appsmith/wds";
-import { WDS_MODAL_WIDGET_CLASSNAME } from "modules/ui-builder/ui/wds/constants";
+import { WDS_MODAL_WIDGET_CLASSNAME } from "ee/modules/ui-builder/ui/wds/constants";
 import type { CanvasWidgetsReduxState } from "reducers/entityReducers/canvasWidgetsReducer";
 import type {
   CopiedWidgetData,
@@ -21,7 +21,7 @@ import { ModalLayoutProvider } from "layoutSystems/anvil/layoutComponents/ModalL
 import styles from "./styles.module.css";
 import { getAnvilWidgetDOMId } from "layoutSystems/common/utils/LayoutElementPositionsObserver/utils";
 import { widgetTypeClassname } from "widgets/WidgetUtils";
-import { AnvilDataAttributes } from "modules/ui-builder/ui/wds/constants";
+import { AnvilDataAttributes } from "ee/modules/ui-builder/ui/wds/constants";
 
 class WDSModalWidget extends BaseWidget<ModalWidgetProps, WidgetState> {
   static type = "WDS_MODAL_WIDGET";
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSSectionWidget/widget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSSectionWidget/widget/index.tsx
index 0597440d9f4..9ba2cafff9b 100644
--- a/app/client/src/modules/ui-builder/ui/wds/WDSSectionWidget/widget/index.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/WDSSectionWidget/widget/index.tsx
@@ -25,7 +25,10 @@ import type { ReactNode } from "react";
 import React from "react";
 import { ContainerComponent } from "modules/ui-builder/ui/wds/Container";
 import { LayoutProvider } from "layoutSystems/anvil/layoutComponents/LayoutProvider";
-import { Elevations, anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import {
+  Elevations,
+  anvilWidgets,
+} from "ee/modules/ui-builder/ui/wds/constants";
 import type { CanvasWidgetsReduxState } from "reducers/entityReducers/canvasWidgetsReducer";
 import type {
   CopiedWidgetData,
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSStatsWidget/widget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSStatsWidget/widget/index.tsx
index dbc79ec87e8..f6ed7085f07 100644
--- a/app/client/src/modules/ui-builder/ui/wds/WDSStatsWidget/widget/index.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/WDSStatsWidget/widget/index.tsx
@@ -7,7 +7,7 @@ import * as config from "../config";
 import { StatsComponent } from "../component";
 import type { StatsWidgetProps } from "./types";
 import type { AnvilConfig } from "WidgetProvider/constants";
-import { Elevations } from "modules/ui-builder/ui/wds/constants";
+import { Elevations } from "ee/modules/ui-builder/ui/wds/constants";
 import { ContainerComponent } from "modules/ui-builder/ui/wds/Container";
 
 class WDSStatsWidget extends BaseWidget<StatsWidgetProps, WidgetState> {
diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSZoneWidget/widget/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSZoneWidget/widget/index.tsx
index 3c252e8d910..270f66dc613 100644
--- a/app/client/src/modules/ui-builder/ui/wds/WDSZoneWidget/widget/index.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/WDSZoneWidget/widget/index.tsx
@@ -24,7 +24,10 @@ import type { LayoutProps } from "layoutSystems/anvil/utils/anvilTypes";
 import type { ContainerWidgetProps } from "widgets/ContainerWidget/widget";
 import { ContainerComponent } from "modules/ui-builder/ui/wds/Container";
 import { LayoutProvider } from "layoutSystems/anvil/layoutComponents/LayoutProvider";
-import { Elevations, anvilWidgets } from "modules/ui-builder/ui/wds/constants";
+import {
+  Elevations,
+  anvilWidgets,
+} from "ee/modules/ui-builder/ui/wds/constants";
 import type { CanvasWidgetsReduxState } from "reducers/entityReducers/canvasWidgetsReducer";
 import type {
   CopiedWidgetData,
diff --git a/app/client/src/pages/Editor/QueryEditor/BindDataButton.tsx b/app/client/src/pages/Editor/QueryEditor/BindDataButton.tsx
index 56271b6f31d..79dfce14dab 100644
--- a/app/client/src/pages/Editor/QueryEditor/BindDataButton.tsx
+++ b/app/client/src/pages/Editor/QueryEditor/BindDataButton.tsx
@@ -26,7 +26,7 @@ import { FEATURE_FLAG } from "ee/entities/FeatureFlag";
 import { getHasManagePagePermission } from "ee/utils/BusinessFeatures/permissionPageHelpers";
 import { getWidgets } from "sagas/selectors";
 import type { FlattenedWidgetProps } from "reducers/entityReducers/canvasWidgetsStructureReducer";
-import { WDS_V2_WIDGET_MAP } from "modules/ui-builder/ui/wds/constants";
+import { WDS_V2_WIDGET_MAP } from "ee/modules/ui-builder/ui/wds/constants";
 import { getNextWidgetName } from "sagas/WidgetOperationUtils";
 import AnalyticsUtil from "ee/utils/AnalyticsUtil";
 import { addSuggestedWidget } from "actions/widgetActions";
diff --git a/app/client/src/selectors/editorSelectors.tsx b/app/client/src/selectors/editorSelectors.tsx
index ee3560dd503..5022fcc13df 100644
--- a/app/client/src/selectors/editorSelectors.tsx
+++ b/app/client/src/selectors/editorSelectors.tsx
@@ -41,7 +41,7 @@ import { isAutoHeightEnabledForWidget } from "widgets/WidgetUtils";
 import WidgetFactory from "WidgetProvider/factory";
 import { isAirgapped } from "ee/utils/airgapHelpers";
 import { getIsAnonymousDataPopupVisible } from "./onboardingSelectors";
-import { WDS_V2_WIDGET_MAP } from "modules/ui-builder/ui/wds/constants";
+import { WDS_V2_WIDGET_MAP } from "ee/modules/ui-builder/ui/wds/constants";
 import { LayoutSystemTypes } from "layoutSystems/types";
 import { getLayoutSystemType } from "./layoutSystemSelectors";
 import { protectedModeSelector } from "./gitSyncSelectors";

From d551c15126d8246145a9a1fd7a945ebebadecfc4 Mon Sep 17 00:00:00 2001
From: Ilia Znamenskii <ilia@appsmith.com>
Date: Tue, 8 Oct 2024 15:09:13 +0200
Subject: [PATCH 5/6] remove

---
 .../AIChat/stories/AIChat.stories.tsx         |  78 ------------
 .../components/AIChat/tests/AIChat.test.tsx   | 113 ------------------
 2 files changed, 191 deletions(-)
 delete mode 100644 app/client/packages/design-system/widgets/src/components/AIChat/stories/AIChat.stories.tsx
 delete mode 100644 app/client/packages/design-system/widgets/src/components/AIChat/tests/AIChat.test.tsx

diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/stories/AIChat.stories.tsx b/app/client/packages/design-system/widgets/src/components/AIChat/stories/AIChat.stories.tsx
deleted file mode 100644
index 6b0abe2853b..00000000000
--- a/app/client/packages/design-system/widgets/src/components/AIChat/stories/AIChat.stories.tsx
+++ /dev/null
@@ -1,78 +0,0 @@
-import { AIChat } from "@appsmith/wds";
-import type { Meta, StoryObj } from "@storybook/react";
-
-const meta: Meta<typeof AIChat> = {
-  component: AIChat,
-  title: "WDS/Widgets/AIChat",
-};
-
-export default meta;
-type Story = StoryObj<typeof AIChat>;
-
-export const Main: Story = {
-  args: {
-    thread: [],
-    prompt: "",
-    username: "John Doe",
-    promptInputPlaceholder: "Type your message here",
-    chatTitle: "Chat with Assistant",
-    assistantName: "",
-    isWaitingForResponse: false,
-  },
-};
-
-export const EmptyHistory: Story = {
-  args: {
-    thread: [],
-    prompt: "",
-    username: "John Doe",
-    promptInputPlaceholder: "Type your message here",
-    chatTitle: "Chat with Assistant",
-  },
-};
-
-export const Loading: Story = {
-  args: {
-    thread: [],
-    prompt: "",
-    username: "John Doe",
-    promptInputPlaceholder: "Type your message here",
-    chatTitle: "Chat with Assistant",
-    isWaitingForResponse: true,
-  },
-};
-
-export const WithHistory: Story = {
-  args: {
-    thread: [
-      {
-        id: "1",
-        content: "Hi, how can I help you?",
-        isAssistant: true,
-      },
-      {
-        id: "2",
-        content: "Find stuck support requests",
-        isAssistant: false,
-      },
-      {
-        id: "3",
-        content: `Certainly! Here's what I can do to help you:
-* Search our customers database
-* Find stuck support requests
-* Provide advice on how to resolve customer issues
-`,
-        isAssistant: true,
-      },
-      {
-        id: "4",
-        content: "Thank you",
-        isAssistant: false,
-      },
-    ],
-    prompt: "",
-    username: "John Doe",
-    promptInputPlaceholder: "Type your message here",
-    chatTitle: "Chat with Assistant",
-  },
-};
diff --git a/app/client/packages/design-system/widgets/src/components/AIChat/tests/AIChat.test.tsx b/app/client/packages/design-system/widgets/src/components/AIChat/tests/AIChat.test.tsx
deleted file mode 100644
index ce868dd72c7..00000000000
--- a/app/client/packages/design-system/widgets/src/components/AIChat/tests/AIChat.test.tsx
+++ /dev/null
@@ -1,113 +0,0 @@
-import React from "react";
-import "@testing-library/jest-dom";
-import { render, screen, within } from "@testing-library/react";
-import { faker } from "@faker-js/faker";
-
-import { AIChat, type AIChatProps } from "..";
-import userEvent from "@testing-library/user-event";
-
-const renderComponent = (props: Partial<AIChatProps> = {}) => {
-  const defaultProps: AIChatProps = {
-    username: "",
-    thread: [],
-    prompt: "",
-    onPromptChange: jest.fn(),
-  };
-
-  return render(<AIChat {...defaultProps} {...props} />);
-};
-
-describe("@appsmith/wds/AIChat", () => {
-  it("should render chat's title", () => {
-    const chatTitle = faker.lorem.words(2);
-
-    renderComponent({ chatTitle });
-
-    expect(screen.getByTestId("t--aichat-chat-title")).toHaveTextContent(
-      chatTitle,
-    );
-  });
-
-  it("should render username", () => {
-    const username = faker.name.firstName();
-
-    renderComponent({ username });
-
-    expect(screen.getByTestId("t--aichat-username")).toHaveTextContent(
-      username,
-    );
-  });
-
-  it("should render thread", () => {
-    const thread = [
-      {
-        id: faker.datatype.uuid(),
-        content: faker.lorem.paragraph(1),
-        isAssistant: false,
-      },
-      {
-        id: faker.datatype.uuid(),
-        content: faker.lorem.paragraph(2),
-        isAssistant: true,
-      },
-    ];
-
-    renderComponent({ thread });
-
-    const messages = within(
-      screen.getByTestId("t--aichat-thread"),
-    ).getAllByRole("listitem");
-
-    expect(messages).toHaveLength(thread.length);
-    expect(messages[0]).toHaveTextContent(thread[0].content);
-    expect(messages[1]).toHaveTextContent(thread[1].content);
-  });
-
-  it("should render prompt input placeholder", () => {
-    const promptInputPlaceholder = faker.lorem.words(3);
-
-    renderComponent({
-      promptInputPlaceholder,
-    });
-
-    expect(screen.getByRole("textbox")).toHaveAttribute(
-      "placeholder",
-      promptInputPlaceholder,
-    );
-  });
-
-  it("should render prompt input value", () => {
-    const prompt = faker.lorem.words(3);
-
-    renderComponent({
-      prompt,
-    });
-
-    expect(screen.getByRole("textbox")).toHaveValue(prompt);
-  });
-
-  it("should trigger user's prompt", async () => {
-    const onPromptChange = jest.fn();
-
-    renderComponent({
-      onPromptChange,
-    });
-
-    await userEvent.type(screen.getByRole("textbox"), "A");
-
-    expect(onPromptChange).toHaveBeenCalledWith("A");
-  });
-
-  it("should submit user's prompt", async () => {
-    const onSubmit = jest.fn();
-
-    renderComponent({
-      prompt: "ABCD",
-      onSubmit,
-    });
-
-    await userEvent.click(screen.getByRole("button", { name: "Send" }));
-
-    expect(onSubmit).toHaveBeenCalled();
-  });
-});

From 72115ea094040735f6bbf1f2fd1c976261fa67d9 Mon Sep 17 00:00:00 2001
From: Ilia Znamenskii <ilia@appsmith.com>
Date: Tue, 8 Oct 2024 15:10:20 +0200
Subject: [PATCH 6/6] unregister ai chat widget

---
 app/client/src/ce/modules/ui-builder/ui/wds/constants.ts | 1 -
 1 file changed, 1 deletion(-)

diff --git a/app/client/src/ce/modules/ui-builder/ui/wds/constants.ts b/app/client/src/ce/modules/ui-builder/ui/wds/constants.ts
index 45f141e139c..f1d62d284a2 100644
--- a/app/client/src/ce/modules/ui-builder/ui/wds/constants.ts
+++ b/app/client/src/ce/modules/ui-builder/ui/wds/constants.ts
@@ -59,7 +59,6 @@ export const WDS_V2_WIDGET_MAP = {
   MULTILINE_INPUT_WIDGET: "WDS_MULTILINE_INPUT_WIDGET",
   WDS_SELECT_WIDGET: "WDS_SELECT_WIDGET",
   WDS_COMBOBOX_WIDGET: "WDS_COMBOBOX_WIDGET",
-  WDS_AI_CHAT_WIDGET: "WDS_AI_CHAT_WIDGET",
 
   // Anvil layout widgets
   ZONE_WIDGET: anvilWidgets.ZONE_WIDGET,