From dc14fb397489e5b9f8eb30ab5476e7d1f2796d8f Mon Sep 17 00:00:00 2001
From: Wonder <hezhaohui0807@163.com>
Date: Sat, 8 Nov 2025 23:08:17 +0800
Subject: [PATCH] =?UTF-8?q?=20=E2=99=BB=EF=B8=8FRefactor:=20=E7=9F=A5?=
 =?UTF-8?q?=E8=AF=86=E5=BA=93=E8=A7=84=E5=88=99=E9=9B=86=E5=AF=BC=E5=85=A5?=
 =?UTF-8?q?=E4=B8=8A=E4=B8=8B=E6=96=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/wonder/bi/rag/MyMarkdownReader.java   |  2 +-
 .../bi/manager/InsightAgentManagerTest.java   | 29 +++++++++++++++++++
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/Backend/src/main/java/com/wonder/bi/rag/MyMarkdownReader.java b/Backend/src/main/java/com/wonder/bi/rag/MyMarkdownReader.java
index 01c690f..ceab624 100644
--- a/Backend/src/main/java/com/wonder/bi/rag/MyMarkdownReader.java
+++ b/Backend/src/main/java/com/wonder/bi/rag/MyMarkdownReader.java
@@ -21,7 +21,7 @@ public class MyMarkdownReader {
         this.docs = resolver.getResources("classpath:rag/*.md");
     }
 
-    List<Document> loadMarkdown() {
+    public List<Document> loadMarkdown() {
         MarkdownDocumentReaderConfig config = MarkdownDocumentReaderConfig.builder()
                 .withHorizontalRuleCreateDocument(true)
                 .withIncludeCodeBlock(false)
diff --git a/Backend/src/test/java/com/wonder/bi/manager/InsightAgentManagerTest.java b/Backend/src/test/java/com/wonder/bi/manager/InsightAgentManagerTest.java
index faffd22..f5efeab 100644
--- a/Backend/src/test/java/com/wonder/bi/manager/InsightAgentManagerTest.java
+++ b/Backend/src/test/java/com/wonder/bi/manager/InsightAgentManagerTest.java
@@ -1,10 +1,14 @@
 package com.wonder.bi.manager;
 
+import com.wonder.bi.rag.MyMarkdownReader;
 import jakarta.annotation.Resource;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
+import org.springframework.ai.document.Document;
 import org.springframework.boot.test.context.SpringBootTest;
 
+import java.util.List;
+
 import static org.junit.jupiter.api.Assertions.*;
 
 @SpringBootTest
@@ -13,6 +17,9 @@ class InsightAgentManagerTest {
     @Resource
     private InsightAgentManager insightAgent;
 
+    @Resource
+    private MyMarkdownReader reader;
+
     @Test
     void getAnalysisInsight() {
         String insight = insightAgent.getAnalysisInsight("分析需求：\n" +
@@ -25,4 +32,26 @@ class InsightAgentManagerTest {
                 "定义增长 50% 及以上为快速增长");
         Assertions.assertNotNull(insight);
     }
+
+    @Test
+    void getAnalysisInsightWithRag() {
+        List<Document> documents = reader.loadMarkdown();
+        StringBuilder ragText = new StringBuilder();
+        for (Document document: documents) {
+            ragText.append(document.toString());
+        }
+        String insight = insightAgent.getAnalysisInsight("分析需求：\n" +
+                        "分析班级成绩分布\n" +
+                        "原始数据：\n" +
+                        "序号,分数\n" +
+                        "1号,90\n" +
+                        "2号,90\n" +
+                        "3号,80\n" +
+                        "4号,20\n" +
+                        "5号,40\n" +
+                        "6号,89\n" +
+                        "7号,59\n",
+                ragText.toString());
+        Assertions.assertNotNull(insight);
+    }
 }
\ No newline at end of file