wangmiaomiao 11 ヶ月 前
コミット
4f3d321f7c

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

@@ -138,7 +138,7 @@ public class GRPCController extends BaseController {
                     .usePlaintext()
                     .usePlaintext()
                     .build();
                     .build();
             InferenceAPIsServiceGrpc.InferenceAPIsServiceBlockingStub stub = InferenceAPIsServiceGrpc.newBlockingStub(channel);
             InferenceAPIsServiceGrpc.InferenceAPIsServiceBlockingStub stub = InferenceAPIsServiceGrpc.newBlockingStub(channel);
-            dataJson = "{\"bot_id\":\"b00001\",\"exp_id\":\"721\",\"norm\":\"" + chatReq.getCategory() + "\",\"feedback\":" + feedback + "\",\"simulate\":" + simulate + ",\"session_id\":" + "\"" + sessionId + "\"" + ",\"generate_args\":{\"max_new_tokens\":1024,\"max_length\":4096,\"num_beams\":1,\"do_sample\":true,\"top_p\":0.7,\"temperature\":0.95},\"extra\":{\"rows\":" +  rows + "}}";
+            dataJson = "{\"bot_id\":\"b00001\",\"exp_id\":\"721\",\"norm\":\"" + chatReq.getCategory() + "\",\"feedback\":" + feedback + ",\"simulate\":" + simulate + ",\"session_id\":" + "\"" + sessionId + "\"" + ",\"generate_args\":{\"max_new_tokens\":1024,\"max_length\":4096,\"num_beams\":1,\"do_sample\":true,\"top_p\":0.7,\"temperature\":0.95},\"extra\":{\"rows\":" +  rows + "}}";
             log.info("请求大模型的决策的参数为{}", dataJson);
             log.info("请求大模型的决策的参数为{}", dataJson);
             PredictionsRequest request = PredictionsRequest.newBuilder()
             PredictionsRequest request = PredictionsRequest.newBuilder()
                     .setModelName("slibra_bot")
                     .setModelName("slibra_bot")
@@ -188,7 +188,7 @@ public class GRPCController extends BaseController {
                 throw new RuntimeException(ex);
                 throw new RuntimeException(ex);
             }
             }
         } finally {
         } finally {
-            log.info("决策最终要保存的数据是{}", JSON.toJSONString(resultData));
+            log.info("sessionId是{}\n决策最终要保存的数据是{}", sessionId, JSON.toJSONString(resultData));
             //保存聊天记录
             //保存聊天记录
             //将问答更新到数据库中
             //将问答更新到数据库中
             chatReq.setSessionId(sessionId);
             chatReq.setSessionId(sessionId);

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

@@ -587,6 +587,7 @@ public class RyTask
             chatReq.setShowVal(showVal);
             chatReq.setShowVal(showVal);
             chatReq.setQuestion(shWarningPrompt);
             chatReq.setQuestion(shWarningPrompt);
             chatReq.setAnswer(sb.toString());
             chatReq.setAnswer(sb.toString());
+            chatReq.setWarningId(String.valueOf(xinyiWarningRecord.getId()));
             chatReq.setCounts(counts);//问答次数
             chatReq.setCounts(counts);//问答次数
             chatReq.setUserId(WARNING_DEFAULT_CREATE);
             chatReq.setUserId(WARNING_DEFAULT_CREATE);
             chatReq.setCreateBy(WARNING_DEFAULT_CREATE);
             chatReq.setCreateBy(WARNING_DEFAULT_CREATE);