use Chat Agent UI

Author: alexcheng1982

pom.xml

@@ -7,7 +7,7 @@
   <parent>
     <groupId>org.springframework.boot</groupId>
     <artifactId>spring-boot-starter-parent</artifactId>
-    <version>3.3.7</version>
+    <version>3.4.4</version>
     <relativePath/>
   </parent>
 
@@ -16,10 +16,13 @@
   <version>1.0.0-SNAPSHOT</version>
 
   <properties>
-    <maven.compiler.source>21</maven.compiler.source>
-    <maven.compiler.target>21</maven.compiler.target>
+    <java.version>21</java.version>
+    <maven.compiler.source>${java.version}</maven.compiler.source>
+    <maven.compiler.target>${java.version}</maven.compiler.target>
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
-    <spring-ai.version>1.0.0-M5</spring-ai.version>
+    <spring-ai.version>1.0.0-M6</spring-ai.version>
+    <chat-agent-ui.version>0.11.0</chat-agent-ui.version>
+    <llm-agent-spec.version>0.1.2</llm-agent-spec.version>
   </properties>
 
   <dependencyManagement>
@@ -60,6 +63,16 @@
       <artifactId>springdoc-openapi-starter-webmvc-ui</artifactId>
       <version>2.5.0</version>
     </dependency>
+    <dependency>
+      <groupId>com.javaaidev.chatagentui</groupId>
+      <artifactId>chat-agent-ui</artifactId>
+      <version>${chat-agent-ui.version}</version>
+    </dependency>
+    <dependency>
+      <groupId>com.javaaidev.llmagentspec</groupId>
+      <artifactId>spring-ai-adapter</artifactId>
+      <version>${llm-agent-spec.version}</version>
+    </dependency>
   </dependencies>
 
   <repositories>
@@ -71,14 +84,6 @@
         <enabled>false</enabled>
       </snapshots>
     </repository>
-    <repository>
-      <id>spring-snapshots</id>
-      <name>Spring Snapshots</name>
-      <url>https://repo.spring.io/snapshot</url>
-      <releases>
-        <enabled>false</enabled>
-      </releases>
-    </repository>
   </repositories>
 
 </project>

src/main/java/com/javaaidev/pdfqa/QaController.java

@@ -1,11 +1,17 @@
 package com.javaaidev.pdfqa;
 
+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.client.advisor.QuestionAnswerAdvisor;
 import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
+import org.springframework.ai.chat.messages.Message;
+import org.springframework.http.codec.ServerSentEvent;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
+import reactor.core.publisher.Flux;
 
 @RestController
 public class QaController {
@@ -20,16 +26,13 @@ public QaController(ChatClient.Builder builder,
         simpleLoggerAdvisor).build();
   }
 
-  @PostMapping("/qa")
-  public QaResponse qa(@RequestBody QaRequest request) {
-    return new QaResponse(chatClient.prompt().user(request.input()).call().content());
+  @PostMapping("/chat")
+  public Flux<ServerSentEvent<ChatAgentResponse>> qa(@RequestBody ChatAgentRequest request) {
+    return ModelAdapter.toStreamingResponse(
+        chatClient.prompt()
+            .messages(ModelAdapter.fromRequest(request).toArray(new Message[0]))
+            .stream()
+            .chatResponse());
   }
 
-  public record QaRequest(String input) {
-
-  }
-
-  public record QaResponse(String output) {
-
-  }
 }