diff --git a/packages/core/resources/sagemaker_connect b/packages/core/resources/sagemaker_connect index 19d0e1984cc..ede46c1c4b3 100755 --- a/packages/core/resources/sagemaker_connect +++ b/packages/core/resources/sagemaker_connect @@ -46,7 +46,7 @@ _get_ssm_session_info_async() { # Generate unique temporary file name to avoid conflicts local temp_file="/tmp/ssm_session_response_$$_$(date +%s%N).json" - local max_retries=60 + local max_retries=8 local retry_interval=5 local attempt=1 diff --git a/packages/core/resources/sagemaker_connect.ps1 b/packages/core/resources/sagemaker_connect.ps1 index 034f9f09754..0e593d65b85 100644 --- a/packages/core/resources/sagemaker_connect.ps1 +++ b/packages/core/resources/sagemaker_connect.ps1 @@ -54,7 +54,7 @@ function Get-SSMSessionInfoAsync { $url = "http://localhost:$LocalEndpointPort/get_session_async?connection_identifier=$AwsResourceArn&credentials_type=$CredentialsType&request_id=$requestId" Write-Host "Calling Get-SSMSessionInfoAsync with URL: $url" - $maxRetries = 60 + $maxRetries = 8 $retryInterval = 5 for ($attempt = 1; $attempt -le $maxRetries; $attempt++) {