topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 11:07:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][chat]Introduce new chat workflow state.
Some checks are pending
supersonic CentOS CI / build (21) (push) Waiting to run
Details
supersonic mac CI / build (21) (push) Waiting to run
Details
supersonic ubuntu CI / build (21) (push) Waiting to run
Details
supersonic windows CI / build (21) (push) Waiting to run
Details

Browse Source
This commit is contained in:
jerryjzhang
2025-02-08 09:24:23 +08:00
parent be2e380b4c
commit c34b85c8a4
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/enums/ChatWorkflowState.java
View File

@@ -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
}

7
headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ChatWorkflowEngine.java
View File

@@ -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;
}
}