文件保存修改

takai-bigmodel/src/main/java/com/takai/bigmodel/service/impl/BigModelServiceImpl.java

@@ -598,7 +598,7 @@ public class BigModelServiceImpl implements IBigModelService
             }
 
             if(dto.getDocument() != null && !"".equals(dto.getDocument().getId())){
-                FileInfo fileInfo = fileInfoMapper.getFileInfoById(dto.getDocument().getId(), dialogId, dto.getId());
+                FileInfo fileInfo = fileInfoMapper.getFileInfoById(dto.getDocument().getId(), dialogId, null);
                 if(fileInfo == null && "user".equals(dto.getRole())) {
                     FileInfo params = new FileInfo();
                     params.setId(dto.getDocument().getId());
@@ -646,11 +646,13 @@ public class BigModelServiceImpl implements IBigModelService
         List<DialogRespDTO> detail = dialogMapper.selectDialogDetail(dialogId);
         if(detail != null && detail.size() != 0){
             for(DialogRespDTO dto : detail){
-                FileInfo fileInfo = fileInfoMapper.getFileInfoById(null, dto.getDialog_id(), dto.getDid());
-                if(fileInfo != null){
-                    DocumentDTO fileInfoDTO = new DocumentDTO();
-                    BeanUtils.copyProperties(fileInfo, fileInfoDTO);
-                    dto.setDocument(fileInfoDTO);
+                if("user".equals(dto.getType())){
+                    FileInfo fileInfo = fileInfoMapper.getFileInfoById(null, dto.getDialog_id(), null);
+                    if(fileInfo != null){
+                        DocumentDTO fileInfoDTO = new DocumentDTO();
+                        BeanUtils.copyProperties(fileInfo, fileInfoDTO);
+                        dto.setDocument(fileInfoDTO);
+                    }
                 }
             }
         }