diff --git a/src/main/java/com/knowledge/base/infrastructure/util/http/FilteredSseOutputAdapter.java b/src/main/java/com/knowledge/base/infrastructure/util/http/FilteredSseOutputAdapter.java index d0ce7c0..03267d0 100644 --- a/src/main/java/com/knowledge/base/infrastructure/util/http/FilteredSseOutputAdapter.java +++ b/src/main/java/com/knowledge/base/infrastructure/util/http/FilteredSseOutputAdapter.java @@ -46,6 +46,11 @@ public class FilteredSseOutputAdapter implements WriterAdapter { lastChunk.put("sources", original.get("sources")); } + if(original.containsKey("close")) { + filtered.put("close", original.get("close")); + lastChunk.put("close", original.get("close")); + } + // 最后一条才合并 sources 输出 if (Boolean.TRUE.equals(original.get("close"))) { // 合并缓存中的 sources