AIP-C01 practice question: Query expansion, decomposition and rewriting

Foundation Model Integration, Data Management, and Compliance A free question from our original AIP-C01 practice exam. Try it, then open the answer to see why each option is right or wrong.

A travel company's assistant answers questions from its fare rules, baggage policy and pet travel documents, all stored in one knowledge base.

A customer asks: compare the cancellation terms and the baggage allowance for the Lisbon fare, and tell me whether my dog can ride in the cabin. The assistant returns a long answer that covers only the cancellation terms, because a single search for the whole question ranks those passages highest.

The team wants every part of a multi-part question answered from the right passages, with the answer still grounded in the retrieved text.

Which TWO approaches meet these requirements? (Choose two.)

Choose two answers.

A. Retrieve the top 100 passages for the full question in one search, and send all of them to a larger generation model, which can sort out which passages answer each part.
B. Ask the customer to send each part of the question in a separate message, and answer each message with its own search, so the assistant never needs to split a question.
C. Turn on query decomposition in the knowledge base's RetrieveAndGenerate settings, so that the question is split into sub-queries that are each retrieved before the answer is written.
D. Convert the question into a SQL query and run it against a structured data knowledge base connected to Amazon Redshift, so that each policy is looked up as a table row.
E. Build an agent on AgentCore Runtime with a knowledge base search tool, and let the agent call the tool once for each part of the question before it writes the answer.

Show the answer and explanation

Answers: C, E

A. Wrong: The right passages may still be ranked below the cancellation text, and a larger model cannot read passages that were never retrieved. The requirement for per-part retrieval is not met.
B. Wrong: This works, but it moves the splitting work to the customer and changes the product experience. The requirement is for the assistant to answer multi-part questions by itself.
C. Right: Query decomposition breaks a complex question into smaller sub-queries, such as cancellation, baggage and pet cabin rules, and retrieves passages for each one. The baggage and pet passages are no longer crowded out, and the answer is still generated from the retrieved text. Learn more: Configure and customize queries and response generation - Amazon Bedrock (opens in a new tab)
D. Wrong: Structured data knowledge bases answer questions over tables, but the policies here are unstructured documents. The approach cannot find the pet travel rules in free text, so it misses the requirement. Learn more: Create a knowledge base by connecting to a structured data store (opens in a new tab)
E. Right: The agent decides the parts, calls the search tool separately for each, and answers only from the returned passages. Each part gets its own retrieval, which meets both the completeness and grounding requirements. Learn more: Overview of Amazon Bedrock AgentCore (opens in a new tab)

Explanation

Both correct approaches split the question into parts and retrieve for each part, so the baggage and pet passages are not crowded out. Query decomposition does the splitting inside the knowledge base request, while the agent chooses its own searches, and both answer from the retrieved text. A single large search, a structured knowledge base over free text and pushing the work to the customer each fail a requirement.

AIP-C01 practice question: Query expansion, decomposition and rewriting

Explanation

Study

Keep practicing