diff --git a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SchemaElement.java b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SchemaElement.java index 2fb27fafb..f7105b86b 100644 --- a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SchemaElement.java +++ b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SchemaElement.java @@ -1,6 +1,7 @@ package com.tencent.supersonic.chat.api.pojo; import com.google.common.base.Objects; +import com.google.common.collect.Lists; import lombok.AllArgsConstructor; import lombok.Builder; import lombok.Data; @@ -51,4 +52,7 @@ public class SchemaElement implements Serializable { return Objects.hashCode(view, id, name, bizName, type); } + public List getModelNames() { + return Lists.newArrayList(name); + } } diff --git a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SemanticParseInfo.java b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SemanticParseInfo.java index f4f8b783d..36e146d29 100644 --- a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SemanticParseInfo.java +++ b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/SemanticParseInfo.java @@ -78,4 +78,8 @@ public class SemanticParseInfo { return view.getView(); } + public SchemaElement getModel() { + return view; + } + } diff --git a/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/ViewResp.java b/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/ViewResp.java index c6cbd8824..143bae90e 100644 --- a/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/ViewResp.java +++ b/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/ViewResp.java @@ -36,7 +36,7 @@ public class ViewResp extends SchemaItem { } public List getAllDimensions() { - return getViewModelConfigs().stream().map(ViewModelConfig::getMetrics) + return getViewModelConfigs().stream().map(ViewModelConfig::getDimensions) .flatMap(Collection::stream).collect(Collectors.toList()); }