@flyteorg/flyteidl2 2.0.15 → 2.0.46

package/flyteidl2/notification/definition_pb.ts

@@ -5,7 +5,7 @@
 import type { GenEnum, GenFile, GenMessage } from "@bufbuild/protobuf/codegenv1";
 import { enumDesc, fileDesc, messageDesc } from "@bufbuild/protobuf/codegenv1";
 import { file_buf_validate_validate } from "../../buf/validate/validate_pb.ts";
-import type { RunIdentifier } from "../common/identifier_pb.ts";
+import type { ActionIdentifier, RunIdentifier } from "../common/identifier_pb.ts";
 import { file_flyteidl2_common_identifier } from "../common/identifier_pb.ts";
 import type { ActionPhase } from "../common/phase_pb.ts";
 import { file_flyteidl2_common_phase } from "../common/phase_pb.ts";
@@ -15,7 +15,7 @@ import type { Message } from "@bufbuild/protobuf";
  * Describes the file flyteidl2/notification/definition.proto.
  */
 export const file_flyteidl2_notification_definition: GenFile = /*@__PURE__*/
-  fileDesc("CidmbHl0ZWlkbDIvbm90aWZpY2F0aW9uL2RlZmluaXRpb24ucHJvdG8SFmZseXRlaWRsMi5ub3RpZmljYXRpb24ijgIKFkRlbGl2ZXJ5Q29uZmlnVGVtcGxhdGUSQAoHd2ViaG9vaxgBIAEoCzIvLmZseXRlaWRsMi5ub3RpZmljYXRpb24uV2ViaG9va0RlbGl2ZXJ5VGVtcGxhdGUSPAoFZW1haWwYAiABKAsyLS5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLkVtYWlsRGVsaXZlcnlUZW1wbGF0ZTp0ukhxGm8KFWF0X2xlYXN0X29uZV9yZXF1aXJlZBIwYXQgbGVhc3Qgb25lIG9mIHRoZSBkZWxpdmVyeSBvcHRpb25zIG11c3QgYmUgc2V0GiRoYXModGhpcy53ZWJob29rKSB8fCBoYXModGhpcy5lbWFpbCkimQEKJFJ1bkNvbXBsZXRlZE5vdGlmaWNhdGlvblRlbXBsYXRlRGF0YRI0CgNydW4YASABKAsyHy5mbHl0ZWlkbDIuY29tbW9uLlJ1bklkZW50aWZpZXJCBrpIA8gBARIsCgVwaGFzZRgCIAEoDjIdLmZseXRlaWRsMi5jb21tb24uQWN0aW9uUGhhc2USDQoFZXJyb3IYAyABKAkioAIKF1dlYmhvb2tEZWxpdmVyeVRlbXBsYXRlEhcKA3VybBgBIAEoCUIKukgHcgUQARjQDxI8CgZtZXRob2QYAiABKA4yIi5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLkh0dHBNZXRob2RCCLpIBYIBAiAAEmcKB2hlYWRlcnMYAyADKAsyPC5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLldlYmhvb2tEZWxpdmVyeVRlbXBsYXRlLkhlYWRlcnNFbnRyeUIYukgVmgESEBQiBnIEEAEYFCoGcgQQARgyEhUKDWJvZHlfdGVtcGxhdGUYBCABKAkaLgoMSGVhZGVyc0VudHJ5EgsKA2tleRgBIAEoCRINCgV2YWx1ZRgCIAEoCToCOAEiLwoORW1haWxSZWNpcGllbnQSDAoEbmFtZRgBIAEoCRIPCgdhZGRyZXNzGAIgASgJImYKE0lubGluZUVtYWlsVGVtcGxhdGUSGAoHc3ViamVjdBgBIAEoCUIHukgEcgIQARIeCg1odG1sX3RlbXBsYXRlGAIgASgJQge6SARyAhABEhUKDXRleHRfdGVtcGxhdGUYAyABKAki3gEKFVByb3ZpZGVyRW1haWxUZW1wbGF0ZRIfCgt0ZW1wbGF0ZV9pZBgBIAEoCUIKukgHcgUQARjIARJvCg10ZW1wbGF0ZV9kYXRhGAIgAygLMj8uZmx5dGVpZGwyLm5vdGlmaWNhdGlvbi5Qcm92aWRlckVtYWlsVGVtcGxhdGUuVGVtcGxhdGVEYXRhRW50cnlCF7pIFJoBERBkIgZyBBABGGQqBXIDGKAfGjMKEVRlbXBsYXRlRGF0YUVudHJ5EgsKA2tleRgBIAEoCRINCgV2YWx1ZRgCIAEoCToCOAEi2wIKFUVtYWlsRGVsaXZlcnlUZW1wbGF0ZRI8CgJ0bxgBIAMoCzImLmZseXRlaWRsMi5ub3RpZmljYXRpb24uRW1haWxSZWNpcGllbnRCCLpIBZIBAggBEjIKAmNjGAIgAygLMiYuZmx5dGVpZGwyLm5vdGlmaWNhdGlvbi5FbWFpbFJlY2lwaWVudBIzCgNiY2MYAyADKAsyJi5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLkVtYWlsUmVjaXBpZW50Ej0KBmlubGluZRgEIAEoCzIrLmZseXRlaWRsMi5ub3RpZmljYXRpb24uSW5saW5lRW1haWxUZW1wbGF0ZUgAEkoKEXByb3ZpZGVyX3RlbXBsYXRlGAUgASgLMi0uZmx5dGVpZGwyLm5vdGlmaWNhdGlvbi5Qcm92aWRlckVtYWlsVGVtcGxhdGVIAEIQCgdjb250ZW50EgW6SAIIASpFCglFdmVudFR5cGUSGgoWRVZFTlRfVFlQRV9VTlNQRUNJRklFRBAAEhwKGEVWRU5UX1RZUEVfUlVOX0NPTVBMRVRFRBABKvcBCgpIdHRwTWV0aG9kEhsKF0hUVFBfTUVUSE9EX1VOU1BFQ0lGSUVEEAASEwoPSFRUUF9NRVRIT0RfR0VUEAESFAoQSFRUUF9NRVRIT0RfSEVBRBACEhQKEEhUVFBfTUVUSE9EX1BPU1QQAxITCg9IVFRQX01FVEhPRF9QVVQQBBIWChJIVFRQX01FVEhPRF9ERUxFVEUQBRIXChNIVFRQX01FVEhPRF9DT05ORUNUEAYSFwoTSFRUUF9NRVRIT0RfT1BUSU9OUxAHEhUKEUhUVFBfTUVUSE9EX1RSQUNFEAgSFQoRSFRUUF9NRVRIT0RfUEFUQ0gQCULkAQoaY29tLmZseXRlaWRsMi5ub3RpZmljYXRpb25CD0RlZmluaXRpb25Qcm90b0gCUAFaOmdpdGh1Yi5jb20vZmx5dGVvcmcvZmx5dGUvdjIvZ2VuL2dvL2ZseXRlaWRsMi9ub3RpZmljYXRpb26iAgNGTliqAhZGbHl0ZWlkbDIuTm90aWZpY2F0aW9uygIWRmx5dGVpZGwyXE5vdGlmaWNhdGlvbuICIkZseXRlaWRsMlxOb3RpZmljYXRpb25cR1BCTWV0YWRhdGHqAhdGbHl0ZWlkbDI6Ok5vdGlmaWNhdGlvbmIGcHJvdG8z", [file_buf_validate_validate, file_flyteidl2_common_identifier, file_flyteidl2_common_phase]);
+  fileDesc("CidmbHl0ZWlkbDIvbm90aWZpY2F0aW9uL2RlZmluaXRpb24ucHJvdG8SFmZseXRlaWRsMi5ub3RpZmljYXRpb24ijgIKFkRlbGl2ZXJ5Q29uZmlnVGVtcGxhdGUSQAoHd2ViaG9vaxgBIAEoCzIvLmZseXRlaWRsMi5ub3RpZmljYXRpb24uV2ViaG9va0RlbGl2ZXJ5VGVtcGxhdGUSPAoFZW1haWwYAiABKAsyLS5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLkVtYWlsRGVsaXZlcnlUZW1wbGF0ZTp0ukhxGm8KFWF0X2xlYXN0X29uZV9yZXF1aXJlZBIwYXQgbGVhc3Qgb25lIG9mIHRoZSBkZWxpdmVyeSBvcHRpb25zIG11c3QgYmUgc2V0GiRoYXModGhpcy53ZWJob29rKSB8fCBoYXModGhpcy5lbWFpbCkimQEKJFJ1bkNvbXBsZXRlZE5vdGlmaWNhdGlvblRlbXBsYXRlRGF0YRI0CgNydW4YASABKAsyHy5mbHl0ZWlkbDIuY29tbW9uLlJ1bklkZW50aWZpZXJCBrpIA8gBARIsCgVwaGFzZRgCIAEoDjIdLmZseXRlaWRsMi5jb21tb24uQWN0aW9uUGhhc2USDQoFZXJyb3IYAyABKAkinAEKJEFjdGlvblBhdXNlZE5vdGlmaWNhdGlvblRlbXBsYXRlRGF0YRI6CgZhY3Rpb24YASABKAsyIi5mbHl0ZWlkbDIuY29tbW9uLkFjdGlvbklkZW50aWZpZXJCBrpIA8gBARIOCgZwcm9tcHQYAiABKAkSEwoLZGVzY3JpcHRpb24YAyABKAkSEwoLcHJvbXB0X3R5cGUYBCABKAkioAIKF1dlYmhvb2tEZWxpdmVyeVRlbXBsYXRlEhcKA3VybBgBIAEoCUIKukgHcgUQARjQDxI8CgZtZXRob2QYAiABKA4yIi5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLkh0dHBNZXRob2RCCLpIBYIBAiAAEmcKB2hlYWRlcnMYAyADKAsyPC5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLldlYmhvb2tEZWxpdmVyeVRlbXBsYXRlLkhlYWRlcnNFbnRyeUIYukgVmgESEBQiBnIEEAEYFCoGcgQQARgyEhUKDWJvZHlfdGVtcGxhdGUYBCABKAkaLgoMSGVhZGVyc0VudHJ5EgsKA2tleRgBIAEoCRINCgV2YWx1ZRgCIAEoCToCOAEiLwoORW1haWxSZWNpcGllbnQSDAoEbmFtZRgBIAEoCRIPCgdhZGRyZXNzGAIgASgJImYKE0lubGluZUVtYWlsVGVtcGxhdGUSGAoHc3ViamVjdBgBIAEoCUIHukgEcgIQARIeCg1odG1sX3RlbXBsYXRlGAIgASgJQge6SARyAhABEhUKDXRleHRfdGVtcGxhdGUYAyABKAki3gEKFVByb3ZpZGVyRW1haWxUZW1wbGF0ZRIfCgt0ZW1wbGF0ZV9pZBgBIAEoCUIKukgHcgUQARjIARJvCg10ZW1wbGF0ZV9kYXRhGAIgAygLMj8uZmx5dGVpZGwyLm5vdGlmaWNhdGlvbi5Qcm92aWRlckVtYWlsVGVtcGxhdGUuVGVtcGxhdGVEYXRhRW50cnlCF7pIFJoBERBkIgZyBBABGGQqBXIDGKAfGjMKEVRlbXBsYXRlRGF0YUVudHJ5EgsKA2tleRgBIAEoCRINCgV2YWx1ZRgCIAEoCToCOAEi2wIKFUVtYWlsRGVsaXZlcnlUZW1wbGF0ZRI8CgJ0bxgBIAMoCzImLmZseXRlaWRsMi5ub3RpZmljYXRpb24uRW1haWxSZWNpcGllbnRCCLpIBZIBAggBEjIKAmNjGAIgAygLMiYuZmx5dGVpZGwyLm5vdGlmaWNhdGlvbi5FbWFpbFJlY2lwaWVudBIzCgNiY2MYAyADKAsyJi5mbHl0ZWlkbDIubm90aWZpY2F0aW9uLkVtYWlsUmVjaXBpZW50Ej0KBmlubGluZRgEIAEoCzIrLmZseXRlaWRsMi5ub3RpZmljYXRpb24uSW5saW5lRW1haWxUZW1wbGF0ZUgAEkoKEXByb3ZpZGVyX3RlbXBsYXRlGAUgASgLMi0uZmx5dGVpZGwyLm5vdGlmaWNhdGlvbi5Qcm92aWRlckVtYWlsVGVtcGxhdGVIAEIQCgdjb250ZW50EgW6SAIIASpjCglFdmVudFR5cGUSGgoWRVZFTlRfVFlQRV9VTlNQRUNJRklFRBAAEhwKGEVWRU5UX1RZUEVfUlVOX0NPTVBMRVRFRBABEhwKGEVWRU5UX1RZUEVfQUNUSU9OX1BBVVNFRBACKvcBCgpIdHRwTWV0aG9kEhsKF0hUVFBfTUVUSE9EX1VOU1BFQ0lGSUVEEAASEwoPSFRUUF9NRVRIT0RfR0VUEAESFAoQSFRUUF9NRVRIT0RfSEVBRBACEhQKEEhUVFBfTUVUSE9EX1BPU1QQAxITCg9IVFRQX01FVEhPRF9QVVQQBBIWChJIVFRQX01FVEhPRF9ERUxFVEUQBRIXChNIVFRQX01FVEhPRF9DT05ORUNUEAYSFwoTSFRUUF9NRVRIT0RfT1BUSU9OUxAHEhUKEUhUVFBfTUVUSE9EX1RSQUNFEAgSFQoRSFRUUF9NRVRIT0RfUEFUQ0gQCULkAQoaY29tLmZseXRlaWRsMi5ub3RpZmljYXRpb25CD0RlZmluaXRpb25Qcm90b0gCUAFaOmdpdGh1Yi5jb20vZmx5dGVvcmcvZmx5dGUvdjIvZ2VuL2dvL2ZseXRlaWRsMi9ub3RpZmljYXRpb26iAgNGTliqAhZGbHl0ZWlkbDIuTm90aWZpY2F0aW9uygIWRmx5dGVpZGwyXE5vdGlmaWNhdGlvbuICIkZseXRlaWRsMlxOb3RpZmljYXRpb25cR1BCTWV0YWRhdGHqAhdGbHl0ZWlkbDI6Ok5vdGlmaWNhdGlvbmIGcHJvdG8z", [file_buf_validate_validate, file_flyteidl2_common_identifier, file_flyteidl2_common_phase]);
 
 /**
  * @generated from message flyteidl2.notification.DeliveryConfigTemplate
@@ -66,6 +66,52 @@ export type RunCompletedNotificationTemplateData = Message<"flyteidl2.notificati
 export const RunCompletedNotificationTemplateDataSchema: GenMessage<RunCompletedNotificationTemplateData> = /*@__PURE__*/
   messageDesc(file_flyteidl2_notification_definition, 1);
 
+/**
+ * ActionPausedNotificationTemplateData is the template data available to
+ * delivery templates on rules subscribed to ACTION_PHASE_PAUSED.
+ *
+ * @generated from message flyteidl2.notification.ActionPausedNotificationTemplateData
+ */
+export type ActionPausedNotificationTemplateData = Message<"flyteidl2.notification.ActionPausedNotificationTemplateData"> & {
+  /**
+   * Identifier of the condition action that paused awaiting a signal.
+   *
+   * @generated from field: flyteidl2.common.ActionIdentifier action = 1;
+   */
+  action?: ActionIdentifier;
+
+  /**
+   * Prompt shown to the user when the condition is awaited.
+   *
+   * @generated from field: string prompt = 2;
+   */
+  prompt: string;
+
+  /**
+   * Description of the condition.
+   *
+   * @generated from field: string description = 3;
+   */
+  description: string;
+
+  /**
+   * How to render the prompt: "TEXT", "MARKDOWN", or "" when unspecified.
+   * Mirrors flyteidl2.workflow.ConditionPromptType with the enum prefix
+   * trimmed (the workflow package cannot be imported here without an import
+   * cycle through flyteidl2/task/run.proto).
+   *
+   * @generated from field: string prompt_type = 4;
+   */
+  promptType: string;
+};
+
+/**
+ * Describes the message flyteidl2.notification.ActionPausedNotificationTemplateData.
+ * Use `create(ActionPausedNotificationTemplateDataSchema)` to create a new message.
+ */
+export const ActionPausedNotificationTemplateDataSchema: GenMessage<ActionPausedNotificationTemplateData> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_notification_definition, 2);
+
 /**
  * @generated from message flyteidl2.notification.WebhookDeliveryTemplate
  */
@@ -98,7 +144,7 @@ export type WebhookDeliveryTemplate = Message<"flyteidl2.notification.WebhookDel
  * Use `create(WebhookDeliveryTemplateSchema)` to create a new message.
  */
 export const WebhookDeliveryTemplateSchema: GenMessage<WebhookDeliveryTemplate> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_notification_definition, 2);
+  messageDesc(file_flyteidl2_notification_definition, 3);
 
 /**
  * @generated from message flyteidl2.notification.EmailRecipient
@@ -120,7 +166,7 @@ export type EmailRecipient = Message<"flyteidl2.notification.EmailRecipient"> &
  * Use `create(EmailRecipientSchema)` to create a new message.
  */
 export const EmailRecipientSchema: GenMessage<EmailRecipient> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_notification_definition, 3);
+  messageDesc(file_flyteidl2_notification_definition, 4);
 
 /**
  * InlineEmailTemplate contains subject and body content rendered by the
@@ -156,7 +202,7 @@ export type InlineEmailTemplate = Message<"flyteidl2.notification.InlineEmailTem
  * Use `create(InlineEmailTemplateSchema)` to create a new message.
  */
 export const InlineEmailTemplateSchema: GenMessage<InlineEmailTemplate> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_notification_definition, 4);
+  messageDesc(file_flyteidl2_notification_definition, 5);
 
 /**
  * ProviderEmailTemplate references an externally managed email template owned
@@ -187,7 +233,7 @@ export type ProviderEmailTemplate = Message<"flyteidl2.notification.ProviderEmai
  * Use `create(ProviderEmailTemplateSchema)` to create a new message.
  */
 export const ProviderEmailTemplateSchema: GenMessage<ProviderEmailTemplate> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_notification_definition, 5);
+  messageDesc(file_flyteidl2_notification_definition, 6);
 
 /**
  * @generated from message flyteidl2.notification.EmailDeliveryTemplate
@@ -233,7 +279,7 @@ export type EmailDeliveryTemplate = Message<"flyteidl2.notification.EmailDeliver
  * Use `create(EmailDeliveryTemplateSchema)` to create a new message.
  */
 export const EmailDeliveryTemplateSchema: GenMessage<EmailDeliveryTemplate> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_notification_definition, 6);
+  messageDesc(file_flyteidl2_notification_definition, 7);
 
 /**
  * @generated from enum flyteidl2.notification.EventType
@@ -248,6 +294,11 @@ export enum EventType {
    * @generated from enum value: EVENT_TYPE_RUN_COMPLETED = 1;
    */
   RUN_COMPLETED = 1,
+
+  /**
+   * @generated from enum value: EVENT_TYPE_ACTION_PAUSED = 2;
+   */
+  ACTION_PAUSED = 2,
 }
 
 /**

package/flyteidl2/plugins/clustered_pb.ts

@@ -0,0 +1,232 @@
+// @generated by protoc-gen-es v2.2.5 with parameter "target=ts,import_extension=.ts"
+// @generated from file flyteidl2/plugins/clustered.proto (package flyteidl2.plugins, syntax proto3)
+/* eslint-disable */
+
+import type { GenEnum, GenFile, GenMessage } from "@bufbuild/protobuf/codegenv1";
+import { enumDesc, fileDesc, messageDesc } from "@bufbuild/protobuf/codegenv1";
+import { file_buf_validate_validate } from "../../buf/validate/validate_pb.ts";
+import { file_google_protobuf_wrappers } from "@bufbuild/protobuf/wkt";
+import type { Message } from "@bufbuild/protobuf";
+
+/**
+ * Describes the file flyteidl2/plugins/clustered.proto.
+ */
+export const file_flyteidl2_plugins_clustered: GenFile = /*@__PURE__*/
+  fileDesc("CiFmbHl0ZWlkbDIvcGx1Z2lucy9jbHVzdGVyZWQucHJvdG8SEWZseXRlaWRsMi5wbHVnaW5zIrYCChFDbHVzdGVyZWRUYXNrU3BlYxIZCghyZXBsaWNhcxgBIAEoBUIHukgEGgIgABIfCg5ucHJvY19wZXJfbm9kZRgCIAEoBUIHukgEGgIgABIrCgdydW50aW1lGAMgASgLMhouZmx5dGVpZGwyLnBsdWdpbnMuUnVudGltZRI1CgxpbnRlcmNvbm5lY3QYBCABKA4yHy5mbHl0ZWlkbDIucGx1Z2lucy5JbnRlcmNvbm5lY3QSPwoOZmFpbHVyZV9wb2xpY3kYBSABKAsyJy5mbHl0ZWlkbDIucGx1Z2lucy5DbHVzdGVyRmFpbHVyZVBvbGljeRJAChp0dGxfc2Vjb25kc19hZnRlcl9maW5pc2hlZBgGIAEoCzIcLmdvb2dsZS5wcm90b2J1Zi5VSW50MzJWYWx1ZSJGCgdSdW50aW1lEjMKCHRvcmNocnVuGAEgASgLMh8uZmx5dGVpZGwyLnBsdWdpbnMuVG9yY2hSdW50aW1lSABCBgoEa2luZCJaCgxUb3JjaFJ1bnRpbWUSNAoMcmR6dl9iYWNrZW5kGAEgASgOMh4uZmx5dGVpZGwyLnBsdWdpbnMuUmR6dkJhY2tlbmQSFAoMbWF4X3Jlc3RhcnRzGAIgASgFIlEKFENsdXN0ZXJGYWlsdXJlUG9saWN5EhQKDG1heF9yZXN0YXJ0cxgBIAEoBRIjChtyZXN0YXJ0X29uX2hvc3RfbWFpbnRlbmFuY2UYAiABKAgqIwoLUmR6dkJhY2tlbmQSCgoGU1RBVElDEAASCAoEQzEwRBABKjoKDEludGVyY29ubmVjdBIHCgNUQ1AQABIHCgNFRkEQARIOCgpJTkZJTklCQU5EEAISCAoEUk9DRRADQsUBChVjb20uZmx5dGVpZGwyLnBsdWdpbnNCDkNsdXN0ZXJlZFByb3RvSAJQAVo1Z2l0aHViLmNvbS9mbHl0ZW9yZy9mbHl0ZS92Mi9nZW4vZ28vZmx5dGVpZGwyL3BsdWdpbnOiAgNGUFiqAhFGbHl0ZWlkbDIuUGx1Z2luc8oCEUZseXRlaWRsMlxQbHVnaW5z4gIdRmx5dGVpZGwyXFBsdWdpbnNcR1BCTWV0YWRhdGHqAhJGbHl0ZWlkbDI6OlBsdWdpbnNiBnByb3RvMw", [file_buf_validate_validate, file_google_protobuf_wrappers]);
+
+/**
+ * Top-level spec for a clustered task execution. Packed into TaskTemplate.custom.
+ *
+ * @generated from message flyteidl2.plugins.ClusteredTaskSpec
+ */
+export type ClusteredTaskSpec = Message<"flyteidl2.plugins.ClusteredTaskSpec"> & {
+  /**
+   * Number of pods (== number of nodes). Required, must be >= 1.
+   *
+   * @generated from field: int32 replicas = 1;
+   */
+  replicas: number;
+
+  /**
+   * Number of processes per node passed to torchrun as --nproc-per-node (typically one per GPU). Required, must be >= 1.
+   *
+   * @generated from field: int32 nproc_per_node = 2;
+   */
+  nprocPerNode: number;
+
+  /**
+   * Selects the launcher and its per-launcher configuration.
+   *
+   * @generated from field: flyteidl2.plugins.Runtime runtime = 3;
+   */
+  runtime?: Runtime;
+
+  /**
+   * Network fabric for collective communications (NCCL all-reduce etc). Defaults to TCP.
+   *
+   * @generated from field: flyteidl2.plugins.Interconnect interconnect = 4;
+   */
+  interconnect: Interconnect;
+
+  /**
+   * Controls restart behaviour on rank failure and host maintenance.
+   *
+   * @generated from field: flyteidl2.plugins.ClusterFailurePolicy failure_policy = 5;
+   */
+  failurePolicy?: ClusterFailurePolicy;
+
+  /**
+   * If set, the JobSet is deleted this many seconds after it finishes (success or failure).
+   *
+   * @generated from field: google.protobuf.UInt32Value ttl_seconds_after_finished = 6;
+   */
+  ttlSecondsAfterFinished?: number;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.ClusteredTaskSpec.
+ * Use `create(ClusteredTaskSpecSchema)` to create a new message.
+ */
+export const ClusteredTaskSpecSchema: GenMessage<ClusteredTaskSpec> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_clustered, 0);
+
+/**
+ * Tagged union of supported launchers. Phase 1: TorchRun only.
+ * Future variants (DeepSpeed, Accelerate, JAX) add new oneof fields here;
+ * the JobSet shape and Go plugin are unchanged.
+ *
+ * @generated from message flyteidl2.plugins.Runtime
+ */
+export type Runtime = Message<"flyteidl2.plugins.Runtime"> & {
+  /**
+   * @generated from oneof flyteidl2.plugins.Runtime.kind
+   */
+  kind: {
+    /**
+     * @generated from field: flyteidl2.plugins.TorchRuntime torchrun = 1;
+     */
+    value: TorchRuntime;
+    case: "torchrun";
+  } | { case: undefined; value?: undefined };
+};
+
+/**
+ * Describes the message flyteidl2.plugins.Runtime.
+ * Use `create(RuntimeSchema)` to create a new message.
+ */
+export const RuntimeSchema: GenMessage<Runtime> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_clustered, 1);
+
+/**
+ * Configuration for the torchrun launcher.
+ *
+ * @generated from message flyteidl2.plugins.TorchRuntime
+ */
+export type TorchRuntime = Message<"flyteidl2.plugins.TorchRuntime"> & {
+  /**
+   * Rendezvous backend. Defaults to STATIC.
+   *
+   * @generated from field: flyteidl2.plugins.RdzvBackend rdzv_backend = 1;
+   */
+  rdzvBackend: RdzvBackend;
+
+  /**
+   * Number of in-pod torchrun restarts before the pod is considered failed.
+   * Defaults to 0 — whole-cluster restart via JobSet failurePolicy is the
+   * recovery mechanism, so in-pod retries are disabled.
+   *
+   * @generated from field: int32 max_restarts = 2;
+   */
+  maxRestarts: number;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.TorchRuntime.
+ * Use `create(TorchRuntimeSchema)` to create a new message.
+ */
+export const TorchRuntimeSchema: GenMessage<TorchRuntime> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_clustered, 2);
+
+/**
+ * Controls how the JobSet responds to rank failures and host-maintenance evictions.
+ *
+ * @generated from message flyteidl2.plugins.ClusterFailurePolicy
+ */
+export type ClusterFailurePolicy = Message<"flyteidl2.plugins.ClusterFailurePolicy"> & {
+  /**
+   * Maximum number of whole-cluster restarts before the JobSet is marked failed.
+   * Each restart increments JOBSET_RESTART_ATTEMPT and rotates the torchrun RDZV_ID.
+   *
+   * @generated from field: int32 max_restarts = 1;
+   */
+  maxRestarts: number;
+
+  /**
+   * If true, pods evicted due to host maintenance (DisruptionTarget condition)
+   * trigger a free restart that does not count against max_restarts.
+   *
+   * @generated from field: bool restart_on_host_maintenance = 2;
+   */
+  restartOnHostMaintenance: boolean;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.ClusterFailurePolicy.
+ * Use `create(ClusterFailurePolicySchema)` to create a new message.
+ */
+export const ClusterFailurePolicySchema: GenMessage<ClusterFailurePolicy> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_clustered, 3);
+
+/**
+ * Rendezvous backend for torchrun.
+ *
+ * @generated from enum flyteidl2.plugins.RdzvBackend
+ */
+export enum RdzvBackend {
+  /**
+   * Static rendezvous: full membership declared upfront via --nnodes and
+   * --master-addr. Simple and deterministic. Recovery is whole-job restart
+   * (JobSet failurePolicy), not in-job re-rendezvous.
+   *
+   * @generated from enum value: STATIC = 0;
+   */
+  STATIC = 0,
+
+  /**
+   * TCPStore-based dynamic rendezvous. Enables elastic membership
+   * (--nnodes=MIN:MAX) and in-job re-rendezvous on partial failure.
+   *
+   * @generated from enum value: C10D = 1;
+   */
+  C10D = 1,
+}
+
+/**
+ * Describes the enum flyteidl2.plugins.RdzvBackend.
+ */
+export const RdzvBackendSchema: GenEnum<RdzvBackend> = /*@__PURE__*/
+  enumDesc(file_flyteidl2_plugins_clustered, 0);
+
+/**
+ * Network fabric for gradient all-reduce and other collective ops.
+ * Non-TCP values cause the plugin to inject RDMA device resources,
+ * capabilities, and NCCL env vars into the pod spec.
+ *
+ * @generated from enum flyteidl2.plugins.Interconnect
+ */
+export enum Interconnect {
+  /**
+   * Plain TCP over the pod CNI network. Works on any cluster. Default.
+   *
+   * @generated from enum value: TCP = 0;
+   */
+  TCP = 0,
+
+  /**
+   * AWS Elastic Fabric Adapter (EC2 p4d / p5 / trn1). ~100-400 Gbps GPUDirect RDMA.
+   *
+   * @generated from enum value: EFA = 1;
+   */
+  EFA = 1,
+
+  /**
+   * Mellanox / NVIDIA InfiniBand HCA (CoreWeave, Lambda, on-prem H100 clusters).
+   *
+   * @generated from enum value: INFINIBAND = 2;
+   */
+  INFINIBAND = 2,
+
+  /**
+   * RDMA over Converged Ethernet (Azure ND-series, Oracle Cloud, on-prem Ethernet RDMA).
+   *
+   * @generated from enum value: ROCE = 3;
+   */
+  ROCE = 3,
+}
+
+/**
+ * Describes the enum flyteidl2.plugins.Interconnect.
+ */
+export const InterconnectSchema: GenEnum<Interconnect> = /*@__PURE__*/
+  enumDesc(file_flyteidl2_plugins_clustered, 1);
+

package/flyteidl2/plugins/ray_pb.ts

@@ -2,9 +2,11 @@
 // @generated from file flyteidl2/plugins/ray.proto (package flyteidl2.plugins, syntax proto3)
 /* eslint-disable */
 
-import type { GenFile, GenMessage } from "@bufbuild/protobuf/codegenv1";
-import { fileDesc, messageDesc } from "@bufbuild/protobuf/codegenv1";
-import type { K8sPod } from "../core/tasks_pb.ts";
+import type { GenEnum, GenFile, GenMessage } from "@bufbuild/protobuf/codegenv1";
+import { enumDesc, fileDesc, messageDesc } from "@bufbuild/protobuf/codegenv1";
+import type { KeyValuePair } from "../core/literals_pb.ts";
+import { file_flyteidl2_core_literals } from "../core/literals_pb.ts";
+import type { ExtendedResources, K8sPod, Resources } from "../core/tasks_pb.ts";
 import { file_flyteidl2_core_tasks } from "../core/tasks_pb.ts";
 import type { Message } from "@bufbuild/protobuf";
 
@@ -12,7 +14,7 @@ import type { Message } from "@bufbuild/protobuf";
  * Describes the file flyteidl2/plugins/ray.proto.
  */
 export const file_flyteidl2_plugins_ray: GenFile = /*@__PURE__*/
-  fileDesc("ChtmbHl0ZWlkbDIvcGx1Z2lucy9yYXkucHJvdG8SEWZseXRlaWRsMi5wbHVnaW5zIrgBCgZSYXlKb2ISMgoLcmF5X2NsdXN0ZXIYASABKAsyHS5mbHl0ZWlkbDIucGx1Z2lucy5SYXlDbHVzdGVyEhcKC3J1bnRpbWVfZW52GAIgASgJQgIYARIjChtzaHV0ZG93bl9hZnRlcl9qb2JfZmluaXNoZXMYAyABKAgSIgoadHRsX3NlY29uZHNfYWZ0ZXJfZmluaXNoZWQYBCABKAUSGAoQcnVudGltZV9lbnZfeWFtbBgFIAEoCSKiAQoKUmF5Q2x1c3RlchI5Cg9oZWFkX2dyb3VwX3NwZWMYASABKAsyIC5mbHl0ZWlkbDIucGx1Z2lucy5IZWFkR3JvdXBTcGVjEj0KEXdvcmtlcl9ncm91cF9zcGVjGAIgAygLMiIuZmx5dGVpZGwyLnBsdWdpbnMuV29ya2VyR3JvdXBTcGVjEhoKEmVuYWJsZV9hdXRvc2NhbGluZxgDIAEoCCK/AQoNSGVhZEdyb3VwU3BlYxJOChByYXlfc3RhcnRfcGFyYW1zGAEgAygLMjQuZmx5dGVpZGwyLnBsdWdpbnMuSGVhZEdyb3VwU3BlYy5SYXlTdGFydFBhcmFtc0VudHJ5EicKB2s4c19wb2QYAiABKAsyFi5mbHl0ZWlkbDIuY29yZS5LOHNQb2QaNQoTUmF5U3RhcnRQYXJhbXNFbnRyeRILCgNrZXkYASABKAkSDQoFdmFsdWUYAiABKAk6AjgBIpUCCg9Xb3JrZXJHcm91cFNwZWMSEgoKZ3JvdXBfbmFtZRgBIAEoCRIQCghyZXBsaWNhcxgCIAEoBRIUCgxtaW5fcmVwbGljYXMYAyABKAUSFAoMbWF4X3JlcGxpY2FzGAQgASgFElAKEHJheV9zdGFydF9wYXJhbXMYBSADKAsyNi5mbHl0ZWlkbDIucGx1Z2lucy5Xb3JrZXJHcm91cFNwZWMuUmF5U3RhcnRQYXJhbXNFbnRyeRInCgdrOHNfcG9kGAYgASgLMhYuZmx5dGVpZGwyLmNvcmUuSzhzUG9kGjUKE1JheVN0YXJ0UGFyYW1zRW50cnkSCwoDa2V5GAEgASgJEg0KBXZhbHVlGAIgASgJOgI4AUK/AQoVY29tLmZseXRlaWRsMi5wbHVnaW5zQghSYXlQcm90b0gCUAFaNWdpdGh1Yi5jb20vZmx5dGVvcmcvZmx5dGUvdjIvZ2VuL2dvL2ZseXRlaWRsMi9wbHVnaW5zogIDRlBYqgIRRmx5dGVpZGwyLlBsdWdpbnPKAhFGbHl0ZWlkbDJcUGx1Z2luc+ICHUZseXRlaWRsMlxQbHVnaW5zXEdQQk1ldGFkYXRh6gISRmx5dGVpZGwyOjpQbHVnaW5zYgZwcm90bzM", [file_flyteidl2_core_tasks]);
+  fileDesc("ChtmbHl0ZWlkbDIvcGx1Z2lucy9yYXkucHJvdG8SEWZseXRlaWRsMi5wbHVnaW5zIrgBCgZSYXlKb2ISMgoLcmF5X2NsdXN0ZXIYASABKAsyHS5mbHl0ZWlkbDIucGx1Z2lucy5SYXlDbHVzdGVyEhcKC3J1bnRpbWVfZW52GAIgASgJQgIYARIjChtzaHV0ZG93bl9hZnRlcl9qb2JfZmluaXNoZXMYAyABKAgSIgoadHRsX3NlY29uZHNfYWZ0ZXJfZmluaXNoZWQYBCABKAUSGAoQcnVudGltZV9lbnZfeWFtbBgFIAEoCSLzAgoRQXV0b3NjYWxlck9wdGlvbnMSSgoOdXBzY2FsaW5nX21vZGUYASABKA4yMi5mbHl0ZWlkbDIucGx1Z2lucy5BdXRvc2NhbGVyT3B0aW9ucy5VcHNjYWxpbmdNb2RlEhwKFGlkbGVfdGltZW91dF9zZWNvbmRzGAIgASgFEikKA2VudhgDIAMoCzIcLmZseXRlaWRsMi5jb3JlLktleVZhbHVlUGFpchINCgVpbWFnZRgEIAEoCRIsCglyZXNvdXJjZXMYBSABKAsyGS5mbHl0ZWlkbDIuY29yZS5SZXNvdXJjZXMiiwEKDVVwc2NhbGluZ01vZGUSHgoaVVBTQ0FMSU5HX01PREVfVU5TUEVDSUZJRUQQABIaChZVUFNDQUxJTkdfTU9ERV9ERUZBVUxUEAESHQoZVVBTQ0FMSU5HX01PREVfQUdHUkVTU0lWRRACEh8KG1VQU0NBTElOR19NT0RFX0NPTlNFUlZBVElWRRADIuQBCgpSYXlDbHVzdGVyEjkKD2hlYWRfZ3JvdXBfc3BlYxgBIAEoCzIgLmZseXRlaWRsMi5wbHVnaW5zLkhlYWRHcm91cFNwZWMSPQoRd29ya2VyX2dyb3VwX3NwZWMYAiADKAsyIi5mbHl0ZWlkbDIucGx1Z2lucy5Xb3JrZXJHcm91cFNwZWMSGgoSZW5hYmxlX2F1dG9zY2FsaW5nGAMgASgIEkAKEmF1dG9zY2FsZXJfb3B0aW9ucxgEIAEoCzIkLmZseXRlaWRsMi5wbHVnaW5zLkF1dG9zY2FsZXJPcHRpb25zIv4BCg1IZWFkR3JvdXBTcGVjEk4KEHJheV9zdGFydF9wYXJhbXMYASADKAsyNC5mbHl0ZWlkbDIucGx1Z2lucy5IZWFkR3JvdXBTcGVjLlJheVN0YXJ0UGFyYW1zRW50cnkSJwoHazhzX3BvZBgCIAEoCzIWLmZseXRlaWRsMi5jb3JlLks4c1BvZBI9ChJleHRlbmRlZF9yZXNvdXJjZXMYAyABKAsyIS5mbHl0ZWlkbDIuY29yZS5FeHRlbmRlZFJlc291cmNlcxo1ChNSYXlTdGFydFBhcmFtc0VudHJ5EgsKA2tleRgBIAEoCRINCgV2YWx1ZRgCIAEoCToCOAEi1AIKD1dvcmtlckdyb3VwU3BlYxISCgpncm91cF9uYW1lGAEgASgJEhAKCHJlcGxpY2FzGAIgASgFEhQKDG1pbl9yZXBsaWNhcxgDIAEoBRIUCgxtYXhfcmVwbGljYXMYBCABKAUSUAoQcmF5X3N0YXJ0X3BhcmFtcxgFIAMoCzI2LmZseXRlaWRsMi5wbHVnaW5zLldvcmtlckdyb3VwU3BlYy5SYXlTdGFydFBhcmFtc0VudHJ5EicKB2s4c19wb2QYBiABKAsyFi5mbHl0ZWlkbDIuY29yZS5LOHNQb2QSPQoSZXh0ZW5kZWRfcmVzb3VyY2VzGAcgASgLMiEuZmx5dGVpZGwyLmNvcmUuRXh0ZW5kZWRSZXNvdXJjZXMaNQoTUmF5U3RhcnRQYXJhbXNFbnRyeRILCgNrZXkYASABKAkSDQoFdmFsdWUYAiABKAk6AjgBQr8BChVjb20uZmx5dGVpZGwyLnBsdWdpbnNCCFJheVByb3RvSAJQAVo1Z2l0aHViLmNvbS9mbHl0ZW9yZy9mbHl0ZS92Mi9nZW4vZ28vZmx5dGVpZGwyL3BsdWdpbnOiAgNGUFiqAhFGbHl0ZWlkbDIuUGx1Z2luc8oCEUZseXRlaWRsMlxQbHVnaW5z4gIdRmx5dGVpZGwyXFBsdWdpbnNcR1BCTWV0YWRhdGHqAhJGbHl0ZWlkbDI6OlBsdWdpbnNiBnByb3RvMw", [file_flyteidl2_core_literals, file_flyteidl2_core_tasks]);
 
 /**
  * RayJobSpec defines the desired state of RayJob
@@ -66,6 +68,80 @@ export type RayJob = Message<"flyteidl2.plugins.RayJob"> & {
 export const RayJobSchema: GenMessage<RayJob> = /*@__PURE__*/
   messageDesc(file_flyteidl2_plugins_ray, 0);
 
+/**
+ * @generated from message flyteidl2.plugins.AutoscalerOptions
+ */
+export type AutoscalerOptions = Message<"flyteidl2.plugins.AutoscalerOptions"> & {
+  /**
+   * @generated from field: flyteidl2.plugins.AutoscalerOptions.UpscalingMode upscaling_mode = 1;
+   */
+  upscalingMode: AutoscalerOptions_UpscalingMode;
+
+  /**
+   * @generated from field: int32 idle_timeout_seconds = 2;
+   */
+  idleTimeoutSeconds: number;
+
+  /**
+   * autoscaler sidecar env vars
+   *
+   * @generated from field: repeated flyteidl2.core.KeyValuePair env = 3;
+   */
+  env: KeyValuePair[];
+
+  /**
+   * custom autoscaler image
+   *
+   * @generated from field: string image = 4;
+   */
+  image: string;
+
+  /**
+   * autoscaler container resources
+   *
+   * @generated from field: flyteidl2.core.Resources resources = 5;
+   */
+  resources?: Resources;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.AutoscalerOptions.
+ * Use `create(AutoscalerOptionsSchema)` to create a new message.
+ */
+export const AutoscalerOptionsSchema: GenMessage<AutoscalerOptions> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_ray, 1);
+
+/**
+ * @generated from enum flyteidl2.plugins.AutoscalerOptions.UpscalingMode
+ */
+export enum AutoscalerOptions_UpscalingMode {
+  /**
+   * @generated from enum value: UPSCALING_MODE_UNSPECIFIED = 0;
+   */
+  UNSPECIFIED = 0,
+
+  /**
+   * @generated from enum value: UPSCALING_MODE_DEFAULT = 1;
+   */
+  DEFAULT = 1,
+
+  /**
+   * @generated from enum value: UPSCALING_MODE_AGGRESSIVE = 2;
+   */
+  AGGRESSIVE = 2,
+
+  /**
+   * @generated from enum value: UPSCALING_MODE_CONSERVATIVE = 3;
+   */
+  CONSERVATIVE = 3,
+}
+
+/**
+ * Describes the enum flyteidl2.plugins.AutoscalerOptions.UpscalingMode.
+ */
+export const AutoscalerOptions_UpscalingModeSchema: GenEnum<AutoscalerOptions_UpscalingMode> = /*@__PURE__*/
+  enumDesc(file_flyteidl2_plugins_ray, 1, 0);
+
 /**
  * Define Ray cluster defines the desired state of RayCluster
  *
@@ -92,6 +168,11 @@ export type RayCluster = Message<"flyteidl2.plugins.RayCluster"> & {
    * @generated from field: bool enable_autoscaling = 3;
    */
   enableAutoscaling: boolean;
+
+  /**
+   * @generated from field: flyteidl2.plugins.AutoscalerOptions autoscaler_options = 4;
+   */
+  autoscalerOptions?: AutoscalerOptions;
 };
 
 /**
@@ -99,7 +180,7 @@ export type RayCluster = Message<"flyteidl2.plugins.RayCluster"> & {
  * Use `create(RayClusterSchema)` to create a new message.
  */
 export const RayClusterSchema: GenMessage<RayCluster> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_plugins_ray, 1);
+  messageDesc(file_flyteidl2_plugins_ray, 2);
 
 /**
  * HeadGroupSpec are the spec for the head pod
@@ -121,6 +202,14 @@ export type HeadGroupSpec = Message<"flyteidl2.plugins.HeadGroupSpec"> & {
    * @generated from field: flyteidl2.core.K8sPod k8s_pod = 2;
    */
   k8sPod?: K8sPod;
+
+  /**
+   * Encapsulates all non-standard resources, not captured by
+   * v1.ResourceRequirements, to allocate to a task.
+   *
+   * @generated from field: flyteidl2.core.ExtendedResources extended_resources = 3;
+   */
+  extendedResources?: ExtendedResources;
 };
 
 /**
@@ -128,7 +217,7 @@ export type HeadGroupSpec = Message<"flyteidl2.plugins.HeadGroupSpec"> & {
  * Use `create(HeadGroupSpecSchema)` to create a new message.
  */
 export const HeadGroupSpecSchema: GenMessage<HeadGroupSpec> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_plugins_ray, 2);
+  messageDesc(file_flyteidl2_plugins_ray, 3);
 
 /**
  * WorkerGroupSpec are the specs for the worker pods
@@ -178,6 +267,14 @@ export type WorkerGroupSpec = Message<"flyteidl2.plugins.WorkerGroupSpec"> & {
    * @generated from field: flyteidl2.core.K8sPod k8s_pod = 6;
    */
   k8sPod?: K8sPod;
+
+  /**
+   * Encapsulates all non-standard resources, not captured by
+   * v1.ResourceRequirements, to allocate to a task.
+   *
+   * @generated from field: flyteidl2.core.ExtendedResources extended_resources = 7;
+   */
+  extendedResources?: ExtendedResources;
 };
 
 /**
@@ -185,5 +282,5 @@ export type WorkerGroupSpec = Message<"flyteidl2.plugins.WorkerGroupSpec"> & {
  * Use `create(WorkerGroupSpecSchema)` to create a new message.
  */
 export const WorkerGroupSpecSchema: GenMessage<WorkerGroupSpec> = /*@__PURE__*/
-  messageDesc(file_flyteidl2_plugins_ray, 3);
+  messageDesc(file_flyteidl2_plugins_ray, 4);