topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 04:27:39 +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 // get primaryEntityBizName
SemanticService semanticService = ContextUtils.getBean(SemanticService.class); SemanticService semanticService = ContextUtils.getBean(SemanticService.class);
EntityInfo entityInfo = semanticService.getEntityInfo(parseInfo, user); EntityInfo entityInfo = semanticService.getEntityInfo(parseInfo, user);
String primaryEntityBizName = semanticService.getPrimaryEntityBizName(entityInfo); if (Objects.nonNull(entityInfo) && Objects.nonNull(entityInfo.getModelInfo()) && StringUtils.isNotEmpty(
if (StringUtils.isNotEmpty(primaryEntityBizName)) { entityInfo.getModelInfo().getPrimaryEntityName())) {
//if exist primaryEntityBizName in parseInfo's dimensions, set nativeQuery to true String primaryEntityName = entityInfo.getModelInfo().getPrimaryEntityName();
boolean existPrimaryEntityBizName = parseInfo.getDimensions().stream() //if exist primaryEntityName in S2SQL select.
.anyMatch(schemaElement -> primaryEntityBizName.equalsIgnoreCase(schemaElement.getBizName())); List<String> selectFields = SqlParserSelectHelper.getSelectFields(sqlInfo.getS2SQL());
if (existPrimaryEntityBizName) { boolean existPrimaryEntityName = selectFields.stream()
.anyMatch(fieldName -> primaryEntityName.equalsIgnoreCase(fieldName));
if (existPrimaryEntityName) {
return QueryType.ENTITY; return QueryType.ENTITY;
} }
} }