Add a Knowledge Base component to a Prompt Flow app - Amazon Bedrock Studio

Amazon Bedrock Studio is in preview release and is subject to change.

Add a Knowledge Base component to a Prompt Flow app

In this procedure, you add a Knowledge Base component to an existing Prompt Flow app.

  1. Open the Propmt Flows app in Amazon Bedrock Studio.

  2. In the Prompt Flow builder pane, select the Nodes tab.

  3. From the Data section, drag a Knowledge Base node onto the flow builder canvas.

  4. The circles on the nodes are connection points. Draw a line from the circle on the upstream node (such as the Flow input node) to the circle on the Input section of the Knowledge Base node that you just added.

  5. Connect the Output of the Knowledge Base node to the downstream node that you want the Knowledge Base to send its output to. The Prompt Flow should look similar to the following image:

    Connect an Amazon Bedrock Studio Knowledge Base node to a downstream node.

  6. In the the flow builder, select the Knowledge Base node that you just added.

  7. In the Prompt Flow builder pane, choose the Configure tab and do the following:

    1. For Node name, enter a name for the Knowledge Base node.

    2. For Select Knowledge Base in the Knowledge Base Details section, select the Knowledge Base that you just created.

    3. For Select response generation model, select the model that you want the Knowledge Base to generate responses with.

  8. Choose Save to save your changes.