update streaming response

Author: alexcheng1982

pom.xml

@@ -17,7 +17,7 @@
   <properties>
     <java.version>21</java.version>
     <spring-ai.version>1.0.0-M5</spring-ai.version>
-    <chat-agent-ui.version>0.9.0</chat-agent-ui.version>
+    <chat-agent-ui.version>0.10.0</chat-agent-ui.version>
   </properties>
   <dependencies>
     <dependency>

src/main/java/com/javaaidev/agent/AbstractChatAgentController.java

@@ -11,8 +11,7 @@ public abstract class AbstractChatAgentController {
       You are a chef who is proficient in various cuisines. Please answer users' questions about cooking.
       For other unrelated inputs, simply tell the user that you don't know.
       """;
-
-
+  
   protected List<Message> chatRequestToMessages(ChatAgentRequest request) {
     return ModelAdapter.fromRequest(request);
   }

src/main/java/com/javaaidev/agent/ChatAgentController.java

@@ -1,6 +1,8 @@
 package com.javaaidev.agent;
 
 import com.javaaidev.chatagent.model.ChatAgentRequest;
+import com.javaaidev.chatagent.model.ChatAgentResponse;
+import com.javaaidev.chatagent.springai.ModelAdapter;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.messages.Message;
 import org.springframework.http.codec.ServerSentEvent;
@@ -21,17 +23,18 @@ public ChatAgentController(ChatClient.Builder builder) {
   }
 
   @PostMapping
-  public Flux<ServerSentEvent<String>> chat(@RequestBody ChatAgentRequest request) {
+  public Flux<ServerSentEvent<ChatAgentResponse>> chat(@RequestBody ChatAgentRequest request) {
     if (request == null) {
       return Flux.empty();
     }
     var messages = chatRequestToMessages(request);
-    var output = chatClient.prompt().system(SYSTEM_TEXT)
+    var chatResponse = chatClient.prompt().system(SYSTEM_TEXT)
         .messages(messages.toArray(new Message[0]))
         .call()
-        .content();
-    return Flux.just(ServerSentEvent.<String>builder()
-        .data("#" + output + "#")
-        .build());
+        .chatResponse();
+    if (chatResponse == null) {
+      return Flux.empty();
+    }
+    return ModelAdapter.toStreamingResponse(Flux.just(chatResponse));
   }
 }

src/main/java/com/javaaidev/agent/ChatAgentStreamingController.java

@@ -1,6 +1,8 @@
 package com.javaaidev.agent;
 
 import com.javaaidev.chatagent.model.ChatAgentRequest;
+import com.javaaidev.chatagent.model.ChatAgentResponse;
+import com.javaaidev.chatagent.springai.ModelAdapter;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.messages.Message;
 import org.springframework.http.MediaType;
@@ -22,12 +24,14 @@ public ChatAgentStreamingController(ChatClient.Builder builder) {
   }
 
   @PostMapping(produces = MediaType.TEXT_EVENT_STREAM_VALUE)
-  public Flux<ServerSentEvent<String>> chatStreaming(@RequestBody ChatAgentRequest request) {
+  public Flux<ServerSentEvent<ChatAgentResponse>> chatStreaming(
+      @RequestBody ChatAgentRequest request) {
     var messages = chatRequestToMessages(request);
-    return chatClient.prompt().system(SYSTEM_TEXT).messages(messages.toArray(new Message[0]))
-        .stream().content()
-        .map(output -> ServerSentEvent.<String>builder()
-            .data("#" + output + "#")
-            .build());
+    return ModelAdapter.toStreamingResponse(
+        chatClient.prompt()
+            .system(SYSTEM_TEXT)
+            .messages(messages.toArray(new Message[0]))
+            .stream()
+            .chatResponse());
   }
 }