fix: Increase retries for Gateway client

Author: clareliguori

examples/servers/book-search/gateway_setup/setup_gateway.py

@@ -3,6 +3,7 @@
 import json
 import os
 import boto3
+from botocore.config import Config
 from bedrock_agentcore_starter_toolkit.operations.gateway.client import GatewayClient
 
 
@@ -53,6 +54,16 @@ def main():
 
     # Create Gateway client for us-west-2
     gateway_client = GatewayClient(region_name="us-west-2")
+    retry_config = Config(
+        retries={
+            "max_attempts": 10,
+            "mode": "standard",
+        }
+    )
+    boto_client = boto3.client(
+        "bedrock-agentcore-control", region_name="us-west-2", config=retry_config
+    )
+    gateway_client.client = boto_client
 
     # Create Gateway
     role_arn = f"arn:aws:iam::{account_id}:role/mcp-lambda-example-agentcore-gateways"

examples/servers/book-search/gateway_setup/teardown_gateway.py

@@ -3,6 +3,7 @@
 import json
 import os
 import boto3
+from botocore.config import Config
 
 
 def main():
@@ -23,8 +24,14 @@ def main():
         return
 
     # Delete all gateway targets
+    retry_config = Config(
+        retries={
+            "max_attempts": 10,
+            "mode": "standard",
+        }
+    )
     agentcore_client = boto3.client(
-        "bedrock-agentcore-control", region_name="us-west-2"
+        "bedrock-agentcore-control", region_name="us-west-2", config=retry_config
     )
 
     targets = agentcore_client.list_gateway_targets(gatewayIdentifier=gateway_id)

examples/servers/dictionary/gateway_setup/setup-gateway.ts

@@ -72,7 +72,10 @@ async function main() {
   };
 
   // Create Gateway
-  const agentCoreClient = new BedrockAgentCoreControlClient({ region });
+  const agentCoreClient = new BedrockAgentCoreControlClient({
+    region,
+    maxAttempts: 10,
+  });
   const roleArn = `arn:aws:iam::${accountId}:role/mcp-lambda-example-agentcore-gateways`;
 
   const gateway = await agentCoreClient.send(

examples/servers/dictionary/gateway_setup/teardown-gateway.ts

@@ -38,7 +38,10 @@ async function main() {
     throw error;
   }
 
-  const agentCoreClient = new BedrockAgentCoreControlClient({ region });
+  const agentCoreClient = new BedrockAgentCoreControlClient({
+    region,
+    maxAttempts: 10,
+  });
 
   // Delete all gateway targets
   const targets = await agentCoreClient.send(

examples/servers/inspiration/setup_gateway.py

@@ -3,6 +3,7 @@
 import json
 import os
 import boto3
+from botocore.config import Config
 from bedrock_agentcore_starter_toolkit.operations.gateway.client import GatewayClient
 
 
@@ -39,6 +40,16 @@ def main():
 
     # Create Gateway client for us-west-2
     gateway_client = GatewayClient(region_name="us-west-2")
+    retry_config = Config(
+        retries={
+            "max_attempts": 10,
+            "mode": "standard",
+        }
+    )
+    boto_client = boto3.client(
+        "bedrock-agentcore-control", region_name="us-west-2", config=retry_config
+    )
+    gateway_client.client = boto_client
 
     # Create Gateway
     role_arn = f"arn:aws:iam::{account_id}:role/mcp-lambda-example-agentcore-gateways"

examples/servers/inspiration/teardown_gateway.py

@@ -3,6 +3,7 @@
 import json
 import os
 import boto3
+from botocore.config import Config
 
 
 def main():
@@ -23,8 +24,14 @@ def main():
         return
 
     # Delete all gateway targets
+    retry_config = Config(
+        retries={
+            "max_attempts": 10,
+            "mode": "standard",
+        }
+    )
     agentcore_client = boto3.client(
-        "bedrock-agentcore-control", region_name="us-west-2"
+        "bedrock-agentcore-control", region_name="us-west-2", config=retry_config
     )
 
     targets = agentcore_client.list_gateway_targets(gatewayIdentifier=gateway_id)