diff --git a/apps/api/src/app/workflows-v2/shared/schemas/chat-control.schema.ts b/apps/api/src/app/workflows-v2/shared/schemas/chat-control.schema.ts
new file mode 100644
index 00000000000..a90e225ad89
--- /dev/null
+++ b/apps/api/src/app/workflows-v2/shared/schemas/chat-control.schema.ts
@@ -0,0 +1,27 @@
+import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+
+import { JSONSchemaDto, UiComponentEnum, UiSchema, UiSchemaGroupEnum } from '@novu/shared';
+
+export const ChatStepControlZodSchema = z
+  .object({
+    body: z.string(),
+  })
+  .strict();
+
+export type ChatStepControlType = z.infer<typeof ChatStepControlZodSchema>;
+
+export const chatStepControlSchema = zodToJsonSchema(ChatStepControlZodSchema) as JSONSchemaDto;
+export const chatStepUiSchema: UiSchema = {
+  group: UiSchemaGroupEnum.CHAT,
+  properties: {
+    body: {
+      component: UiComponentEnum.CHAT_BODY,
+    },
+  },
+};
+
+export const chatStepControl = {
+  uiSchema: chatStepUiSchema,
+  schema: chatStepControlSchema,
+};
diff --git a/apps/api/src/app/workflows-v2/shared/step-type-to-control.mapper.ts b/apps/api/src/app/workflows-v2/shared/step-type-to-control.mapper.ts
index 0b2abd32b4c..fdcd743eebf 100644
--- a/apps/api/src/app/workflows-v2/shared/step-type-to-control.mapper.ts
+++ b/apps/api/src/app/workflows-v2/shared/step-type-to-control.mapper.ts
@@ -4,6 +4,7 @@ import { emailStepControlSchema, emailStepUiSchema, inAppControlSchema, inAppUiS
 import { DelayTimeControlSchema, delayUiSchema } from './schemas/delay-control.schema';
 import { DigestOutputJsonSchema, digestUiSchema } from './schemas/digest-control.schema';
 import { smsStepControl } from './schemas/sms-control.schema';
+import { chatStepControl } from './schemas/chat-control.schema';
 
 export const PERMISSIVE_EMPTY_SCHEMA = {
   type: 'object',
@@ -29,7 +30,8 @@ export const stepTypeToControlSchema: Record<ChannelStepEnum | ActionStepEnum, C
     schema: channelStepSchemas[ChannelStepEnum.PUSH].output as unknown as JSONSchemaDto,
   },
   [ChannelStepEnum.CHAT]: {
-    schema: channelStepSchemas[ChannelStepEnum.CHAT].output as unknown as JSONSchemaDto,
+    schema: chatStepControl.schema,
+    uiSchema: chatStepControl.uiSchema,
   },
   [ActionStepEnum.DELAY]: {
     schema: DelayTimeControlSchema,
diff --git a/packages/shared/src/dto/workflows/step-data.dto.ts b/packages/shared/src/dto/workflows/step-data.dto.ts
index aa309f89dc5..c8bcb8f51e1 100644
--- a/packages/shared/src/dto/workflows/step-data.dto.ts
+++ b/packages/shared/src/dto/workflows/step-data.dto.ts
@@ -21,6 +21,7 @@ export enum UiSchemaGroupEnum {
   DIGEST = 'DIGEST',
   DELAY = 'DELAY',
   SMS = 'SMS',
+  CHAT = 'CHAT',
 }
 
 export enum UiComponentEnum {
@@ -40,6 +41,7 @@ export enum UiComponentEnum {
   DELAY_AMOUNT = 'DELAY_AMOUNT',
   DELAY_UNIT = 'DELAY_UNIT',
   SMS_BODY = 'SMS_BODY',
+  CHAT_BODY = 'CHAT_BODY',
 }
 
 export class UiSchemaProperty {