feature: 增加片段返回

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

@@ -631,19 +631,23 @@ public class AppChatCompletionService {
                             if (referenceFlag[0] == false && CollUtil.isEmpty(references) && payload.containsKey("knowledge")) {
                                 referenceFlag[0] = true;
                                 JSONArray knowledge = payload.getJSONArray("knowledge");
+                                List<String> segmentIds = CollUtil.newArrayList();
                                 if (ObjectUtil.isNotNull(knowledge) && knowledge.size() > 0) {
                                     for (Object kb : knowledge) {
                                         JSONObject item = (JSONObject) kb;
                                         if (ObjectUtil.equals(2, item.getInteger("type"))) {
-                                            references.add(item.getString("seg_id"));
+                                            segmentIds.add(item.getString("seg_id"));
                                         }
                                     }
                                 }
-                                if (CollUtil.isNotEmpty(references)) {
+                                if (CollUtil.isNotEmpty(segmentIds)) {
                                     try {
-                                        List<String> docs =  tencentCloudFacade.getKbSegment(references);
-                                        log.info("get kb docs:{}", docs.size());
-                                        result.setReferences(docs);
+                                        List<String> docs =  tencentCloudFacade.getKbSegment(segmentIds);
+                                        if (CollUtil.isNotEmpty(docs)) {
+                                            for (String doc : docs) {
+                                                references.add(doc);
+                                            }
+                                        }
                                     } catch (TencentCloudSDKException e) {
                                         log.error("getKbSegment error,{}", e.getMessage());
                                     }
@@ -655,7 +659,7 @@ public class AppChatCompletionService {
                     if(replyFlag[0]) {
                         finalContent += replyContent[0];
                     }
-
+                    result.setReferences(references);
                     result.setReplyContent(finalContent);
                     result.setChatCompletionId(chatCompletionId);
                     result.setMessageId(userQaRecordId);