新增自然灾害打标签接口

src/main/java/com/nanyan/securitylink/controller/AIController.java

 package com.nanyan.securitylink.controller;
 
-import com.nanyan.securitylink.dto.TranslateDTO;
+import com.nanyan.securitylink.dto.AIRequestDTO;
 import com.nanyan.securitylink.service.AIService;
 import com.nanyan.securitylink.vo.AIResponse;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -14,11 +14,15 @@ public class AIController {
     @Autowired
     AIService aiService;
     @PostMapping("/translate")
-    public Response<AIResponse> translate(@RequestBody TranslateDTO translateDTO) {
-        return Response.SUCCESS(aiService.translate(translateDTO));
+    public Response<AIResponse> translate(@RequestBody AIRequestDTO AIRequestDTO) {
+        return Response.SUCCESS(aiService.translate(AIRequestDTO));
     }
-    @PostMapping("/tag")
-    public Response<AIResponse> newsTag(@RequestBody TranslateDTO translateDTO) {
-        return Response.SUCCESS(aiService.newsTags(translateDTO));
+    @PostMapping("/news/tag")
+    public Response<AIResponse> newsTag(@RequestBody AIRequestDTO AIRequestDTO) {
+        return Response.SUCCESS(aiService.newsTags(AIRequestDTO));
+    }
+    @PostMapping("/natural/disaster/tag")
+    public Response<AIResponse> naturalDisasterTag(@RequestBody AIRequestDTO AIRequestDTO) {
+        return Response.SUCCESS(aiService.naturalDisasterTag(AIRequestDTO));
     }
 }

src/main/java/com/nanyan/securitylink/dto/TranslateInputDTO.java → src/main/java/com/nanyan/securitylink/dto/AIInputDTO.java

@@ -3,8 +3,9 @@ package com.nanyan.securitylink.dto;
 import lombok.Data;
 
 @Data
-public class TranslateInputDTO {
+public class AIInputDTO {
     String record_json;
     String msg_info;
+    String warn_info;
     String language;
 }

src/main/java/com/nanyan/securitylink/dto/TranslateDTO.java → src/main/java/com/nanyan/securitylink/dto/AIRequestDTO.java

@@ -3,8 +3,8 @@ package com.nanyan.securitylink.dto;
 import lombok.Data;
 
 @Data
-public class TranslateDTO {
-    TranslateInputDTO inputs;
+public class AIRequestDTO {
+    AIInputDTO inputs;
     String response_mode;
     String user;
 }

src/main/java/com/nanyan/securitylink/service/AIService.java

 package com.nanyan.securitylink.service;
 
-import com.nanyan.securitylink.dto.TranslateDTO;
+import com.nanyan.securitylink.dto.AIRequestDTO;
 import com.nanyan.securitylink.vo.AIResponse;
 
 public interface AIService {
-    AIResponse translate(TranslateDTO translateDTO);
-    AIResponse newsTags(TranslateDTO translateDTO);
+    AIResponse translate(AIRequestDTO AIRequestDTO);
+    AIResponse newsTags(AIRequestDTO AIRequestDTO);
+
+    AIResponse naturalDisasterTag(AIRequestDTO aiRequestDTO);
 }