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

src/main/java/com/zzsn/event/controller/FileController.java

@@ -421,10 +421,11 @@ public class FileController {
             long timestamp = System.currentTimeMillis();
             String filename = subjectId + "_" + timestamp + ".xlsx";
             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.setPageSize(1000);
-                String[] arr = new String[]{"得分", "标题", "标题译文", "摘要", "摘要译文", "正文", "正文译文", "作者", "来源", "发布时间", "网址", "命中词"};
+                //String[] arr = new String[]{"得分", "标题", "标题译文", "摘要", "摘要译文", "正文", "正文译文", "作者", "来源", "发布时间", "网址", "命中词"};
+                String[] arr = new String[]{"标题", "标题译文", "摘要", "摘要译文", "正文", "正文译文", "作者", "来源", "发布时间", "网址", "命中词","通用打分","大模型打分","综合打分(通用:大模型=6:4))"};
                 List<String> headers = Arrays.asList(arr);
                 SXSSFWorkbook workbook = new SXSSFWorkbook();
                 for (int i = 1; ; i++) {

src/main/java/com/zzsn/event/vo/es/SpecialInformation.java

@@ -6,6 +6,7 @@ import org.apache.commons.lang3.ObjectUtils;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Optional;
 import java.util.stream.Collectors;
 
 @Data
@@ -208,7 +209,6 @@ public class SpecialInformation {
 
     public List<String> toExcelList(){
         List<String> list = new ArrayList<>();
-        list.add(getRealValue(score));
         list.add(getRealValue(title));
         list.add(getRealValue(titleRaw));
         list.add(getRealValue(summary));
@@ -224,6 +224,24 @@ public class SpecialInformation {
         } else {
             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;
     }