fix: make pod_name and container_name optional in LoggingContext; add ContainerIdentifier to TailLogsResponse

flyteidl2/dataproxy/dataproxy_service.proto

@@ -203,8 +203,15 @@ message TailLogsRequest {
   // The attempt number.
   uint32 attempt = 2 [(buf.validate.field).uint32.gt = 0];
 
-  // The pod name to tail logs from. If not provided, attempt to find the primary pod, else assume the first pod.
-  string pod_name = 3;
+  // Selects which pod(s) to tail. If unset, defaults to primary_pod.
+  oneof pod_filter {
+    // Tail logs from the primary pod only. This is the default when pod_filter is unset.
+    bool primary_pod = 6;
+    // Tail logs from all pods.
+    bool all_pods = 5;
+    // Tail logs from the named pod only.
+    string pod_name = 3;
+  }
 
   // +optional. The dataproxy uses this to stream logs from the connector via
   // AsyncConnectorService.GetTaskLogs instead of pod logs.
@@ -217,6 +224,8 @@ message TailLogsResponse {
   message Logs {
     // Structured log lines.
     repeated flyteidl2.logs.dataplane.LogLine lines = 1;
+    // Which container these lines are from.
+    flyteidl2.logs.dataplane.ContainerIdentifier container = 2;
   }
 
   // One or more batches of logs.

flyteidl2/logs/dataplane/payload.proto

@@ -34,6 +34,10 @@ message LoggingContext {
   map<string, string> kubernetes_pod_labels = 9;
   common.Sort.Direction order = 10;
   uint64 number_of_batches = 11;
+  // Attempt number for the execution. Used by log sources to compute pod uniqueIDs
+  // when pod names are hashed (e.g. CloudWatch stream name prefix narrowing).
+  // +optional, defaults to 1
+  uint32 execution_attempt = 12;
 }
 
 // Parameters of environment in which logs were collected. Should contain everything

flyteidl2/workflow/run_service.proto

@@ -236,6 +236,12 @@ message GetActionLogContextResponse {
 
   // The cluster where the action attempt is running.
   string cluster = 2;
+
+  // The start time of the action attempt.
+  google.protobuf.Timestamp start_time = 3;
+
+  // The end time of the action attempt. Unset if the attempt is still running.
+  google.protobuf.Timestamp end_time = 4;
 }
 
 // Request message for listing runs.