update to Spring AI 1.0.0-M6

Author: alexcheng1982

README.md

@@ -2,15 +2,13 @@
 
 ## 运行
 
-需要在本地安装 Ollama 和 Docker Desktop。使用 Docker Desktop 来运行
-Redis。参考 `docker-compose.yaml` 文件。
+需要在本地安装 Docker Desktop。使用 Docker Desktop 来运行
+Pgvector。参考 `docker-compose.yaml` 文件。
 
-默认使用通义千问 0.5b 模型（`qwen:0.5b`），可以通过配置文件来修改。
+默认使用通义千问 2.5 0.5b 模型（`qwen2.5:0.5b`），可以通过配置文件来修改。
 
 > 朴素 RAG 的结果，依赖于文档相似性的查询结果。在本地运行时，受限于模型的参数数量，查询结果可能不太准确。
 
-使用 `dev` profile 来运行 Spring，`-Dspring.profiles.active=dev`。
-
 访问 [Swagger UI](http://localhost:8080/swagger-ui/index.html) 来与 API 交互。
 
 ## 加载网页到向量数据库
@@ -25,14 +23,16 @@ Redis。参考 `docker-compose.yaml` 文件。
 
 ## 与大模型交互
 
-发送 POST 请求到 `/chat/query`，提供消息内容。
+发送 POST 请求到 `/chat`，提供消息内容。
 
 ```json
 {
-  "message": "从2024年政府工作报告中，分析人工智能的发展前景"
+  "input": "从2024年政府工作报告中，分析人工智能的发展前景"
 }
 ```
 
+通过用户界面进行交互，访问 http://localhost:8080/webjars/chat-agent-ui/index.html
+
 ## 课程
 
 [![课程](./course.png)](https://www.bilibili.com/cheese/play/ss15645)

docker-compose.yaml

@@ -1,11 +1,13 @@
-version: "3"
 services:
-  redis-stack:
-    image: redis/redis-stack
+  postgres:
+    image: pgvector/pgvector:0.8.0-pg16
     volumes:
-      - redis-data:/data
+      - postgresql-data:/var/lib/postgresql/data
+    environment:
+      POSTGRES_DB: "postgres"
+      POSTGRES_USER: "postgres"
+      POSTGRES_PASSWORD: "postgres"
     ports:
-      - "6379:6379"
-      - "8001:8001"
+      - "5432:5432"
 volumes:
-  redis-data:
+    postgresql-data:

pom.xml

@@ -7,7 +7,7 @@
     <groupId>org.springframework.boot</groupId>
     <artifactId>spring-boot-starter-parent</artifactId>
     <version>3.2.4</version>
-    <relativePath/> <!-- lookup parent from repository -->
+    <relativePath/>
   </parent>
   <groupId>cc.vividcode.ai</groupId>
   <artifactId>spring-ai-naive-rag</artifactId>
@@ -16,7 +16,7 @@
   <description>Spring AI Naive RAG</description>
   <properties>
     <java.version>21</java.version>
-    <spring-ai.version>0.8.1</spring-ai.version>
+    <spring-ai.version>1.0.0-M6</spring-ai.version>
   </properties>
   <dependencies>
     <dependency>
@@ -29,7 +29,7 @@
     </dependency>
     <dependency>
       <groupId>org.springframework.ai</groupId>
-      <artifactId>spring-ai-redis</artifactId>
+      <artifactId>spring-ai-pgvector-store-spring-boot-starter</artifactId>
     </dependency>
 
     <dependency>
@@ -47,6 +47,11 @@
       <artifactId>springdoc-openapi-starter-webmvc-ui</artifactId>
       <version>2.5.0</version>
     </dependency>
+    <dependency>
+      <groupId>com.javaaidev</groupId>
+      <artifactId>chat-agent-ui</artifactId>
+      <version>0.5.1</version>
+    </dependency>
 
     <dependency>
       <groupId>org.springframework.boot</groupId>

src/main/java/cc/vividcode/ai/naiverag/chat/ChatRequest.java

@@ -1,7 +1,7 @@
-package cc.vividcode.ai.naiverag;
+package com.javaaidev.naiverag;
 
-import cc.vividcode.ai.naiverag.chat.ChatModuleConfiguration;
-import cc.vividcode.ai.naiverag.etl.EtlModuleConfiguration;
+import com.javaaidev.naiverag.chat.ChatModuleConfiguration;
+import com.javaaidev.naiverag.etl.EtlModuleConfiguration;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.context.annotation.Import;

src/main/java/cc/vividcode/ai/naiverag/etl/EtlModuleConfiguration.java

@@ -1,12 +1,10 @@
-package cc.vividcode.ai.naiverag.chat;
+package com.javaaidev.naiverag.chat;
 
 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;
 
 @RestController
-@RequestMapping("/chat")
 public class ChatController {
 
   private final ChatService chatService;
@@ -15,8 +13,8 @@ public ChatController(ChatService chatService) {
     this.chatService = chatService;
   }
 
-  @PostMapping("/query")
-  public String chat(@RequestBody ChatRequest request) {
-    return chatService.chat(request);
+  @PostMapping("/chat")
+  public ChatResponse chat(@RequestBody ChatRequest request) {
+    return new ChatResponse(chatService.chat(request));
   }
 }

src/main/java/cc/vividcode/ai/naiverag/etl/RedisConfig.java

@@ -1,13 +1,18 @@
-package cc.vividcode.ai.naiverag.chat;
+package com.javaaidev.naiverag.chat;
 
-import org.springframework.ai.chat.ChatClient;
+import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.vectorstore.VectorStore;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 
 @Configuration
 public class ChatModuleConfiguration {
 
+  @Bean
+  public ChatClient chatClient(ChatClient.Builder builder) {
+    return builder.build();
+  }
+
   @Bean
   public ChatService chatService(ChatClient chatClient,
       VectorStore vectorStore) {