Databricks Generative AI Engineer Associate · 30% practice allocation

Databricks Generative AI Engineer Associate: Application development Practice Questions

5 free sample questions from the Application development domain, with answers and explanations. The full Generative AI Engineer Associate bank has 250 questions across 6 domains. Questions are scenario-based and written from the official Databricks exam guide.

1.

A developer is building a LangChain-based RAG pipeline on Databricks and wants to use a Mosaic AI Vector Search index as the retriever. After creating the index, which LangChain integration class should they instantiate to wrap the index as a LangChain-compatible retriever?

  • A. FAISS.from_existing_index() pointed at the Vector Search endpoint
  • B. VectorSearchRetriever, imported directly from langchain_community.retrievers
  • C. DatabricksEmbeddings wrapped around the index object
  • D. DatabricksVectorSearch, then call .as_retriever()✓ Correct
Explanation

DatabricksVectorSearch (from langchain_community.vectorstores) is the correct class to wrap a Mosaic AI Vector Search index, and calling .as_retriever() on it produces a standard LangChain retriever interface. There is no standalone VectorSearchRetriever class in langchain_community.retrievers for Databricks. DatabricksEmbeddings is an embeddings wrapper used for generating embeddings, not for wrapping a vector store as a retriever. FAISS.from_existing_index() is used for local FAISS indexes and has no connection to Mosaic AI Vector Search.

2.

A Generative AI engineer is designing a conversational assistant that must remember the context of a multi-turn conversation. They are implementing this with LangChain on Databricks. Which LangChain memory component is most appropriate when the primary concern is staying within the model's context window limit across many turns?

  • A. ConversationTokenBufferMemory, because it retains the most recent turns based on a strict token count limit
  • B. VectorStoreRetrieverMemory, because it retrieves only the semantically relevant past turns from a vector store
  • C. ConversationBufferMemory, because it stores the complete raw conversation history
  • D. ConversationSummaryMemory, because it condenses older turns into a summary to manage token usage✓ Correct
Explanation

ConversationSummaryMemory progressively summarizes older parts of the conversation, allowing the assistant to maintain coherent context across many turns while keeping the total token count manageable — directly addressing the concern of staying within the context window. ConversationBufferMemory retains the full raw history and will grow unbounded, eventually exceeding context limits. ConversationTokenBufferMemory does enforce a token limit but does so by truncating old turns entirely, losing that context rather than summarizing it. VectorStoreRetrieverMemory retrieves relevant past turns but may miss sequential context needed for coherent multi-turn dialogue.

3.

A Generative AI engineer is using MLflow's LLM-as-judge framework to evaluate a RAG application. They configure mlflow.evaluate() with model_type='databricks-agent'. Which of the following metrics are automatically computed by this evaluation preset? (Select TWO)

  • A. ROUGE-L
  • B. exact_match
  • C. perplexity
  • D. groundedness✓ Correct
  • E. BLEU score
  • F. relevance_to_query✓ Correct
Explanation

The databricks-agent evaluation preset in MLflow automatically computes LLM-as-judge metrics including groundedness (whether the answer is supported by the retrieved context) and relevance_to_query (whether the answer addresses the user's question). BLEU (E) and ROUGE-L (A) are n-gram overlap metrics suited for translation/summarization, not built into the agent preset. Perplexity (C) requires access to model logits and is not part of this preset. Exact_match (B) is a lexical metric not included in the agent evaluation suite.

4.

A Generative AI engineer has deployed a RAG application on Databricks Model Serving. After deployment, end users report that some answers are correct but lack the key contextual detail found in the source documents. Retrieval logs show the correct documents are being retrieved. Which of the following are the MOST likely causes and should be investigated first? (Select TWO)

  • A. The prompt template does not instruct the model to use the provided context and may be causing the model to answer from parametric memory instead.✓ Correct
  • B. The Foundation Model API endpoint is using pay-per-token pricing, which limits output token length.
  • C. The embedding model used to create the index is different from the one used at query time, causing mismatched vector representations.
  • D. The vector index type (Delta Sync vs. Direct Vector Access) is causing retrieval latency that truncates the response.
  • E. The retrieved document chunks are too small (aggressive chunking), and relevant detail is split across chunks that fall outside the top-k results.✓ Correct
  • F. The model serving endpoint's environment does not have the correct MLflow flavor installed.
Explanation

When correct documents are retrieved but answers still lack detail, two common causes are: (1) the prompt template does not explicitly instruct the model to ground its answer in the provided context, causing the LLM to rely on its parametric knowledge instead of the retrieved chunks; and (2) over-aggressive chunking splits relevant detail across multiple small chunks, so critical information lands in chunks ranked outside the top-k and is never passed to the LLM. Option D is incorrect — index type affects sync latency, not response content truncation. Option B is a misconception — pay-per-token pricing does not impose output token length limits; output length is controlled by the max_tokens parameter. Option C describes an embedding model mismatch, which would degrade retrieval quality (wrong documents returned), but retrieval logs already confirm correct documents are being retrieved, ruling this out. Option F is an infrastructure/deployment issue unrelated to content quality in responses.

5.

A Generative AI engineer is evaluating whether to use pay-per-token Foundation Model API endpoints or Provisioned Throughput endpoints for a new internal tool used by approximately 50 employees during business hours. Usage is intermittent and unpredictable, with occasional bursts. Which THREE of the following are valid reasons to choose pay-per-token endpoints for this workload? (Select THREE)

  • A. Pay-per-token endpoints have no idle cost; the team only pays when requests are actually made.✓ Correct
  • B. Pay-per-token endpoints are better suited for workloads with unpredictable, bursty traffic patterns where sustained high throughput is not required.✓ Correct
  • C. Pay-per-token endpoints guarantee lower latency than Provisioned Throughput endpoints under all conditions.
  • D. Pay-per-token endpoints require no capacity planning or throughput tier selection, reducing operational overhead for small teams.✓ Correct
  • E. Pay-per-token endpoints automatically support fine-tuned private models without any additional configuration.
  • F. Pay-per-token endpoints are managed by Databricks, eliminating the need to manage GPU infrastructure for a small internal tool.
Explanation

Options A, B, and D are correct. A: Pay-per-token has no reserved capacity, so there is zero cost when the tool is idle, which is advantageous for low-utilization internal tools. B: Bursty, unpredictable traffic is the classic use case for pay-per-token; Provisioned Throughput is designed for sustained, predictable high load. D: Pay-per-token requires no upfront capacity decisions, which reduces operational burden for a small team. Option C is wrong because Provisioned Throughput generally offers more consistent and lower latency under high load; pay-per-token can experience higher latency during shared-capacity peaks. Option E is wrong because pay-per-token Foundation Model API endpoints serve Databricks-hosted base models; deploying fine-tuned private models requires Provisioned Throughput or a custom serving endpoint. Option F, while partially true, is not a distinguishing reason to choose pay-per-token over Provisioned Throughput specifically, since both are managed by Databricks.

Test yourself across every Generative AI Engineer Associate domain

The free Generative AI Engineer Associate practice test mixes about 20 questions weighted like the exam. The full 250-question bank adds instant grading, flashcards and a timed mock exam.

Study this domain

Application development study guide (Generative AI Engineer Associate) →

Other Generative AI Engineer Associate domains

← All Databricks Generative AI Engineer Associate study resources