From 579394af404bf0cd7ff1fa7adab0b78bdf34aa2b Mon Sep 17 00:00:00 2001 From: jerryjzhang Date: Fri, 13 Sep 2024 14:54:34 +0800 Subject: [PATCH] [fix][headless-chat]Fix NPE problem. --- .../supersonic/headless/chat/query/llm/s2sql/LLMReq.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/query/llm/s2sql/LLMReq.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/query/llm/s2sql/LLMReq.java index dcf2cad40..064e36ee4 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/query/llm/s2sql/LLMReq.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/query/llm/s2sql/LLMReq.java @@ -11,6 +11,7 @@ import org.apache.commons.collections4.CollectionUtils; import java.util.ArrayList; import java.util.List; +import java.util.Objects; import java.util.stream.Collectors; @Data @@ -56,8 +57,12 @@ public class LLMReq { .map(dimension -> dimension.getName()) .collect(Collectors.toList())); } - fieldNameList.add(partitionTime.getName()); - fieldNameList.add(primaryKey.getName()); + if (Objects.nonNull(partitionTime)) { + fieldNameList.add(partitionTime.getName()); + } + if (Objects.nonNull(primaryKey)) { + fieldNameList.add(primaryKey.getName()); + } return fieldNameList; } }