DatabricksDesign Applications

RAG & Vector Search — Databricks GenAI Associate Practice Question

A representative Databricks Certified Generative AI Engineer Associate (Databricks GenAI Associate) exam question on RAG & Vector Search. Work through it below, then read why each option is right or wrong.

This question is part of the free Databricks Certified Generative AI Engineer Associate practice test.

Short answer

The correct answer is A. Retrieval-augmented generation that queries a Mosaic AI Vector Search index of policy documents at inference time.

Retrieval-augmented generation (RAG) separates knowledge from the model: policy documents are chunked, embedded, and indexed in Mosaic AI Vector Search, so weekly changes only require re-indexing, not retraining. Retrieved passages also provide citable grounding. Continued pre-training every week is exactly the slow, costly retraining cycle the company rejected, and it cannot guarantee citations. Static few-shot prompts freeze knowledge at prompt-authoring time, so weekly policy changes would make answers stale, and prompt size limits how much policy text fits. A SQL-writing agent suits structured analytics, but return policies are unstructured prose, and text-to-SQL adds latency and failure modes without solving grounding on documents.

The Question

A retail company wants a customer-support assistant on Databricks that answers questions about its internal return policies, which change weekly. Fine-tuning a model on each policy update is considered too slow and costly. Latency must stay low and answers must cite current policy text. Which application pattern best fits these requirements?

ARetrieval-augmented generation that queries a Mosaic AI Vector Search index of policy documents at inference timeCorrect
BContinued pre-training of an open-source LLM on the full policy corpus every week
CA code-generation agent that writes SQL against the policy tables for every user question
DPrompt-only few-shot examples of past support tickets embedded in a static system prompt

Why A is correct

Retrieval-augmented generation (RAG) separates knowledge from the model: policy documents are chunked, embedded, and indexed in Mosaic AI Vector Search, so weekly changes only require re-indexing, not retraining. Retrieved passages also provide citable grounding. Continued pre-training every week is exactly the slow, costly retraining cycle the company rejected, and it cannot guarantee citations. Static few-shot prompts freeze knowledge at prompt-authoring time, so weekly policy changes would make answers stale, and prompt size limits how much policy text fits. A SQL-writing agent suits structured analytics, but return policies are unstructured prose, and text-to-SQL adds latency and failure modes without solving grounding on documents.

Why the other options are wrong

Option B: Continued pre-training of an open-source LLM on the full policy corpus every week

Option B does not satisfy the requirement in the scenario. Review the explanation above: the correct choice (A) is the only one that fully meets every constraint stated in the question.

Option C: A code-generation agent that writes SQL against the policy tables for every user question

Option C does not satisfy the requirement in the scenario. Review the explanation above: the correct choice (A) is the only one that fully meets every constraint stated in the question.

Option D: Prompt-only few-shot examples of past support tickets embedded in a static system prompt

Option D does not satisfy the requirement in the scenario. Review the explanation above: the correct choice (A) is the only one that fully meets every constraint stated in the question.

Key idea: RAG & Vector Search

Retrieval-augmented generation (RAG) separates knowledge from the model: policy documents are chunked, embedded, and indexed in Mosaic AI Vector Search, so weekly changes only require re-indexing, not retraining. Retrieved passages also provide citable grounding. Continued pre-training every week is exactly the slow, costly retraining cycle the company rejected, and it cannot guarantee citations. Static few-shot prompts freeze knowledge at prompt-authoring time, so weekly policy changes would make answers stale, and prompt size limits how much policy text fits. A SQL-writing agent suits structured analytics, but return policies are unstructured prose, and text-to-SQL adds latency and failure modes without solving grounding on documents. On the Databricks GenAI Associate exam, questions in the "Design Applications" domain test whether you can map a scenario's constraints to the right choice. Read the requirement carefully, eliminate options that violate any single constraint, and pick the one that satisfies all of them with the least operational overhead.

Ready to see how you'd score?

Take the free practice quiz and find out which Databricks Certified Generative AI Engineer Associate domains you need to focus on. No signup required.

Practice 5 similar questions

Same cert, same or adjacent domain. Use these after reviewing the explanation.

Related Databricks Certified Generative AI Engineer Associate Practice Questions

Official sources used

Verify program and exam details at the source before scheduling.