diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/AiChatController.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/AiChatController.java index 78a6d04..4c46ee3 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/AiChatController.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/AiChatController.java @@ -1,6 +1,7 @@ package com.electromagnetic.industry.software.manage.controller; import cn.hutool.core.io.FileUtil; +import cn.hutool.core.util.StrUtil; import com.electromagnetic.industry.software.common.resp.ElectromagneticResult; import com.electromagnetic.industry.software.common.util.ElectromagneticResultUtil; import com.electromagnetic.industry.software.manage.pojo.req.ChatQueryDTO; @@ -29,7 +30,7 @@ public class AiChatController { @RequestMapping(value = "addKnowledge") public ElectromagneticResult addKnowledge(@RequestParam("file") MultipartFile file) throws IOException { String fileType = FileUtil.extName(file.getOriginalFilename()); - if (!fileType.equals("pdf")) { + if (!StrUtil.equals(fileType, "pdf")) { return ElectromagneticResultUtil.fail("-1", "当前仅支持pdf格式文件"); } return aiChatService.addKnowledge(file);