Agents for Amazon Bedrock Test UI

A generic Streamlit UI for testing generative AI agents built using Agents for Amazon Bedrock. For more information, refer to the blog post Developing a Generic Streamlit UI to Test Amazon Bedrock Agents.

Prequisites

• AWS CLI
• Python 3

Running Locally

1. Run the following pip command to install the dependencies:

   pip install -r requirements.txt
   

2. Set the following environment variables:

   • BEDROCK_AGENT_ID - The ID of the agent
   • BEDROCK_AGENT_ALIAS_ID - The ID of the agent alias. The default TSTALIASID will be used if it is not set.
   • The AWS environment variables that provides the credentials to your account. The principal must have the necessary permissions to invoke the Bedrock agent.

3. (Optional) Set the folliowing environment variables to customize the UI:

   • BEDROCK_AGENT_TEST_UI_TITLE - The page title. The default Agents for Amazon Bedrock Test UI will used if it is not set.
   • BEDROCK_AGENT_TEST_UI_ICON - The favicon, such as :bar_chart:. The default Streamlit icon will be used if it is not set.

4. Run the following command to start the Streamlit app:

   streamlit run app.py --server.port=8080 --server.address=localhost