refactor: single input value

Author: kevinkim-ogp

packages/backend/src/graphql/__tests__/mutations/ai/generate-ai-steps.test.ts

@@ -72,9 +72,8 @@ const DEFAULT_MOCKED_OUTPUT = {
 }
 
 const DEFAULT_INPUT = {
-  trigger: 'New FormSG submission',
-  actions:
-    'Send a welcome email to the new user\nSend a welcome SMS to the new user',
+  prompt:
+    '#### Start the workflow\nNew FormSG submission\n\n#### Actions\nSend a welcome email to the new user\nSend a welcome SMS to the new user',
 }
 
 describe('generateAiSteps mutation', () => {
@@ -115,28 +114,12 @@ describe('generateAiSteps mutation', () => {
         null,
         {
           input: {
-            trigger: 'gibberish',
-            actions: 'gibberish',
+            prompt: 'gibberish',
           },
         },
         context,
       ),
-    ).rejects.toThrow('Description must be at least 15 characters')
-  })
-
-  it('should throw an error if the input is invalid', async () => {
-    await expect(
-      generateAiSteps(
-        null,
-        {
-          input: {
-            trigger: 'New user submit a new form response',
-            actions: 'gibberish',
-          },
-        },
-        context,
-      ),
-    ).rejects.toThrow('Description must be at least 30 characters')
+    ).rejects.toThrow('Prompt must be at least 15 characters')
   })
 
   it('should map step positions correctly', async () => {

packages/backend/src/graphql/mutations/ai/generate-ai-steps.ts

@@ -33,8 +33,7 @@ const generateAiSteps: MutationResolvers['generateAiSteps'] = async (
 
   try {
     const validatedInput = INPUT_SCHEMA.parse(params.input)
-    const { trigger: triggerInput, actions: actionsInput } = validatedInput
-    const userPrompt = `**trigger**:\n${triggerInput}\n\n**actions**:\n${actionsInput}`
+    const { prompt: userPrompt } = validatedInput
 
     if (!context.currentUser) {
       throw new ForbiddenError('Not authorised!')

packages/backend/src/graphql/mutations/ai/schemas/input.zod.ts

@@ -1,14 +1,9 @@
 import { z } from 'zod/v3'
 
 export const INPUT_SCHEMA = z.object({
-  trigger: z
+  prompt: z
     .string()
     .trim()
-    .min(15, 'Description must be at least 15 characters')
-    .max(250, 'Description must not exceed 250 characters'),
-  actions: z
-    .string()
-    .trim()
-    .min(30, 'Description must be at least 30 characters')
-    .max(3000, 'Description must not exceed 3000 characters'),
+    .min(15, 'Prompt must be at least 15 characters')
+    .max(3000, 'Prompt must not exceed 3000 characters'),
 })

packages/backend/src/graphql/schema.graphql

@@ -127,8 +127,7 @@ type Mutation {
 }
 
 input GenerateAiStepsInput {
-  trigger: String!
-  actions: String!
+  prompt: String!
 }
 
 type AdminMutation {