Ver código fonte

问答模块新增新的字段区分哪个模型

王苗苗 1 mês atrás
pai
commit
cb658f2b03

+ 4 - 4
slibra-admin/src/main/java/com/slibra/web/controller/business/GRPCController.java

@@ -42,8 +42,6 @@ import java.util.*;
 
 import static com.slibra.common.constant.MyConstants.*;
 import static com.slibra.common.enums.BusinessEnum.BigModelBizEnum.DECISION_REPORT;
-import static com.slibra.common.enums.BusinessEnum.WarningCategoryEnum.*;
-import static com.slibra.common.enums.BusinessEnum.WarningCategoryEnum.CS_ZL;
 
 /**
  * 告警相关
@@ -640,7 +638,7 @@ public class GRPCController extends BaseController {
 //                else
 //                    tools = tools;
 //            }
-            String dataJson = buildBigModelReqForChat(sessionId, historyDates, ipAddr, isStrong, chatReq.getTopP(), chatReq.getTemperature(), tools, useRag, extraStr, chatReq.getPrompt());
+            String dataJson = buildBigModelReqForChat(sessionId, historyDates, ipAddr, isStrong, chatReq.getTopP(), chatReq.getTemperature(), tools, useRag, extraStr, chatReq.getPrompt(), chatReq.getModelType());
             log.info("******请求大模型的问答参数为{}", dataJson);
             PredictionsRequest request = PredictionsRequest.newBuilder()
                     .setModelName("slibra_bot")
@@ -838,7 +836,7 @@ public class GRPCController extends BaseController {
         return JSON.toJSONString(map);
     }
 
-    private String buildBigModelReqForChat(String sessionId, List<String> historyDates, String ipAddr, int isStrong, Double topP, Double temperature, String tools, boolean useRag, String extraStr, String prompt) {
+    private String buildBigModelReqForChat(String sessionId, List<String> historyDates, String ipAddr, int isStrong, Double topP, Double temperature, String tools, boolean useRag, String extraStr, String prompt, int modelType) {
         ChatRequest chatRequest = new ChatRequest();
         chatRequest.setSessionId(sessionId);
         chatRequest.setHistoryDia(historyDates);
@@ -861,6 +859,8 @@ public class GRPCController extends BaseController {
             chatRequest.setUseRag(false);
         //2024年08月11日16:04:21 允许用户自定义prompt
         chatRequest.setPrompt(StringUtils.isBlank(prompt) ? DEFAULT_PROMPT : prompt);
+        //2025年02月24日11:41:38 新增的 区分模型的
+        chatRequest.setModelType(modelType);
         return JSON.toJSONString(chatRequest);
     }
 

+ 2 - 1
slibra-quartz/src/main/java/com/slibra/quartz/task/RyTask.java

@@ -664,12 +664,13 @@ public class RyTask
         List<TXinyiRobot> tXinyiRobots;
         try {
             result = HttpUtil.get(HTTP_GET_URL_ROBOT_DATA + lastTime, HTTP_GET_TIME_OUT_MILLION_SECONDS);
+            log.info("调用化验室同步接口返回的结果为{}", result);
             tXinyiRobots = JSON.parseArray(result, TXinyiRobot.class);
         } catch (Exception e) {
             log.error("调用获取化验室数据接口异常,异常信息为{}", e.getMessage());
             return;
         }
-        log.info("查询到此次需要同步的数据为{}", result);
+        log.info("查询到此次需要同步的数据为{}", JSON.toJSONString(tXinyiRobots));
 //        System.out.println("-------------");
         //主
         List<TXinyiRobot> needHandleList = new ArrayList<>();

+ 12 - 0
slibra-system/src/main/java/com/slibra/business/req/ChatReq.java

@@ -121,6 +121,18 @@ public class ChatReq extends TXinyiChatRecord {
     //2024年08月11日16:01:27 允许用户自定义prompt
     private String prompt;
 
+    //2025年02月24日11:36:25 新增字段 尚辉那边切换模型使用  1 deepseek  2 wenxin  ,默认是0【调用决策使用】
+    @JSONField(name = "modelType")
+    private int modelType = 0;
+
+    public int getModelType() {
+        return modelType;
+    }
+
+    public void setModelType(int modelType) {
+        this.modelType = modelType;
+    }
+
     public String getPrompt() {
         return prompt;
     }

+ 4 - 0
slibra-system/src/main/java/com/slibra/business/req/ChatRequest.java

@@ -49,6 +49,10 @@ public class ChatRequest {
     @JSONField(name = "strengthen")
     private Boolean strengthen;
 
+    //2025年02月24日11:36:25 新增字段 尚辉那边切换模型使用  1 deepseek  2 wenxin  ,默认是0【调用决策使用】
+    @JSONField(name = "modelType")
+    private int modelType = 0;
+
     //2024年7月3日16:18:48 新增参数
     //2024年7月8日13:37:58 逻辑调整:截止到目前,只有专家问答才传true(可能走工具调用),其他都是false(都不走工具)。
     //2024年08月11日15:49:04 调整:专家问答目前也是传false了,只有具体工具才传工具名称