提交 054916b6 作者: 925993793@qq.com

【ai资讯/自定义专题-研究中心】导出excel 增加通用打分、大模型打分字段

上级 d7fa29bb
...@@ -421,10 +421,11 @@ public class FileController { ...@@ -421,10 +421,11 @@ public class FileController {
long timestamp = System.currentTimeMillis(); long timestamp = System.currentTimeMillis();
String filename = subjectId + "_" + timestamp + ".xlsx"; String filename = subjectId + "_" + timestamp + ".xlsx";
try { try {
String[] fetchFields = new String[]{"score", "title", "titleRaw", "summary", "summaryRaw", "content", "contentRaw", "author", "origin", "publishDate", "sourceAddress", "hitWords"}; String[] fetchFields = new String[]{"score", "title", "titleRaw", "summary", "summaryRaw", "content", "contentRaw", "author", "origin", "publishDate", "sourceAddress", "hitWords","modelScores"};
searchCondition.setFetchFields(fetchFields); searchCondition.setFetchFields(fetchFields);
searchCondition.setPageSize(1000); searchCondition.setPageSize(1000);
String[] arr = new String[]{"得分", "标题", "标题译文", "摘要", "摘要译文", "正文", "正文译文", "作者", "来源", "发布时间", "网址", "命中词"}; //String[] arr = new String[]{"得分", "标题", "标题译文", "摘要", "摘要译文", "正文", "正文译文", "作者", "来源", "发布时间", "网址", "命中词"};
String[] arr = new String[]{"标题", "标题译文", "摘要", "摘要译文", "正文", "正文译文", "作者", "来源", "发布时间", "网址", "命中词","通用打分","大模型打分","综合打分(通用:大模型=6:4))"};
List<String> headers = Arrays.asList(arr); List<String> headers = Arrays.asList(arr);
SXSSFWorkbook workbook = new SXSSFWorkbook(); SXSSFWorkbook workbook = new SXSSFWorkbook();
for (int i = 1; ; i++) { for (int i = 1; ; i++) {
......
...@@ -6,6 +6,7 @@ import org.apache.commons.lang3.ObjectUtils; ...@@ -6,6 +6,7 @@ import org.apache.commons.lang3.ObjectUtils;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@Data @Data
...@@ -208,7 +209,6 @@ public class SpecialInformation { ...@@ -208,7 +209,6 @@ public class SpecialInformation {
public List<String> toExcelList(){ public List<String> toExcelList(){
List<String> list = new ArrayList<>(); List<String> list = new ArrayList<>();
list.add(getRealValue(score));
list.add(getRealValue(title)); list.add(getRealValue(title));
list.add(getRealValue(titleRaw)); list.add(getRealValue(titleRaw));
list.add(getRealValue(summary)); list.add(getRealValue(summary));
...@@ -224,6 +224,24 @@ public class SpecialInformation { ...@@ -224,6 +224,24 @@ public class SpecialInformation {
} else { } else {
list.add(""); list.add("");
} }
if (CollectionUtils.isNotEmpty(modelScores)) {
Optional<ModelScore> commonFirst = modelScores.stream().filter(modelScore -> "Mark-Service".equals(modelScore.getServiceName())).findFirst();
if (commonFirst.isPresent()) {
list.add(getRealValue(commonFirst.get().getScore()));
} else {
list.add("");
}
Optional<ModelScore> llmFirst = modelScores.stream().filter(modelScore -> "llm-score-model".equals(modelScore.getServiceName())).findFirst();
if (llmFirst.isPresent()) {
list.add(getRealValue(llmFirst.get().getScore()));
} else {
list.add("");
}
} else {
list.add("");
list.add("");
}
list.add(getRealValue(score));
return list; return list;
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论