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 all_pods.
+  oneof pod_selector {
+    // Tail logs from the primary pod only.
+    bool primary_pod = 3;
+    // Tail logs from all pods. This is the default when pod_selector is unset.
+    bool all_pods = 4;
+    // Tail logs from the named pod only.
+    string pod_name = 5;
+  }
 }
 
 // Reponse message for tailing logs.
@@ -213,6 +220,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

@@ -27,8 +27,8 @@ message LoggingContext {
 
   string cluster_name = 3 [(buf.validate.field).string.min_len = 1];
   string kubernetes_namespace = 4 [(buf.validate.field).string.min_len = 1];
-  string kubernetes_pod_name = 5 [(buf.validate.field).string.min_len = 1];
-  string kubernetes_container_name = 6 [(buf.validate.field).string.min_len = 1];
+  string kubernetes_pod_name = 5;
+  string kubernetes_container_name = 6;
   google.protobuf.Timestamp execution_attempt_start_time = 7;
   google.protobuf.Timestamp execution_attempt_end_time = 8;
   map<string, string> kubernetes_pod_labels = 9;