@flyteorg/flyteidl2 2.0.0-alpha10

package/flyteidl2/logs/dataplane/payload_pb.ts

@@ -0,0 +1,396 @@
+// @generated by protoc-gen-es v2.2.5 with parameter "target=ts,import_extension=.ts"
+// @generated from file flyteidl2/logs/dataplane/payload.proto (package flyteidl2.logs.dataplane, 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 type { Timestamp } from "@bufbuild/protobuf/wkt";
+import { file_google_protobuf_timestamp } from "@bufbuild/protobuf/wkt";
+import type { Message } from "@bufbuild/protobuf";
+
+/**
+ * Describes the file flyteidl2/logs/dataplane/payload.proto.
+ */
+export const file_flyteidl2_logs_dataplane_payload: GenFile = /*@__PURE__*/
+  fileDesc("CiZmbHl0ZWlkbDIvbG9ncy9kYXRhcGxhbmUvcGF5bG9hZC5wcm90bxIYZmx5dGVpZGwyLmxvZ3MuZGF0YXBsYW5lIlMKC1BvZFJlc291cmNlEhoKCW5hbWVzcGFjZRgBIAEoCUIHukgEcgIQARIVCgRuYW1lGAIgASgJQge6SARyAhABEhEKCWNvbnRhaW5lchgDIAEoCSLUAwoOTG9nZ2luZ0NvbnRleHQSHQoMY2x1c3Rlcl9uYW1lGAMgASgJQge6SARyAhABEiUKFGt1YmVybmV0ZXNfbmFtZXNwYWNlGAQgASgJQge6SARyAhABEiQKE2t1YmVybmV0ZXNfcG9kX25hbWUYBSABKAlCB7pIBHICEAESKgoZa3ViZXJuZXRlc19jb250YWluZXJfbmFtZRgGIAEoCUIHukgEcgIQARJAChxleGVjdXRpb25fYXR0ZW1wdF9zdGFydF90aW1lGAcgASgLMhouZ29vZ2xlLnByb3RvYnVmLlRpbWVzdGFtcBI+ChpleGVjdXRpb25fYXR0ZW1wdF9lbmRfdGltZRgIIAEoCzIaLmdvb2dsZS5wcm90b2J1Zi5UaW1lc3RhbXASYAoVa3ViZXJuZXRlc19wb2RfbGFiZWxzGAkgAygLMkEuZmx5dGVpZGwyLmxvZ3MuZGF0YXBsYW5lLkxvZ2dpbmdDb250ZXh0Lkt1YmVybmV0ZXNQb2RMYWJlbHNFbnRyeRo6ChhLdWJlcm5ldGVzUG9kTGFiZWxzRW50cnkSCwoDa2V5GAEgASgJEg0KBXZhbHVlGAIgASgJOgI4AUoECAEQAkoECAIQAyKkAQoTQ29udGFpbmVySWRlbnRpZmllchIdCgxjbHVzdGVyX25hbWUYASABKAlCB7pIBHICEAESJQoUa3ViZXJuZXRlc19uYW1lc3BhY2UYAiABKAlCB7pIBHICEAESJAoTa3ViZXJuZXRlc19wb2RfbmFtZRgDIAEoCUIHukgEcgIQARIhChlrdWJlcm5ldGVzX2NvbnRhaW5lcl9uYW1lGAQgASgJIscBChFDb250YWluZXJTZWxlY3RvchIdCgxjbHVzdGVyX25hbWUYASABKAlCB7pIBHICEAESJQoUa3ViZXJuZXRlc19uYW1lc3BhY2UYAiABKAlCB7pIBHICEAESIgoaa3ViZXJuZXRlc19wb2RfbmFtZV9wcmVmaXgYAyABKAkSIQoZa3ViZXJuZXRlc19jb250YWluZXJfbmFtZRgEIAEoCRIlCh1rdWJlcm5ldGVzX3BvZF9sYWJlbF9zZWxlY3RvchgFIAEoCSJBCg9MaXZlTG9nc09wdGlvbnMSFgoObG9nX3BvZF9zdGF0dXMYASABKAgSFgoObG9nX3RpbWVzdGFtcHMYAiABKAgiigEKB0xvZ0xpbmUSLQoJdGltZXN0YW1wGAEgASgLMhouZ29vZ2xlLnByb3RvYnVmLlRpbWVzdGFtcBIPCgdtZXNzYWdlGAIgASgJEj8KCm9yaWdpbmF0b3IYAyABKA4yKy5mbHl0ZWlkbDIubG9ncy5kYXRhcGxhbmUuTG9nTGluZU9yaWdpbmF0b3IivQEKCExvZ0xpbmVzEhEKBWxpbmVzGAEgAygJQgIYARIXCg9jb250YWluZXJfaW5kZXgYAiABKA0SSAoJY29udGFpbmVyGAMgASgLMi0uZmx5dGVpZGwyLmxvZ3MuZGF0YXBsYW5lLkNvbnRhaW5lcklkZW50aWZpZXJCBrpIA8gBARI7ChBzdHJ1Y3R1cmVkX2xpbmVzGAQgAygLMiEuZmx5dGVpZGwyLmxvZ3MuZGF0YXBsYW5lLkxvZ0xpbmUiVgoRTG9nQ29udGFpbmVyc0xpc3QSQQoKY29udGFpbmVycxgBIAMoCzItLmZseXRlaWRsMi5sb2dzLmRhdGFwbGFuZS5Db250YWluZXJJZGVudGlmaWVyIkEKDUxvZ0xpbmVzQmF0Y2gSMAoEbG9ncxgBIAMoCzIiLmZseXRlaWRsMi5sb2dzLmRhdGFwbGFuZS5Mb2dMaW5lcyo2ChFMb2dMaW5lT3JpZ2luYXRvchILCgdVTktOT1dOEAASCAoEVVNFUhABEgoKBlNZU1RFTRACKkYKCkxvZ3NTb3VyY2USFQoRTElWRV9PUl9QRVJTSVNURUQQABINCglMSVZFX09OTFkQARISCg5QRVJTSVNURURfT05MWRACQu4BChxjb20uZmx5dGVpZGwyLmxvZ3MuZGF0YXBsYW5lQgxQYXlsb2FkUHJvdG9IAlABWjxnaXRodWIuY29tL2ZseXRlb3JnL2ZseXRlL3YyL2dlbi9nby9mbHl0ZWlkbDIvbG9ncy9kYXRhcGxhbmWiAgNGTESqAhhGbHl0ZWlkbDIuTG9ncy5EYXRhcGxhbmXKAhhGbHl0ZWlkbDJcTG9nc1xEYXRhcGxhbmXiAiRGbHl0ZWlkbDJcTG9nc1xEYXRhcGxhbmVcR1BCTWV0YWRhdGHqAhpGbHl0ZWlkbDI6OkxvZ3M6OkRhdGFwbGFuZWIGcHJvdG8z", [file_buf_validate_validate, file_google_protobuf_timestamp]);
+
+/**
+ * @generated from message flyteidl2.logs.dataplane.PodResource
+ */
+export type PodResource = Message<"flyteidl2.logs.dataplane.PodResource"> & {
+  /**
+   * The namespace of the pod.
+   *
+   * @generated from field: string namespace = 1;
+   */
+  namespace: string;
+
+  /**
+   * The pod name.
+   *
+   * @generated from field: string name = 2;
+   */
+  name: string;
+
+  /**
+   * The container name. If not provided, attempt to find the primary container, else assume the first container.
+   * +optional
+   *
+   * @generated from field: string container = 3;
+   */
+  container: string;
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.PodResource.
+ * Use `create(PodResourceSchema)` to create a new message.
+ */
+export const PodResourceSchema: GenMessage<PodResource> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 0);
+
+/**
+ * Parameters of environment in which logs were collected. Should contain everything
+ * necessary to identify location of task execution logs in cloud providers.
+ *
+ * @generated from message flyteidl2.logs.dataplane.LoggingContext
+ */
+export type LoggingContext = Message<"flyteidl2.logs.dataplane.LoggingContext"> & {
+  /**
+   * @generated from field: string cluster_name = 3;
+   */
+  clusterName: string;
+
+  /**
+   * @generated from field: string kubernetes_namespace = 4;
+   */
+  kubernetesNamespace: string;
+
+  /**
+   * @generated from field: string kubernetes_pod_name = 5;
+   */
+  kubernetesPodName: string;
+
+  /**
+   * @generated from field: string kubernetes_container_name = 6;
+   */
+  kubernetesContainerName: string;
+
+  /**
+   * @generated from field: google.protobuf.Timestamp execution_attempt_start_time = 7;
+   */
+  executionAttemptStartTime?: Timestamp;
+
+  /**
+   * @generated from field: google.protobuf.Timestamp execution_attempt_end_time = 8;
+   */
+  executionAttemptEndTime?: Timestamp;
+
+  /**
+   * @generated from field: map<string, string> kubernetes_pod_labels = 9;
+   */
+  kubernetesPodLabels: { [key: string]: string };
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.LoggingContext.
+ * Use `create(LoggingContextSchema)` to create a new message.
+ */
+export const LoggingContextSchema: GenMessage<LoggingContext> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 1);
+
+/**
+ * Parameters of environment in which logs were collected. Should contain everything
+ * necessary to identify location of task execution logs in cloud providers.
+ *
+ * @generated from message flyteidl2.logs.dataplane.ContainerIdentifier
+ */
+export type ContainerIdentifier = Message<"flyteidl2.logs.dataplane.ContainerIdentifier"> & {
+  /**
+   * The name of the cluster.
+   *
+   * @generated from field: string cluster_name = 1;
+   */
+  clusterName: string;
+
+  /**
+   * The namespace in Kubernetes.
+   *
+   * @generated from field: string kubernetes_namespace = 2;
+   */
+  kubernetesNamespace: string;
+
+  /**
+   * The name of the pod in Kubernetes.
+   *
+   * @generated from field: string kubernetes_pod_name = 3;
+   */
+  kubernetesPodName: string;
+
+  /**
+   * The name of the container in Kubernetes.
+   *
+   * @generated from field: string kubernetes_container_name = 4;
+   */
+  kubernetesContainerName: string;
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.ContainerIdentifier.
+ * Use `create(ContainerIdentifierSchema)` to create a new message.
+ */
+export const ContainerIdentifierSchema: GenMessage<ContainerIdentifier> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 2);
+
+/**
+ * @generated from message flyteidl2.logs.dataplane.ContainerSelector
+ */
+export type ContainerSelector = Message<"flyteidl2.logs.dataplane.ContainerSelector"> & {
+  /**
+   * The name of the cluster.
+   *
+   * @generated from field: string cluster_name = 1;
+   */
+  clusterName: string;
+
+  /**
+   * The namespace in Kubernetes.
+   *
+   * @generated from field: string kubernetes_namespace = 2;
+   */
+  kubernetesNamespace: string;
+
+  /**
+   * The prefix of the name of the pod in Kubernetes. This will only apply to persisted pods' logs because listing by
+   * prefix is the supported way to filter pods.
+   *
+   * @generated from field: string kubernetes_pod_name_prefix = 3;
+   */
+  kubernetesPodNamePrefix: string;
+
+  /**
+   * The name of the container in Kubernetes. If not specified, logs for all containers
+   * will be streamed.
+   *
+   * @generated from field: string kubernetes_container_name = 4;
+   */
+  kubernetesContainerName: string;
+
+  /**
+   * The label selector to filter pods. This will only apply to live pods' logs because Listing by prefix
+   * isn't supported.
+   *
+   * @generated from field: string kubernetes_pod_label_selector = 5;
+   */
+  kubernetesPodLabelSelector: string;
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.ContainerSelector.
+ * Use `create(ContainerSelectorSchema)` to create a new message.
+ */
+export const ContainerSelectorSchema: GenMessage<ContainerSelector> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 3);
+
+/**
+ * @generated from message flyteidl2.logs.dataplane.LiveLogsOptions
+ */
+export type LiveLogsOptions = Message<"flyteidl2.logs.dataplane.LiveLogsOptions"> & {
+  /**
+   * LogPodStatus indicates whether to log the pod status along with the logs.
+   *
+   * @generated from field: bool log_pod_status = 1;
+   */
+  logPodStatus: boolean;
+
+  /**
+   * LogTimestamps indicates whether to log the timestamps along with the logs. It prepends RFC3339 or RFC3339Nano
+   * format in the beginning of each log line.
+   *
+   * @generated from field: bool log_timestamps = 2;
+   */
+  logTimestamps: boolean;
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.LiveLogsOptions.
+ * Use `create(LiveLogsOptionsSchema)` to create a new message.
+ */
+export const LiveLogsOptionsSchema: GenMessage<LiveLogsOptions> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 4);
+
+/**
+ * @generated from message flyteidl2.logs.dataplane.LogLine
+ */
+export type LogLine = Message<"flyteidl2.logs.dataplane.LogLine"> & {
+  /**
+   * @generated from field: google.protobuf.Timestamp timestamp = 1;
+   */
+  timestamp?: Timestamp;
+
+  /**
+   * Each line is separated by either CRLF, CR or LF, which are included
+   * at the ends of the lines. This lets clients know whether log emitter
+   * wanted to overwrite the previous line (LF) or append a new line (CRLF).
+   *
+   * @generated from field: string message = 2;
+   */
+  message: string;
+
+  /**
+   * @generated from field: flyteidl2.logs.dataplane.LogLineOriginator originator = 3;
+   */
+  originator: LogLineOriginator;
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.LogLine.
+ * Use `create(LogLineSchema)` to create a new message.
+ */
+export const LogLineSchema: GenMessage<LogLine> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 5);
+
+/**
+ * @generated from message flyteidl2.logs.dataplane.LogLines
+ */
+export type LogLines = Message<"flyteidl2.logs.dataplane.LogLines"> & {
+  /**
+   * Each line is separated by either CRLF, CR or LF, which are included
+   * at the ends of the lines. This lets clients know whether log emitter
+   * wanted to overwrite the previous line (LF) or append a new line (CRLF).
+   *
+   * @generated from field: repeated string lines = 1 [deprecated = true];
+   * @deprecated
+   */
+  lines: string[];
+
+  /**
+   * The index of the container in the list of containers. If the request was made with a single container identifier,
+   * this value will always be 0. Otherwise, it'll be an index into the last list of containers sent in the stream.
+   *
+   * @generated from field: uint32 container_index = 2;
+   */
+  containerIndex: number;
+
+  /**
+   * The container identifier.
+   *
+   * @generated from field: flyteidl2.logs.dataplane.ContainerIdentifier container = 3;
+   */
+  container?: ContainerIdentifier;
+
+  /**
+   * Each line is separated by either CRLF, CR or LF, which are included
+   * at the ends of the lines. This lets clients know whether log emitter
+   * wanted to overwrite the previous line (LF) or append a new line (CRLF).
+   *
+   * @generated from field: repeated flyteidl2.logs.dataplane.LogLine structured_lines = 4;
+   */
+  structuredLines: LogLine[];
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.LogLines.
+ * Use `create(LogLinesSchema)` to create a new message.
+ */
+export const LogLinesSchema: GenMessage<LogLines> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 6);
+
+/**
+ * @generated from message flyteidl2.logs.dataplane.LogContainersList
+ */
+export type LogContainersList = Message<"flyteidl2.logs.dataplane.LogContainersList"> & {
+  /**
+   * @generated from field: repeated flyteidl2.logs.dataplane.ContainerIdentifier containers = 1;
+   */
+  containers: ContainerIdentifier[];
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.LogContainersList.
+ * Use `create(LogContainersListSchema)` to create a new message.
+ */
+export const LogContainersListSchema: GenMessage<LogContainersList> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 7);
+
+/**
+ * @generated from message flyteidl2.logs.dataplane.LogLinesBatch
+ */
+export type LogLinesBatch = Message<"flyteidl2.logs.dataplane.LogLinesBatch"> & {
+  /**
+   * @generated from field: repeated flyteidl2.logs.dataplane.LogLines logs = 1;
+   */
+  logs: LogLines[];
+};
+
+/**
+ * Describes the message flyteidl2.logs.dataplane.LogLinesBatch.
+ * Use `create(LogLinesBatchSchema)` to create a new message.
+ */
+export const LogLinesBatchSchema: GenMessage<LogLinesBatch> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_logs_dataplane_payload, 8);
+
+/**
+ * @generated from enum flyteidl2.logs.dataplane.LogLineOriginator
+ */
+export enum LogLineOriginator {
+  /**
+   * The originator of the log line is unknown.
+   *
+   * @generated from enum value: UNKNOWN = 0;
+   */
+  UNKNOWN = 0,
+
+  /**
+   * The originator of the log line is the user application.
+   *
+   * @generated from enum value: USER = 1;
+   */
+  USER = 1,
+
+  /**
+   * The originator of the log line is the system.
+   *
+   * @generated from enum value: SYSTEM = 2;
+   */
+  SYSTEM = 2,
+}
+
+/**
+ * Describes the enum flyteidl2.logs.dataplane.LogLineOriginator.
+ */
+export const LogLineOriginatorSchema: GenEnum<LogLineOriginator> = /*@__PURE__*/
+  enumDesc(file_flyteidl2_logs_dataplane_payload, 0);
+
+/**
+ * @generated from enum flyteidl2.logs.dataplane.LogsSource
+ */
+export enum LogsSource {
+  /**
+   * Return live logs and fall back to persisted if not available.
+   *
+   * @generated from enum value: LIVE_OR_PERSISTED = 0;
+   */
+  LIVE_OR_PERSISTED = 0,
+
+  /**
+   * Return live logs only or error if pod is no longer around.
+   *
+   * @generated from enum value: LIVE_ONLY = 1;
+   */
+  LIVE_ONLY = 1,
+
+  /**
+   * Return persisted logs only.
+   *
+   * @generated from enum value: PERSISTED_ONLY = 2;
+   */
+  PERSISTED_ONLY = 2,
+}
+
+/**
+ * Describes the enum flyteidl2.logs.dataplane.LogsSource.
+ */
+export const LogsSourceSchema: GenEnum<LogsSource> = /*@__PURE__*/
+  enumDesc(file_flyteidl2_logs_dataplane_payload, 1);
+

package/flyteidl2/plugins/dask_pb.ts

@@ -0,0 +1,109 @@
+// @generated by protoc-gen-es v2.2.5 with parameter "target=ts,import_extension=.ts"
+// @generated from file flyteidl2/plugins/dask.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 { Resources } from "../core/tasks_pb.ts";
+import { file_flyteidl2_core_tasks } from "../core/tasks_pb.ts";
+import type { Message } from "@bufbuild/protobuf";
+
+/**
+ * Describes the file flyteidl2/plugins/dask.proto.
+ */
+export const file_flyteidl2_plugins_dask: GenFile = /*@__PURE__*/
+  fileDesc("ChxmbHl0ZWlkbDIvcGx1Z2lucy9kYXNrLnByb3RvEhFmbHl0ZWlkbDIucGx1Z2lucyJzCgdEYXNrSm9iEjMKCXNjaGVkdWxlchgBIAEoCzIgLmZseXRlaWRsMi5wbHVnaW5zLkRhc2tTY2hlZHVsZXISMwoHd29ya2VycxgCIAEoCzIiLmZseXRlaWRsMi5wbHVnaW5zLkRhc2tXb3JrZXJHcm91cCJMCg1EYXNrU2NoZWR1bGVyEg0KBWltYWdlGAEgASgJEiwKCXJlc291cmNlcxgCIAEoCzIZLmZseXRlaWRsMi5jb3JlLlJlc291cmNlcyJpCg9EYXNrV29ya2VyR3JvdXASGQoRbnVtYmVyX29mX3dvcmtlcnMYASABKA0SDQoFaW1hZ2UYAiABKAkSLAoJcmVzb3VyY2VzGAMgASgLMhkuZmx5dGVpZGwyLmNvcmUuUmVzb3VyY2VzQsABChVjb20uZmx5dGVpZGwyLnBsdWdpbnNCCURhc2tQcm90b0gCUAFaNWdpdGh1Yi5jb20vZmx5dGVvcmcvZmx5dGUvdjIvZ2VuL2dvL2ZseXRlaWRsMi9wbHVnaW5zogIDRlBYqgIRRmx5dGVpZGwyLlBsdWdpbnPKAhFGbHl0ZWlkbDJcUGx1Z2luc+ICHUZseXRlaWRsMlxQbHVnaW5zXEdQQk1ldGFkYXRh6gISRmx5dGVpZGwyOjpQbHVnaW5zYgZwcm90bzM", [file_flyteidl2_core_tasks]);
+
+/**
+ * Custom Proto for Dask Plugin.
+ *
+ * @generated from message flyteidl2.plugins.DaskJob
+ */
+export type DaskJob = Message<"flyteidl2.plugins.DaskJob"> & {
+  /**
+   * Spec for the scheduler pod.
+   *
+   * @generated from field: flyteidl2.plugins.DaskScheduler scheduler = 1;
+   */
+  scheduler?: DaskScheduler;
+
+  /**
+   * Spec of the default worker group.
+   *
+   * @generated from field: flyteidl2.plugins.DaskWorkerGroup workers = 2;
+   */
+  workers?: DaskWorkerGroup;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.DaskJob.
+ * Use `create(DaskJobSchema)` to create a new message.
+ */
+export const DaskJobSchema: GenMessage<DaskJob> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_dask, 0);
+
+/**
+ * Specification for the scheduler pod.
+ *
+ * @generated from message flyteidl2.plugins.DaskScheduler
+ */
+export type DaskScheduler = Message<"flyteidl2.plugins.DaskScheduler"> & {
+  /**
+   * Optional image to use. If unset, will use the default image.
+   *
+   * @generated from field: string image = 1;
+   */
+  image: string;
+
+  /**
+   * Resources assigned to the scheduler pod.
+   *
+   * @generated from field: flyteidl2.core.Resources resources = 2;
+   */
+  resources?: Resources;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.DaskScheduler.
+ * Use `create(DaskSchedulerSchema)` to create a new message.
+ */
+export const DaskSchedulerSchema: GenMessage<DaskScheduler> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_dask, 1);
+
+/**
+ * @generated from message flyteidl2.plugins.DaskWorkerGroup
+ */
+export type DaskWorkerGroup = Message<"flyteidl2.plugins.DaskWorkerGroup"> & {
+  /**
+   * Number of workers in the group.
+   *
+   * @generated from field: uint32 number_of_workers = 1;
+   */
+  numberOfWorkers: number;
+
+  /**
+   * Optional image to use for the pods of the worker group. If unset, will use the default image.
+   *
+   * @generated from field: string image = 2;
+   */
+  image: string;
+
+  /**
+   * Resources assigned to the all pods of the worker group.
+   * As per https://kubernetes.dask.org/en/latest/kubecluster.html?highlight=limit#best-practices
+   * it is advised to only set limits. If requests are not explicitly set, the plugin will make
+   * sure to set requests==limits.
+   * The plugin sets ` --memory-limit` as well as `--nthreads` for the workers according to the limit.
+   *
+   * @generated from field: flyteidl2.core.Resources resources = 3;
+   */
+  resources?: Resources;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.DaskWorkerGroup.
+ * Use `create(DaskWorkerGroupSchema)` to create a new message.
+ */
+export const DaskWorkerGroupSchema: GenMessage<DaskWorkerGroup> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_dask, 2);
+

package/flyteidl2/plugins/ray_pb.ts

@@ -0,0 +1,189 @@
+// @generated by protoc-gen-es v2.2.5 with parameter "target=ts,import_extension=.ts"
+// @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 { file_flyteidl2_core_tasks } from "../core/tasks_pb.ts";
+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]);
+
+/**
+ * RayJobSpec defines the desired state of RayJob
+ *
+ * @generated from message flyteidl2.plugins.RayJob
+ */
+export type RayJob = Message<"flyteidl2.plugins.RayJob"> & {
+  /**
+   * RayClusterSpec is the cluster template to run the job
+   *
+   * @generated from field: flyteidl2.plugins.RayCluster ray_cluster = 1;
+   */
+  rayCluster?: RayCluster;
+
+  /**
+   * runtime_env is base64 encoded.
+   * Ray runtime environments: https://docs.ray.io/en/latest/ray-core/handling-dependencies.html#runtime-environments
+   *
+   * @generated from field: string runtime_env = 2 [deprecated = true];
+   * @deprecated
+   */
+  runtimeEnv: string;
+
+  /**
+   * shutdown_after_job_finishes specifies whether the RayCluster should be deleted after the RayJob finishes.
+   *
+   * @generated from field: bool shutdown_after_job_finishes = 3;
+   */
+  shutdownAfterJobFinishes: boolean;
+
+  /**
+   * ttl_seconds_after_finished specifies the number of seconds after which the RayCluster will be deleted after the RayJob finishes.
+   *
+   * @generated from field: int32 ttl_seconds_after_finished = 4;
+   */
+  ttlSecondsAfterFinished: number;
+
+  /**
+   * RuntimeEnvYAML represents the runtime environment configuration
+   * provided as a multi-line YAML string.
+   *
+   * @generated from field: string runtime_env_yaml = 5;
+   */
+  runtimeEnvYaml: string;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.RayJob.
+ * Use `create(RayJobSchema)` to create a new message.
+ */
+export const RayJobSchema: GenMessage<RayJob> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_ray, 0);
+
+/**
+ * Define Ray cluster defines the desired state of RayCluster
+ *
+ * @generated from message flyteidl2.plugins.RayCluster
+ */
+export type RayCluster = Message<"flyteidl2.plugins.RayCluster"> & {
+  /**
+   * HeadGroupSpecs are the spec for the head pod
+   *
+   * @generated from field: flyteidl2.plugins.HeadGroupSpec head_group_spec = 1;
+   */
+  headGroupSpec?: HeadGroupSpec;
+
+  /**
+   * WorkerGroupSpecs are the specs for the worker pods
+   *
+   * @generated from field: repeated flyteidl2.plugins.WorkerGroupSpec worker_group_spec = 2;
+   */
+  workerGroupSpec: WorkerGroupSpec[];
+
+  /**
+   * Whether to enable autoscaling.
+   *
+   * @generated from field: bool enable_autoscaling = 3;
+   */
+  enableAutoscaling: boolean;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.RayCluster.
+ * Use `create(RayClusterSchema)` to create a new message.
+ */
+export const RayClusterSchema: GenMessage<RayCluster> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_ray, 1);
+
+/**
+ * HeadGroupSpec are the spec for the head pod
+ *
+ * @generated from message flyteidl2.plugins.HeadGroupSpec
+ */
+export type HeadGroupSpec = Message<"flyteidl2.plugins.HeadGroupSpec"> & {
+  /**
+   * Optional. RayStartParams are the params of the start command: address, object-store-memory.
+   * Refer to https://docs.ray.io/en/latest/ray-core/package-ref.html#ray-start
+   *
+   * @generated from field: map<string, string> ray_start_params = 1;
+   */
+  rayStartParams: { [key: string]: string };
+
+  /**
+   * Pod Spec for the ray head pod
+   *
+   * @generated from field: flyteidl2.core.K8sPod k8s_pod = 2;
+   */
+  k8sPod?: K8sPod;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.HeadGroupSpec.
+ * Use `create(HeadGroupSpecSchema)` to create a new message.
+ */
+export const HeadGroupSpecSchema: GenMessage<HeadGroupSpec> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_ray, 2);
+
+/**
+ * WorkerGroupSpec are the specs for the worker pods
+ *
+ * @generated from message flyteidl2.plugins.WorkerGroupSpec
+ */
+export type WorkerGroupSpec = Message<"flyteidl2.plugins.WorkerGroupSpec"> & {
+  /**
+   * Required. RayCluster can have multiple worker groups, and it distinguishes them by name
+   *
+   * @generated from field: string group_name = 1;
+   */
+  groupName: string;
+
+  /**
+   * Required. Desired replicas of the worker group. Defaults to 1.
+   *
+   * @generated from field: int32 replicas = 2;
+   */
+  replicas: number;
+
+  /**
+   * Optional. Min replicas of the worker group. MinReplicas defaults to 1.
+   *
+   * @generated from field: int32 min_replicas = 3;
+   */
+  minReplicas: number;
+
+  /**
+   * Optional. Max replicas of the worker group. MaxReplicas defaults to maxInt32
+   *
+   * @generated from field: int32 max_replicas = 4;
+   */
+  maxReplicas: number;
+
+  /**
+   * Optional. RayStartParams are the params of the start command: address, object-store-memory.
+   * Refer to https://docs.ray.io/en/latest/ray-core/package-ref.html#ray-start
+   *
+   * @generated from field: map<string, string> ray_start_params = 5;
+   */
+  rayStartParams: { [key: string]: string };
+
+  /**
+   * Pod Spec for ray worker pods
+   *
+   * @generated from field: flyteidl2.core.K8sPod k8s_pod = 6;
+   */
+  k8sPod?: K8sPod;
+};
+
+/**
+ * Describes the message flyteidl2.plugins.WorkerGroupSpec.
+ * Use `create(WorkerGroupSpecSchema)` to create a new message.
+ */
+export const WorkerGroupSpecSchema: GenMessage<WorkerGroupSpec> = /*@__PURE__*/
+  messageDesc(file_flyteidl2_plugins_ray, 3);
+