Fix PDF document grouping logic to ensure correct page distribution according to pagesPerDocument

Signed-off-by: stroller <[email protected]>

Author: jiafu1115

document-readers/pdf-reader/src/main/java/org/springframework/ai/reader/pdf/PagePdfDocumentReader.java

@@ -44,6 +44,7 @@
  * pageBottomMargin = 0
  *
  * @author Christian Tzolov
+ * @author Fu Jian
  */
 public class PagePdfDocumentReader implements DocumentReader {
 
@@ -119,7 +120,7 @@ public List<Document> get() {
 				pagesPerDocument++;
 
 				if (this.config.pagesPerDocument != PdfDocumentReaderConfig.ALL_PAGES
-						&& pagesPerDocument >= this.config.pagesPerDocument) {
+						&& pagesPerDocument > this.config.pagesPerDocument) {
 					pagesPerDocument = 0;
 
 					var aggregatedPageTextGroup = pageTextGroupList.stream().collect(Collectors.joining());

document-readers/pdf-reader/src/test/java/org/springframework/ai/reader/pdf/PagePdfDocumentReaderTests.java

@@ -30,6 +30,7 @@
 /**
  * @author Christian Tzolov
  * @author Tibor Tarnai
+ * @author Fu Jian
  */
 class PagePdfDocumentReaderTests {
 
@@ -71,4 +72,16 @@ void testIndexOutOfBound() {
 		assertThat(documents).hasSize(64);
 	}
 
+	@Test
+	void testPagesPerDocument() {
+		var documents = new PagePdfDocumentReader("classpath:/sample2.pdf",
+				PdfDocumentReaderConfig.builder()
+					.withPageExtractedTextFormatter(ExtractedTextFormatter.builder().build())
+					.withPagesPerDocument(32)
+					.build())
+			.get();
+
+		assertThat(documents).hasSize(2);
+	}
+
 }