@flyteorg/flyteidl 1.2.8 → 1.2.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -5104,6 +5104,9 @@ export namespace flyteidl {
5104
5104
 
5105
5105
  /** TaskMetadata tags */
5106
5106
  tags?: ({ [k: string]: string }|null);
5107
+
5108
+ /** TaskMetadata podTemplateName */
5109
+ podTemplateName?: (string|null);
5107
5110
  }
5108
5111
 
5109
5112
  /** Represents a TaskMetadata. */
@@ -5145,6 +5148,9 @@ export namespace flyteidl {
5145
5148
  /** TaskMetadata tags. */
5146
5149
  public tags: { [k: string]: string };
5147
5150
 
5151
+ /** TaskMetadata podTemplateName. */
5152
+ public podTemplateName: string;
5153
+
5148
5154
  /** TaskMetadata interruptibleValue. */
5149
5155
  public interruptibleValue?: "interruptible";
5150
5156
 
@@ -6534,6 +6540,9 @@ export namespace flyteidl {
6534
6540
  /** NodeExecutionEvent inputUri */
6535
6541
  inputUri?: (string|null);
6536
6542
 
6543
+ /** NodeExecutionEvent inputData */
6544
+ inputData?: (flyteidl.core.ILiteralMap|null);
6545
+
6537
6546
  /** NodeExecutionEvent outputUri */
6538
6547
  outputUri?: (string|null);
6539
6548
 
@@ -6601,6 +6610,9 @@ export namespace flyteidl {
6601
6610
  /** NodeExecutionEvent inputUri. */
6602
6611
  public inputUri: string;
6603
6612
 
6613
+ /** NodeExecutionEvent inputData. */
6614
+ public inputData?: (flyteidl.core.ILiteralMap|null);
6615
+
6604
6616
  /** NodeExecutionEvent outputUri. */
6605
6617
  public outputUri: string;
6606
6618
 
@@ -6643,6 +6655,9 @@ export namespace flyteidl {
6643
6655
  /** NodeExecutionEvent deckUri. */
6644
6656
  public deckUri: string;
6645
6657
 
6658
+ /** NodeExecutionEvent inputValue. */
6659
+ public inputValue?: ("inputUri"|"inputData");
6660
+
6646
6661
  /** NodeExecutionEvent outputResult. */
6647
6662
  public outputResult?: ("outputUri"|"error"|"outputData");
6648
6663
 
@@ -6999,6 +7014,9 @@ export namespace flyteidl {
6999
7014
  /** TaskExecutionEvent inputUri */
7000
7015
  inputUri?: (string|null);
7001
7016
 
7017
+ /** TaskExecutionEvent inputData */
7018
+ inputData?: (flyteidl.core.ILiteralMap|null);
7019
+
7002
7020
  /** TaskExecutionEvent outputUri */
7003
7021
  outputUri?: (string|null);
7004
7022
 
@@ -7060,6 +7078,9 @@ export namespace flyteidl {
7060
7078
  /** TaskExecutionEvent inputUri. */
7061
7079
  public inputUri: string;
7062
7080
 
7081
+ /** TaskExecutionEvent inputData. */
7082
+ public inputData?: (flyteidl.core.ILiteralMap|null);
7083
+
7063
7084
  /** TaskExecutionEvent outputUri. */
7064
7085
  public outputUri: string;
7065
7086
 
@@ -7087,6 +7108,9 @@ export namespace flyteidl {
7087
7108
  /** TaskExecutionEvent eventVersion. */
7088
7109
  public eventVersion: number;
7089
7110
 
7111
+ /** TaskExecutionEvent inputValue. */
7112
+ public inputValue?: ("inputUri"|"inputData");
7113
+
7090
7114
  /** TaskExecutionEvent outputResult. */
7091
7115
  public outputResult?: ("outputUri"|"error"|"outputData");
7092
7116
 
@@ -19160,6 +19184,9 @@ export namespace flyteidl {
19160
19184
 
19161
19185
  /** UserInfoResponse picture */
19162
19186
  picture?: (string|null);
19187
+
19188
+ /** UserInfoResponse additionalClaims */
19189
+ additionalClaims?: (google.protobuf.IStruct|null);
19163
19190
  }
19164
19191
 
19165
19192
  /** Represents a UserInfoResponse. */
@@ -19192,6 +19219,9 @@ export namespace flyteidl {
19192
19219
  /** UserInfoResponse picture. */
19193
19220
  public picture: string;
19194
19221
 
19222
+ /** UserInfoResponse additionalClaims. */
19223
+ public additionalClaims?: (google.protobuf.IStruct|null);
19224
+
19195
19225
  /**
19196
19226
  * Creates a new UserInfoResponse instance using the specified properties.
19197
19227
  * @param [properties] Properties to set
@@ -12048,6 +12048,7 @@
12048
12048
  * @property {boolean|null} [cacheSerializable] TaskMetadata cacheSerializable
12049
12049
  * @property {boolean|null} [generatesDeck] TaskMetadata generatesDeck
12050
12050
  * @property {Object.<string,string>|null} [tags] TaskMetadata tags
12051
+ * @property {string|null} [podTemplateName] TaskMetadata podTemplateName
12051
12052
  */
12052
12053
 
12053
12054
  /**
@@ -12146,6 +12147,14 @@
12146
12147
  */
12147
12148
  TaskMetadata.prototype.tags = $util.emptyObject;
12148
12149
 
12150
+ /**
12151
+ * TaskMetadata podTemplateName.
12152
+ * @member {string} podTemplateName
12153
+ * @memberof flyteidl.core.TaskMetadata
12154
+ * @instance
12155
+ */
12156
+ TaskMetadata.prototype.podTemplateName = "";
12157
+
12149
12158
  // OneOf field names bound to virtual getters and setters
12150
12159
  var $oneOfFields;
12151
12160
 
@@ -12205,6 +12214,8 @@
12205
12214
  if (message.tags != null && message.hasOwnProperty("tags"))
12206
12215
  for (var keys = Object.keys(message.tags), i = 0; i < keys.length; ++i)
12207
12216
  writer.uint32(/* id 11, wireType 2 =*/90).fork().uint32(/* id 1, wireType 2 =*/10).string(keys[i]).uint32(/* id 2, wireType 2 =*/18).string(message.tags[keys[i]]).ldelim();
12217
+ if (message.podTemplateName != null && message.hasOwnProperty("podTemplateName"))
12218
+ writer.uint32(/* id 12, wireType 2 =*/98).string(message.podTemplateName);
12208
12219
  return writer;
12209
12220
  };
12210
12221
 
@@ -12261,6 +12272,9 @@
12261
12272
  reader.pos++;
12262
12273
  message.tags[key] = reader.string();
12263
12274
  break;
12275
+ case 12:
12276
+ message.podTemplateName = reader.string();
12277
+ break;
12264
12278
  default:
12265
12279
  reader.skipType(tag & 7);
12266
12280
  break;
@@ -12324,6 +12338,9 @@
12324
12338
  if (!$util.isString(message.tags[key[i]]))
12325
12339
  return "tags: string{k:string} expected";
12326
12340
  }
12341
+ if (message.podTemplateName != null && message.hasOwnProperty("podTemplateName"))
12342
+ if (!$util.isString(message.podTemplateName))
12343
+ return "podTemplateName: string expected";
12327
12344
  return null;
12328
12345
  };
12329
12346
 
@@ -15665,6 +15682,7 @@
15665
15682
  * @property {flyteidl.core.NodeExecution.Phase|null} [phase] NodeExecutionEvent phase
15666
15683
  * @property {google.protobuf.ITimestamp|null} [occurredAt] NodeExecutionEvent occurredAt
15667
15684
  * @property {string|null} [inputUri] NodeExecutionEvent inputUri
15685
+ * @property {flyteidl.core.ILiteralMap|null} [inputData] NodeExecutionEvent inputData
15668
15686
  * @property {string|null} [outputUri] NodeExecutionEvent outputUri
15669
15687
  * @property {flyteidl.core.IExecutionError|null} [error] NodeExecutionEvent error
15670
15688
  * @property {flyteidl.core.ILiteralMap|null} [outputData] NodeExecutionEvent outputData
@@ -15736,6 +15754,14 @@
15736
15754
  */
15737
15755
  NodeExecutionEvent.prototype.inputUri = "";
15738
15756
 
15757
+ /**
15758
+ * NodeExecutionEvent inputData.
15759
+ * @member {flyteidl.core.ILiteralMap|null|undefined} inputData
15760
+ * @memberof flyteidl.event.NodeExecutionEvent
15761
+ * @instance
15762
+ */
15763
+ NodeExecutionEvent.prototype.inputData = null;
15764
+
15739
15765
  /**
15740
15766
  * NodeExecutionEvent outputUri.
15741
15767
  * @member {string} outputUri
@@ -15851,6 +15877,17 @@
15851
15877
  // OneOf field names bound to virtual getters and setters
15852
15878
  var $oneOfFields;
15853
15879
 
15880
+ /**
15881
+ * NodeExecutionEvent inputValue.
15882
+ * @member {"inputUri"|"inputData"|undefined} inputValue
15883
+ * @memberof flyteidl.event.NodeExecutionEvent
15884
+ * @instance
15885
+ */
15886
+ Object.defineProperty(NodeExecutionEvent.prototype, "inputValue", {
15887
+ get: $util.oneOfGetter($oneOfFields = ["inputUri", "inputData"]),
15888
+ set: $util.oneOfSetter($oneOfFields)
15889
+ });
15890
+
15854
15891
  /**
15855
15892
  * NodeExecutionEvent outputResult.
15856
15893
  * @member {"outputUri"|"error"|"outputData"|undefined} outputResult
@@ -15935,6 +15972,8 @@
15935
15972
  writer.uint32(/* id 18, wireType 0 =*/144).bool(message.isDynamic);
15936
15973
  if (message.deckUri != null && message.hasOwnProperty("deckUri"))
15937
15974
  writer.uint32(/* id 19, wireType 2 =*/154).string(message.deckUri);
15975
+ if (message.inputData != null && message.hasOwnProperty("inputData"))
15976
+ $root.flyteidl.core.LiteralMap.encode(message.inputData, writer.uint32(/* id 20, wireType 2 =*/162).fork()).ldelim();
15938
15977
  return writer;
15939
15978
  };
15940
15979
 
@@ -15971,6 +16010,9 @@
15971
16010
  case 5:
15972
16011
  message.inputUri = reader.string();
15973
16012
  break;
16013
+ case 20:
16014
+ message.inputData = $root.flyteidl.core.LiteralMap.decode(reader, reader.uint32());
16015
+ break;
15974
16016
  case 6:
15975
16017
  message.outputUri = reader.string();
15976
16018
  break;
@@ -16063,9 +16105,21 @@
16063
16105
  if (error)
16064
16106
  return "occurredAt." + error;
16065
16107
  }
16066
- if (message.inputUri != null && message.hasOwnProperty("inputUri"))
16108
+ if (message.inputUri != null && message.hasOwnProperty("inputUri")) {
16109
+ properties.inputValue = 1;
16067
16110
  if (!$util.isString(message.inputUri))
16068
16111
  return "inputUri: string expected";
16112
+ }
16113
+ if (message.inputData != null && message.hasOwnProperty("inputData")) {
16114
+ if (properties.inputValue === 1)
16115
+ return "inputValue: multiple values";
16116
+ properties.inputValue = 1;
16117
+ {
16118
+ var error = $root.flyteidl.core.LiteralMap.verify(message.inputData);
16119
+ if (error)
16120
+ return "inputData." + error;
16121
+ }
16122
+ }
16069
16123
  if (message.outputUri != null && message.hasOwnProperty("outputUri")) {
16070
16124
  properties.outputResult = 1;
16071
16125
  if (!$util.isString(message.outputUri))
@@ -16825,6 +16879,7 @@
16825
16879
  * @property {Array.<flyteidl.core.ITaskLog>|null} [logs] TaskExecutionEvent logs
16826
16880
  * @property {google.protobuf.ITimestamp|null} [occurredAt] TaskExecutionEvent occurredAt
16827
16881
  * @property {string|null} [inputUri] TaskExecutionEvent inputUri
16882
+ * @property {flyteidl.core.ILiteralMap|null} [inputData] TaskExecutionEvent inputData
16828
16883
  * @property {string|null} [outputUri] TaskExecutionEvent outputUri
16829
16884
  * @property {flyteidl.core.IExecutionError|null} [error] TaskExecutionEvent error
16830
16885
  * @property {flyteidl.core.ILiteralMap|null} [outputData] TaskExecutionEvent outputData
@@ -16916,6 +16971,14 @@
16916
16971
  */
16917
16972
  TaskExecutionEvent.prototype.inputUri = "";
16918
16973
 
16974
+ /**
16975
+ * TaskExecutionEvent inputData.
16976
+ * @member {flyteidl.core.ILiteralMap|null|undefined} inputData
16977
+ * @memberof flyteidl.event.TaskExecutionEvent
16978
+ * @instance
16979
+ */
16980
+ TaskExecutionEvent.prototype.inputData = null;
16981
+
16919
16982
  /**
16920
16983
  * TaskExecutionEvent outputUri.
16921
16984
  * @member {string} outputUri
@@ -16991,6 +17054,17 @@
16991
17054
  // OneOf field names bound to virtual getters and setters
16992
17055
  var $oneOfFields;
16993
17056
 
17057
+ /**
17058
+ * TaskExecutionEvent inputValue.
17059
+ * @member {"inputUri"|"inputData"|undefined} inputValue
17060
+ * @memberof flyteidl.event.TaskExecutionEvent
17061
+ * @instance
17062
+ */
17063
+ Object.defineProperty(TaskExecutionEvent.prototype, "inputValue", {
17064
+ get: $util.oneOfGetter($oneOfFields = ["inputUri", "inputData"]),
17065
+ set: $util.oneOfSetter($oneOfFields)
17066
+ });
17067
+
16994
17068
  /**
16995
17069
  * TaskExecutionEvent outputResult.
16996
17070
  * @member {"outputUri"|"error"|"outputData"|undefined} outputResult
@@ -17061,6 +17135,8 @@
17061
17135
  $root.flyteidl.core.LiteralMap.encode(message.outputData, writer.uint32(/* id 17, wireType 2 =*/138).fork()).ldelim();
17062
17136
  if (message.eventVersion != null && message.hasOwnProperty("eventVersion"))
17063
17137
  writer.uint32(/* id 18, wireType 0 =*/144).int32(message.eventVersion);
17138
+ if (message.inputData != null && message.hasOwnProperty("inputData"))
17139
+ $root.flyteidl.core.LiteralMap.encode(message.inputData, writer.uint32(/* id 19, wireType 2 =*/154).fork()).ldelim();
17064
17140
  return writer;
17065
17141
  };
17066
17142
 
@@ -17108,6 +17184,9 @@
17108
17184
  case 8:
17109
17185
  message.inputUri = reader.string();
17110
17186
  break;
17187
+ case 19:
17188
+ message.inputData = $root.flyteidl.core.LiteralMap.decode(reader, reader.uint32());
17189
+ break;
17111
17190
  case 9:
17112
17191
  message.outputUri = reader.string();
17113
17192
  break;
@@ -17199,9 +17278,21 @@
17199
17278
  if (error)
17200
17279
  return "occurredAt." + error;
17201
17280
  }
17202
- if (message.inputUri != null && message.hasOwnProperty("inputUri"))
17281
+ if (message.inputUri != null && message.hasOwnProperty("inputUri")) {
17282
+ properties.inputValue = 1;
17203
17283
  if (!$util.isString(message.inputUri))
17204
17284
  return "inputUri: string expected";
17285
+ }
17286
+ if (message.inputData != null && message.hasOwnProperty("inputData")) {
17287
+ if (properties.inputValue === 1)
17288
+ return "inputValue: multiple values";
17289
+ properties.inputValue = 1;
17290
+ {
17291
+ var error = $root.flyteidl.core.LiteralMap.verify(message.inputData);
17292
+ if (error)
17293
+ return "inputData." + error;
17294
+ }
17295
+ }
17205
17296
  if (message.outputUri != null && message.hasOwnProperty("outputUri")) {
17206
17297
  properties.outputResult = 1;
17207
17298
  if (!$util.isString(message.outputUri))
@@ -44528,6 +44619,7 @@
44528
44619
  * @property {string|null} [familyName] UserInfoResponse familyName
44529
44620
  * @property {string|null} [email] UserInfoResponse email
44530
44621
  * @property {string|null} [picture] UserInfoResponse picture
44622
+ * @property {google.protobuf.IStruct|null} [additionalClaims] UserInfoResponse additionalClaims
44531
44623
  */
44532
44624
 
44533
44625
  /**
@@ -44601,6 +44693,14 @@
44601
44693
  */
44602
44694
  UserInfoResponse.prototype.picture = "";
44603
44695
 
44696
+ /**
44697
+ * UserInfoResponse additionalClaims.
44698
+ * @member {google.protobuf.IStruct|null|undefined} additionalClaims
44699
+ * @memberof flyteidl.service.UserInfoResponse
44700
+ * @instance
44701
+ */
44702
+ UserInfoResponse.prototype.additionalClaims = null;
44703
+
44604
44704
  /**
44605
44705
  * Creates a new UserInfoResponse instance using the specified properties.
44606
44706
  * @function create
@@ -44639,6 +44739,8 @@
44639
44739
  writer.uint32(/* id 6, wireType 2 =*/50).string(message.email);
44640
44740
  if (message.picture != null && message.hasOwnProperty("picture"))
44641
44741
  writer.uint32(/* id 7, wireType 2 =*/58).string(message.picture);
44742
+ if (message.additionalClaims != null && message.hasOwnProperty("additionalClaims"))
44743
+ $root.google.protobuf.Struct.encode(message.additionalClaims, writer.uint32(/* id 8, wireType 2 =*/66).fork()).ldelim();
44642
44744
  return writer;
44643
44745
  };
44644
44746
 
@@ -44681,6 +44783,9 @@
44681
44783
  case 7:
44682
44784
  message.picture = reader.string();
44683
44785
  break;
44786
+ case 8:
44787
+ message.additionalClaims = $root.google.protobuf.Struct.decode(reader, reader.uint32());
44788
+ break;
44684
44789
  default:
44685
44790
  reader.skipType(tag & 7);
44686
44791
  break;
@@ -44721,6 +44826,11 @@
44721
44826
  if (message.picture != null && message.hasOwnProperty("picture"))
44722
44827
  if (!$util.isString(message.picture))
44723
44828
  return "picture: string expected";
44829
+ if (message.additionalClaims != null && message.hasOwnProperty("additionalClaims")) {
44830
+ var error = $root.google.protobuf.Struct.verify(message.additionalClaims);
44831
+ if (error)
44832
+ return "additionalClaims." + error;
44833
+ }
44724
44834
  return null;
44725
44835
  };
44726
44836
 
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@flyteorg/flyteidl",
3
- "version": "1.2.8",
3
+ "version": "1.2.9",
4
4
  "description": "Compiled protocol buffers and gRPC service clients/servers for Flyte IDLs",
5
5
  "repository": {
6
6
  "type": "git",
@@ -38,6 +38,100 @@ will be executed concurrently.
38
38
 
39
39
 
40
40
 
41
+ ..
42
+ end messages
43
+
44
+
45
+ ..
46
+ end enums
47
+
48
+
49
+ ..
50
+ end HasExtensions
51
+
52
+
53
+ ..
54
+ end services
55
+
56
+
57
+
58
+
59
+ .. _ref_flyteidl/plugins/dask.proto:
60
+
61
+ flyteidl/plugins/dask.proto
62
+ ==================================================================
63
+
64
+
65
+
66
+
67
+
68
+ .. _ref_flyteidl.plugins.DaskCluster:
69
+
70
+ DaskCluster
71
+ ------------------------------------------------------------------
72
+
73
+
74
+
75
+
76
+
77
+ .. csv-table:: DaskCluster type fields
78
+ :header: "Field", "Type", "Label", "Description"
79
+ :widths: auto
80
+
81
+ "image", ":ref:`ref_string`", "", "Optional image to use for the scheduler as well as the default worker group. If unset, will use the default image."
82
+ "nWorkers", ":ref:`ref_int32`", "", "Number of workers in the default worker group"
83
+ "resources", ":ref:`ref_flyteidl.core.Resources`", "", "Resources assigned to the scheduler as well as all pods of the default 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."
84
+
85
+
86
+
87
+
88
+
89
+
90
+
91
+ .. _ref_flyteidl.plugins.DaskJob:
92
+
93
+ DaskJob
94
+ ------------------------------------------------------------------
95
+
96
+ Custom Proto for Dask Plugin
97
+
98
+
99
+
100
+ .. csv-table:: DaskJob type fields
101
+ :header: "Field", "Type", "Label", "Description"
102
+ :widths: auto
103
+
104
+ "namespace", ":ref:`ref_string`", "", "Optional namespace to use for the dask pods. If none is given, the namespace of the Flyte task is used"
105
+ "jobPodSpec", ":ref:`ref_flyteidl.plugins.JobPodSpec`", "", "Spec for the job pod"
106
+ "cluster", ":ref:`ref_flyteidl.plugins.DaskCluster`", "", "Cluster"
107
+
108
+
109
+
110
+
111
+
112
+
113
+
114
+ .. _ref_flyteidl.plugins.JobPodSpec:
115
+
116
+ JobPodSpec
117
+ ------------------------------------------------------------------
118
+
119
+ Specification for the job pod
120
+
121
+
122
+
123
+ .. csv-table:: JobPodSpec type fields
124
+ :header: "Field", "Type", "Label", "Description"
125
+ :widths: auto
126
+
127
+ "image", ":ref:`ref_string`", "", "Optional image to use. If unset, will use the default image."
128
+ "resources", ":ref:`ref_flyteidl.core.Resources`", "", "Resources assigned to the job pod."
129
+
130
+
131
+
132
+
133
+
134
+
41
135
  ..
42
136
  end messages
43
137
 
@@ -501,6 +595,7 @@ Custom Proto for Spark Plugin.
501
595
  "sparkConf", ":ref:`ref_flyteidl.plugins.SparkJob.SparkConfEntry`", "repeated", ""
502
596
  "hadoopConf", ":ref:`ref_flyteidl.plugins.SparkJob.HadoopConfEntry`", "repeated", ""
503
597
  "executorPath", ":ref:`ref_string`", "", "Executor path for Python jobs."
598
+ "databricksConf", ":ref:`ref_string`", "", "databricksConf is base64 encoded string which stores databricks job configuration. Config structure can be found here. https://docs.databricks.com/dev-tools/api/2.0/jobs.html#request-structure The config is automatically encoded by flytekit, and decoded in the propeller."
504
599
 
505
600
 
506
601
 
@@ -101,6 +101,11 @@ message TaskMetadata {
101
101
 
102
102
  // Arbitrary tags that allow users and the platform to store small but arbitrary labels
103
103
  map<string, string> tags = 11;
104
+
105
+ // pod_template_name is the unique name of a PodTemplate k8s resource to be used as the base configuration if this
106
+ // task creates a k8s Pod. If this value is set, the specified PodTemplate will be used instead of, but applied
107
+ // identically as, the default PodTemplate configured in FlytePropeller.
108
+ string pod_template_name = 12;
104
109
  }
105
110
 
106
111
  // A Task structure that uniquely identifies a task in the system
@@ -203,34 +203,55 @@ message UpdateArtifactResponse {
203
203
  * ReservationID message that is composed of several string fields.
204
204
  */
205
205
  message ReservationID {
206
+ // The unique ID for the reserved dataset
206
207
  DatasetID dataset_id = 1;
208
+
209
+ // The specific artifact tag for the reservation
207
210
  string tag_name = 2;
208
211
  }
209
212
 
210
213
  // Try to acquire or extend an artifact reservation. If an active reservation exists, retreive that instance.
211
214
  message GetOrExtendReservationRequest {
215
+ // The unique ID for the reservation
212
216
  ReservationID reservation_id = 1;
217
+
218
+ // The unique ID of the owner for the reservation
213
219
  string owner_id = 2;
214
- google.protobuf.Duration heartbeat_interval = 3; // Requested reservation extension heartbeat interval
220
+
221
+ // Requested reservation extension heartbeat interval
222
+ google.protobuf.Duration heartbeat_interval = 3;
215
223
  }
216
224
 
217
225
  // A reservation including owner, heartbeat interval, expiration timestamp, and various metadata.
218
226
  message Reservation {
227
+ // The unique ID for the reservation
219
228
  ReservationID reservation_id = 1;
229
+
230
+ // The unique ID of the owner for the reservation
220
231
  string owner_id = 2;
221
- google.protobuf.Duration heartbeat_interval = 3; // Recommended heartbeat interval to extend reservation
222
- google.protobuf.Timestamp expires_at = 4; // Expiration timestamp of this reservation
232
+
233
+ // Recommended heartbeat interval to extend reservation
234
+ google.protobuf.Duration heartbeat_interval = 3;
235
+
236
+ // Expiration timestamp of this reservation
237
+ google.protobuf.Timestamp expires_at = 4;
238
+
239
+ // Free-form metadata associated with the artifact
223
240
  Metadata metadata = 6;
224
241
  }
225
242
 
226
243
  // Response including either a newly minted reservation or the existing reservation
227
244
  message GetOrExtendReservationResponse {
245
+ // The reservation to be acquired or extended
228
246
  Reservation reservation = 1;
229
247
  }
230
248
 
231
249
  // Request to release reservation
232
250
  message ReleaseReservationRequest {
251
+ // The unique ID for the reservation
233
252
  ReservationID reservation_id = 1;
253
+
254
+ // The unique ID of the owner for the reservation
234
255
  string owner_id = 2;
235
256
  }
236
257
 
@@ -51,7 +51,12 @@ message NodeExecutionEvent {
51
51
  // by the executor of the node.
52
52
  google.protobuf.Timestamp occurred_at = 4;
53
53
 
54
- string input_uri = 5;
54
+ oneof input_value {
55
+ string input_uri = 5;
56
+
57
+ // Raw input data consumed by this node execution.
58
+ core.LiteralMap input_data = 20;
59
+ }
55
60
 
56
61
  oneof output_result {
57
62
  // URL to the output of the execution, it encodes all the information
@@ -165,9 +170,15 @@ message TaskExecutionEvent {
165
170
  // by the executor of the task.
166
171
  google.protobuf.Timestamp occurred_at = 7;
167
172
 
168
- // URI of the input file, it encodes all the information
169
- // including Cloud source provider. ie., s3://...
170
- string input_uri = 8;
173
+
174
+ oneof input_value {
175
+ // URI of the input file, it encodes all the information
176
+ // including Cloud source provider. ie., s3://...
177
+ string input_uri = 8;
178
+
179
+ // Raw input data consumed by this task execution.
180
+ core.LiteralMap input_data = 19;
181
+ }
171
182
 
172
183
  oneof output_result {
173
184
  // URI to the output of the execution, it will be in a format that encodes all the information
@@ -0,0 +1,41 @@
1
+ syntax = "proto3";
2
+
3
+ import "flyteidl/core/tasks.proto";
4
+
5
+ package flyteidl.plugins;
6
+
7
+ option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/plugins";
8
+
9
+
10
+ // Custom Proto for Dask Plugin.
11
+ message DaskJob {
12
+ // Spec for the scheduler pod.
13
+ DaskScheduler scheduler = 1;
14
+
15
+ // Spec of the default worker group.
16
+ DaskWorkerGroup workers = 2;
17
+ }
18
+
19
+ // Specification for the scheduler pod.
20
+ message DaskScheduler {
21
+ // Optional image to use. If unset, will use the default image.
22
+ string image = 1;
23
+
24
+ // Resources assigned to the scheduler pod.
25
+ core.Resources resources = 2;
26
+ }
27
+
28
+ message DaskWorkerGroup {
29
+ // Number of workers in the group.
30
+ uint32 number_of_workers = 1;
31
+
32
+ // Optional image to use for the pods of the worker group. If unset, will use the default image.
33
+ string image = 2;
34
+
35
+ // Resources assigned to the all pods of the worker group.
36
+ // As per https://kubernetes.dask.org/en/latest/kubecluster.html?highlight=limit#best-practices
37
+ // it is advised to only set limits. If requests are not explicitly set, the plugin will make
38
+ // sure to set requests==limits.
39
+ // The plugin sets ` --memory-limit` as well as `--nthreads` for the workers according to the limit.
40
+ core.Resources resources = 3;
41
+ }
@@ -1,6 +1,7 @@
1
1
  syntax = "proto3";
2
2
 
3
3
  package flyteidl.plugins;
4
+ import "google/protobuf/struct.proto";
4
5
 
5
6
  option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/plugins";
6
7
 
@@ -21,4 +22,13 @@ message SparkJob {
21
22
  map<string, string> sparkConf = 4;
22
23
  map<string, string> hadoopConf = 5;
23
24
  string executorPath = 6; // Executor path for Python jobs.
25
+ // Databricks job configuration.
26
+ // Config structure can be found here. https://docs.databricks.com/dev-tools/api/2.0/jobs.html#request-structure.
27
+ google.protobuf.Struct databricksConf = 7;
28
+ // Databricks access token. https://docs.databricks.com/dev-tools/api/latest/authentication.html
29
+ // This token can be set in either flytepropeller or flytekit.
30
+ string databricksToken = 8;
31
+ // Domain name of your deployment. Use the form <account>.cloud.databricks.com.
32
+ // This instance name can be set in either flytepropeller or flytekit.
33
+ string databricksInstance = 9;
24
34
  }