feat: add module rag example

Signed-off-by: yuluo-yx <yuluo08290126@gmail.com>

Author: yuluo-yx

docker-compose/mysql/.keep renamed to docker-compose/mysql/conf/my.cnf

@@ -2,4 +2,10 @@
 
 此文件夹中用于存放一些 pgvector 服务常用的 Docker Compose 启动文件。
 
-- postgres16版本同时安装age和pgvector [DockerFile](./postgres16-age/DockerFile)
+- postgres16版本同时安装age和pgvector [DockerFile](./postgres16-age/DockerFile)
+
+执行以下命令启动 pg：
+
+```shell
+docker compose up -d --build
+```

docker-compose/mysql/docker-compose.yml

@@ -0,0 +1,18 @@
+services:
+
+  postgres:
+    build:
+      context: .
+      dockerfile: ./postgres16-age/DockerFile
+    container_name: spring-ai-alibaba-postgres
+    healthcheck:
+      test: [ "CMD", "pg_isready", "-q", "-d", "postgres", "-U", "root" ]
+      timeout: 45s
+      interval: 10s
+      retries: 10
+    environment:
+      - POSTGRES_DB=vector_store
+      - POSTGRES_USER=root
+      - POSTGRES_PASSWORD=password
+    ports:
+      - "5432:5432"

docker-compose/pgvector/README.md

@@ -52,7 +52,5 @@ RUN echo "shared_preload_libraries = 'age,vector'" >> /usr/share/postgresql/post
 
 # Start PostgreSQL
 
-
-
 #CMD ["/cmd_command.sh"]
-CMD ["postgres"]
+CMD ["postgres"]

docker-compose/pgvector/docker-compose.yml

@@ -17,45 +17,70 @@
 -->
 
 <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
-		 xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
+	xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
 	<modelVersion>4.0.0</modelVersion>
-
 	<parent>
 		<groupId>com.alibaba.cloud.ai</groupId>
 		<artifactId>spring-ai-alibaba-rag-example</artifactId>
 		<version>${revision}</version>
 		<relativePath>../pom.xml</relativePath>
 	</parent>
-
-	<artifactId>web-search</artifactId>
-	<version>2025.03.01</version>
-
-	<name>internet-search</name>
-	<description>Internet Search project for Spring AI Alibaba</description>
+	<groupId>com.alibaba.cloud.ai</groupId>
+	<artifactId>rag-pgvector-example</artifactId>
+	<version>0.0.1-SNAPSHOT</version>
+	<name>rag-pgvector-example</name>
+	<description>Demo project for Spring AI Alibaba</description>
 
 	<properties>
+		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+		<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
 		<maven.compiler.source>17</maven.compiler.source>
 		<maven.compiler.target>17</maven.compiler.target>
 		<maven-deploy-plugin.version>3.1.1</maven-deploy-plugin.version>
 
 		<!-- Spring AI -->
-		<spring-ai-alibaba.version>1.0.0-M5.1</spring-ai-alibaba.version>
-		<spring-ai.version>1.0.0-M5</spring-ai.version>
+		<spring-ai-alibaba.version>1.0.0-M3.1</spring-ai-alibaba.version>
+		<spring-ai.version>1.0.0-M3</spring-ai.version>
 	</properties>
 
 	<dependencies>
+		<dependency>
+			<groupId>com.alibaba.cloud.ai</groupId>
+			<artifactId>spring-ai-alibaba-starter</artifactId>
+			<version>${spring-ai-alibaba.version}</version>
+		</dependency>
+
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-web</artifactId>
 		</dependency>
 
+
 		<dependency>
-			<groupId>com.alibaba.cloud.ai</groupId>
-			<artifactId>spring-ai-alibaba-starter</artifactId>
-			<version>${spring-ai-alibaba.version}</version>
+			<groupId>org.springframework.ai</groupId>
+			<artifactId>spring-ai-pdf-document-reader</artifactId>
+			<version>${spring-ai.version}</version>
 		</dependency>
+
+		<dependency>
+			<groupId>org.springframework.ai</groupId>
+			<artifactId>spring-ai-pgvector-store-spring-boot-starter</artifactId>
+		</dependency>
+
 	</dependencies>
 
+	<dependencyManagement>
+		<dependencies>
+			<dependency>
+				<groupId>org.springframework.ai</groupId>
+				<artifactId>spring-ai-bom</artifactId>
+				<version>${spring-ai.version}</version>
+				<type>pom</type>
+				<scope>import</scope>
+			</dependency>
+		</dependencies>
+	</dependencyManagement>
+
 	<build>
 		<plugins>
 			<plugin>

docker-compose/pgvector/postgres16-age/DockerFile

@@ -0,0 +1,5 @@
+package com.alibaba.cloud.ai.example.rag;/**
+ * @author yuluo
+ * @author <a href="mailto:yuluo08290126@gmail.com">yuluo</a>
+ */public class ModuleRAGApplication {
+}

spring-ai-alibaba-rag-example/module-rag/README.md

@@ -0,0 +1,17 @@
+package com.alibaba.cloud.ai.example.rag.controller;
+
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * @author yuluo
+ * @author <a href="mailto:yuluo08290126@gmail.com">yuluo</a>
+ */
+
+@RestController
+@RequestMapping("/module-rag")
+public class ModuleRAGController {
+
+
+
+}

spring-ai-alibaba-rag-example/web-search/pom.xml renamed to spring-ai-alibaba-rag-example/module-rag/pom.xml

@@ -0,0 +1,45 @@
+package com.alibaba.cloud.ai.example.rag.controller;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.advisor.RetrievalAugmentationAdvisor;
+import org.springframework.ai.rag.retrieval.search.VectorStoreDocumentRetriever;
+import org.springframework.ai.vectorstore.VectorStore;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * @author yuluo
+ * @author <a href="mailto:yuluo08290126@gmail.com">yuluo</a>
+ */
+
+@RestController
+@RequestMapping("/module-rag")
+public class ModuleRAGRewriteController {
+
+	private final ChatClient chatClient;
+	private final RetrievalAugmentationAdvisor retrievalAugmentationAdvisor;
+
+	public ModuleRAGRewriteController(ChatClient.Builder chatClientBuilder, VectorStore vectorStore) {
+
+		this.chatClient = chatClientBuilder.build();
+		this.retrievalAugmentationAdvisor = RetrievalAugmentationAdvisor.builder()
+				.documentRetriever(VectorStoreDocumentRetriever.builder()
+						.similarityThreshold(0.50)
+						.vectorStore(vectorStore)
+						.build()
+				).build();
+	}
+
+	@PostMapping("/rag/basic")
+	String chatWithDocument(@RequestParam("prompt") String prompt) {
+
+		return chatClient.prompt()
+				.advisors(retrievalAugmentationAdvisor)
+				.user(prompt)
+				.call()
+				.content();
+	}
+
+}