Skip to content

Generative AI Engineer Associate · Domain 1: Design Applications · 14% of the exam

Task 1.3: Define and order tools that gather knowledge or take actions for multi-stage reasoning; Determine how and when to use Agent Bricks (Knowledge Assistant, Multiagent Supervisor, Information Extraction) to solve problems

Tool order in an agent loop, Unity Catalog functions and MCP tools, and where today's Agent Bricks offerings fit: Information Extraction, Genie Agents for low-code agents, and the legacy Knowledge Assistant and Supervisor Agent.

Study it

  • Tools for multi-stage reasoning and the order they run in

    Partly covered by: The Tool-Use Loop, Agents and Tools

  • Agent Bricks offerings, Information Extraction and Genie Agents

    Lesson coming

Sample questions

Try each one before opening the answer. Every option is explained, with the Databricks documentation page that proves it.

Question 1 · choose 1

An HR agent serves two kinds of requests. Employees ask how many leave days they have left, and an approved SQL query returns that from a Unity Catalog table given an employee ID. HR managers ask open-ended business questions, such as how overtime changed by region last quarter, across the workspace's curated data. Which tool definitions follow Databricks guidance for connecting agents to structured data?

  1. AOne Unity Catalog function that runs any SQL text the model writes, used for both kinds of request
  2. BThe Genie One MCP for both kinds of request, including the employee leave-balance lookup
  3. CA Unity Catalog SQL function on the employee ID for balances, and the Genie One MCP for the rest
  4. DThe Databricks SQL MCP for both kinds of request, so that the model writes and runs every query
Show the answer and why
  • AOne Unity Catalog function that runs any SQL text the model writes, used for both kinds of request

    Incorrect

    Unity Catalog function tools fit a query that is known ahead of time and takes parameters from the agent, not SQL that the model writes.

  • BThe Genie One MCP for both kinds of request, including the employee leave-balance lookup

    Incorrect

    Genie One fits open-ended business questions, but the balance must come from the approved query, which belongs in a parameterized function.

  • CA Unity Catalog SQL function on the employee ID for balances, and the Genie One MCP for the rest

    Correct

    Databricks maps a predefined query with parameters to Unity Catalog function tools, and natural-language business questions across the workspace to the Genie One MCP.

  • DThe Databricks SQL MCP for both kinds of request, so that the model writes and runs every query

    Incorrect

    That MCP is for writing and running SQL, so it would bypass the approved balance query, and system.ai.dbsql also needs the Unity Gateway beta.

Choose a structured-data tool by who writes the query. A Unity Catalog SQL function runs a known query with parameters that the agent supplies; the Genie One MCP interprets business questions across the workspace and queries the data with the caller's permissions.

Question 2 · choose 1

Finance wants vendor name, total amount and due date pulled from thousands of invoice PDFs in a Unity Catalog volume every night, with field-level accuracy measured against a labeled table before go-live. Which Agent Bricks offering fits?

  1. AKnowledge Assistant pointed at the invoice volume
  2. BInformation Extraction, then ai_extract in a scheduled pipeline
  3. CSupervisor Agent coordinating one extractor per field
  4. DA Genie Agent over the Unity Catalog table that lists the invoice files and paths
Show the answer and why
  • AKnowledge Assistant pointed at the invoice volume

    Incorrect

    Knowledge Assistant is a legacy document chatbot; it answers questions rather than filling a schema, and it is not recommended for new agents.

  • BInformation Extraction, then ai_extract in a scheduled pipeline

    Correct

    Information Extraction defines a schema, scores each field against a ground-truth table, and hands off to ai_extract in SQL or a Lakeflow pipeline for scheduled runs.

  • CSupervisor Agent coordinating one extractor per field

    Incorrect

    Supervisor Agent is a legacy multi-agent coordinator and adds orchestration that a single schema extraction does not need.

  • DA Genie Agent over the Unity Catalog table that lists the invoice files and paths

    Incorrect

    A Genie Agent answers natural-language questions with SQL over curated tables; it does not read field values out of PDFs.

Information Extraction turns unstructured documents into structured fields with a defined schema, evaluates extraction quality against labeled data, and runs at scale through ai_extract. Knowledge Assistant and Supervisor Agent are legacy offerings.

Question 3 · choose 1

An old design document says to coordinate several agents through the Agent Bricks Supervisor API. What is its status today?

  1. AIt is in Public Preview and open to new projects through the Previews page
  2. BIt reached end of life and is no longer available
  3. CIt is generally available and recommended
  4. DIt moved into Genie One unchanged
Show the answer and why
  • AIt is in Public Preview and open to new projects through the Previews page

    Incorrect

    It is not in preview; it has been retired.

  • BIt reached end of life and is no longer available

    Correct

    The Supervisor API reached end of life on September 30, 2026; Databricks recommends writing your own agent loop.

  • CIt is generally available and recommended

    Incorrect

    It is no longer available at all.

  • DIt moved into Genie One unchanged

    Incorrect

    There is no such move; the API simply ended.

The Supervisor API was deprecated and then reached end of life on September 30, 2026. Databricks recommends writing your own agent loop and deploying it.

Question 4 · choose 1

Before parsing thousands of contracts in SQL, an engineer wants to view one parsed document side by side with the original, as formatted text or raw JSON. Which Agent Bricks offering does this?

  1. ADocument Parsing
  2. BClassification
  3. CInformation Extraction
  4. DGenie Agents
Show the answer and why
  • ADocument Parsing

    Correct

    Document Parsing shows the source document next to the parsed result and can hand off the ai_parse_document query.

  • BClassification

    Incorrect

    Classification assigns documents to labels; it does not display parsed layouts.

  • CInformation Extraction

    Incorrect

    Information Extraction fills a schema of fields, not a visual parse of the layout.

  • DGenie Agents

    Incorrect

    Genie Agents answer questions over data with SQL.

Document Parsing extracts and visualizes structured content from PDFs, images, Word and PowerPoint files and offers the ai_parse_document query to run at scale.

Question 5 · choose 1

A team sets up an Agent Bricks Classification agent for call transcripts. It defines only the label "complaint" and wants transcripts that are both complaints and cancellations to get both labels. What must it do?

  1. AKeep the single label and turn on confidence scores so both cases are reported
  2. BSwitch to Information Extraction
  3. CAdd at least one more label and enable Multiple labels
  4. DEnable rationales to add labels
Show the answer and why
  • AKeep the single label and turn on confidence scores so both cases are reported

    Incorrect

    One label is not enough to run classification, and confidence scores do not add labels.

  • BSwitch to Information Extraction

    Incorrect

    Extraction fills schema fields; it is not label assignment.

  • CAdd at least one more label and enable Multiple labels

    Correct

    Classification needs at least two labels to run, and the Multiple labels option lets it return more than one label.

  • DEnable rationales to add labels

    Incorrect

    Rationales explain predictions; they do not add labels.

In Classification you add labels manually or import them from a table, need at least two labels, and can enable Multiple labels, confidence scores and rationales.

Question 6 · choose 1

A LangGraph agent should call one specific, curated Genie Agent as a tool. Which approach does Databricks describe for this?

  1. ARegister the Genie Agent as an external MCP server in Unity Gateway
  2. BExport its example SQL into the prompt
  3. CGive the agent a raw JDBC connection
  4. DWrap AI Bridge's GenieAgent helper as a tool
Show the answer and why
  • ARegister the Genie Agent as an external MCP server in Unity Gateway

    Incorrect

    Registering Genie as your own MCP is not supported; use a Databricks-provided MCP or AI Bridge.

  • BExport its example SQL into the prompt

    Incorrect

    Static SQL in a prompt loses Genie's interpretation of new questions.

  • CGive the agent a raw JDBC connection

    Incorrect

    Raw SQL access skips the curated context that makes the Genie Agent reliable.

  • DWrap AI Bridge's GenieAgent helper as a tool

    Correct

    AI Bridge exposes a specific Genie Agent as a Python tool and handles requests, polling and results through the Genie API.

Databricks picks structured-data tools by what the agent needs and who writes the query; for one specific Genie Agent in a Python agent, AI Bridge wraps the Genie API.

Practice domain 1 →Practice all domains →