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

(improvement)(chat) Fix the issue of duplicate addition of the 'data date' in the rule pattern. (#1597)

Browse Source
This commit is contained in:
lexluo09
2024-08-23 16:33:23 +08:00
committed by GitHub
parent be9a8bbc27
commit 59c50176c5
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java
View File

@@ -57,7 +57,8 @@ public class TimeCorrector extends BaseSemanticCorrector {
DataSetSchema dataSetSchema = chatQueryContext.getSemanticSchema().getDataSetSchemaMap().get(dataSetId); DataSetSchema dataSetSchema = chatQueryContext.getSemanticSchema().getDataSetSchemaMap().get(dataSetId);
if (Objects.isNull(dataSetSchema) if (Objects.isNull(dataSetSchema)
|| Objects.isNull(dataSetSchema.getPartitionDimension()) || Objects.isNull(dataSetSchema.getPartitionDimension())
|| Objects.isNull(dataSetSchema.getPartitionDimension().getName())) { || Objects.isNull(dataSetSchema.getPartitionDimension().getName())
|| TimeDimensionEnum.containsZhTimeDimension(whereFields)) {
return; return;
} }
String partitionDimension = dataSetSchema.getPartitionDimension().getName(); String partitionDimension = dataSetSchema.getPartitionDimension().getName();

3
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/utils/QueryReqBuilder.java
View File

@@ -212,7 +212,8 @@ public class QueryReqBuilder {
private static boolean isDateFieldAlreadyPresent(SemanticParseInfo parseInfo, String dateField) { private static boolean isDateFieldAlreadyPresent(SemanticParseInfo parseInfo, String dateField) {
return parseInfo.getDimensions().stream() return parseInfo.getDimensions().stream()
.anyMatch(dimension -> dimension.getBizName().equalsIgnoreCase(dateField)); .anyMatch(dimension -> dimension.getBizName().equalsIgnoreCase(dateField)
|| dimension.containsPartitionTime());
} }
private static void addDimension(SemanticParseInfo parseInfo, SchemaElement dimension) { private static void addDimension(SemanticParseInfo parseInfo, SchemaElement dimension) {