Use streaming mode by default

Author: alexcheng1982

pom.xml

@@ -21,7 +21,7 @@
   <dependencies>
     <dependency>
       <groupId>org.springframework.boot</groupId>
-      <artifactId>spring-boot-starter-web</artifactId>
+      <artifactId>spring-boot-starter-webflux</artifactId>
     </dependency>
     <dependency>
       <groupId>org.springframework.ai</groupId>
@@ -31,7 +31,7 @@
     <dependency>
       <groupId>com.javaaidev</groupId>
       <artifactId>chat-agent-ui</artifactId>
-      <version>0.6.0</version>
+      <version>0.7.0</version>
     </dependency>
 
     <dependency>

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

@@ -11,7 +11,7 @@
 import reactor.core.publisher.Flux;
 
 @RestController
-@RequestMapping("/chat")
+@RequestMapping("/chat_non_streaming")
 public class ChatAgentController extends AbstractChatAgentController {
 
   private final ChatClient chatClient;

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

@@ -3,6 +3,7 @@
 import com.javaaidev.chatagent.model.ChatRequest;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.messages.Message;
+import org.springframework.http.MediaType;
 import org.springframework.http.codec.ServerSentEvent;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
@@ -11,7 +12,7 @@
 import reactor.core.publisher.Flux;
 
 @RestController
-@RequestMapping("/chat_streaming")
+@RequestMapping("/chat")
 public class ChatAgentStreamingController extends AbstractChatAgentController {
 
   private final ChatClient chatClient;
@@ -20,7 +21,7 @@ public ChatAgentStreamingController(ChatClient.Builder builder) {
     chatClient = builder.build();
   }
 
-  @PostMapping
+  @PostMapping(produces = MediaType.TEXT_EVENT_STREAM_VALUE)
   public Flux<ServerSentEvent<String>> chatStreaming(@RequestBody ChatRequest request) {
     var messages = chatRequestToMessages(request);
     return chatClient.prompt().system(SYSTEM_TEXT).messages(messages.toArray(new Message[0]))

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

@@ -0,0 +1,37 @@
+package com.javaaidev.agent;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+
+public class SimpleChatController {
+
+  public static final String SYSTEM_TEXT = """
+      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.
+      """;
+
+  private final ChatClient chatClient;
+
+  public SimpleChatController(ChatClient.Builder builder) {
+    chatClient = builder.build();
+  }
+
+  @PostMapping("/simple_chat")
+  public ChatOutput chat(@RequestBody ChatInput chatInput) {
+    return new ChatOutput(
+        chatClient.prompt()
+            .system(SYSTEM_TEXT)
+            .user(chatInput.input())
+            .call().content());
+  }
+
+
+  public record ChatInput(String input) {
+
+  }
+
+  public record ChatOutput(String output) {
+
+  }
+}