diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/config/ElePropertyConfig.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/config/ElePropertyConfig.java index 4fdacd5..944b81c 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/config/ElePropertyConfig.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/config/ElePropertyConfig.java @@ -6,6 +6,8 @@ import lombok.Getter; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Component; +import java.util.List; + @Component public class ElePropertyConfig { @@ -73,6 +75,13 @@ public class ElePropertyConfig { @Value("${prjRootPath}") private String prjRootPath; + @Value("${allow.upload.file.types}") + private String allowUploadFileTypes; + + public List getAiAllowUploadFileTypes() { + return List.of(allowUploadFileTypes.split(",")); + } + public String getEleTmpPath() { if (EleCommonUtil.isWinOs()) { return winPrefix + prjRootPath + eleTmpPath; diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java index 0cab5ad..4d14872 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java @@ -49,7 +49,6 @@ import java.util.*; @Service public class ChatService { - private static final List ALLOWS_FILE_TYPES = List.of("doc", "docx", "txt", "csv", "xls", "xlsx", "pdf", "ppt", "pptx"); @Resource private VectorStore vectorStore; @Resource @@ -90,7 +89,7 @@ public class ChatService { } String fileType = FileUtil.extName(file.getOriginalFilename()); - if (!ALLOWS_FILE_TYPES.contains(fileType)) { + if (!elePropertyConfig.getAiAllowUploadFileTypes().contains(fileType)) { return ElectromagneticResultUtil.fail("-1", StrFormatter.format("当前格式 {} 不支持", fileType)); } diff --git a/electrmangnetic/src/main/resources/application.properties b/electrmangnetic/src/main/resources/application.properties index 4822c69..815d69f 100644 --- a/electrmangnetic/src/main/resources/application.properties +++ b/electrmangnetic/src/main/resources/application.properties @@ -24,6 +24,7 @@ spring.ai.vectorstore.elasticsearch.initialize-schema=true spring.ai.vectorstore.elasticsearch.index-name=rag spring.ai.vectorstore.elasticsearch.dimensions=768 spring.ai.vectorstore.elasticsearch.similarity=cosine +allow.upload.file.types=doc, docx, txt, csv, xls, xlsx, pdf, ppt", pptx # mybatis mybatis-plus.mapper-locations=classpath:sqlmapper/*.xml mybatis-plus.configuration.log-impl=org.apache.ibatis.logging.stdout.StdOutImpl