collabse stdout and stderr delta events into one (#1787)

codex-rs/core/src/exec.rs

@@ -22,8 +22,8 @@ use crate::error::Result;
 use crate::error::SandboxErr;
 use crate::protocol::Event;
 use crate::protocol::EventMsg;
-use crate::protocol::ExecCommandStderrDeltaEvent;
-use crate::protocol::ExecCommandStdoutDeltaEvent;
+use crate::protocol::ExecCommandOutputDeltaEvent;
+use crate::protocol::ExecOutputStream;
 use crate::protocol::SandboxPolicy;
 use crate::seatbelt::spawn_command_under_seatbelt;
 use crate::spawn::StdioPolicy;
@@ -359,17 +359,15 @@ async fn read_capped<R: AsyncRead + Unpin + Send + 'static>(
 
         if let Some(stream) = &stream {
             let chunk = tmp[..n].to_vec();
-            let msg = if is_stderr {
-                EventMsg::ExecCommandStderrDelta(ExecCommandStderrDeltaEvent {
-                    call_id: stream.call_id.clone(),
-                    chunk: ByteBuf::from(chunk),
-                })
-            } else {
-                EventMsg::ExecCommandStdoutDelta(ExecCommandStdoutDeltaEvent {
-                    call_id: stream.call_id.clone(),
-                    chunk: ByteBuf::from(chunk),
-                })
-            };
+            let msg = EventMsg::ExecCommandOutputDelta(ExecCommandOutputDeltaEvent {
+                call_id: stream.call_id.clone(),
+                stream: if is_stderr {
+                    ExecOutputStream::Stderr
+                } else {
+                    ExecOutputStream::Stdout
+                },
+                chunk: ByteBuf::from(chunk),
+            });
             let event = Event {
                 id: stream.sub_id.clone(),
                 msg,

codex-rs/core/src/protocol.rs

@@ -324,11 +324,8 @@ pub enum EventMsg {
     /// Notification that the server is about to execute a command.
     ExecCommandBegin(ExecCommandBeginEvent),
 
-    /// Incremental chunk of stdout from a running command.
-    ExecCommandStdoutDelta(ExecCommandStdoutDeltaEvent),
-
-    /// Incremental chunk of stderr from a running command.
-    ExecCommandStderrDelta(ExecCommandStderrDeltaEvent),
+    /// Incremental chunk of output from a running command.
+    ExecCommandOutputDelta(ExecCommandOutputDeltaEvent),
 
     ExecCommandEnd(ExecCommandEndEvent),
 
@@ -484,19 +481,19 @@ pub struct ExecCommandEndEvent {
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize)]
-pub struct ExecCommandStdoutDeltaEvent {
-    /// Identifier for the ExecCommandBegin that produced this chunk.
-    pub call_id: String,
-    /// Raw stdout bytes (may not be valid UTF-8).
-    #[serde(with = "serde_bytes")]
-    pub chunk: ByteBuf,
+#[serde(rename_all = "snake_case")]
+pub enum ExecOutputStream {
+    Stdout,
+    Stderr,
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize)]
-pub struct ExecCommandStderrDeltaEvent {
+pub struct ExecCommandOutputDeltaEvent {
     /// Identifier for the ExecCommandBegin that produced this chunk.
     pub call_id: String,
-    /// Raw stderr bytes (may not be valid UTF-8).
+    /// Which stream produced this chunk.
+    pub stream: ExecOutputStream,
+    /// Raw bytes from the stream (may not be valid UTF-8).
     #[serde(with = "serde_bytes")]
     pub chunk: ByteBuf,
 }