diff --git a/src/lambdas/agent-runner/handler.py b/src/lambdas/agent-runner/handler.py index 20c3998..ff5226e 100644 --- a/src/lambdas/agent-runner/handler.py +++ b/src/lambdas/agent-runner/handler.py @@ -21,7 +21,12 @@ def get_ddb(): def get_agentcore(): global _agentcore if _agentcore is None: - _agentcore = boto3.client('bedrock-agentcore', region_name='us-east-1') + from botocore.config import Config + _agentcore = boto3.client( + 'bedrock-agentcore', + region_name='us-east-1', + config=Config(read_timeout=600, connect_timeout=10) + ) return _agentcore