diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java b/chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java index d2e201e5b..6fce496e8 100644 --- a/chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java +++ b/chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java @@ -160,7 +160,12 @@ public class QueryServiceImpl implements QueryService { return candidateParses; } int selectParseSize = selectedParses.size(); + Set selectParseScoreSet = selectedParses.stream() + .map(SemanticParseInfo::getScore).collect(Collectors.toSet()); int candidateParseSize = 5 - selectParseSize; + candidateParses = candidateParses.stream() + .filter(candidateParse -> !selectParseScoreSet.contains(candidateParse.getScore())) + .collect(Collectors.toList()); SemanticParseInfo semanticParseInfo = selectedParses.get(0); Long modelId = semanticParseInfo.getModelId(); if (modelId == null || modelId <= 0) {