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 loadMarkdown() { + public List 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 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