Aurora DSQL MCP Demo

This package provides demo scripts for the Aurora DSQL MCP server.

Running these scripts can incur charges in your AWS account.

Prerequisites:

• An AWS Account and local credentials for a role with dsql:*, and bedrock:* permissions.
• An Aurora DSQL cluster, you can follow the user guide to get started.
• Access to models in Bedrock, you can see which models are used in the scripts for requesting access.

simplest_agent.py

This script demonstrates an agent able to make queriesto an Aurora DSQL cluster through the DSQL MCP server. It assumes that the IMDB database has been loaded to the cluster and runs queries against it.

Example usage:

DSQL_CLUSTER=<your-cluster-id>.dsql.us-east-1.on.aws \
AWS_REGION=us-east-1 \
uv run src/simplest_agent.py "how many movies start with Aurora in the title"

dynamic_prompt.py

This demonstrates how you can use agents to create dynamic prompts for other agents.

Example usage:

DSQL_CLUSTER=<your-cluster-id>.dsql.us-east-1.on.aws \
AWS_REGION=us-east-1 \
uv run src/dynamic_prompt.py

multi_agent.py

This demonstrates how you can setup multiple agents to perform different tasks. In this example, one agent is created to create the schema of the database before saving information about it to a local file; another uses that schema to upload information about news events; and another allows users to query the database about those news events with natural language.

Example:

DSQL_CLUSTER=<your-cluster-id>.dsql.us-east-1.on.aws \
AWS_REGION=us-east-1 \
uv run src/multi_agent.py