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

[fix][chat]Fix minor logic issue.

Browse Source
This commit is contained in:
jerryjzhang
2024-12-21 18:49:07 +08:00
parent caf03588c6
commit 933b54085c
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
chat/server/src/main/java/com/tencent/supersonic/chat/server/executor/SqlExecutor.java
View File

@@ -77,6 +77,7 @@ public class SqlExecutor implements ChatQueryExecutor {
long startTime = System.currentTimeMillis(); long startTime = System.currentTimeMillis();
QueryResult queryResult = new QueryResult(); QueryResult queryResult = new QueryResult();
queryResult.setQueryId(executeContext.getRequest().getQueryId());
queryResult.setChatContext(parseInfo); queryResult.setChatContext(parseInfo);
queryResult.setQueryMode(parseInfo.getQueryMode()); queryResult.setQueryMode(parseInfo.getQueryMode());
queryResult.setQueryTimeCost(System.currentTimeMillis() - startTime); queryResult.setQueryTimeCost(System.currentTimeMillis() - startTime);

2
chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/parse/ErrorMsgRewriteProcessor.java
View File

@@ -47,7 +47,7 @@ public class ErrorMsgRewriteProcessor implements ParseResultProcessor {
public boolean accept(ParseContext parseContext) { public boolean accept(ParseContext parseContext) {
ChatApp chatApp = parseContext.getAgent().getChatAppConfig().get(APP_KEY_ERROR_MESSAGE); ChatApp chatApp = parseContext.getAgent().getChatAppConfig().get(APP_KEY_ERROR_MESSAGE);
return StringUtils.isNotBlank(parseContext.getResponse().getErrorMsg()) return StringUtils.isNotBlank(parseContext.getResponse().getErrorMsg())
&& Objects.nonNull(chatApp) || chatApp.isEnable(); && Objects.nonNull(chatApp) && chatApp.isEnable();
} }
@Override @Override

2
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/LLMSqlCorrector.java
View File

@@ -65,7 +65,7 @@ public class LLMSqlCorrector extends BaseSemanticCorrector {
return; return;
} }
Text2SQLExemplar exemplar = (Text2SQLExemplar)semanticParseInfo.getProperties() Text2SQLExemplar exemplar = (Text2SQLExemplar) semanticParseInfo.getProperties()
.get(Text2SQLExemplar.PROPERTY_KEY); .get(Text2SQLExemplar.PROPERTY_KEY);
ChatLanguageModel chatLanguageModel = ChatLanguageModel chatLanguageModel =