Commit 5e985d57 authored by zhouwei's avatar zhouwei

Merge branch 'dev' into 'master'

新增新闻翻译和标签统一接口

See merge request !8
parents f8e6cacc 66bdacbf
...@@ -25,4 +25,9 @@ public class AIController { ...@@ -25,4 +25,9 @@ public class AIController {
public Response<AIResponse> naturalDisasterTag(@RequestBody AIRequestDTO AIRequestDTO) { public Response<AIResponse> naturalDisasterTag(@RequestBody AIRequestDTO AIRequestDTO) {
return Response.SUCCESS(aiService.naturalDisasterTag(AIRequestDTO)); return Response.SUCCESS(aiService.naturalDisasterTag(AIRequestDTO));
} }
@PostMapping("/news/tag/translate")
public Response<AIResponse> newsTagAndTranslate(@RequestBody AIRequestDTO AIRequestDTO) {
return Response.SUCCESS(aiService.newsTagAndTranslate(AIRequestDTO));
}
} }
...@@ -7,5 +7,5 @@ public class Choice { ...@@ -7,5 +7,5 @@ public class Choice {
private int index; private int index;
private Message message; private Message message;
private Object logprobs; // logprobs is null in the JSON private Object logprobs; // logprobs is null in the JSON
private String finishReason; private String finish_reason;
} }
package com.nanyan.securitylink.service; package com.nanyan.securitylink.service;
import com.alibaba.fastjson.JSONObject;
import com.nanyan.securitylink.dto.AIRequestDTO; import com.nanyan.securitylink.dto.AIRequestDTO;
import com.nanyan.securitylink.vo.AIResponse; import com.nanyan.securitylink.vo.AIResponse;
import com.nanyan.securitylink.vo.CodeVO;
public interface AIService { public interface AIService {
AIResponse translate(AIRequestDTO AIRequestDTO); AIResponse<String> translate(AIRequestDTO AIRequestDTO);
AIResponse newsTags(AIRequestDTO AIRequestDTO); AIResponse<CodeVO> newsTags(AIRequestDTO AIRequestDTO);
AIResponse naturalDisasterTag(AIRequestDTO aiRequestDTO); AIResponse<CodeVO> naturalDisasterTag(AIRequestDTO aiRequestDTO);
AIResponse<JSONObject> newsTagAndTranslate(AIRequestDTO aiRequestDTO);
} }
...@@ -3,13 +3,14 @@ package com.nanyan.securitylink.vo; ...@@ -3,13 +3,14 @@ package com.nanyan.securitylink.vo;
import lombok.Data; import lombok.Data;
@Data @Data
public class AIResponse { public class AIResponse<T> {
private String event; private String event;
private String task_id; private String task_id;
private String id; private String id;
private String message_id; private String message_id;
private String mode; private String mode;
private String answer; private String answer;
private ResultVO outputs; private ResultVO<T> outputs;
private long created_at; private long created_at;
private String finish_reason;
} }
package com.nanyan.securitylink.vo;
import lombok.Data;
import java.util.List;
@Data
public class AnswerVO<T> {
List<T> result;
String finish_reason;
}
...@@ -5,6 +5,7 @@ import lombok.Data; ...@@ -5,6 +5,7 @@ import lombok.Data;
import java.util.List; import java.util.List;
@Data @Data
public class ResultVO { public class ResultVO<T> {
List<CodeVO> result; List<T> result;
AnswerVO<T> answer;
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment