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

(improvement)(chat) recall history solved query in every parse

Browse Source
This commit is contained in:
jolunoluo
2023-09-25 17:11:21 +08:00
parent e1772c25c4
commit 34816451c0
9 changed files with 352 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/response/QueryResult.java
View File

@@ -21,5 +21,4 @@ public class QueryResult {
private SemanticParseInfo chatContext;
private Object response;
private List<Map<String, Object>> queryResults;
private List<SolvedQueryRecallResp> similarSolvedQuery;
}

6
chat/core/src/main/java/com/tencent/supersonic/chat/corrector/GlobalCorrector.java
View File

@@ -6,6 +6,7 @@ import com.tencent.supersonic.chat.query.llm.dsl.LLMReq;
import com.tencent.supersonic.chat.query.llm.dsl.LLMReq.ElementValue;
import com.tencent.supersonic.common.pojo.Constants;
import com.tencent.supersonic.common.util.JsonUtil;
import com.tencent.supersonic.common.util.jsqlparser.SqlParserSelectHelper;
import com.tencent.supersonic.common.util.jsqlparser.SqlParserUpdateHelper;
import java.util.List;
import java.util.Map;
@@ -32,6 +33,11 @@ public class GlobalCorrector extends BaseSemanticCorrector {
private void addAggregateToMetric(SemanticCorrectInfo semanticCorrectInfo) {
if (SqlParserSelectHelper.hasGroupBy(semanticCorrectInfo.getSql())) {
return;
}
}
private void replaceAlias(SemanticCorrectInfo semanticCorrectInfo) {

2
chat/core/src/main/java/com/tencent/supersonic/chat/queryresponder/DefaultQueryResponder.java
View File

@@ -95,7 +95,7 @@ public class DefaultQueryResponder implements QueryResponder {
}
}
} catch (Exception e) {
log.warn("recall similar solved query failed", e);
log.warn("recall similar solved query failed, queryText:{}", queryText);
}
return solvedQueryRecallResps;
}

6
chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java
View File

@@ -143,15 +143,15 @@ public class QueryServiceImpl implements QueryService {
saveInfo(timeCostDOList, queryReq.getQueryText(), parseResult.getQueryId(),
queryReq.getUser().getName(), queryReq.getChatId().longValue());
} else {
List<SolvedQueryRecallResp> solvedQueryRecallResps =
queryResponder.recallSolvedQuery(queryCtx.getRequest().getQueryText());
parseResult = ParseResp.builder()
.chatId(queryReq.getChatId())
.queryText(queryReq.getQueryText())
.state(ParseResp.ParseState.FAILED)
.similarSolvedQuery(solvedQueryRecallResps)
.build();
}
List<SolvedQueryRecallResp> solvedQueryRecallResps =
queryResponder.recallSolvedQuery(queryCtx.getRequest().getQueryText());
parseResult.setSimilarSolvedQuery(solvedQueryRecallResps);
return parseResult;
}