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

[improvement](chat) When determining the query type, S2SQL is uniformly used for parsing. (#393)

Browse Source
This commit is contained in:
lexluo09
2023-11-16 17:39:46 +08:00
committed by GitHub
parent 46a9e5b097
commit 11cdcb29fa
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
chat/core/src/main/java/com/tencent/supersonic/chat/parser/QueryTypeParser.java
View File

@@ -58,12 +58,14 @@ public class QueryTypeParser implements SemanticParser {
// get primaryEntityBizName
SemanticService semanticService = ContextUtils.getBean(SemanticService.class);
EntityInfo entityInfo = semanticService.getEntityInfo(parseInfo, user);
String primaryEntityBizName = semanticService.getPrimaryEntityBizName(entityInfo);
if (StringUtils.isNotEmpty(primaryEntityBizName)) {
//if exist primaryEntityBizName in parseInfo's dimensions, set nativeQuery to true
boolean existPrimaryEntityBizName = parseInfo.getDimensions().stream()
.anyMatch(schemaElement -> primaryEntityBizName.equalsIgnoreCase(schemaElement.getBizName()));
if (existPrimaryEntityBizName) {
if (Objects.nonNull(entityInfo) && Objects.nonNull(entityInfo.getModelInfo()) && StringUtils.isNotEmpty(
entityInfo.getModelInfo().getPrimaryEntityName())) {
String primaryEntityName = entityInfo.getModelInfo().getPrimaryEntityName();
//if exist primaryEntityName in S2SQL select.
List<String> selectFields = SqlParserSelectHelper.getSelectFields(sqlInfo.getS2SQL());
boolean existPrimaryEntityName = selectFields.stream()
.anyMatch(fieldName -> primaryEntityName.equalsIgnoreCase(fieldName));
if (existPrimaryEntityName) {
return QueryType.ENTITY;
}
}