topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 11:07:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(project) dsl support get topN dimension/metric filter by modelId (#106)

Browse Source
This commit is contained in:
lexluo09
2023-09-20 12:09:42 +08:00
committed by GitHub
parent 98656eb445
commit eee82dea07
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/dsl/LLMDslParser.java
View File

@@ -411,12 +411,14 @@ public class LLMDslParser implements SemanticParser {
Map<Long, String> itemIdToName = getItemIdToName(modelId, semanticSchema); Map<Long, String> itemIdToName = getItemIdToName(modelId, semanticSchema);
Set<String> results = semanticSchema.getDimensions().stream() Set<String> results = semanticSchema.getDimensions().stream()
.filter(schemaElement -> modelId.equals(schemaElement.getModel()))
.sorted(Comparator.comparing(SchemaElement::getUseCnt).reversed()) .sorted(Comparator.comparing(SchemaElement::getUseCnt).reversed())
.limit(llmParserConfig.getDimensionTopN()) .limit(llmParserConfig.getDimensionTopN())
.map(entry -> entry.getName()) .map(entry -> entry.getName())
.collect(Collectors.toSet()); .collect(Collectors.toSet());
Set<String> metrics = semanticSchema.getMetrics().stream() Set<String> metrics = semanticSchema.getMetrics().stream()
.filter(schemaElement -> modelId.equals(schemaElement.getModel()))
.sorted(Comparator.comparing(SchemaElement::getUseCnt).reversed()) .sorted(Comparator.comparing(SchemaElement::getUseCnt).reversed())
.limit(llmParserConfig.getMetricTopN()) .limit(llmParserConfig.getMetricTopN())
.map(entry -> entry.getName()) .map(entry -> entry.getName())