From b453856864d44879befd7558b68b0239b4be7870 Mon Sep 17 00:00:00 2001
From: YunaiV <zhijiantianya@gmail.com>
Date: Fri, 9 Aug 2024 00:09:19 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BB=A3=E7=A0=81=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E3=80=91AI=EF=BC=9A=E5=90=91=E9=87=8F=E5=8C=96=E7=9A=84?=
 =?UTF-8?q?=E9=9B=86=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../yudao/module/ai/service/knowledge/DocService.java    | 1 -
 .../module/ai/service/knowledge/DocServiceImpl.java      | 9 ++++-----
 yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml     | 4 +++-
 .../redis/RedisVectorStoreAutoConfiguration.java         | 4 ++--
 yudao-server/src/main/resources/application.yaml         | 2 +-
 5 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocService.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocService.java
index 2e7f792e8e..47905d4b15 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocService.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocService.java
@@ -7,7 +7,6 @@ package cn.iocoder.yudao.module.ai.service.knowledge;
  */
 public interface DocService {
 
-
     /**
      * 向量化文档
      */
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocServiceImpl.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocServiceImpl.java
index eeffebf448..b0f4afaf82 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocServiceImpl.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/knowledge/DocServiceImpl.java
@@ -21,24 +21,23 @@ import java.util.List;
 public class DocServiceImpl implements DocService {
 
     @Resource
-    RedisVectorStore vectorStore;
+    private RedisVectorStore vectorStore;
     @Resource
-    TokenTextSplitter tokenTextSplitter;
+    private TokenTextSplitter tokenTextSplitter;
 
     // TODO @xin 临时测试用，后续删
     @Value("classpath:/webapp/test/Fel.pdf")
     private org.springframework.core.io.Resource data;
 
-
     @Override
     public void embeddingDoc() {
         // 读取文件
-        org.springframework.core.io.Resource file = data;
-        TikaDocumentReader loader = new TikaDocumentReader(file);
+        TikaDocumentReader loader = new TikaDocumentReader(data);
         List<Document> documents = loader.get();
         // 文档分段
         List<Document> segments = tokenTextSplitter.apply(documents);
         // 向量化并存储
         vectorStore.add(segments);
     }
+
 }
diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml b/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml
index 95895e9b0b..ae1f37948f 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/pom.xml
@@ -40,6 +40,7 @@
             <version>${spring-ai.version}</version>
         </dependency>
 
+        <!-- 向量化，基于 Redis 存储，Tika 解析内容 -->
         <dependency>
             <groupId>org.springframework.ai</groupId>
             <artifactId>spring-ai-transformers-spring-boot-starter</artifactId>
@@ -55,13 +56,14 @@
             <artifactId>spring-ai-redis-store</artifactId>
             <version>${spring-ai.version}</version>
         </dependency>
+
+        <!-- TODO @xin：引入我们项目的 starter -->
         <dependency>
             <groupId>org.springframework.data</groupId>
             <artifactId>spring-data-redis</artifactId>
             <optional>true</optional>
         </dependency>
 
-
         <dependency>
             <groupId>cn.iocoder.boot</groupId>
             <artifactId>yudao-common</artifactId>
diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/org/springframework/ai/autoconfigure/vectorstore/redis/RedisVectorStoreAutoConfiguration.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/org/springframework/ai/autoconfigure/vectorstore/redis/RedisVectorStoreAutoConfiguration.java
index 03dc1c19b6..61c38dd1d3 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/org/springframework/ai/autoconfigure/vectorstore/redis/RedisVectorStoreAutoConfiguration.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/org/springframework/ai/autoconfigure/vectorstore/redis/RedisVectorStoreAutoConfiguration.java
@@ -30,6 +30,8 @@ import redis.clients.jedis.JedisPooled;
 /**
  * TODO @xin 先拿 spring-ai 最新代码覆盖，1.0.0-M1 跟 redis 自动配置会冲突
  *
+ * TODO 这个官方，有说啥时候 fix 哇？
+ *
  * @author Christian Tzolov
  * @author Eddú Meléndez
  */
@@ -39,8 +41,6 @@ import redis.clients.jedis.JedisPooled;
 @EnableConfigurationProperties(RedisVectorStoreProperties.class)
 public class RedisVectorStoreAutoConfiguration {
 
-
-
     @Bean
     @ConditionalOnMissingBean
     public RedisVectorStore vectorStore(EmbeddingModel embeddingModel, RedisVectorStoreProperties properties,
diff --git a/yudao-server/src/main/resources/application.yaml b/yudao-server/src/main/resources/application.yaml
index 8677a5b719..804ceb71f2 100644
--- a/yudao-server/src/main/resources/application.yaml
+++ b/yudao-server/src/main/resources/application.yaml
@@ -153,7 +153,7 @@ spring:
 
 spring:
   ai:
-    vectorstore:
+    vectorstore: # 向量存储
       redis:
         index: default-index
         prefix: "default:"