From 4b1ace9a54e4e0644d93203a0b6d3104b99caf1b Mon Sep 17 00:00:00 2001
From: Agus Zubiaga <agus@zed.dev>
Date: Mon, 21 Jul 2025 18:29:13 -0300
Subject: [PATCH] Handle Agent Reasoning

---
 crates/agent_servers/src/codex.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/crates/agent_servers/src/codex.rs b/crates/agent_servers/src/codex.rs
index 519954620d83fc1570fcc6262ff36dfcd54818a9..9e301b33fcbb2bd2f725a2a56e565f69ef5f1313 100644
--- a/crates/agent_servers/src/codex.rs
+++ b/crates/agent_servers/src/codex.rs
@@ -233,6 +233,15 @@ impl CodexAgentConnection {
                     })
                     .await?;
             }
+            AcpNotification::AgentReasoning(message) => {
+                delegate
+                    .stream_assistant_message_chunk(acp::StreamAssistantMessageChunkParams {
+                        chunk: acp::AssistantMessageChunk::Thought {
+                            thought: message.text,
+                        },
+                    })
+                    .await?
+            }
             AcpNotification::Other => {}
         }
 
@@ -258,6 +267,7 @@ struct CodexEvent {
 #[serde(tag = "type", rename_all = "snake_case")]
 pub enum AcpNotification {
     AgentMessage(AgentMessageEvent),
+    AgentReasoning(AgentReasoningEvent),
     #[serde(other)]
     Other,
 }
@@ -266,3 +276,8 @@ pub enum AcpNotification {
 pub struct AgentMessageEvent {
     pub message: String,
 }
+
+#[derive(Debug, Clone, Deserialize, Serialize)]
+pub struct AgentReasoningEvent {
+    pub text: String,
+}