From c34b85c8a4909aec1e6b03c0c6c338a65ed76d4d Mon Sep 17 00:00:00 2001
From: jerryjzhang <jerryjzhang@tencent.com>
Date: Sat, 8 Feb 2025 09:24:23 +0800
Subject: [PATCH] [improvement][chat]Introduce new chat workflow state.

---
 .../headless/api/pojo/enums/ChatWorkflowState.java       | 9 ++++++++-
 .../headless/server/utils/ChatWorkflowEngine.java        | 7 ++++---
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/enums/ChatWorkflowState.java b/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/enums/ChatWorkflowState.java
index 429a3bcdb..953f1f020 100644
--- a/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/enums/ChatWorkflowState.java
+++ b/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/enums/ChatWorkflowState.java
@@ -1,5 +1,12 @@
 package com.tencent.supersonic.headless.api.pojo.enums;
 
 public enum ChatWorkflowState {
-    MAPPING, PARSING, CORRECTING, TRANSLATING, PROCESSING, FINISHED
+    MAPPING,
+    PARSING,
+    S2SQL_CORRECTING,
+    TRANSLATING,
+    VALIDATING,
+    SQL_CORRECTING,
+    PROCESSING,
+    FINISHED
 }
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ChatWorkflowEngine.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ChatWorkflowEngine.java
index 3ac3e6d8e..937fd5678 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ChatWorkflowEngine.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ChatWorkflowEngine.java
@@ -61,14 +61,14 @@ public class ChatWorkflowEngine {
                                 .map(SemanticQuery::getParseInfo).collect(Collectors.toList());
                         parseResult.setSelectedParses(parseInfos);
                         if (queryCtx.needSQL()) {
-                            queryCtx.setChatWorkflowState(ChatWorkflowState.CORRECTING);
+                            queryCtx.setChatWorkflowState(ChatWorkflowState.S2SQL_CORRECTING);
                         } else {
                             parseResult.setState(ParseResp.ParseState.COMPLETED);
                             queryCtx.setChatWorkflowState(ChatWorkflowState.FINISHED);
                         }
                     }
                     break;
-                case CORRECTING:
+                case S2SQL_CORRECTING:
                     performCorrecting(queryCtx);
                     queryCtx.setChatWorkflowState(ChatWorkflowState.TRANSLATING);
                     break;
@@ -109,7 +109,8 @@ public class ChatWorkflowEngine {
             for (SemanticQuery semanticQuery : candidateQueries) {
                 for (SemanticCorrector corrector : semanticCorrectors) {
                     corrector.correct(queryCtx, semanticQuery.getParseInfo());
-                    if (!ChatWorkflowState.CORRECTING.equals(queryCtx.getChatWorkflowState())) {
+                    if (!ChatWorkflowState.S2SQL_CORRECTING
+                            .equals(queryCtx.getChatWorkflowState())) {
                         break;
                     }
                 }