nico-martin / ask-my-pdf

A Webapp that uses Retrieval Augmented Generation (RAG) and Large Language Models to interact with a PDF directly in the browser.
https://pdf.nico.dev
MIT License
58 stars 5 forks source link
rag transformers-js webai webllm

Ask my PDF

Ask my PDF

A Webapp that uses Retrieval Augmented Generation (RAG) and Large Language Models to interact with a PDF directly in the browser.

pdf.nico.dev

There are three main pillars this tool is based on:

1. Read the PDF

Ask my PDF uses PDF.js to process a PDF locally and split the content up into separate lines.

2. Vector Search

Individual paragraphs and then sentences are extracted from the document and will be mapped to a multi-dimensional dense vector space using all-MiniLM-L6-v2 with TransformersJS. Those entries are then stored in an in-memory VectorDB directly in the browser.

As soon as a query is submitted, it is also vectorized and the cosine similarity search is used to find the most similar text sections, together with the lines surrounding them, so as not to lose the context.

3. LLM answer generation

The text sections found this way together with the query and a few instructions are then used as the input prompt to the Gemma2-2B LLM, compiled to WebAssembly and WebGPU using MLC LLM, which will then generate a response.