Bläddra i källkod

feat "接入字典服务+学院AI服务"

Leonezhurui 2 år sedan
förälder
incheckning
a798ce4491

+ 2 - 0
src/main/java/com/njuzr/eaibackend/service/AIDialogueService.java

@@ -171,6 +171,8 @@ public class AIDialogueService {
         AIRequestService.AIResponse res;
         if (Objects.equals(modelType, "ChatGLM3")) {
             res = aiRequestService.requestChatGLM(aidto.getMessages());
+        } else if (modelType.equals("ChatGLM4")) {
+            res = aiRequestService.requestChatGLM4(aidto.getMessages());
         } else if (modelType.equals("QWen14B")) {
             res = aiRequestService.requestQWen(aidto.getMessages());
         } else if (modelType.equals("ChatGPT")) {

+ 14 - 0
src/main/java/com/njuzr/eaibackend/service/AIRequestService.java

@@ -64,6 +64,20 @@ public class AIRequestService {
         }
     }
 
+
+    public AIResponse requestChatGLM4(List<AIEntry> messages) {
+        MyRequestObject requestObject = new MyRequestObject("Qwen2-7B", messages);
+        String chatglmUrl = "http://10.58.0.2:8000";
+
+        try {
+            return chatglmClient.post(chatglmUrl+prefix, requestObject, AIResponse.class);
+        } catch (Exception e) {
+            log.error("WebClient请求失败,AI请求失败~");
+            throw new MyException(HttpStatus.INTERNAL_SERVER_ERROR.value(), HttpStatus.INTERNAL_SERVER_ERROR.getReasonPhrase()+":"+"服务器请求AI出错");
+        }
+    }
+
+
     public AIResponse requestQWen(List<AIEntry> messages) {
         MyRequestObject requestObject = new MyRequestObject("QWen14B", messages);
         String qwenUrl = "http://10.58.0.2:6679";

+ 1 - 1
src/main/java/com/njuzr/eaibackend/service/TranslateService.java

@@ -40,7 +40,7 @@ public class TranslateService {
     }
 
     public String translate(Long assignmentId, Long userId, TranslationDTO translationDTO) {
-        MyResponseObject response = webClientUtil.post("http://localhost:5000/api/translate", translationDTO, MyResponseObject.class);
+        MyResponseObject response = webClientUtil.post("http://139.196.252.184:5000/api/translate", translationDTO, MyResponseObject.class);
 
         if (Objects.equals(response.getCode(), 200)) {
             Translation translation = new Translation();

+ 1 - 1
src/main/resources/application-dev.yaml

@@ -5,7 +5,7 @@ spring:
   datasource:
     driver-class-name: com.mysql.cj.jdbc.Driver
     username: root
-    password: 123321123b
+    password: Zr200012
     url: jdbc:mysql://localhost:3306/eai?useSSL=false&serverTimezone=UTC&useUnicode=true&characterEncoding=utf8
   data:
     mongodb: