diff --git a/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnector.java b/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnector.java index bb9fa032..d54d6371 100644 --- a/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnector.java +++ b/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnector.java @@ -4,7 +4,6 @@ import com.google.common.flogger.FluentLogger; import java.io.File; -import java.io.FileNotFoundException; import java.io.IOException; import java.util.Optional; import javax.inject.Inject; @@ -30,14 +29,14 @@ public PDFConnector(@NonNull PDFConnectorOption readOption) { public Optional read() { try { PDDocument selectedDocument; - Optional bytes = readOption.getBytes(); + Optional bytes = readOption.getFileBytes(); Optional filePath = readOption.getFilePath(); if (bytes.isPresent()) { selectedDocument = PDDocument.load(bytes.get()); } else if (filePath.isPresent()) { selectedDocument = PDDocument.load(new File(filePath.get())); } else { - throw new PDFConnectorOptionNotFoundException(); + throw new PDFConnectorOptionNotFoundException("No suitable read option provided"); } @Cleanup PDDocument document = selectedDocument; diff --git a/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOption.java b/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOption.java index c0cc965f..51a43b32 100644 --- a/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOption.java +++ b/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOption.java @@ -10,6 +10,14 @@ @Data @Builder(toBuilder = true, setterPrefix = "set") public class PDFConnectorOption implements ConnectorOption { - @Builder.Default private Optional filePath = Optional.empty(); - @Builder.Default private Optional bytes = Optional.empty(); + private String filePath; + private byte[] fileBytes; + + public Optional getFilePath() { + return Optional.ofNullable(filePath); + } + + public Optional getFileBytes() { + return Optional.ofNullable(fileBytes); + } } diff --git a/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOptionNotFoundException.java b/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOptionNotFoundException.java index 70e53261..5bb5a565 100644 --- a/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOptionNotFoundException.java +++ b/src/main/java/ai/knowly/langtorch/connector/pdf/PDFConnectorOptionNotFoundException.java @@ -1,7 +1,7 @@ package ai.knowly.langtorch.connector.pdf; public class PDFConnectorOptionNotFoundException extends RuntimeException { - public PDFConnectorOptionNotFoundException() { - super(); + public PDFConnectorOptionNotFoundException(String msg) { + super(msg); } }