Merge branch 'refs/heads/feature/20241217_收尾' into sandbox

src/main/java/cn/breeze/elleai/application/service/AppChatCompletionService.java

@@ -593,6 +593,7 @@ public class AppChatCompletionService {
         final Boolean[] replyFlag = {false};
         String finalSessionId = sessionId;
         List<String> references = Lists.newArrayList();
+        final Boolean[] referenceFlag = {false};
         return webClient.post().uri(dsApiBase).accept(MediaType.TEXT_EVENT_STREAM).bodyValue(reqBody.toJSONString()).exchangeToFlux(r -> r.bodyToFlux(String.class))
                 .mapNotNull(v -> {
                     log.info("ds:{}", v);
@@ -628,7 +629,8 @@ public class AppChatCompletionService {
                             if (StrUtil.isBlank(recordId[0]) && StrUtil.isNotBlank(payload.getString("record_id"))) {
                                 recordId[0] = payload.getString("record_id");
                             }
-                            if (CollUtil.isEmpty(references) && payload.containsKey("knowledge")) {
+                            if (referenceFlag[0] == false && CollUtil.isEmpty(references) && payload.containsKey("knowledge")) {
+                                referenceFlag[0] = true;
                                 JSONArray knowledge = payload.getJSONArray("knowledge");
                                 if (ObjectUtil.isNotNull(knowledge) && knowledge.size() > 0) {
                                     for (Object kb : knowledge) {
@@ -641,6 +643,7 @@ public class AppChatCompletionService {
                                 if (CollUtil.isNotEmpty(references)) {
                                     try {
                                         List<String> docs =  tencentCloudFacade.getKbSegment(references);
+                                        log.info("get kb docs:{}", docs.size());
                                         result.setReferences(docs);
                                     } catch (TencentCloudSDKException e) {
                                         log.error("getKbSegment error,{}", e.getMessage());