Explorar o código

大模型文档新增请求参数处理

王苗苗 hai 1 mes
pai
achega
233d14dadf

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

@@ -638,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(), chatReq.getModelType());
+            String dataJson = buildBigModelReqForChat(sessionId, historyDates, ipAddr, isStrong, chatReq.getTopP(), chatReq.getTemperature(), tools, useRag, extraStr, chatReq.getPrompt(), chatReq.getModelType(), chatReq.getOnlineSearch());
             log.info("******请求大模型的问答参数为{}", dataJson);
             PredictionsRequest request = PredictionsRequest.newBuilder()
                     .setModelName("slibra_bot")
@@ -836,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, int modelType) {
+    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, Boolean onlineSearch) {
         ChatRequest chatRequest = new ChatRequest();
         chatRequest.setSessionId(sessionId);
         chatRequest.setHistoryDia(historyDates);
@@ -861,6 +861,8 @@ public class GRPCController extends BaseController {
         chatRequest.setPrompt(StringUtils.isBlank(prompt) ? DEFAULT_PROMPT : prompt);
         //2025年02月24日11:41:38 新增的 区分模型的
         chatRequest.setModelType(modelType);
+        //2025年02月27日10:58:16 新增的 是否使用在线搜索
+        chatRequest.setOnlineSearch(onlineSearch);
         return JSON.toJSONString(chatRequest);
     }
 

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

@@ -125,6 +125,18 @@ public class ChatReq extends TXinyiChatRecord {
     @JSONField(name = "modelType")
     private int modelType = 0;
 
+    //2025年02月27日10:55:27  新增字段 是否使用在线搜索
+    @JSONField(name = "online_search")
+    private Boolean onlineSearch = false;
+
+    public Boolean getOnlineSearch() {
+        return onlineSearch;
+    }
+
+    public void setOnlineSearch(Boolean onlineSearch) {
+        this.onlineSearch = onlineSearch;
+    }
+
     public int getModelType() {
         return modelType;
     }

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

@@ -49,16 +49,20 @@ 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了,只有具体工具才传工具名称
     @JSONField(name = "tools")
     private String tools = "false";
 
+    //2025年02月24日11:36:25 新增字段 尚辉那边切换模型使用  1 deepseek  2 wenxin  ,默认是0【调用决策使用】
+    @JSONField(name = "modelType")
+    private int modelType = 0;
+
+    //2025年02月27日10:55:27  新增字段 是否使用在线搜索
+    @JSONField(name = "online_search")
+    private Boolean onlineSearch = false;
+
 
     public static void main(String[] args) {
         //2024年7月26日10:37:50  自定义工单的传参demo给永创