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

[improvement][corrector] Abstract the RuleSqlCorrector component. (#1748)

Browse Source
This commit is contained in:
lexluo09
2024-09-30 19:09:06 +08:00
committed by GitHub
parent 8693e7f8a7
commit 44d1cde34f
4 changed files with 29 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/RuleSqlCorrector.java Normal file
View File

@@ -0,0 +1,27 @@
package com.tencent.supersonic.headless.chat.corrector;
import com.tencent.supersonic.headless.api.pojo.SemanticParseInfo;
import com.tencent.supersonic.headless.chat.ChatQueryContext;
import lombok.extern.slf4j.Slf4j;
import java.util.ArrayList;
import java.util.List;
@Slf4j
public class RuleSqlCorrector extends BaseSemanticCorrector {
private List<BaseSemanticCorrector> correctors;
public RuleSqlCorrector() {
correctors = new ArrayList<>();
correctors.add(new SchemaCorrector());
correctors.add(new TimeCorrector());
correctors.add(new GrammarCorrector());
}
@Override
public void doCorrect(ChatQueryContext chatQueryContext, SemanticParseInfo semanticParseInfo) {
for (BaseSemanticCorrector corrector : correctors) {
corrector.correct(chatQueryContext, semanticParseInfo);
}
}
}