AWS KB Retrieval MCP Integration

Guide for integrating AWS Knowledge Base retrieval capabilities with PraisonAI agents using MCP

Add AWS KB Retrieval Tool to AI Agent

Quick Start

1. Install Dependencies

   Make sure you have Node.js installed, as the MCP server requires it:

   pip install praisonaiagents mcp
   
2. Set API Keys

   Set your AWS credentials as environment variables in your terminal:

   export AWS_ACCESS_KEY_ID=your_aws_access_key_id_here
   export AWS_SECRET_ACCESS_KEY=your_aws_secret_access_key_here
   export AWS_REGION=your_aws_region_here
   export OPENAI_API_KEY=your_openai_api_key_here
   
3. Create a file

   Create a new file aws_kb_retrieval.py with the following code:

   from praisonaiagents import Agent, MCP
   import os
   
   # Get AWS credentials from environment
   aws_access_key = os.getenv("AWS_ACCESS_KEY_ID")
   aws_secret_key = os.getenv("AWS_SECRET_ACCESS_KEY")
   aws_region = os.getenv("AWS_REGION")
   
   # Use a single string command with AWS KB Retrieval configuration
   aws_kb_agent = Agent(
       instructions="""You are a helpful assistant that can interact with AWS Knowledge Base.
       Use the available tools when relevant to retrieve and process AWS information.""",
       llm="gpt-4o-mini",
       tools=MCP("npx -y @modelcontextprotocol/server-aws-kb-retrieval",
               env={
                   "AWS_ACCESS_KEY_ID": aws_access_key,
                   "AWS_SECRET_ACCESS_KEY": aws_secret_key,
                   "AWS_REGION": aws_region
               })
   )
   
   aws_kb_agent.start("Search AWS documentation about EC2 instances")
   
4. Run the Agent

   Execute your script:

   python aws_kb_retrieval.py
   

Requirements

• Python 3.10 or higher
• Node.js installed on your system
• AWS credentials (Access Key ID, Secret Access Key, and Region)
• OpenAI API key (for the agent’s LLM)