promptfoo/examples/rag-full

This RAG example allows you to ask questions over a number of public company SEC filings. It uses LangChain, but the flow is representative of any RAG solution.

There are 3 parts:

1. ingest.py: Chunks and loads PDFs into a vector database (PDFs are pulled from a public Google Cloud bucket)

2. retrieve.py: Promptfoo-compatible provider that answers RAG questions using the database.

3. promptfooconfig.yaml: Test inputs and requirements.

To get started:

1. Set the OPENAI_API_KEY environment variable.

2. Create a python virtual environment: python3 -m venv venv

3. Enter the environment: source venv/bin/activate

4. Install python dependencies: pip install -r requirements.txt

5. Run ingest.py to create the vector database: python ingest.py

Now we're ready to go.

• Edit promptfooconfig.yaml to your liking to configure the questions you'd like to ask in your tests. Then run:
• Edit retrieve.py to control how context is loaded and questions are answered.

promptfoo eval

Afterwards, you can view the results by running promptfoo view

See promptfooconfig.with-asserts.yaml for a more complete example that compares the performance of two RAG configurations.