Architecting Multi-Region AI Agent High Availability & Automated Failover with Amazon Route 53, DynamoDB Global Tables, and Bedrock Agents

import time
import boto3
from botocore.exceptions import ClientError

class MultiRegionAgentFailoverController:
    def __init__(self, primary_region: str, secondary_region: str, routing_control_arn: str):
        self.primary_region = primary_region
        self.secondary_region = secondary_region
        self.routing_control_arn = routing_control_arn

        self.arc_client = boto3.client('route53-recovery-control-config', region_name='us-east-1')
        self.bedrock_primary = boto3.client('bedrock-agent-runtime', region_name=primary_region)
        self.bedrock_secondary = boto3.client('bedrock-agent-runtime', region_name=secondary_region)

    def evaluate_primary_agent_health(self, agent_id: str, agent_alias_id: str) -> bool:
        # Synthetic health probe against primary region Bedrock agent
        try:
            start_time = time.time()
            response = self.bedrock_primary.invoke_agent(
                agentId=agent_id,
                agentAliasId=agent_alias_id,
                sessionId='health-check-session',
                inputText='ping health check'
            )
            elapsed = time.time() - start_time
            print(f"✓ [Primary Agent Probe] Response received in {elapsed:.2f}s")
            return elapsed < 5.0
        except ClientError as e:
            print(f"🚨 [Primary Agent Error] {e.response['Error']['Message']}")
            return False

    def trigger_region_failover(self) -> None:
        # Flips Route 53 ARC routing control state to divert traffic to secondary region
        print(f"🔄 [Failover Triggered] Disabling primary region {self.primary_region} routing control...")
        try:
            self.arc_client.update_routing_control_states(
                UpdateRoutingControlStatesEntries=[
                    {
                        'RoutingControlArn': self.routing_control_arn,
                        'RoutingControlState': 'Off'
                    }
                ]
            )
            print(f"✅ [Failover Complete] Traffic diverted to secondary region {self.secondary_region}.")
        except Exception as e:
            print(f"❌ [Failover Failed] {str(e)}")

if __name__ == "__main__":
    controller = MultiRegionAgentFailoverController(
        primary_region='us-east-1',
        secondary_region='us-west-2',
        routing_control_arn='arn:aws:route53-recovery-control::123456789012:control/primary-gate'
    )

    is_healthy = controller.evaluate_primary_agent_health('AGENT123', 'ALIAS123')
    if not is_healthy:
        controller.trigger_region_failover()