f2f update

src/main/java/cn/com/poc/ai_data_audit/entity/AIDataAuditMatchFileNameResponse.java

@@ -10,6 +10,7 @@ public class AIDataAuditMatchFileNameResponse {
     String templateCode;
     String templateName;
     String uploadFileName;
+    String uploadFileCode;
 
     public String getTemplateCode() {
         return templateCode;
@@ -34,4 +35,12 @@ public class AIDataAuditMatchFileNameResponse {
     public void setUploadFileName(String uploadFileName) {
         this.uploadFileName = uploadFileName;
     }
+
+    public String getUploadFileCode() {
+        return uploadFileCode;
+    }
+
+    public void setUploadFileCode(String uploadFileCode) {
+        this.uploadFileCode = uploadFileCode;
+    }
 }

src/main/java/cn/com/poc/ai_data_audit/repository/FileNameResult.java

+package cn.com.poc.ai_data_audit.repository;
+
+/**
+ * @Author：Roger Wu
+ * @name：FileNameResult
+ * @Date：2025-10-31 21:51
+ */
+public class FileNameResult {
+
+    String uploadFileName;
+
+    String uploadFileCode;
+
+
+    public String getUploadFileName() {
+        return uploadFileName;
+    }
+
+    public void setUploadFileName(String uploadFileName) {
+        this.uploadFileName = uploadFileName;
+    }
+
+    public String getUploadFileCode() {
+        return uploadFileCode;
+    }
+
+    public void setUploadFileCode(String uploadFileCode) {
+        this.uploadFileCode = uploadFileCode;
+    }
+}

src/main/java/cn/com/poc/ai_data_audit/service/impl/DataAuditServiceImpl.java

@@ -4,12 +4,14 @@ import cn.com.poc.agent_application.entity.BizAgentApplicationDialoguesRecordEnt
 import cn.com.poc.agent_application.entity.BizAgentApplicationGcConfigEntity;
 import cn.com.poc.agent_application.service.BizAgentApplicationDialoguesRecordService;
 import cn.com.poc.agent_application.service.BizAgentApplicationGcConfigService;
+import cn.com.poc.ai_data_audit.constants.DataAuditEnum;
 import cn.com.poc.ai_data_audit.dto.DataAuditCallDto;
 import cn.com.poc.ai_data_audit.dto.DataAuditResult;
 import cn.com.poc.ai_data_audit.dto.FileUploadDto;
 import cn.com.poc.ai_data_audit.dto.FileUploadReturnDto;
 import cn.com.poc.ai_data_audit.entity.AIDataAuditMatchFileNameResponse;
 import cn.com.poc.ai_data_audit.entity.BizAiDataAuditFileEntity;
+import cn.com.poc.ai_data_audit.repository.FileNameResult;
 import cn.com.poc.ai_data_audit.service.BizAiDataAuditFileService;
 import cn.com.poc.ai_data_audit.service.DataAuditService;
 import cn.com.poc.common.constant.CommonConstant;
@@ -73,6 +75,10 @@ public class DataAuditServiceImpl extends BaseServiceImpl
                 .map(item -> item.getFileName())
                 .collect(Collectors.toList());
 
+        if (!fileNames.contains("Form.pdf")) {
+            throw new Exception("please upload the Form.pdf");
+        }
+
         CompletableFuture<List<AIDataAuditMatchFileNameResponse>> extractFileFuture = CompletableFuture.supplyAsync(() -> {
             try {
                 return matchOriginalTemplateAndUploadFile(fileNames);
@@ -135,7 +141,7 @@ public class DataAuditServiceImpl extends BaseServiceImpl
             fileUploadExtensionDto.setOrginalTemplateName("Work Permit Extension Details");
             fileUploadExtensionDto.setUploadFileName("");
             fileUploadReturnDtoList.add(fileUploadExtensionDto);
-        }else{
+        } else {
             throw new Exception("please upload the Form.pdf");
         }
 
@@ -185,9 +191,9 @@ public class DataAuditServiceImpl extends BaseServiceImpl
         BizAiDataAuditFileEntity bizAiDataAuditFileEntity = bizAiDataAuditFileService.get(dataAuditCallDto.getFileId());
         BizFileOcrCacheEntity example = new BizFileOcrCacheEntity();
         example.setMd5(bizAiDataAuditFileEntity.getMd5());
-        List<BizFileOcrCacheEntity> bizFileOcrCacheEntityList = bizFileOcrCacheService.findByExample(example,null);
+        List<BizFileOcrCacheEntity> bizFileOcrCacheEntityList = bizFileOcrCacheService.findByExample(example, null);
 
-        if(CollectionUtils.isEmpty(bizFileOcrCacheEntityList)){
+        if (CollectionUtils.isEmpty(bizFileOcrCacheEntityList)) {
             throw new BusinessException("can not file the file");
         }
 
@@ -289,7 +295,18 @@ public class DataAuditServiceImpl extends BaseServiceImpl
         }
 
         fileNames = fileNames.stream().filter(item -> !item.contains("Form.pdf")).collect(Collectors.toList());
-        String prompt = documentDialoguePrompt.getConfigSystem().replace("${uploadFileName}", JSON.toJSONString(fileNames));
+
+        List<FileNameResult> fileNameList = new ArrayList<>();
+        for (String fileName : fileNames) {
+            FileNameResult fileNameResult = new FileNameResult();
+            fileNameResult.setUploadFileName(fileName);
+            fileNameResult.setUploadFileCode(RandomUtil.produceString(10));
+            fileNameList.add(fileNameResult);
+        }
+        Map<String, List<FileNameResult>> fileCodeMap = fileNameList.stream().collect(Collectors.groupingBy(item -> item.getUploadFileCode()));
+
+
+        String prompt = documentDialoguePrompt.getConfigSystem().replace("${#uploadFileName}", JSON.toJSONString(fileNameList));
 
         //组装请求参数
         List<Message> messages = buildMessages("", null, prompt);
@@ -309,6 +326,15 @@ public class DataAuditServiceImpl extends BaseServiceImpl
         if (ObjectUtil.isEmpty(plaintiff)) {
             throw new BusinessException("提取失败");
         }
+
+        for (AIDataAuditMatchFileNameResponse aIDataAuditMatchFileNameResponse : aIDataAuditMatchFileNameResponses) {
+            if (StringUtils.isNotBlank(aIDataAuditMatchFileNameResponse.getUploadFileCode())) {
+                String uploadFileName = fileCodeMap.get(aIDataAuditMatchFileNameResponse.getUploadFileCode()).get(0).getUploadFileName();
+                aIDataAuditMatchFileNameResponse.setUploadFileName(StringUtils.isBlank(uploadFileName) ? aIDataAuditMatchFileNameResponse.getUploadFileName() : uploadFileName);
+                aIDataAuditMatchFileNameResponse.setUploadFileCode("");
+            }
+            aIDataAuditMatchFileNameResponse.setTemplateName(DataAuditEnum.getTemplateNameByCode(aIDataAuditMatchFileNameResponse.getTemplateCode()));
+        }
         return aIDataAuditMatchFileNameResponses;
     }