update chat agent ui

Author: alexcheng1982

pom.xml

@@ -31,7 +31,7 @@
     <dependency>
       <groupId>com.javaaidev</groupId>
       <artifactId>chat-agent-ui</artifactId>
-      <version>0.5.1</version>
+      <version>0.6.0</version>
     </dependency>
 
     <dependency>

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

@@ -1,6 +1,16 @@
 package com.javaaidev.agent;
 
+import com.javaaidev.chatagent.model.ChatRequest;
+import com.javaaidev.chatagent.model.ChatResponse;
+import com.javaaidev.chatagent.model.TextContentPart;
+import com.javaaidev.chatagent.model.ThreadAssistantMessage;
+import com.javaaidev.chatagent.model.ThreadUserMessage;
+import java.util.List;
+import java.util.stream.Stream;
 import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.messages.AssistantMessage;
+import org.springframework.ai.chat.messages.Message;
+import org.springframework.ai.chat.messages.UserMessage;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -10,13 +20,6 @@
 @RequestMapping("/chat")
 public class ChatAgentController {
 
-  public record ChatRequest(String input) {
-
-  }
-
-  public record ChatResponse(String output) {
-
-  }
 
   private static final String SYSTEM_TEXT = "You are a chef who is proficient in various cuisines. Please answer users' questions about cooking.";
   private final ChatClient chatClient;
@@ -27,10 +30,28 @@ public ChatAgentController(ChatClient.Builder builder) {
 
   @PostMapping
   public ChatResponse chat(@RequestBody ChatRequest request) {
+    var messages = request.messages().stream().flatMap(message -> {
+      if (message instanceof ThreadUserMessage userMessage) {
+        return userMessage.content().stream().map(part -> {
+          if (part instanceof TextContentPart textContentPart) {
+            return new UserMessage(textContentPart.text());
+          }
+          return null;
+        });
+      } else if (message instanceof ThreadAssistantMessage assistantMessage) {
+        return assistantMessage.content().stream().map(part -> {
+          if (part instanceof TextContentPart textContentPart) {
+            return new AssistantMessage(textContentPart.text());
+          }
+          return null;
+        });
+      }
+      return Stream.of();
+    }).toList();
     var output = chatClient.prompt().system(SYSTEM_TEXT)
-        .user(request.input())
+        .messages(messages.toArray(new Message[0]))
         .call()
         .content();
-    return new ChatResponse(output);
+    return new ChatResponse(List.of(new TextContentPart(output)));
   }
 }