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

Amazon Bedrock Studio, renamed to Amazon Bedrock IDE, is now available in Amazon SageMaker Unified Studio. Amazon Bedrock Studio will be available until February 28, 2025. You may access existing workspaces in this previous version through February 28, 2025, but you may not create new workspaces. To access the enhanced GA version of Amazon Bedrock Studio with additional features and capabilities, you can create a new Amazon SageMaker Unified Studio domain. To learn about Amazon Bedrock Studio IDE, see the documentation.

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.