From 7bfe312157517ab5d2d328dd72eda1566d85f33b Mon Sep 17 00:00:00 2001
From: Christian Tzolov <ctzolov@vmware.com>
Date: Tue, 12 Mar 2024 16:58:33 +0100
Subject: [PATCH] Convinence StreamingChatClient stream default

   Add StreamingChatClient stream default that takes String and returns stream of String
---
 .../org/springframework/ai/chat/StreamingChatClient.java   | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/StreamingChatClient.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/StreamingChatClient.java
index b259c5da4..a6e4a0b76 100644
--- a/spring-ai-core/src/main/java/org/springframework/ai/chat/StreamingChatClient.java
+++ b/spring-ai-core/src/main/java/org/springframework/ai/chat/StreamingChatClient.java
@@ -23,6 +23,13 @@ import org.springframework.ai.model.StreamingModelClient;
 @FunctionalInterface
 public interface StreamingChatClient extends StreamingModelClient<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().getContent() == null) ? ""
+						: response.getResult().getOutput().getContent());
+	}
+
 	@Override
 	Flux<ChatResponse> stream(Prompt prompt);