Use Optional to check null.

Signed-off-by: Mengqi Xu <[email protected]>

Author: remeio

spring-ai-model/src/main/java/org/springframework/ai/chat/model/StreamingChatModel.java

@@ -17,7 +17,9 @@
 package org.springframework.ai.chat.model;
 
 import java.util.Arrays;
+import java.util.Optional;
 
+import org.springframework.ai.chat.messages.AssistantMessage;
 import reactor.core.publisher.Flux;
 
 import org.springframework.ai.chat.messages.Message;
@@ -29,16 +31,20 @@ public interface StreamingChatModel extends StreamingModel<Prompt, ChatResponse>
 
 	default Flux<String> stream(String message) {
 		Prompt prompt = new Prompt(message);
-		return stream(prompt).map(response -> (response.getResult() == null || response.getResult().getOutput() == null
-				|| response.getResult().getOutput().getText() == null) ? ""
-						: response.getResult().getOutput().getText());
+		return stream(prompt).map(response -> Optional
+				.ofNullable(response.getResult())
+				.map(Generation::getOutput)
+				.map(AssistantMessage::getText)
+				.orElse(""));
 	}
 
 	default Flux<String> stream(Message... messages) {
 		Prompt prompt = new Prompt(Arrays.asList(messages));
-		return stream(prompt).map(response -> (response.getResult() == null || response.getResult().getOutput() == null
-				|| response.getResult().getOutput().getText() == null) ? ""
-						: response.getResult().getOutput().getText());
+		return stream(prompt).map(response -> Optional
+				.ofNullable(response.getResult())
+				.map(Generation::getOutput)
+				.map(AssistantMessage::getText)
+				.orElse(""));
 	}
 
 	@Override