diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java index 41854ef05..52022edb3 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java @@ -57,7 +57,8 @@ public class TimeCorrector extends BaseSemanticCorrector { DataSetSchema dataSetSchema = chatQueryContext.getSemanticSchema().getDataSetSchemaMap().get(dataSetId); if (Objects.isNull(dataSetSchema) || Objects.isNull(dataSetSchema.getPartitionDimension()) - || Objects.isNull(dataSetSchema.getPartitionDimension().getName())) { + || Objects.isNull(dataSetSchema.getPartitionDimension().getName()) + || TimeDimensionEnum.containsZhTimeDimension(whereFields)) { return; } String partitionDimension = dataSetSchema.getPartitionDimension().getName(); diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/utils/QueryReqBuilder.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/utils/QueryReqBuilder.java index 0d5aaefee..8b3636263 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/utils/QueryReqBuilder.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/utils/QueryReqBuilder.java @@ -212,7 +212,8 @@ public class QueryReqBuilder { private static boolean isDateFieldAlreadyPresent(SemanticParseInfo parseInfo, String dateField) { 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) {