(fix)(chat)多个数据集也只保留召回的第一个结果

这里的多数据集其实对召回的结果没有影响,取第一个就好了
This commit is contained in:
QJ_wonder
2025-05-27 14:42:34 +08:00
committed by GitHub
parent d275a145d5
commit 449fdf180f

View File

@@ -22,6 +22,9 @@ public class NL2PluginParser implements ChatQueryParser {
public void parse(ParseContext parseContext) {
pluginRecognizers.forEach(pluginRecognizer -> {
pluginRecognizer.recognize(parseContext);
if (parseContext.getResponse().getSelectedParses().size()>=2){
parseContext.getResponse().setSelectedParses(parseContext.getResponse().getSelectedParses().subList(0, 1));
}
log.info("{} recallResult:{}", pluginRecognizer.getClass().getSimpleName(),
JsonUtil.toString(parseContext.getResponse()));
});