update chat agent ui

Author: alexcheng1982

pom.xml

@@ -17,6 +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>
   </properties>
   <dependencies>
     <dependency>
@@ -29,9 +30,14 @@
     </dependency>
 
     <dependency>
-      <groupId>com.javaaidev</groupId>
+      <groupId>com.javaaidev.chatagentui</groupId>
       <artifactId>chat-agent-ui</artifactId>
-      <version>0.7.0</version>
+      <version>${chat-agent-ui.version}</version>
+    </dependency>
+    <dependency>
+      <groupId>com.javaaidev.chatagentui</groupId>
+      <artifactId>spring-ai-adapter</artifactId>
+      <version>${chat-agent-ui.version}</version>
     </dependency>
 
     <dependency>

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

@@ -1,14 +1,9 @@
 package com.javaaidev.agent;
 
-import com.javaaidev.chatagent.model.ChatRequest;
-import com.javaaidev.chatagent.model.TextContentPart;
-import com.javaaidev.chatagent.model.ThreadAssistantMessage;
-import com.javaaidev.chatagent.model.ThreadUserMessage;
+import com.javaaidev.chatagent.model.ChatAgentRequest;
+import com.javaaidev.chatagent.springai.ModelAdapter;
 import java.util.List;
-import java.util.stream.Stream;
-import org.springframework.ai.chat.messages.AssistantMessage;
 import org.springframework.ai.chat.messages.Message;
-import org.springframework.ai.chat.messages.UserMessage;
 
 public abstract class AbstractChatAgentController {
 
@@ -18,24 +13,7 @@ public abstract class AbstractChatAgentController {
       """;
 
 
-  protected List<Message> chatRequestToMessages(ChatRequest request) {
-    return request.messages().stream().flatMap(message -> {
-      if (message instanceof ThreadUserMessage userMessage) {
-        return userMessage.content().stream().map(part -> {
-          if (part instanceof TextContentPart(String text)) {
-            return new UserMessage(text);
-          }
-          return null;
-        });
-      } else if (message instanceof ThreadAssistantMessage assistantMessage) {
-        return assistantMessage.content().stream().map(part -> {
-          if (part instanceof TextContentPart(String text)) {
-            return new AssistantMessage(text);
-          }
-          return null;
-        });
-      }
-      return Stream.<Message>of();
-    }).toList();
+  protected List<Message> chatRequestToMessages(ChatAgentRequest request) {
+    return ModelAdapter.fromRequest(request);
   }
 }

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

@@ -1,6 +1,6 @@
 package com.javaaidev.agent;
 
-import com.javaaidev.chatagent.model.ChatRequest;
+import com.javaaidev.chatagent.model.ChatAgentRequest;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.messages.Message;
 import org.springframework.http.codec.ServerSentEvent;
@@ -21,7 +21,7 @@ public ChatAgentController(ChatClient.Builder builder) {
   }
 
   @PostMapping
-  public Flux<ServerSentEvent<String>> chat(@RequestBody ChatRequest request) {
+  public Flux<ServerSentEvent<String>> chat(@RequestBody ChatAgentRequest request) {
     if (request == null) {
       return Flux.empty();
     }

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

@@ -1,6 +1,6 @@
 package com.javaaidev.agent;
 
-import com.javaaidev.chatagent.model.ChatRequest;
+import com.javaaidev.chatagent.model.ChatAgentRequest;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.messages.Message;
 import org.springframework.http.MediaType;
@@ -22,7 +22,7 @@ public ChatAgentStreamingController(ChatClient.Builder builder) {
   }
 
   @PostMapping(produces = MediaType.TEXT_EVENT_STREAM_VALUE)
-  public Flux<ServerSentEvent<String>> chatStreaming(@RequestBody ChatRequest request) {
+  public Flux<ServerSentEvent<String>> chatStreaming(@RequestBody ChatAgentRequest request) {
     var messages = chatRequestToMessages(request);
     return chatClient.prompt().system(SYSTEM_TEXT).messages(messages.toArray(new Message[0]))
         .stream().content()