From eee82dea077599d214c028ba722bafff3735a35d Mon Sep 17 00:00:00 2001 From: lexluo09 <39718951+lexluo09@users.noreply.github.com> Date: Wed, 20 Sep 2023 12:09:42 +0800 Subject: [PATCH] (improvement)(project) dsl support get topN dimension/metric filter by modelId (#106) --- .../tencent/supersonic/chat/parser/llm/dsl/LLMDslParser.java | 2 ++ 1 file changed, 2 insertions(+) diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/dsl/LLMDslParser.java b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/dsl/LLMDslParser.java index 6acceb3d0..f286c1177 100644 --- a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/dsl/LLMDslParser.java +++ b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/dsl/LLMDslParser.java @@ -411,12 +411,14 @@ public class LLMDslParser implements SemanticParser { Map itemIdToName = getItemIdToName(modelId, semanticSchema); Set results = semanticSchema.getDimensions().stream() + .filter(schemaElement -> modelId.equals(schemaElement.getModel())) .sorted(Comparator.comparing(SchemaElement::getUseCnt).reversed()) .limit(llmParserConfig.getDimensionTopN()) .map(entry -> entry.getName()) .collect(Collectors.toSet()); Set metrics = semanticSchema.getMetrics().stream() + .filter(schemaElement -> modelId.equals(schemaElement.getModel())) .sorted(Comparator.comparing(SchemaElement::getUseCnt).reversed()) .limit(llmParserConfig.getMetricTopN()) .map(entry -> entry.getName())