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

[improvement][chat]Make a few code restructure.
Some checks failed
supersonic CentOS CI / build (21) (push) Has been cancelled
Details
supersonic mac CI / build (21) (push) Has been cancelled
Details
supersonic ubuntu CI / build (21) (push) Has been cancelled
Details
supersonic windows CI / build (21) (push) Has been cancelled
Details

Browse Source
This commit is contained in:
jerryjzhang
2025-02-08 14:39:23 +08:00
parent c34b85c8a4
commit eef7b3c443
5 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/service/impl/S2ChatLayerService.java
View File

@@ -65,14 +65,15 @@ public class S2ChatLayerService implements ChatLayerService {
@Override
public ParseResp parse(QueryNLReq queryNLReq) {
ParseResp parseResult = new ParseResp(queryNLReq.getQueryText());
ParseResp parseResp = new ParseResp(queryNLReq.getQueryText());
ChatQueryContext queryCtx = buildChatQueryContext(queryNLReq);
queryCtx.setParseResp(parseResp);
if (queryCtx.getMapInfo().isEmpty()) {
chatWorkflowEngine.start(ChatWorkflowState.MAPPING, queryCtx, parseResult);
chatWorkflowEngine.start(ChatWorkflowState.MAPPING, queryCtx);
} else {
chatWorkflowEngine.start(ChatWorkflowState.PARSING, queryCtx, parseResult);
chatWorkflowEngine.start(ChatWorkflowState.PARSING, queryCtx);
}
return parseResult;
return parseResp;
}
public void correct(QuerySqlReq querySqlReq, User user) {

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

@@ -34,8 +34,8 @@ public class ChatWorkflowEngine {
private final List<SemanticCorrector> semanticCorrectors =
CoreComponentFactory.getSemanticCorrectors();
public void start(ChatWorkflowState initialState, ChatQueryContext queryCtx,
ParseResp parseResult) {
public void start(ChatWorkflowState initialState, ChatQueryContext queryCtx) {
ParseResp parseResult = queryCtx.getParseResp();
queryCtx.setChatWorkflowState(initialState);
while (queryCtx.getChatWorkflowState() != ChatWorkflowState.FINISHED) {
switch (queryCtx.getChatWorkflowState()) {