diff --git a/examples/java/src/main/java/com/xebia/functional/xef/java/auto/PDFDocument.java b/examples/java/src/main/java/com/xebia/functional/xef/java/auto/PDFDocument.java index 71f9f8f88..58b0fb68e 100644 --- a/examples/java/src/main/java/com/xebia/functional/xef/java/auto/PDFDocument.java +++ b/examples/java/src/main/java/com/xebia/functional/xef/java/auto/PDFDocument.java @@ -18,18 +18,18 @@ public static class AIResponse { public String source; } - private static final String pdfUrl = "https://people.cs.ksu.edu/~schmidt/705a/Scala/Programming-in-Scala.pdf"; + private static final String PDF_URL = "https://people.cs.ksu.edu/~schmidt/705a/Scala/Programming-in-Scala.pdf"; private static final BufferedReader sysin = new BufferedReader(new InputStreamReader(System.in)); private static CompletableFuture askQuestion(AIScope scope) { - System.out.println("Enter your question: "); + System.out.println("Enter your question ( to exit): "); String line = readLine(); if (line == null || line.isBlank()) { return CompletableFuture.completedFuture(null); } else { scope.prompt(line, AIResponse.class) - .thenAccept((aiRes) -> System.out.println(aiRes.answer + "\n---\n" + + .thenAccept(aiRes -> System.out.println(aiRes.answer + "\n---\n" + aiRes.source + "\n---\n")); return askQuestion(scope); @@ -39,7 +39,7 @@ private static CompletableFuture askQuestion(AIScope scope) { public static void main(String[] args) throws ExecutionException, InterruptedException { TextSplitter textSplitter = TokenTextSplitter(ModelType.getDEFAULT_SPLITTER_MODEL(), 100, 50); try (AIScope scope = new AIScope()) { - scope.contextScope(scope.pdf(pdfUrl, textSplitter).get(), PDFDocument::askQuestion).get(); + scope.contextScope(scope.pdf(PDF_URL, textSplitter).get(), PDFDocument::askQuestion).get(); } }