import boto3 def check_endpoint(endpoint_arn): comprehend = boto3.client("comprehend") describe_response = comprehend.describe_endpoint( EndpointArn=endpoint_arn ) endpoint_is_complete = False status = describe_response["EndpointProperties"]["Status"] if status != "IN_SERVICE": if status == "FAILED": message = describe_response["EndpointProperties"]["Message"] raise ValueError(f"The endpoint is in error:", message) else: endpoint_is_complete = True return endpoint_is_complete def lambda_handler(event, context): endpoint_arn = event["endpoint_arn"] endpoint_is_complete = check_endpoint(endpoint_arn) return { 'endpoint_is_complete': endpoint_is_complete, 'endpoint_arn': endpoint_arn }