Amazon Redshift

Use Amazon Redshift tools with Agno agents.

Enable Agno agents to access cloud warehousing functionality and interface directly with Amazon Redshift. This allows them to perform petabyte-scale data analysis and schema exploration without requiring manual SQL coding from the user.

Prerequisites

• Set environment variables for IAM authentication:

export AWS_ACCESS_KEY_ID="your-access-key" 
export AWS_SECRET_ACCESS_KEY="your-secret-key" 
export AWS_SESSION_TOKEN="your-session-token" 
export REDSHIFT_HOST="your-workgroup.123456789.us-east-1.redshift-serverless.amazonaws.com" 
export REDSHIFT_DATABASE="dev"

from agno.agent import Agent
from agno.tools.redshift import RedshiftTools

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------


# Example 1: Standard username/password authentication
agent = Agent(
    tools=[
        RedshiftTools(
            user="your-username",
            password="your-password",
        )
    ]
)

# Example 2: IAM authentication with environment variables (Serverless)
agent_iam = Agent(
    tools=[
        RedshiftTools(
            iam=True,
        )
    ]
)

# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
    agent.print_response(
        "List the tables in the database and describe one of the tables", markdown=True
    )

    agent_iam.print_response("Run a query to select 1 + 1 as result", markdown=True)

Run the Example

# Clone and setup repo
git clone https://github.com/agno-agi/agno.git
cd agno/cookbook/91_tools

# Create and activate virtual environment
./scripts/demo_setup.sh
source .venvs/demo/bin/activate

python redshift_tools.py

For details, see Amazon Redshift cookbook.