After upgrading to version 2.0.1, streaming is no longer supported

my core code:
@RequestMapping(value="/chat-stream",produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public Flux chatStream(String request) {
UserMessage userMsg = new UserMessage(request);
Flux reply = harnessAgent.streamEvents(
userMsg,
RuntimeContext
.builder()
.sessionId("sdfhisdfhioqwupore-2324e32-fdfdsdfds6")
.userId("3333")
.build())
.map(event -> {
if (event.getType() == AgentEventType.TEXT_BLOCK_DELTA) {
return ((TextBlockDeltaEvent) event).getDelta();
} else if (event.getType() == AgentEventType.TOOL_CALL_START) {
return ((ToolCallStartEvent) event).getToolCallName();
}else if (event.getType() == AgentEventType.TOOL_CALL_DELTA) {
return ((ToolCallDeltaEvent) event).getDelta();
}else if(event.getType() == AgentEventType.THINKING_BLOCK_DELTA){
return ((ThinkingBlockDeltaEvent) event).getDelta();
}else if(event.getType() == AgentEventType.TOOL_RESULT_TEXT_DELTA){
return ((ToolResultTextDeltaEvent) event).getDelta();
}
return "";
});
return reply;
}

When I was in version 2.0.0, I was able to stream, but upgrading to version 2.0.1 returned the results all at once. The model is the same, is it because the code writing method has changed