(improvement)(Chat) Fix integrating test (#824)

Co-authored-by: jolunoluo
This commit is contained in:
LXW
2024-03-15 13:04:05 +08:00
committed by GitHub
parent 36136e4c15
commit 7f3cb5812c
2 changed files with 42 additions and 10 deletions

View File

@@ -24,9 +24,15 @@ public class ParseResp {
FAILED FAILED
} }
public ParseResp(Integer chatId, String queryText) {
this.chatId = chatId;
this.queryText = queryText;
parseTimeCost.setParseStartTime(System.currentTimeMillis());
}
public List<SemanticParseInfo> getSelectedParses() { public List<SemanticParseInfo> getSelectedParses() {
selectedParses = selectedParses.stream() selectedParses = selectedParses.stream()
.sorted(Comparator.comparingDouble(SemanticParseInfo::getScore)) .sorted(Comparator.comparingDouble(SemanticParseInfo::getScore).reversed())
.collect(Collectors.toList()); .collect(Collectors.toList());
generateParseInfoId(selectedParses); generateParseInfoId(selectedParses);
return selectedParses; return selectedParses;
@@ -48,9 +54,4 @@ public class ParseResp {
} }
} }
public ParseResp(Integer chatId, String queryText) {
this.chatId = chatId;
this.queryText = queryText;
parseTimeCost.setParseStartTime(System.currentTimeMillis());
}
} }

View File

@@ -9,17 +9,40 @@ com.tencent.supersonic.headless.core.chat.parser.SemanticParser=\
com.tencent.supersonic.headless.core.chat.parser.llm.LLMSqlParser, \ com.tencent.supersonic.headless.core.chat.parser.llm.LLMSqlParser, \
com.tencent.supersonic.headless.core.chat.parser.QueryTypeParser com.tencent.supersonic.headless.core.chat.parser.QueryTypeParser
com.tencent.supersonic.chat.server.parser.ChatParser=\
com.tencent.supersonic.chat.server.parser.NL2PluginParser, \
com.tencent.supersonic.chat.server.parser.NL2SQLParser
com.tencent.supersonic.chat.server.executor.ChatExecutor=\
com.tencent.supersonic.chat.server.executor.PluginExecutor, \
com.tencent.supersonic.chat.server.executor.SqlExecutor
com.tencent.supersonic.headless.core.chat.corrector.SemanticCorrector=\ com.tencent.supersonic.headless.core.chat.corrector.SemanticCorrector=\
com.tencent.supersonic.headless.core.chat.corrector.SchemaCorrector, \ com.tencent.supersonic.headless.core.chat.corrector.SchemaCorrector, \
com.tencent.supersonic.headless.core.chat.corrector.TimeCorrector, \ com.tencent.supersonic.headless.core.chat.corrector.TimeCorrector, \
com.tencent.supersonic.headless.core.chat.corrector.GrammarCorrector com.tencent.supersonic.headless.core.chat.corrector.GrammarCorrector
com.tencent.supersonic.headless.core.parser.converter.HeadlessConverter=\
com.tencent.supersonic.headless.core.parser.converter.DefaultDimValueConverter,\
com.tencent.supersonic.headless.core.parser.converter.SqlVariableParseConverter,\
com.tencent.supersonic.headless.core.parser.converter.CalculateAggConverter,\
com.tencent.supersonic.headless.core.parser.converter.ParserDefaultConverter
com.tencent.supersonic.headless.core.planner.QueryOptimizer=\
com.tencent.supersonic.headless.core.planner.DetailQueryOptimizer
com.tencent.supersonic.headless.core.executor.QueryExecutor=\
com.tencent.supersonic.headless.core.executor.JdbcExecutor
com.tencent.supersonic.headless.core.parser.SqlParser=\
com.tencent.supersonic.headless.core.parser.calcite.CalciteSqlParser
com.tencent.supersonic.headless.core.cache.QueryCache=\
com.tencent.supersonic.headless.core.cache.DefaultQueryCache
com.tencent.supersonic.headless.server.processor.ResultProcessor=\ com.tencent.supersonic.headless.server.processor.ResultProcessor=\
com.tencent.supersonic.headless.server.processor.ParseInfoProcessor, \ com.tencent.supersonic.headless.server.processor.ParseInfoProcessor, \
com.tencent.supersonic.headless.server.processor.QueryRankProcessor, \ com.tencent.supersonic.headless.server.processor.SqlInfoProcessor
com.tencent.supersonic.headless.server.processor.SqlInfoProcessor, \
com.tencent.supersonic.chat.server.processor.parse.TimeCostProcessor, \
com.tencent.supersonic.headless.server.processor.RespBuildProcessor
com.tencent.supersonic.headless.core.chat.parser.llm.DataSetResolver=\ com.tencent.supersonic.headless.core.chat.parser.llm.DataSetResolver=\
com.tencent.supersonic.headless.core.chat.parser.llm.HeuristicDataSetResolver com.tencent.supersonic.headless.core.chat.parser.llm.HeuristicDataSetResolver
@@ -30,6 +53,14 @@ com.tencent.supersonic.auth.authentication.interceptor.AuthenticationInterceptor
com.tencent.supersonic.auth.api.authentication.adaptor.UserAdaptor=\ com.tencent.supersonic.auth.api.authentication.adaptor.UserAdaptor=\
com.tencent.supersonic.auth.authentication.adaptor.DefaultUserAdaptor com.tencent.supersonic.auth.authentication.adaptor.DefaultUserAdaptor
com.tencent.supersonic.chat.server.plugin.recognize.PluginRecognizer=\
com.tencent.supersonic.chat.server.plugin.recognize.embedding.EmbeddingRecallRecognizer
com.tencent.supersonic.chat.server.processor.parse.ParseResultProcessor=\
com.tencent.supersonic.chat.server.processor.parse.QueryRecommendProcessor,\
com.tencent.supersonic.chat.server.processor.parse.EntityInfoProcessor,\
com.tencent.supersonic.chat.server.processor.parse.TimeCostProcessor
com.tencent.supersonic.chat.server.processor.execute.ExecuteResultProcessor=\ com.tencent.supersonic.chat.server.processor.execute.ExecuteResultProcessor=\
com.tencent.supersonic.chat.server.processor.execute.MetricRecommendProcessor,\ com.tencent.supersonic.chat.server.processor.execute.MetricRecommendProcessor,\
com.tencent.supersonic.chat.server.processor.execute.DimensionRecommendProcessor,\ com.tencent.supersonic.chat.server.processor.execute.DimensionRecommendProcessor,\