ragulate 0.0.11

A tool for evaluating RAG pipelines

RAGulate

A tool for evaluating RAG pipelines

The Metrics

The RAGulate currently reports 4 relevancy metrics: Answer Correctness, Answer Relevance, Context Relevance, and Groundedness.

• Answer Correctness
  • How well does the generated answer match the ground-truth answer?
  • This confirms how well the full system performed.
• Answer Relevance
  • Is the generated answer relevant to the query?
  • This shows if the LLM is responding in a way that is helpful to answer the query.
• Context Relevance:
  • Does the retrieved context contain information to answer the query?
  • This shows how well the retrieval part of the process is performing.
• Groundedness:
  • Is the generated response supported by the context?
  • Low scores here indicate that the LLM is hallucinating.

Example Output

The tool outputs results as images like this:

These images show distribution box plots of the metrics for different test runs.

Installation

pip install ragulate

Initial Setup

1. Set your environment variables or create a .env file. You will need to set OPENAI_API_KEY and any other environment variables needed by your ingest and query pipelines.

2. Wrap your ingest pipeline in a single python method. The method should take a file_path parameter and any other variables that you will pass during your experimentation. The method should ingest the passed file into your vector store.

   See the ingest() method in experiment_chunk_size_and_k.py as an example. This method configures an ingest pipeline using the parameter chunk_size and ingests the file passed.

3. Wrap your query pipeline in a single python method, and return it. The method should have parameters for any variables that you will pass during your experimentation. Currently only LangChain LCEL query pipelines are supported.

   See the query() method in experiment_chunk_size_and_k.py as an example. This method returns a LangChain LCEL pipeline configured by the parameters chunk_size and k.

Note: It is helpful to have a **kwargs param in your pipeline method definitions, so that if extra params are passed, they can be safely ignored.

Usage

Summary

usage: ragulate [-h] {download,ingest,query,compare} ...

RAGu-late CLI tool.

options:
  -h, --help            show this help message and exit

commands:
    download            Download a dataset
    ingest              Run an ingest pipeline
    query               Run an query pipeline
    compare             Compare results from 2 (or more) recipes

Example

For the examples below, we will use the example experiment experiment_chunk_size_and_k.py and see how the RAG metrics change for changes in chunk_size and k (number of documents retrieved).

1. Download a dataset. See available datasets here: https://llamahub.ai/?tab=llama_datasets

• If you are unsure where to start, recommended datasets are:

  • BraintrustCodaHelpDesk
  • BlockchainSolana

  Examples:

  • ragulate download -k llama BraintrustCodaHelpDesk
  • ragulate download -k llama BlockchainSolana

2. Ingest the datasets using different methods:

   Examples:

   • Ingest with chunk_size=500:

     ragulate ingest -n chunk_size_500 -s experiment_chunk_size_and_k.py -m ingest \
     --var-name chunk_size --var-value 500 --dataset BraintrustCodaHelpDesk --dataset BlockchainSolana
     

   • Ingest with chunk_size=1000:

     ragulate ingest -n chunk_size_1000 -s experiment_chunk_size_and_k.py -m ingest \
     --var-name chunk_size --var-value 1000 --dataset BraintrustCodaHelpDesk --dataset BlockchainSolana
     

3. Run query and evaluations on the datasets using methods:

   Examples:

   • Query with chunk_size=500 and k=2

     ragulate query -n chunk_size_500_k_2 -s experiment_chunk_size_and_k.py -m query_pipeline \
     --var-name chunk_size --var-value 500  --var-name k --var-value 2 --dataset BraintrustCodaHelpDesk --dataset BlockchainSolana
     

   • Query with chunk_size=1000 and k=2

     ragulate query -n chunk_size_1000_k_2 -s experiment_chunk_size_and_k.py -m query_pipeline \
     --var-name chunk_size --var-value 1000  --var-name k --var-value 2 --dataset BraintrustCodaHelpDesk --dataset BlockchainSolana
     

   • Query with chunk_size=500 and k=5

     ragulate query -n chunk_size_500_k_5 -s experiment_chunk_size_and_k.py -m query_pipeline \
     --var-name chunk_size --var-value 500  --var-name k --var-value 5 --dataset BraintrustCodaHelpDesk --dataset BlockchainSolana
     

   • Query with chunk_size=1000 and k=25

     ragulate query -n chunk_size_1000_k_5 -s experiment_chunk_size_and_k.py -m query_pipeline \
     --var-name chunk_size --var-value 1000  --var-name k --var-value 5 --dataset BraintrustCodaHelpDesk --dataset BlockchainSolana
     

4. Run a compare to get the results:

   Example:

   ragulate compare -r chunk_size_500_k_2 -r chunk_size_1000_k_2 -r chunk_size_500_k_5 -r chunk_size_1000_k_5
   

   This will output 2 png files. one for each dataset.

Current Limitations

• The evaluation model is locked to OpenAI gpt3.5
• Only LangChain query pipelines are supported
• Only LlamaIndex datasets are supported
• There is no way to specify which metrics to evaluate.