topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-14 22:25:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(Chat) Simplify processor in Headless and Chat (#822)

Browse Source 
Co-authored-by: jolunoluo
This commit is contained in:
LXW
2024-03-15 12:39:14 +08:00
committed by GitHub
parent aad8bb4367
commit 988a025cdf
19 changed files with 85 additions and 186 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/ParseResp.java
View File

@@ -3,7 +3,11 @@ package com.tencent.supersonic.headless.api.pojo.response;
import com.google.common.collect.Lists;
import com.tencent.supersonic.headless.api.pojo.SemanticParseInfo;
import lombok.Data;
import org.apache.commons.collections4.CollectionUtils;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
@Data
public class ParseResp {
@@ -20,4 +24,33 @@ public class ParseResp {
FAILED
}
public List<SemanticParseInfo> getSelectedParses() {
selectedParses = selectedParses.stream()
.sorted(Comparator.comparingDouble(SemanticParseInfo::getScore))
.collect(Collectors.toList());
generateParseInfoId(selectedParses);
return selectedParses;
}
public ParseState getState() {
if (CollectionUtils.isNotEmpty(selectedParses)) {
this.state = ParseResp.ParseState.COMPLETED;
} else {
this.state = ParseState.FAILED;
}
return this.state;
}
private void generateParseInfoId(List<SemanticParseInfo> selectedParses) {
for (int i = 0; i < selectedParses.size(); i++) {
SemanticParseInfo parseInfo = selectedParses.get(i);
parseInfo.setId(i + 1);
}
}
public ParseResp(Integer chatId, String queryText) {
this.chatId = chatId;
this.queryText = queryText;
parseTimeCost.setParseStartTime(System.currentTimeMillis());
}
}