From 881e5bf5c9b2a158c487e3a03205d8f420916175 Mon Sep 17 00:00:00 2001
From: chenxudong <chenxudong@cqida.cn>
Date: Tue, 22 Apr 2025 10:02:19 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=86AI=E4=B8=AD=E6=94=AF=E6=8C=81=E4=B8=8A?=
 =?UTF-8?q?=E4=BC=A0=E7=9A=84=E6=96=87=E4=BB=B6=E7=B1=BB=E5=9E=8B=E8=BD=AC?=
 =?UTF-8?q?=E7=A7=BB=E5=88=B0=E9=85=8D=E7=BD=AE=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../software/manage/config/ElePropertyConfig.java        | 9 +++++++++
 .../software/manage/service/serviceimpl/ChatService.java | 3 +--
 .../src/main/resources/application.properties            | 1 +
 3 files changed, 11 insertions(+), 2 deletions(-)

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<String> 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<String> 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