Skip to content

Commit 69f4f47

Browse files
committed
style: fix code style error
Signed-off-by: yuluo-yx <[email protected]>
1 parent 9bdc49b commit 69f4f47

File tree

2 files changed

+50
-49
lines changed

2 files changed

+50
-49
lines changed

document-readers/markdown-reader/src/main/java/org/springframework/ai/reader/markdown/MarkdownDocumentReader.java

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -52,11 +52,11 @@
5252
* Reads the given Markdown resource and groups headers, paragraphs, or text divided by
5353
* horizontal lines (depending on the
5454
* {@link MarkdownDocumentReaderConfig#horizontalRuleCreateDocument} configuration) into
55-
* {@link Document}s.
56-
* Currently, only Markdown resource files in the ClassPath path are supported,
57-
* and Markdown files can be read in the way of directory path configuration.
58-
* Use {@See org.springframework.ai.reader.markdown.MarkdownDocumentReaderTest#testDirPathSingle()}
59-
* {@See org.springframework.ai.reader.markdown.MarkdownDocumentReaderTest#testMultipleMarkdownFiles()}
55+
* {@link Document}s. Currently, only Markdown resource files in the ClassPath path are
56+
* supported, and Markdown files can be read in the way of directory path configuration.
57+
* Use
58+
* {@See org.springframework.ai.reader.markdown.MarkdownDocumentReaderTest#testDirPathSingle()}
59+
* {@See org.springframework.ai.reader.markdown.MarkdownDocumentReaderTest#testMultipleMarkdownFiles()}
6060
*
6161
* @author Piotr Olaszewski
6262
* @auther shown.Ji
@@ -144,8 +144,8 @@ public MarkdownDocumentReader(List<Resource> markdownResource, MarkdownDocumentR
144144
public List<Document> get() {
145145

146146
return this.markdownResources.stream()
147-
.flatMap(markdownResource -> getDocuments(markdownResource).stream())
148-
.collect(Collectors.toList());
147+
.flatMap(markdownResource -> getDocuments(markdownResource).stream())
148+
.collect(Collectors.toList());
149149
}
150150

151151
private List<Document> getDocuments(Resource markdownResource) {
@@ -165,7 +165,8 @@ private List<Document> getDocuments(Resource markdownResource) {
165165

166166
documents = documentVisitor.getDocuments();
167167
}
168-
} catch (IOException e) {
168+
}
169+
catch (IOException e) {
169170
logger.error("Error reading markdown resource: " + e.getMessage(), e);
170171
throw new RuntimeException("Error reading markdown resource", e);
171172
}
@@ -268,7 +269,7 @@ public void visit(FencedCodeBlock fencedCodeBlock) {
268269
public void visit(Text text) {
269270
if (text.getParent() instanceof Heading heading) {
270271
this.currentDocumentBuilder.metadata("category", "header_%d".formatted(heading.getLevel()))
271-
.metadata("title", text.getLiteral());
272+
.metadata("title", text.getLiteral());
272273
}
273274
else {
274275
this.currentParagraphs.add(text.getLiteral());
@@ -317,9 +318,7 @@ private static List<Resource> loadResources(List<String> markdownResourcePaths)
317318

318319
DefaultResourceLoader resourceLoader = new DefaultResourceLoader();
319320

320-
return markdownResourcePaths.stream()
321-
.map(resourceLoader::getResource)
322-
.collect(Collectors.toList());
321+
return markdownResourcePaths.stream().map(resourceLoader::getResource).collect(Collectors.toList());
323322
}
324323

325324
/**
@@ -343,11 +342,13 @@ private static List<String> loadResourcePaths(String resourcePath) {
343342
resources.add("classpath:" + mdFile.getName());
344343
}
345344
}
346-
} else if (file.exists() && file.getName().endsWith(".md")) {
345+
}
346+
else if (file.exists() && file.getName().endsWith(".md")) {
347347
resources.add(resourcePath);
348348
}
349349
}
350-
} else {
350+
}
351+
else {
351352
File file = new File(resourcePath);
352353
if (file.exists() && file.isDirectory()) {
353354
File[] files = file.listFiles((dir, name) -> name.endsWith(".md"));
@@ -356,7 +357,8 @@ private static List<String> loadResourcePaths(String resourcePath) {
356357
resources.add(mdFile.getAbsolutePath());
357358
}
358359
}
359-
} else if (file.exists() && file.getName().endsWith(".md")) {
360+
}
361+
else if (file.exists() && file.getName().endsWith(".md")) {
360362
resources.add(file.getAbsolutePath());
361363
}
362364
}

document-readers/markdown-reader/src/test/java/org/springframework/ai/reader/markdown/MarkdownDocumentReaderTest.java

Lines changed: 33 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -40,11 +40,11 @@ void testDirPathSingle() {
4040
List<Document> documents = reader.get();
4141

4242
assertThat(documents).hasSize(2)
43-
.extracting(Document::getMetadata, Document::getText)
44-
.containsOnly(tuple(Map.of(),
45-
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur diam eros, laoreet sit amet cursus vitae, varius sed nisi. Cras sit amet quam quis velit commodo porta consectetur id nisi. Phasellus tincidunt pulvinar augue."),
46-
tuple(Map.of("category", "blockquote"),
47-
"Proin vel laoreet leo, sed luctus augue. Sed et ligula commodo, commodo lacus at, consequat turpis. Maecenas eget sapien odio. Maecenas urna lectus, pellentesque in accumsan aliquam, congue eu libero. Ut rhoncus nec justo a porttitor. Pellentesque auctor pharetra eros, viverra sodales lorem aliquet id. Curabitur semper nisi vel sem interdum suscipit."));
43+
.extracting(Document::getMetadata, Document::getText)
44+
.containsOnly(tuple(Map.of(),
45+
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur diam eros, laoreet sit amet cursus vitae, varius sed nisi. Cras sit amet quam quis velit commodo porta consectetur id nisi. Phasellus tincidunt pulvinar augue."),
46+
tuple(Map.of("category", "blockquote"),
47+
"Proin vel laoreet leo, sed luctus augue. Sed et ligula commodo, commodo lacus at, consequat turpis. Maecenas eget sapien odio. Maecenas urna lectus, pellentesque in accumsan aliquam, congue eu libero. Ut rhoncus nec justo a porttitor. Pellentesque auctor pharetra eros, viverra sodales lorem aliquet id. Curabitur semper nisi vel sem interdum suscipit."));
4848
}
4949

5050
@Test
@@ -53,42 +53,41 @@ void testDirPathMultiple() {
5353
List<Document> documents = reader.get();
5454

5555
assertThat(documents).hasSize(6)
56-
.extracting(Document::getMetadata, Document::getText)
57-
.containsOnly(
58-
tuple(Map.of("category", "header_1", "title", "This is a fancy header name"),
59-
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Donec tincidunt velit non bibendum gravida. Cras accumsan tincidunt ornare. Donec hendrerit consequat tellus blandit accumsan. Aenean aliquam metus at arcu elementum dignissim."),
60-
tuple(Map.of("category", "header_3", "title", "Header 3"),
61-
"Aenean eu leo eu nibh tristique posuere quis quis massa."),
62-
tuple(Map.of("category", "header_1", "title", "Header 1a"),
63-
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur diam eros, laoreet sit amet cursus vitae, varius sed nisi. Cras sit amet quam quis velit commodo porta consectetur id nisi. Phasellus tincidunt pulvinar augue."),
64-
tuple(Map.of("category", "header_1", "title", "Header 1b"),
65-
"Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia curae; Etiam lobortis risus libero, sed sollicitudin risus cursus in. Morbi enim metus, ornare vel lacinia eget, venenatis vel nibh."),
66-
tuple(Map.of("category", "header_2", "title", "Header 2b"),
67-
"Proin vel laoreet leo, sed luctus augue. Sed et ligula commodo, commodo lacus at, consequat turpis. Maecenas eget sapien odio. Maecenas urna lectus, pellentesque in accumsan aliquam, congue eu libero."),
68-
tuple(Map.of("category", "header_2", "title", "Header 2c"),
69-
"Ut rhoncus nec justo a porttitor. Pellentesque auctor pharetra eros, viverra sodales lorem aliquet id. Curabitur semper nisi vel sem interdum suscipit."));
56+
.extracting(Document::getMetadata, Document::getText)
57+
.containsOnly(tuple(Map.of("category", "header_1", "title", "This is a fancy header name"),
58+
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Donec tincidunt velit non bibendum gravida. Cras accumsan tincidunt ornare. Donec hendrerit consequat tellus blandit accumsan. Aenean aliquam metus at arcu elementum dignissim."),
59+
tuple(Map.of("category", "header_3", "title", "Header 3"),
60+
"Aenean eu leo eu nibh tristique posuere quis quis massa."),
61+
tuple(Map.of("category", "header_1", "title", "Header 1a"),
62+
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur diam eros, laoreet sit amet cursus vitae, varius sed nisi. Cras sit amet quam quis velit commodo porta consectetur id nisi. Phasellus tincidunt pulvinar augue."),
63+
tuple(Map.of("category", "header_1", "title", "Header 1b"),
64+
"Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia curae; Etiam lobortis risus libero, sed sollicitudin risus cursus in. Morbi enim metus, ornare vel lacinia eget, venenatis vel nibh."),
65+
tuple(Map.of("category", "header_2", "title", "Header 2b"),
66+
"Proin vel laoreet leo, sed luctus augue. Sed et ligula commodo, commodo lacus at, consequat turpis. Maecenas eget sapien odio. Maecenas urna lectus, pellentesque in accumsan aliquam, congue eu libero."),
67+
tuple(Map.of("category", "header_2", "title", "Header 2c"),
68+
"Ut rhoncus nec justo a porttitor. Pellentesque auctor pharetra eros, viverra sodales lorem aliquet id. Curabitur semper nisi vel sem interdum suscipit."));
7069
}
7170

7271
@Test
7372
void testMultipleMarkdownFiles() {
74-
MarkdownDocumentReader reader = new MarkdownDocumentReader(List.of("classpath:/only-headers.md", "classpath:/with-formatting.md"));
73+
MarkdownDocumentReader reader = new MarkdownDocumentReader(
74+
List.of("classpath:/only-headers.md", "classpath:/with-formatting.md"));
7575
List<Document> documents = reader.get();
7676

7777
assertThat(documents).hasSize(6)
78-
.extracting(Document::getMetadata, Document::getText)
79-
.containsOnly(
80-
tuple(Map.of("category", "header_1", "title", "This is a fancy header name"),
81-
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Donec tincidunt velit non bibendum gravida. Cras accumsan tincidunt ornare. Donec hendrerit consequat tellus blandit accumsan. Aenean aliquam metus at arcu elementum dignissim."),
82-
tuple(Map.of("category", "header_3", "title", "Header 3"),
83-
"Aenean eu leo eu nibh tristique posuere quis quis massa."),
84-
tuple(Map.of("category", "header_1", "title", "Header 1a"),
85-
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur diam eros, laoreet sit amet cursus vitae, varius sed nisi. Cras sit amet quam quis velit commodo porta consectetur id nisi. Phasellus tincidunt pulvinar augue."),
86-
tuple(Map.of("category", "header_1", "title", "Header 1b"),
87-
"Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia curae; Etiam lobortis risus libero, sed sollicitudin risus cursus in. Morbi enim metus, ornare vel lacinia eget, venenatis vel nibh."),
88-
tuple(Map.of("category", "header_2", "title", "Header 2b"),
89-
"Proin vel laoreet leo, sed luctus augue. Sed et ligula commodo, commodo lacus at, consequat turpis. Maecenas eget sapien odio. Maecenas urna lectus, pellentesque in accumsan aliquam, congue eu libero."),
90-
tuple(Map.of("category", "header_2", "title", "Header 2c"),
91-
"Ut rhoncus nec justo a porttitor. Pellentesque auctor pharetra eros, viverra sodales lorem aliquet id. Curabitur semper nisi vel sem interdum suscipit."));
78+
.extracting(Document::getMetadata, Document::getText)
79+
.containsOnly(tuple(Map.of("category", "header_1", "title", "This is a fancy header name"),
80+
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Donec tincidunt velit non bibendum gravida. Cras accumsan tincidunt ornare. Donec hendrerit consequat tellus blandit accumsan. Aenean aliquam metus at arcu elementum dignissim."),
81+
tuple(Map.of("category", "header_3", "title", "Header 3"),
82+
"Aenean eu leo eu nibh tristique posuere quis quis massa."),
83+
tuple(Map.of("category", "header_1", "title", "Header 1a"),
84+
"Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur diam eros, laoreet sit amet cursus vitae, varius sed nisi. Cras sit amet quam quis velit commodo porta consectetur id nisi. Phasellus tincidunt pulvinar augue."),
85+
tuple(Map.of("category", "header_1", "title", "Header 1b"),
86+
"Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia curae; Etiam lobortis risus libero, sed sollicitudin risus cursus in. Morbi enim metus, ornare vel lacinia eget, venenatis vel nibh."),
87+
tuple(Map.of("category", "header_2", "title", "Header 2b"),
88+
"Proin vel laoreet leo, sed luctus augue. Sed et ligula commodo, commodo lacus at, consequat turpis. Maecenas eget sapien odio. Maecenas urna lectus, pellentesque in accumsan aliquam, congue eu libero."),
89+
tuple(Map.of("category", "header_2", "title", "Header 2c"),
90+
"Ut rhoncus nec justo a porttitor. Pellentesque auctor pharetra eros, viverra sodales lorem aliquet id. Curabitur semper nisi vel sem interdum suscipit."));
9291
}
9392

9493
@Test

0 commit comments

Comments
 (0)