From 2130f7f9a47d4f872fe493214f650e943722223c Mon Sep 17 00:00:00 2001
From: Yogesh Chauhan <yogesh.chauhan@harness.io>
Date: Thu, 30 Jan 2025 14:55:22 +0000
Subject: [PATCH] [fix]: [ML-575]: Ai Agent  Fix stage type enums (#3357)

* Fix stage enums
---
 app/api/controller/aiagent/types/enum/stage_type.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/app/api/controller/aiagent/types/enum/stage_type.go b/app/api/controller/aiagent/types/enum/stage_type.go
index 1ca461753..142c04504 100644
--- a/app/api/controller/aiagent/types/enum/stage_type.go
+++ b/app/api/controller/aiagent/types/enum/stage_type.go
@@ -17,10 +17,10 @@ package enum
 type StageType string
 
 const (
-	Build          StageType = "Build"
-	Deploy         StageType = "Deploy"
+	Build          StageType = "CI"
+	Deploy         StageType = "Deployment"
 	Approval       StageType = "Approval"
-	Security       StageType = "Security"
+	Security       StageType = "SecurityTests"
 	Pipeline       StageType = "Pipeline"
 	Custom         StageType = "Custom"
 	Infrastructure StageType = "IACM"