This is an excellent topic, as it sits at the intersection of a popular framework (Spring AI), a specific resource format (PDF), and a vital developer platform (GitHub).
@Service public class GitHubPdfFetcher private final GitHub github = new GitHubBuilder().withOAuthToken(System.getenv("GITHUB_TOKEN")).build(); public List<byte[]> fetchPdfsFromRepo(String repoName, String path) throws IOException GHRepository repo = github.getRepository(repoName); List<GHContent> pdfs = repo.getDirectoryContent(path).stream() .filter(c -> c.getName().endsWith(".pdf")) .toList(); return pdfs.stream().map(content -> try (InputStream is = content.read()) return is.readAllBytes(); catch (IOException e) throw new RuntimeException(e); ).collect(Collectors.toList());
@Service public class PdfDocumentService public List<Document> parsePdfs(List<byte[]> pdfBytesList) return pdfBytesList.stream() .flatMap(bytes -> ByteArrayInputStream bais = new ByteArrayInputStream(bytes); TikaDocumentReader reader = new TikaDocumentReader(bais); return reader.get().stream(); // Returns List<Document> ) .collect(Collectors.toList());
@RestController public class ChatController private final ChatClient chatClient; private final VectorStore vectorStore; @GetMapping("/ask") public String askAboutGitHubPdfs(@RequestParam String question) // Retrieve relevant PDF chunks List<Document> relevantDocs = vectorStore.similaritySearch(question); // Create system prompt with context String context = relevantDocs.stream() .map(Document::getText) .collect(Collectors.joining("\n---\n")); return chatClient.call(new Prompt( List.of(new SystemMessage("Answer based only on: " + context), new UserMessage(question)) )).getResult().getOutput().getText();