import os
from prefect.agent.fargate import FargateAgent

CLUSTER = "XXX"
SUBNETS = ["subnet-XXX", "subnet-XXX"]
EXECUTION_ROLE_ARN = "XXX"

agent = FargateAgent(
    labels=["XXX],
    launch_type="FARGATE",
    cluster=CLUSTER,
    executionRoleArn=EXECUTION_ROLE_ARN,
    networkConfiguration={
        "awsvpcConfiguration": {
            "assignPublicIp": "ENABLED",
            "subnets": SUBNETS,
        }
    },
    cpu="256",
    memory="512",
    containerDefinitions=[
        {
            "environment": [{"name": "TEST_ENV", "value": "Success!"}],
            "logConfiguration": {
                "logDriver": "awslogs",
                "options": {
                    "awslogs-group": "/my/log/group",
                    "awslogs-region": "us-east-1",
                    "awslogs-stream-prefix": "prefect-flow-runs",
                    "awslogs-create-group": "true",
                },
            },
        }
    ],
)

agent.start()

from datetime import timedelta
import prefect
from prefect import Client, task, Flow
from prefect.schedules import IntervalSchedule


@task
def hello_task():
    logger = prefect.context.get("logger")
    <http://logger.info|logger.info>("Hello, Cloud!")


schedule = IntervalSchedule(interval=timedelta(minutes=1))

with Flow("A New Flow", schedule,) as flow:
    hello_task()

flow.register(project_name="Test Project")

executionRoleArn