diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java index a263096fb..71e9c613c 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java @@ -27,6 +27,7 @@ import org.springframework.validation.annotation.Validated; import javax.annotation.Resource; import java.util.*; +import java.util.regex.Pattern; import java.util.stream.Collectors; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; @@ -313,6 +314,10 @@ public class ModelServiceServiceImpl implements ModelServiceService { List res = new ArrayList<>(); if (fileList != null){ for (String fileName : fileList) { + Pattern extensionPattern = Pattern.compile("\\.[a-zA-Z0-9]+$"); + if (!extensionPattern.matcher(fileName).find()){ + continue; + } res.add(modelFileDownload + baseModelName +"/" + fileName); } }