RAG architecture overview in Prompt Engineering / GenAI - Model Pipeline Trace

Model Pipeline - RAG architecture overview

The RAG (Retrieval-Augmented Generation) architecture combines a search step with a text generation step. It first finds relevant documents from a large collection, then uses those documents to help create better answers.

Data Flow - 4 Stages

1Input Query

1 query string→User provides a question or prompt→1 query string

"What is the capital of France?"

↓

2Document Retrieval

1 query string→Searches a large document database for relevant texts→Top 5 documents (5 texts)

["Paris is the capital of France.", "France is in Europe.", "The Eiffel Tower is in Paris.", "French culture is rich.", "Paris has many museums."]

↓

3Context Preparation

1 query string + 5 documents→Combines query with retrieved documents to form input for generation→1 combined input string

"Question: What is the capital of France? Context: Paris is the capital of France. France is in Europe. The Eiffel Tower is in Paris. French culture is rich. Paris has many museums."

↓

4Text Generation

1 combined input string→Generates an answer using a language model conditioned on the context→1 answer string

"The capital of France is Paris."

Training Trace - Epoch by Epoch

Loss: 2.3 |****     
       1.8 |******   
       1.4 |******** 
       1.1 |*********
       0.9 |*********

Epoch	Loss ↓	Accuracy ↑	Observation
1	2.3	0.25	Model starts learning, loss is high, accuracy low
2	1.8	0.40	Loss decreases, accuracy improves
3	1.4	0.55	Model learns better context usage
4	1.1	0.65	Continued improvement in generation quality
5	0.9	0.72	Model converges with good retrieval and generation

Prediction Trace - 4 Layers

Layer 1: Input Query

Layer 2: Document Retrieval

Layer 3: Context Preparation

Layer 4: Text Generation

Model Quiz - 3 Questions

What is the main role of the document retrieval step in RAG?

A Find relevant documents to help answer the query

B Generate the final answer text

C Clean the input query

D Evaluate model accuracy

Key Insight

RAG improves answer quality by first finding helpful documents, then using them as context for generation. This two-step approach helps the model give more accurate and informative responses.

Practice

1. What is the main purpose of the retriever component in a RAG architecture?

easy

A. To find relevant documents or information from a large dataset

B. To generate natural language answers from scratch

C. To train the model on labeled data

D. To evaluate the accuracy of the answers

Solution

1. Step 1: Understand the role of retriever in RAG

   The retriever searches a large collection of documents to find relevant information related to the question.

2. Step 2: Differentiate retriever from generator

   The generator uses the retrieved information to create a natural language answer, not to find documents.

3. Final Answer:

   To find relevant documents or information from a large dataset -> Option A

4. Quick Check:

   Retriever = Find info [OK]

Hint: Retriever searches data; generator writes answers [OK]

Common Mistakes:

• Confusing retriever with generator
• Thinking retriever generates answers
• Assuming retriever evaluates answers

2. Which of the following correctly describes the sequence of operations in a RAG model?

easy

A. Generate answer first, then retrieve documents

B. Retrieve documents first, then generate answer

C. Train model, then retrieve documents

D. Evaluate answer, then generate documents

Solution

1. Step 1: Recall RAG workflow

   RAG first retrieves relevant documents to provide context for the answer.

2. Step 2: Understand generation step

   After retrieval, the generator uses the documents to produce a final answer.

3. Final Answer:

   Retrieve documents first, then generate answer -> Option B

4. Quick Check:

   Retrieve before generate [OK]

Hint: Retrieve info before writing answer [OK]

Common Mistakes:

• Thinking generation happens before retrieval
• Mixing training with retrieval steps
• Confusing evaluation with generation

3. Consider this simplified Python pseudocode for a RAG-like process:

retrieved_docs = retriever.search(query)
answer = generator.generate(retrieved_docs, query)
print(answer)

What will be printed if the retriever returns an empty list?

medium

A. An answer generated without context, possibly generic or incorrect

B. A runtime error because generator cannot handle empty input

C. The original query string printed

D. An empty string printed

Solution

1. Step 1: Analyze retriever output

   The retriever returns an empty list, meaning no documents found.

2. Step 2: Understand generator behavior

   The generator tries to create an answer without context, so it may produce a generic or less accurate answer, but no error occurs.

3. Final Answer:

   An answer generated without context, possibly generic or incorrect -> Option A

4. Quick Check:

   Empty retrieval leads to generic answer [OK]

Hint: Empty retrieval means generic answer, not error [OK]

Common Mistakes:

• Assuming empty retrieval causes error
• Thinking query is printed directly
• Expecting empty string output

4. You have a RAG model that always returns irrelevant answers. Which of these is the most likely cause?

medium

A. The model is overfitting on training data

B. Generator is not trained on any data

C. Retriever is returning unrelated documents

D. The evaluation metric is incorrect

Solution

1. Step 1: Identify cause of irrelevant answers

   If answers are irrelevant, the source documents are likely unrelated to the question.

2. Step 2: Check retriever role

   The retriever finds documents; if it returns unrelated ones, the generator has poor context to answer.

3. Final Answer:

   Retriever is returning unrelated documents -> Option C

4. Quick Check:

   Bad retrieval causes irrelevant answers [OK]

Hint: Check retriever output first for relevance [OK]

Common Mistakes:

• Blaming generator without checking retrieval
• Confusing overfitting with retrieval errors
• Ignoring data quality issues

5. In a RAG system designed for a constantly updated news database, which advantage does RAG provide compared to a standard language model?

hard

A. It generates answers faster by skipping retrieval

B. It always produces shorter answers

C. It requires no training data at all

D. It can access fresh news by retrieving documents without retraining

Solution

1. Step 1: Understand RAG with dynamic data

   RAG retrieves documents from an external source, so it can use new data without retraining the generator.

2. Step 2: Compare with standard language models

   Standard models need retraining to learn new info, but RAG updates answers by searching fresh documents.

3. Final Answer:

   It can access fresh news by retrieving documents without retraining -> Option D

4. Quick Check:

   RAG updates answers via retrieval [OK]

Hint: RAG uses retrieval to handle new data easily [OK]

Common Mistakes:

• Thinking RAG skips retrieval
• Assuming no training data needed
• Believing RAG limits answer length