from prefect_aws import AwsCredentials
from prefect_aws.ecs import ECSTask

aws_credentials_block = AwsCredentials(
    aws_access_key_id="****",
    aws_secret_access_key="****",
)
aws_credentials_block.save("prod", overwrite=True)

ecs = ECSTask(
    aws_credentials=aws_credentials_block,
    image="<http://631792532656.dkr.ecr.us-east-1.amazonaws.com/prefect2-image:latest|631792532656.dkr.ecr.us-east-1.amazonaws.com/prefect2-image:latest>",  # image pushed to ECR
    cpu="256",
    memory="512",
    stream_output=True,
    configure_cloudwatch_logs=True,
    cluster="prefect2EcsCluster",
    execution_role_arn="arn:aws:iam::631792532656:role/prefectECSAgentTaskExecutionRole",
    task_role_arn="arn:aws:iam::631792532656:role/prefectTaskRole",
)
ecs.save("prod", overwrite=True)