topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-13 13:07:32 +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; package com.tencent.supersonic.headless.api.pojo.enums;
public enum ChatWorkflowState { 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()); .map(SemanticQuery::getParseInfo).collect(Collectors.toList());
parseResult.setSelectedParses(parseInfos); parseResult.setSelectedParses(parseInfos);
if (queryCtx.needSQL()) { if (queryCtx.needSQL()) {
queryCtx.setChatWorkflowState(ChatWorkflowState.CORRECTING); queryCtx.setChatWorkflowState(ChatWorkflowState.S2SQL_CORRECTING);
} else { } else {
parseResult.setState(ParseResp.ParseState.COMPLETED); parseResult.setState(ParseResp.ParseState.COMPLETED);
queryCtx.setChatWorkflowState(ChatWorkflowState.FINISHED); queryCtx.setChatWorkflowState(ChatWorkflowState.FINISHED);
} }
} }
break; break;
case CORRECTING: case S2SQL_CORRECTING:
performCorrecting(queryCtx); performCorrecting(queryCtx);
queryCtx.setChatWorkflowState(ChatWorkflowState.TRANSLATING); queryCtx.setChatWorkflowState(ChatWorkflowState.TRANSLATING);
break; break;
@@ -109,7 +109,8 @@ public class ChatWorkflowEngine {
for (SemanticQuery semanticQuery : candidateQueries) { for (SemanticQuery semanticQuery : candidateQueries) {
for (SemanticCorrector corrector : semanticCorrectors) { for (SemanticCorrector corrector : semanticCorrectors) {
corrector.correct(queryCtx, semanticQuery.getParseInfo()); corrector.correct(queryCtx, semanticQuery.getParseInfo());
if (!ChatWorkflowState.CORRECTING.equals(queryCtx.getChatWorkflowState())) { if (!ChatWorkflowState.S2SQL_CORRECTING
.equals(queryCtx.getChatWorkflowState())) {
break; break;
} }
} }