diff --git a/collections/prefect-aws/workers/v0.4.7.json b/collections/prefect-aws/workers/v0.4.7.json new file mode 100644 index 00000000..cb5bc8c8 --- /dev/null +++ b/collections/prefect-aws/workers/v0.4.7.json @@ -0,0 +1,306 @@ +{ + "prefect-aws": { + "ecs": { + "default_base_job_configuration": { + "job_configuration": { + "command": "{{ command }}", + "env": "{{ env }}", + "labels": "{{ labels }}", + "name": "{{ name }}", + "aws_credentials": "{{ aws_credentials }}", + "task_definition": { + "containerDefinitions": [ + { + "image": "{{ image }}", + "name": "{{ container_name }}" + } + ], + "cpu": "{{ cpu }}", + "family": "{{ family }}", + "memory": "{{ memory }}", + "executionRoleArn": "{{ execution_role_arn }}" + }, + "task_run_request": { + "launchType": "{{ launch_type }}", + "cluster": "{{ cluster }}", + "overrides": { + "containerOverrides": [ + { + "name": "{{ container_name }}", + "command": "{{ command }}", + "environment": "{{ env }}", + "cpu": "{{ cpu }}", + "memory": "{{ memory }}" + } + ], + "cpu": "{{ cpu }}", + "memory": "{{ memory }}", + "taskRoleArn": "{{ task_role_arn }}" + }, + "tags": "{{ labels }}", + "taskDefinition": "{{ task_definition_arn }}" + }, + "configure_cloudwatch_logs": "{{ configure_cloudwatch_logs }}", + "cloudwatch_logs_options": "{{ cloudwatch_logs_options }}", + "network_configuration": "{{ network_configuration }}", + "stream_output": "{{ stream_output }}", + "task_start_timeout_seconds": "{{ task_start_timeout_seconds }}", + "task_watch_poll_interval": "{{ task_watch_poll_interval }}", + "auto_deregister_task_definition": "{{ auto_deregister_task_definition }}", + "vpc_id": "{{ vpc_id }}", + "container_name": "{{ container_name }}", + "cluster": "{{ cluster }}" + }, + "variables": { + "description": "Variables for templating an ECS job.", + "type": "object", + "properties": { + "name": { + "title": "Name", + "description": "Name given to infrastructure created by a worker.", + "type": "string" + }, + "env": { + "title": "Environment Variables", + "description": "Environment variables to provide to the task run. These variables are set on the Prefect container at task runtime. These will not be set on the task definition.", + "type": "object", + "additionalProperties": { + "type": "string" + } + }, + "labels": { + "title": "Labels", + "description": "Labels applied to infrastructure created by a worker.", + "type": "object", + "additionalProperties": { + "type": "string" + } + }, + "command": { + "title": "Command", + "description": "The command to use when starting a flow run. In most cases, this should be left blank and the command will be automatically generated by the worker.", + "type": "string" + }, + "task_definition_arn": { + "title": "Task Definition Arn", + "description": "An identifier for an existing task definition to use. If set, options that require changes to the task definition will be ignored. All contents of the task definition in the job configuration will be ignored.", + "type": "string" + }, + "aws_credentials": { + "title": "AWS Credentials", + "description": "The AWS credentials to use to connect to ECS. If not provided, credentials will be inferred from the local environment following AWS's boto client's rules.", + "allOf": [ + { + "$ref": "#/definitions/AwsCredentials" + } + ] + }, + "cluster": { + "title": "Cluster", + "description": "The ECS cluster to run the task in. An ARN or name may be provided. If not provided, the default cluster will be used.", + "type": "string" + }, + "family": { + "title": "Family", + "description": "A family for the task definition. If not provided, it will be inferred from the task definition. If the task definition does not have a family, the name will be generated. When flow and deployment metadata is available, the generated name will include their names. Values for this field will be slugified to match AWS character requirements.", + "type": "string" + }, + "launch_type": { + "title": "Launch Type", + "description": "The type of ECS task run infrastructure that should be used. Note that 'FARGATE_SPOT' is not a formal ECS launch type, but we will configure the proper capacity provider stategy if set here.", + "default": "FARGATE", + "enum": [ + "FARGATE", + "EC2", + "EXTERNAL", + "FARGATE_SPOT" + ], + "type": "string" + }, + "image": { + "title": "Image", + "description": "The image to use for the Prefect container in the task. If this value is not null, it will override the value in the task definition. This value defaults to a Prefect base image matching your local versions.", + "type": "string" + }, + "cpu": { + "title": "CPU", + "description": "The amount of CPU to provide to the ECS task. Valid amounts are specified in the AWS documentation. If not provided, a default value of 1024 will be used unless present on the task definition.", + "type": "integer" + }, + "memory": { + "title": "Memory", + "description": "The amount of memory to provide to the ECS task. Valid amounts are specified in the AWS documentation. If not provided, a default value of 2048 will be used unless present on the task definition.", + "type": "integer" + }, + "container_name": { + "title": "Container Name", + "description": "The name of the container flow run orchestration will occur in. If not specified, a default value of prefect will be used and if that is not found in the task definition the first container will be used.", + "type": "string" + }, + "task_role_arn": { + "title": "Task Role ARN", + "description": "A role to attach to the task run. This controls the permissions of the task while it is running.", + "type": "string" + }, + "execution_role_arn": { + "title": "Execution Role ARN", + "description": "An execution role to use for the task. This controls the permissions of the task when it is launching. If this value is not null, it will override the value in the task definition. An execution role must be provided to capture logs from the container.", + "type": "string" + }, + "vpc_id": { + "title": "VPC ID", + "description": "The AWS VPC to link the task run to. This is only applicable when using the 'awsvpc' network mode for your task. FARGATE tasks require this network mode, but for EC2 tasks the default network mode is 'bridge'. If using the 'awsvpc' network mode and this field is null, your default VPC will be used. If no default VPC can be found, the task run will fail.", + "type": "string" + }, + "configure_cloudwatch_logs": { + "title": "Configure Cloudwatch Logs", + "description": "If enabled, the Prefect container will be configured to send its output to the AWS CloudWatch logs service. This functionality requires an execution role with logs:CreateLogStream, logs:CreateLogGroup, and logs:PutLogEvents permissions. The default for this field is `False` unless `stream_output` is set.", + "type": "boolean" + }, + "cloudwatch_logs_options": { + "title": "Cloudwatch Logs Options", + "description": "When `configure_cloudwatch_logs` is enabled, this setting may be used to pass additional options to the CloudWatch logs configuration or override the default options. See the [AWS documentation](https://docs.aws.amazon.com/AmazonECS/latest/developerguide/using_awslogs.html#create_awslogs_logdriver_options) for available options. ", + "type": "object", + "additionalProperties": { + "type": "string" + } + }, + "network_configuration": { + "title": "Network Configuration", + "description": "When `network_configuration` is supplied it will override ECS Worker'sawsvpcConfiguration that defined in the ECS task executing your workload. See the [AWS documentation](https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-ecs-service-awsvpcconfiguration.html) for available options.", + "type": "object" + }, + "stream_output": { + "title": "Stream Output", + "description": "If enabled, logs will be streamed from the Prefect container to the local console. Unless you have configured AWS CloudWatch logs manually on your task definition, this requires the same prerequisites outlined in `configure_cloudwatch_logs`.", + "type": "boolean" + }, + "task_start_timeout_seconds": { + "title": "Task Start Timeout Seconds", + "description": "The amount of time to watch for the start of the ECS task before marking it as failed. The task must enter a RUNNING state to be considered started.", + "default": 300, + "type": "integer" + }, + "task_watch_poll_interval": { + "title": "Task Watch Poll Interval", + "description": "The amount of time to wait between AWS API calls while monitoring the state of an ECS task.", + "default": 5.0, + "type": "number" + }, + "auto_deregister_task_definition": { + "title": "Auto Deregister Task Definition", + "description": "If enabled, any task definitions that are created by this block will be deregistered. Existing task definitions linked by ARN will never be deregistered. Deregistering a task definition does not remove it from your AWS account, instead it will be marked as INACTIVE.", + "default": false, + "type": "boolean" + } + }, + "definitions": { + "AwsClientParameters": { + "title": "AwsClientParameters", + "description": "Model used to manage extra parameters that you can pass when you initialize\nthe Client. If you want to find more information, see\n[boto3 docs](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html)\nfor more info about the possible client configurations.\n\nAttributes:\n api_version: The API version to use. By default, botocore will\n use the latest API version when creating a client. You only need\n to specify this parameter if you want to use a previous API version\n of the client.\n use_ssl: Whether or not to use SSL. By default, SSL is used.\n Note that not all services support non-ssl connections.\n verify: Whether or not to verify SSL certificates. By default\n SSL certificates are verified. If False, SSL will still be used\n (unless use_ssl is False), but SSL certificates\n will not be verified. Passing a file path to this is deprecated.\n verify_cert_path: A filename of the CA cert bundle to\n use. You can specify this argument if you want to use a\n different CA cert bundle than the one used by botocore.\n endpoint_url: The complete URL to use for the constructed\n client. Normally, botocore will automatically construct the\n appropriate URL to use when communicating with a service. You\n can specify a complete URL (including the \"http/https\" scheme)\n to override this behavior. If this value is provided,\n then ``use_ssl`` is ignored.\n config: Advanced configuration for Botocore clients. See\n [botocore docs](https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html)\n for more details.", + "type": "object", + "properties": { + "api_version": { + "title": "API Version", + "description": "The API version to use.", + "type": "string" + }, + "use_ssl": { + "title": "Use SSL", + "description": "Whether or not to use SSL.", + "default": true, + "type": "boolean" + }, + "verify": { + "title": "Verify", + "description": "Whether or not to verify SSL certificates.", + "default": true, + "anyOf": [ + { + "type": "boolean" + }, + { + "type": "string", + "format": "file-path" + } + ] + }, + "verify_cert_path": { + "title": "Certificate Authority Bundle File Path", + "description": "Path to the CA cert bundle to use.", + "format": "file-path", + "type": "string" + }, + "endpoint_url": { + "title": "Endpoint URL", + "description": "The complete URL to use for the constructed client.", + "type": "string" + }, + "config": { + "title": "Botocore Config", + "description": "Advanced configuration for Botocore clients.", + "type": "object" + } + } + }, + "AwsCredentials": { + "title": "AwsCredentials", + "description": "Block used to manage authentication with AWS. AWS authentication is\nhandled via the `boto3` module. Refer to the\n[boto3 docs](https://boto3.amazonaws.com/v1/documentation/api/latest/guide/credentials.html)\nfor more info about the possible credential configurations.", + "type": "object", + "properties": { + "aws_access_key_id": { + "title": "AWS Access Key ID", + "description": "A specific AWS access key ID.", + "type": "string" + }, + "aws_secret_access_key": { + "title": "AWS Access Key Secret", + "description": "A specific AWS secret access key.", + "type": "string", + "writeOnly": true, + "format": "password" + }, + "aws_session_token": { + "title": "AWS Session Token", + "description": "The session key for your AWS account. This is only needed when you are using temporary credentials.", + "type": "string" + }, + "profile_name": { + "title": "Profile Name", + "description": "The profile to use when creating your session.", + "type": "string" + }, + "region_name": { + "title": "Region Name", + "description": "The AWS Region where you want to create new connections.", + "type": "string" + }, + "aws_client_parameters": { + "title": "AWS Client Parameters", + "description": "Extra parameters to initialize the Client.", + "allOf": [ + { + "$ref": "#/definitions/AwsClientParameters" + } + ] + } + }, + "block_type_slug": "aws-credentials", + "secret_fields": [ + "aws_secret_access_key" + ], + "block_schema_references": {} + } + } + } + }, + "description": "Execute flow runs within containers on AWS ECS. Works with EC2 and Fargate clusters. Requires an AWS account.", + "display_name": "AWS Elastic Container Service", + "documentation_url": "https://prefecthq.github.io/prefect-aws/ecs_worker/", + "install_command": "pip install prefect-aws", + "is_beta": false, + "logo_url": "https://cdn.sanity.io/images/3ugk85nk/production/d74b16fe84ce626345adf235a47008fea2869a60-225x225.png", + "type": "ecs" + } + } +} \ No newline at end of file