@flyteorg/flyteidl 1.5.21 → 1.10.7

package/protos/flyteidl/admin/workflow_attributes.proto

@@ -1,7 +1,7 @@
 syntax = "proto3";
 
 package flyteidl.admin;
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/admin";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/admin";
 
 import "flyteidl/admin/matchable_resource.proto";
 
@@ -18,6 +18,9 @@ message WorkflowAttributes {
     string workflow                = 3;
 
     MatchingAttributes matching_attributes = 4;
+
+    // Optional, org key applied to the attributes.
+    string org = 5;
 }
 
 // Sets custom attributes for a project, domain and workflow combination.
@@ -48,6 +51,9 @@ message WorkflowAttributesGetRequest {
     // Which type of matchable attributes to return.
     // +required
     MatchableResource resource_type = 4;
+
+    // Optional, org key applied to the attributes.
+    string org = 5;
 }
 
 // Response to get an individual workflow attribute override.
@@ -73,6 +79,9 @@ message WorkflowAttributesDeleteRequest {
     // Which type of matchable attributes to delete.
     // +required
     MatchableResource resource_type = 4;
+
+    // Optional, org key applied to the attributes.
+    string org = 5;
 }
 
 // Purposefully empty, may be populated in the future.

package/protos/flyteidl/core/artifact_id.proto

@@ -0,0 +1,90 @@
+syntax = "proto3";
+
+package flyteidl.core;
+
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
+
+import "google/protobuf/timestamp.proto";
+import "flyteidl/core/identifier.proto";
+
+
+message ArtifactKey {
+  // Project and domain and suffix needs to be unique across a given artifact store.
+  string project = 1;
+  string domain = 2;
+  string name = 3;
+}
+
+// Only valid for triggers
+message ArtifactBindingData {
+  uint32 index = 1;
+
+  // These two fields are only relevant in the partition value case
+  oneof partition_data {
+    string partition_key = 2;
+    bool bind_to_time_partition = 3;
+  }
+
+  // This is only relevant in the time partition case
+  string transform = 4;
+}
+
+message InputBindingData {
+  string var = 1;
+}
+
+message LabelValue {
+  oneof value {
+    // The string static value is for use in the Partitions object
+    string static_value = 1;
+
+    // The time value is for use in the TimePartition case
+    google.protobuf.Timestamp time_value = 2;
+    ArtifactBindingData triggered_binding = 3;
+    InputBindingData input_binding = 4;
+  }
+}
+
+message Partitions {
+  map<string, LabelValue> value = 1;
+}
+
+message TimePartition {
+  LabelValue value = 1;
+}
+
+message ArtifactID {
+  ArtifactKey artifact_key = 1;
+
+  string version = 2;
+
+  // Think of a partition as a tag on an Artifact, except it's a key-value pair.
+  // Different partitions naturally have different versions (execution ids).
+  Partitions partitions = 3;
+
+  // There is no such thing as an empty time partition - if it's not set, then there is no time partition.
+  TimePartition time_partition = 4;
+}
+
+message ArtifactTag {
+  ArtifactKey artifact_key = 1;
+
+  LabelValue value = 2;
+}
+
+// Uniqueness constraints for Artifacts
+//  - project, domain, name, version, partitions
+// Option 2 (tags are standalone, point to an individual artifact id):
+//  - project, domain, name, alias (points to one partition if partitioned)
+//  - project, domain, name, partition key, partition value
+message ArtifactQuery {
+  oneof identifier {
+    ArtifactID artifact_id = 1;
+    ArtifactTag artifact_tag = 2;
+    string uri = 3;
+
+    // This is used in the trigger case, where a user specifies a value for an input that is one of the triggering
+    // artifacts, or a partition value derived from a triggering artifact.
+    ArtifactBindingData binding = 4;
+  }
+}

package/protos/flyteidl/core/catalog.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "flyteidl/core/identifier.proto";
 
@@ -22,6 +22,8 @@ enum CatalogCacheStatus {
   CACHE_PUT_FAILURE = 5;
   // Used to indicate the cache lookup was skipped
   CACHE_SKIPPED = 6;
+  // Used to indicate that the cache was evicted
+  CACHE_EVICTED = 7;
 };
 
 message CatalogArtifactTag {

package/protos/flyteidl/core/compiler.proto

@@ -2,8 +2,10 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
+import "flyteidl/core/identifier.proto";
+import "flyteidl/core/interface.proto";
 import "flyteidl/core/workflow.proto";
 import "flyteidl/core/tasks.proto";
 
@@ -29,6 +31,12 @@ message CompiledWorkflow {
     ConnectionSet connections = 2;
 }
 
+// Output of the compilation step. This object represents one LaunchPlan. We store more metadata at this layer
+message CompiledLaunchPlan {
+    // Completely contained LaunchPlan Template
+    LaunchPlanTemplate template = 1;
+}
+
 // Output of the Compilation step. This object represent one Task. We store more metadata at this layer
 message CompiledTask {
     // Completely contained TaskTemplate
@@ -50,4 +58,7 @@ message CompiledWorkflowClosure {
     // Guaranteed that there will only exist one and only one task with a given id, i.e., every task has a unique id
     //+required (at least 1)
     repeated CompiledTask tasks = 3;
+    // A collection of launch plans that are compiled. Guaranteed that there will only exist one and only one launch plan
+    // with a given id, i.e., every launch plan has a unique id.
+    repeated CompiledLaunchPlan launch_plans = 4;
 }

package/protos/flyteidl/core/condition.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "flyteidl/core/literals.proto";
 

package/protos/flyteidl/core/dynamic_job.proto

@@ -6,7 +6,7 @@ import "flyteidl/core/literals.proto";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 // Describes a set of tasks to execute and how the final outputs are produced.
 message DynamicJobSpec {

package/protos/flyteidl/core/errors.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "flyteidl/core/execution.proto";
 

package/protos/flyteidl/core/execution.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "google/protobuf/duration.proto";
 

package/protos/flyteidl/core/identifier.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 // Indicates a resource type within Flyte.
 enum ResourceType {
@@ -33,6 +33,9 @@ message Identifier {
 
     // Specific version of the resource.
     string version              = 5;
+
+    // Optional, org key applied to the resource.
+    string org = 6;
 }
 
 // Encapsulation of fields that uniquely identifies a Flyte workflow execution
@@ -46,6 +49,9 @@ message WorkflowExecutionIdentifier {
 
     // User or system provided value for the resource.
     string name                 = 4;
+
+    // Optional, org key applied to the resource.
+    string org = 5;
 }
 
 // Encapsulation of fields that identify a Flyte node execution entity.

package/protos/flyteidl/core/interface.proto

@@ -2,10 +2,11 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "flyteidl/core/types.proto";
 import "flyteidl/core/literals.proto";
+import "flyteidl/core/artifact_id.proto";
 
 // Defines a strongly typed variable.
 message Variable {
@@ -14,6 +15,12 @@ message Variable {
 
     //+optional string describing input variable
     string description = 2;
+
+    //+optional This object allows the user to specify how Artifacts are created.
+    // name, tag, partitions can be specified. The other fields (version and project/domain) are ignored.
+    core.ArtifactID artifact_partial_id = 3;
+
+    core.ArtifactTag artifact_tag = 4;
 }
 
 // A map of Variables
@@ -41,6 +48,12 @@ message Parameter {
 
         //+optional, is this value required to be filled.
         bool required = 3;
+
+        // This is an execution time search basically that should result in exactly one Artifact with a Type that
+        // matches the type of the variable.
+        core.ArtifactQuery artifact_query = 4;
+
+        core.ArtifactID artifact_id = 5;
     }
 }
 

package/protos/flyteidl/core/literals.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "google/protobuf/timestamp.proto";
 import "google/protobuf/duration.proto";
@@ -108,6 +108,9 @@ message Literal {
     // This is used for caching purposes. For more details refer to RFC 1893
     // (https://github.com/flyteorg/flyte/blob/master/rfc/system/1893-caching-of-offloaded-objects.md)
     string hash = 4;
+
+    // Additional metadata for literals.
+    map<string, string> metadata = 5;
 }
 
 // A collection of literals. This is a workaround since oneofs in proto messages cannot contain a repeated field.

package/protos/flyteidl/core/metrics.proto

@@ -2,10 +2,11 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "flyteidl/core/identifier.proto";
 import "google/protobuf/timestamp.proto";
+import "google/protobuf/struct.proto";
 
 // Span represents a duration trace of Flyte execution. The id field denotes a Flyte execution entity or an operation
 // which uniquely identifies the Span. The spans attribute allows this Span to be further broken down into more
@@ -34,3 +35,16 @@ message Span {
     // spans defines a collection of Spans that breakdown this execution.
     repeated Span spans = 7;
 }
+
+// ExecutionMetrics is a collection of metrics that are collected during the execution of a Flyte task.
+message ExecutionMetricResult {
+    // The metric this data represents. e.g. EXECUTION_METRIC_USED_CPU_AVG or EXECUTION_METRIC_USED_MEMORY_BYTES_AVG.
+    string metric = 1;
+
+    // The result data in prometheus range query result format
+    // https://prometheus.io/docs/prometheus/latest/querying/api/#expression-query-result-formats.
+    // This may include multiple time series, differentiated by their metric labels.
+    // Start time is greater of (execution attempt start, 48h ago)
+    // End time is lesser of (execution attempt end, now)
+    google.protobuf.Struct data = 2;
+}

package/protos/flyteidl/core/security.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 // Secret encapsulates information about the secret a task needs to proceed. An environment variable
 // FLYTE_SECRETS_ENV_PREFIX will be passed to indicate the prefix of the environment variables that will be present if

package/protos/flyteidl/core/tasks.proto

@@ -9,7 +9,7 @@ import "google/protobuf/struct.proto";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 // A customizable interface to convey resources requested for a container. This can be interpreted differently for different
 // container engines.
@@ -43,6 +43,31 @@ message Resources {
     repeated ResourceEntry limits = 2;
 }
 
+// Metadata associated with the GPU accelerator to allocate to a task. Contains
+// information about device type, and for multi-instance GPUs, the partition size to
+// use.
+message GPUAccelerator {
+    // This can be any arbitrary string, and should be informed by the labels or taints
+    // associated with the nodes in question. Default cloud provider labels typically
+    // use the following values: `nvidia-tesla-t4`, `nvidia-tesla-a100`, etc.
+    string device = 1;
+    oneof partition_size_value {
+        bool unpartitioned = 2;
+        // Like `device`, this can be any arbitrary string, and should be informed by
+        // the labels or taints associated with the nodes in question. Default cloud
+        // provider labels typically use the following values: `1g.5gb`, `2g.10gb`, etc.
+        string partition_size = 3;
+    }
+}
+
+// Encapsulates all non-standard resources, not captured by v1.ResourceRequirements, to
+// allocate to a task.
+message ExtendedResources {
+    // GPU accelerator to select for task. Contains information about device type, and
+    // for multi-instance GPUs, the partition size to use.
+    GPUAccelerator gpu_accelerator = 1;
+}
+
 // Runtime information. This is loosely defined to allow for extensibility.
 message RuntimeMetadata {
     enum RuntimeType {
@@ -106,6 +131,9 @@ message TaskMetadata {
     // task creates a k8s Pod. If this value is set, the specified PodTemplate will be used instead of, but applied
     // identically as, the default PodTemplate configured in FlytePropeller.
     string pod_template_name = 12;
+
+    // cache_ignore_input_vars is the input variables that should not be included when calculating hash for cache.
+    repeated string cache_ignore_input_vars  = 13;
 }
 
 // A Task structure that uniquely identifies a task in the system
@@ -144,6 +172,10 @@ message TaskTemplate {
     // security_context encapsulates security attributes requested to run this task.
     SecurityContext security_context = 8;
 
+    // Encapsulates all non-standard resources, not captured by
+    // v1.ResourceRequirements, to allocate to a task.
+    ExtendedResources extended_resources = 9;
+
     // Metadata about the custom defined for this task. This is extensible to allow various plugins in the system
     // to use as required.
     // reserve the field numbers 1 through 15 for very frequently occurring message elements

package/protos/flyteidl/core/types.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "google/protobuf/struct.proto";
 
@@ -114,6 +114,12 @@ message UnionType {
 message TypeStructure {
     // Must exactly match for types to be castable
     string tag = 1;
+    // dataclass_type only exists for dataclasses.
+    // This is used to resolve the type of the fields of dataclass
+    // The key is the field name, and the value is the literal type of the field
+    // e.g. For dataclass Foo, with fields a, and a is a string
+    // Foo.a will be resolved as a literal type of string from dataclass_type
+    map<string, LiteralType> dataclass_type = 2;
 }
 
 // TypeAnnotation encapsulates registration time information about a type. This can be used for various control-plane operations. TypeAnnotation will not be available at runtime when a task runs.

package/protos/flyteidl/core/workflow.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "flyteidl/core/condition.proto";
 import "flyteidl/core/execution.proto";
@@ -149,6 +149,20 @@ message NodeMetadata {
         bool interruptible = 6;
     };
 
+    // Identify whether a node should have it's outputs cached.
+    oneof cacheable_value {
+        bool cacheable = 7;
+    }
+
+    // The version of the cache to use.
+    oneof cache_version_value {
+        string cache_version = 8;
+    }
+
+    // Identify whether caching operations involving this node should be serialized.
+    oneof cache_serializable_value {
+        bool cache_serializable = 9;
+    }
 }
 
 // Links a variable to an alias.
@@ -279,4 +293,17 @@ message WorkflowTemplate {
 message TaskNodeOverrides {
     // A customizable interface to convey resources requested for a task container. 
     Resources resources = 1;
+
+    // Overrides for all non-standard resources, not captured by
+    // v1.ResourceRequirements, to allocate to a task.
+    ExtendedResources extended_resources = 2;
+}
+
+// A structure that uniquely identifies a launch plan in the system.
+message LaunchPlanTemplate {
+    // A globally unique identifier for the launch plan.
+    Identifier id = 1;
+
+    // The input and output interface for the launch plan
+    TypedInterface interface = 2;
 }

package/protos/flyteidl/core/workflow_closure.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.core;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core";
 
 import "flyteidl/core/workflow.proto";
 import "flyteidl/core/tasks.proto";

package/protos/flyteidl/datacatalog/datacatalog.proto

@@ -6,7 +6,7 @@ import "flyteidl/core/literals.proto";
 import "google/protobuf/duration.proto";
 import "google/protobuf/timestamp.proto";
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/datacatalog";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/datacatalog";
 
 /*
  * Data Catalog service definition
@@ -189,6 +189,9 @@ message UpdateArtifactRequest {
     // List of data to overwrite stored artifact data with. Must contain ALL data for updated Artifact as any missing
     // ArtifactData entries will be removed from the underlying blob storage and database.
     repeated ArtifactData data = 4;
+
+    // Update execution metadata(including execution domain, name, node, project data) when overwriting cache
+    Metadata metadata = 5;
 }
 
 /*
@@ -210,7 +213,7 @@ message ReservationID {
     string tag_name = 2;
 }
 
-// Try to acquire or extend an artifact reservation. If an active reservation exists, retreive that instance.
+// Try to acquire or extend an artifact reservation. If an active reservation exists, retrieve that instance.
 message GetOrExtendReservationRequest {
     // The unique ID for the reservation
     ReservationID reservation_id = 1;
@@ -286,6 +289,9 @@ message DatasetID {
     string domain = 3;   // The domain (eg. environment)
     string version = 4;  // Version of the data schema
     string UUID = 5;     // UUID for the dataset (if set the above fields are optional)
+
+    // Optional, org key applied to the resource.
+    string org = 6;
 }
 
 /*
@@ -383,6 +389,8 @@ message DatasetPropertyFilter {
         string name = 2;
         string domain = 3;
         string version = 4;
+        // Optional, org key applied to the dataset.
+        string org = 5;
     }
 }
 

package/protos/flyteidl/event/cloudevents.proto

@@ -0,0 +1,73 @@
+syntax = "proto3";
+
+package flyteidl.event;
+
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/event";
+
+import "flyteidl/event/event.proto";
+import "flyteidl/core/literals.proto";
+import "flyteidl/core/interface.proto";
+import "flyteidl/core/artifact_id.proto";
+import "flyteidl/core/identifier.proto";
+import "google/protobuf/timestamp.proto";
+
+// This is the cloud event parallel to the raw WorkflowExecutionEvent message. It's filled in with additional
+// information that downstream consumers may find useful.
+message CloudEventWorkflowExecution {
+  event.WorkflowExecutionEvent raw_event = 1;
+
+  core.TypedInterface output_interface = 2;
+
+  // The following are ExecutionMetadata fields
+  // We can't have the ExecutionMetadata object directly because of import cycle
+  repeated core.ArtifactID artifact_ids = 3;
+  core.WorkflowExecutionIdentifier reference_execution = 4;
+  string principal = 5;
+
+  // The ID of the LP that generated the execution that generated the Artifact.
+  // Here for provenance information.
+  // Launch plan IDs are easier to get than workflow IDs so we'll use these for now.
+  core.Identifier launch_plan_id = 6;
+}
+
+message CloudEventNodeExecution {
+  event.NodeExecutionEvent raw_event = 1;
+
+  // The relevant task execution if applicable
+  core.TaskExecutionIdentifier task_exec_id = 2;
+
+  // The typed interface for the task that produced the event.
+  core.TypedInterface output_interface = 3;
+
+  // The following are ExecutionMetadata fields
+  // We can't have the ExecutionMetadata object directly because of import cycle
+  repeated core.ArtifactID artifact_ids = 4;
+  string principal = 5;
+
+  // The ID of the LP that generated the execution that generated the Artifact.
+  // Here for provenance information.
+  // Launch plan IDs are easier to get than workflow IDs so we'll use these for now.
+  core.Identifier launch_plan_id = 6;
+}
+
+message CloudEventTaskExecution {
+  event.TaskExecutionEvent raw_event = 1;
+}
+
+// This event is to be sent by Admin after it creates an execution.
+message CloudEventExecutionStart {
+  // The execution created.
+  core.WorkflowExecutionIdentifier execution_id = 1;
+  // The launch plan used.
+  core.Identifier launch_plan_id = 2;
+
+  core.Identifier workflow_id = 3;
+
+  // Artifact inputs to the workflow execution for which we have the full Artifact ID. These are likely the result of artifact queries that are run.
+  repeated core.ArtifactID artifact_ids = 4;
+
+  // Artifact inputs to the workflow execution for which we only have the tracking bit that's installed into the Literal's metadata by the Artifact service.
+  repeated string artifact_trackers = 5;
+
+  string principal = 6;
+}

package/protos/flyteidl/event/event.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.event;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/event";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/event";
 
 import "flyteidl/core/literals.proto";
 import "flyteidl/core/compiler.proto";
@@ -12,6 +12,7 @@ import "flyteidl/core/catalog.proto";
 import "google/protobuf/timestamp.proto";
 import "google/protobuf/struct.proto";
 
+
 message WorkflowExecutionEvent {
     // Workflow execution id
     core.WorkflowExecutionIdentifier execution_id = 1;
@@ -110,6 +111,9 @@ message NodeExecutionEvent {
     // literal inputs are initially copied. The event however will not be sent until after the copy completes.
     // Extracting both of these timestamps facilitates a more accurate portrayal of the evaluation time-series.
     google.protobuf.Timestamp reported_at = 21;
+
+    // Indicates if this node is an ArrayNode.
+    bool is_array = 22;
 }
 
 // For Workflow Nodes we need to send information about the workflow that's launched

package/protos/flyteidl/plugins/array_job.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.plugins;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/plugins";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins";
 
 // Describes a job that can process independent pieces of data concurrently. Multiple copies of the runnable component
 // will be executed concurrently.

package/protos/flyteidl/plugins/dask.proto

@@ -4,7 +4,7 @@ import "flyteidl/core/tasks.proto";
 
 package flyteidl.plugins;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/plugins";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins";
 
 
 // Custom Proto for Dask Plugin.

package/protos/flyteidl/plugins/kubeflow/common.proto

@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package flyteidl.plugins.kubeflow;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/plugins";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/plugins";
 
 
 enum RestartPolicy {