Add a Knowledge Base component to a flow app

Navigate to the Amazon SageMaker Unified Studio landing page by using the URL from your admininstrator.

Access Amazon SageMaker Unified Studio using your IAM or single sign-on (SSO) credentials. For more information, see Access Amazon SageMaker Unified Studio.

If the project that you want to use isn't already open, do the following:

1. Choose the current project at the top of the page. If a project isn't already open, choose Select a project.

2. Select Browse all projects.

3. In Projects select the project that you want to use.

Choose the Build menu option at the top of the page.

In MACHINE LEARNING & GENERATIVE AI choose My apps.

In Apps choose the flow app that you want to add the knowledge base component to.

In the Flow app builder pane, select the Nodes tab.

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

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.

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

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

In the 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.

Choose Save to save your changes.