Update demo-flow

Author: Yuriy Bezsonov

samples/spring-ai-te-agent/ai-agent/samples/ticket-tram-cz.png

@@ -19,7 +19,7 @@ public class ChatService {
     private final ChatMemoryService chatMemoryService;
 
     public static final String SYSTEM_PROMPT = """
-        You are a helpful AI Agent for travel and expenses.
+        You are a helpful AI Agent for travel and expenses management.
 
         Guidelines:
         1. Use markdown tables for structured data

samples/spring-ai-te-agent/ai-agent/src/main/java/com/example/ai/agent/service/ChatService.java

@@ -28,7 +28,7 @@ spring.ai.bedrock.converse.chat.options.model=global.anthropic.claude-sonnet-4-2
 # spring.ai.bedrock.converse.chat.options.model=us.amazon.nova-premier-v1:0
 
 # Document processing model
-ai.agent.document.model=global.anthropic.claude-sonnet-4-20250514-v1:0
+ai.agent.document.model=global.anthropic.claude-sonnet-4-5-20250929-v1:0
 
 # PostgreSQL Configuration
 spring.datasource.url=jdbc:postgresql://localhost:5432/ai_agent_db

samples/spring-ai-te-agent/ai-agent/src/main/resources/application.properties

@@ -1,5 +1,10 @@
 #!/bin/bash
 
+# Clear current directory completely
+echo "Clearing current directory: $(pwd)"
+rm -rf ./*
+rm -rf ./.[!.]* 2>/dev/null || true
+
 echo "Checking if spring-boot-cli directory exists and remove it if it does..."
 if [ -d "spring-boot-cli" ]; then
     echo "Found existing spring-boot-cli directory, removing it..."
@@ -36,7 +41,7 @@ echo "   --type=maven-project \\"
 echo "   --artifact-id=ai.agent \\"
 echo "   --name=ai-agent \\"
 echo "   --group-id=com.example \\"
-echo -e "   \033[1m--dependencies=spring-ai-bedrock-converse,web,thymeleaf,actuator,devtools,testcontainers,postgresql \033[0m\\"
+echo -e "   \033[1m--dependencies=spring-ai-bedrock-converse,web,thymeleaf \033[0m\\"
 echo "   --extract \\"
 echo "   ai-agent"
 
@@ -53,7 +58,7 @@ echo "Initializing Spring Boot project..."
    --artifact-id=ai.agent \
    --name=ai-agent \
    --group-id=com.example \
-   --dependencies=spring-ai-bedrock-converse,web,thymeleaf,actuator,devtools,testcontainers,postgresql \
+   --dependencies=spring-ai-bedrock-converse,web,thymeleaf \
    --extract \
    ai-agent
 
@@ -89,7 +94,7 @@ spring.thymeleaf.suffix=.html
 # Amazon Bedrock Configuration
 spring.ai.bedrock.aws.region=us-east-1
 spring.ai.bedrock.converse.chat.options.max-tokens=10000
-spring.ai.bedrock.converse.chat.options.model=openai.gpt-oss-120b-1:0
+spring.ai.bedrock.converse.chat.options.model=global.anthropic.claude-sonnet-4-20250514-v1:0
 EOL
 
 echo "Creating necessary directories and files..."
@@ -98,7 +103,7 @@ mkdir -p src/main/java/com/example/ai/agent/controller
 mkdir -p src/main/java/com/example/ai/agent/service
 
 cp "$SOURCES_FOLDER/ai-agent/src/main/resources/templates/chat.html" src/main/resources/templates/
-cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/controller/WebViewController.java" src/main/java/com/example/ai/agent/controller/
+cp "$SOURCES_FOLDER/demo-scripts/Steps/WebViewController.java.1" src/main/java/com/example/ai/agent/controller/WebViewController.java
 cp "$SOURCES_FOLDER/demo-scripts/Steps/ChatService.java.0" src/main/java/com/example/ai/agent/service/ChatService.java
 cp "$SOURCES_FOLDER/demo-scripts/Steps/ChatController.java.0" src/main/java/com/example/ai/agent/controller/ChatController.java
 
@@ -117,10 +122,10 @@ echo "Committing changes to Git repository..."
 git add .
 git commit -m "Update initial files"
 
-./mvnw spring-boot:test-run
+./mvnw spring-boot:run
 
 cp "$SOURCES_FOLDER/demo-scripts/Steps/ChatService.java.1" src/main/java/com/example/ai/agent/service/ChatService.java
 
-./mvnw spring-boot:test-run
+./mvnw spring-boot:run
 
 cd ..

samples/spring-ai-te-agent/demo-scripts/01-create-app.sh

@@ -25,9 +25,17 @@ cd ai-agent
 echo "Copying ChatMemoryService.java..."
 cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/service/ChatMemoryService.java" src/main/java/com/example/ai/agent/service/
 
+echo "Copying ConversationSummaryService.java..."
+cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/service/ConversationSummaryService.java" src/main/java/com/example/ai/agent/service/
+
 echo "Copying ChatService.java from version 2..."
 cp "$SOURCES_FOLDER/demo-scripts/Steps/ChatService.java.2" src/main/java/com/example/ai/agent/service/ChatService.java
-# code src/main/java/com/example/ai/agent/ChatService.java
+
+echo "Copying ChatController.java from version 2..."
+cp "$SOURCES_FOLDER/demo-scripts/Steps/ChatController.java.2" src/main/java/com/example/ai/agent/controller/ChatController.java
+
+echo "Copying WebViewController.java from version 2..."
+cp "$SOURCES_FOLDER/demo-scripts/Steps/WebViewController.java.2" src/main/java/com/example/ai/agent/controller/WebViewController.java
 
 # Add PostgreSQL and JPA configuration to application.properties
 echo "Updating application.properties with database configuration..."
@@ -61,6 +69,21 @@ cat > temp_dependencies.xml << 'EOL'
             <groupId>org.springframework.ai</groupId>
             <artifactId>spring-ai-starter-model-chat-memory-repository-jdbc</artifactId>
         </dependency>
+		<dependency>
+			<groupId>org.postgresql</groupId>
+			<artifactId>postgresql</artifactId>
+			<scope>runtime</scope>
+		</dependency>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-testcontainers</artifactId>
+			<scope>test</scope>
+		</dependency>
+		<dependency>
+			<groupId>org.testcontainers</groupId>
+			<artifactId>postgresql</artifactId>
+			<scope>test</scope>
+		</dependency>
 EOL
 
 # Create a temporary file for the modified pom.xml

samples/spring-ai-te-agent/demo-scripts/02-add-memory.sh

@@ -27,6 +27,9 @@ mkdir -p src/main/java/com/example/ai/agent/tool
 echo "Copying DateTimeService.java..."
 cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/tool/DateTimeService.java" src/main/java/com/example/ai/agent/tool/
 
+echo "Copying WeatherService.java..."
+cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/tool/WeatherService.java" src/main/java/com/example/ai/agent/tool/
+
 echo "Copying ChatService.java from version 4..."
 cp "$SOURCES_FOLDER/demo-scripts/Steps/ChatService.java.4" src/main/java/com/example/ai/agent/service/ChatService.java
 

samples/spring-ai-te-agent/demo-scripts/04-add-tools.sh

@@ -15,24 +15,6 @@ if [ ! -d "ai-agent" ]; then
     exit 1
 fi
 
-# Check if weather folder exists and delete it if it does
-echo "Checking for weather folder..."
-if [ -d "weather" ]; then
-    echo "Found existing weather folder, removing it..."
-    rm -rf weather
-fi
-
-# Copy weather folder from source
-echo "Copying weather folder from source..."
-if [ -d "$SOURCES_FOLDER/weather" ]; then
-    cp -r "$SOURCES_FOLDER/weather" .
-    echo "weather folder copied successfully."
-else
-    echo "Error: weather folder not found at $SOURCES_FOLDER/weather."
-    echo "Please ensure the folder exists at the specified location."
-    exit 1
-fi
-
 # Check if travel folder exists and delete it if it does
 echo "Checking for travel folder..."
 if [ -d "travel" ]; then
@@ -67,7 +49,6 @@ cat >> src/main/resources/application.properties << 'EOL'
 
 # MCP Client Configuration
 spring.ai.mcp.client.toolcallback.enabled=true
-spring.ai.mcp.client.sse.connections.server1.url=http://localhost:8081
 spring.ai.mcp.client.sse.connections.server2.url=http://localhost:8082
 EOL
 

samples/spring-ai-te-agent/demo-scripts/05-add-mcp.sh

@@ -40,15 +40,21 @@ git commit -m "add MCP"
 
 cd ai-agent
 
+echo "Copying ChatController.java with multi-modal support..."
 cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/controller/ChatController.java" src/main/java/com/example/ai/agent/controller/
+
+echo "Copying WebViewController.java with feature flags..."
+cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/controller/WebViewController.java" src/main/java/com/example/ai/agent/controller/
+
+echo "Copying DocumentChatService.java..."
 cp "$SOURCES_FOLDER/ai-agent/src/main/java/com/example/ai/agent/service/DocumentChatService.java" src/main/java/com/example/ai/agent/service/
 
-echo "Updating application.properties with database configuration..."
+echo "Updating application.properties with multi-modal configuration..."
 cat >> src/main/resources/application.properties << 'EOL'
 spring.ai.mcp.client.sse.connections.server3.url=http://localhost:8083
 
 # Document processing model
-ai.agent.document.model=global.anthropic.claude-sonnet-4-20250514-v1:0
+ai.agent.document.model=global.anthropic.claude-sonnet-4-5-20250929-v1:0
 EOL
 
 # echo "Opening application.properties in VS Code..."

samples/spring-ai-te-agent/demo-scripts/06-add-multi.sh

@@ -1,24 +1,23 @@
 package com.example.ai.agent.controller;
 
 import com.example.ai.agent.service.ChatService;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.http.MediaType;
+import org.springframework.web.bind.annotation.*;
+import reactor.core.publisher.Flux;
 
 @RestController
-@RequestMapping("api")
+@RequestMapping("api/chat")
 public class ChatController {
     private final ChatService chatService;
 
     public ChatController(ChatService chatService) {
         this.chatService = chatService;
     }
 
-    @PostMapping("chat")
-    public String chat(@RequestBody ChatRequest request) {
+    @PostMapping(value = "message", produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
+    public Flux<String> chat(@RequestBody ChatRequest request) {
         return chatService.processChat(request.prompt());
     }
 
-    public record ChatRequest(String prompt) {}
+    public record ChatRequest(String prompt, String userId) {}
 }

samples/spring-ai-te-agent/demo-scripts/Steps/ChatController.java.0

@@ -0,0 +1,36 @@
+package com.example.ai.agent.controller;
+
+import com.example.ai.agent.service.ChatService;
+import com.example.ai.agent.service.ChatMemoryService;
+import org.springframework.http.MediaType;
+import org.springframework.web.bind.annotation.*;
+import reactor.core.publisher.Flux;
+import java.security.Principal;
+
+@RestController
+@RequestMapping("api/chat")
+public class ChatController {
+    private final ChatService chatService;
+    private final ChatMemoryService chatMemoryService;
+
+    public ChatController(ChatMemoryService chatMemoryService, ChatService chatService) {
+        this.chatService = chatService;
+        this.chatMemoryService = chatMemoryService;
+    }
+
+    @PostMapping(value = "message", produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
+    public Flux<String> chat(@RequestBody ChatRequest request, Principal principal) {
+        String userId = getUserId(request.userId(), principal);
+        chatMemoryService.setCurrentUserId(userId);
+        return chatService.processChat(request.prompt());
+    }
+
+    private String getUserId(String requestUserId, Principal principal) {
+        if (principal != null) {
+            return principal.getName().toLowerCase();
+        }
+        return requestUserId != null ? requestUserId.toLowerCase() : "user1";
+    }
+
+    public record ChatRequest(String prompt, String userId) {}
+}