Răsfoiți Sursa

主动调用决策 因调用调整,需和之前调用大模型传参norm一致

wangmiaomiao 8 luni în urmă
părinte
comite
9ed05a1e6e

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

@@ -39,6 +39,8 @@ 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;
 
 /**
  * 告警相关
@@ -211,7 +213,17 @@ public class GRPCController extends BaseController {
 
     private String buildBigModelReqForDecision(ChatReq chatReq, String feedback, String simulate, String sessionId, List<DecisionReq> decisionReqs, TXinyiWarningRecord xinyiWarningRecord, Double topP, Double temperature) {
         PolicyReq policyReq = new PolicyReq();
-        policyReq.setNorm(chatReq.getCategory());
+        //2024年7月11日17:57:53 因为化验室的改了  但是 调用决策还得用原来的几个值
+        String category = xinyiWarningRecord.getCategory();
+        if(ROBOT_XSY_1.getCode().equalsIgnoreCase(category)){
+            category = CS_ZD.getCode();
+        } else if (ROBOT_XSY_2.getCode().equalsIgnoreCase(category)) {
+            category = CS_ZD.getCode();
+        }else if (ROBOT_ECCZLSY.getCode().equalsIgnoreCase(category)) {
+            category = CS_ZL.getCode();
+        }
+        policyReq.setNorm(category);
+//        policyReq.setNorm(chatReq.getCategory());//2024年7月12日16:23:24  因为有转义。。。
         policyReq.setFeedback(StringUtils.isBlank(feedback) ? null : JSON.parseObject(feedback, Map.class));
         policyReq.setSimulate(StringUtils.isBlank(simulate) ? null : JSON.parseObject(simulate, Map.class));
         policyReq.setSessionId(sessionId);