feat: add memory citation to agent message (#14821)

Client side to come

codex-rs/protocol/src/items.rs

@@ -1,3 +1,4 @@
+use crate::memory_citation::MemoryCitation;
 use crate::models::MessagePhase;
 use crate::models::WebSearchAction;
 use crate::protocol::AgentMessageEvent;
@@ -58,6 +59,9 @@ pub struct AgentMessageItem {
     #[serde(default, skip_serializing_if = "Option::is_none")]
     #[ts(optional)]
     pub phase: Option<MessagePhase>,
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    #[ts(optional)]
+    pub memory_citation: Option<MemoryCitation>,
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize, TS, JsonSchema)]
@@ -201,6 +205,7 @@ impl AgentMessageItem {
             id: uuid::Uuid::new_v4().to_string(),
             content: content.to_vec(),
             phase: None,
+            memory_citation: None,
         }
     }
 
@@ -211,6 +216,7 @@ impl AgentMessageItem {
                 AgentMessageContent::Text { text } => EventMsg::AgentMessage(AgentMessageEvent {
                     message: text.clone(),
                     phase: self.phase.clone(),
+                    memory_citation: self.memory_citation.clone(),
                 }),
             })
             .collect()

codex-rs/protocol/src/lib.rs

@@ -7,6 +7,7 @@ pub mod custom_prompts;
 pub mod dynamic_tools;
 pub mod items;
 pub mod mcp;
+pub mod memory_citation;
 pub mod message_history;
 pub mod models;
 pub mod num_format;

codex-rs/protocol/src/memory_citation.rs

@@ -0,0 +1,20 @@
+use schemars::JsonSchema;
+use serde::Deserialize;
+use serde::Serialize;
+use ts_rs::TS;
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq, JsonSchema, TS)]
+#[serde(rename_all = "camelCase")]
+pub struct MemoryCitation {
+    pub entries: Vec<MemoryCitationEntry>,
+    pub rollout_ids: Vec<String>,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, JsonSchema, TS)]
+#[serde(rename_all = "camelCase")]
+pub struct MemoryCitationEntry {
+    pub path: String,
+    pub line_start: u32,
+    pub line_end: u32,
+    pub note: String,
+}

codex-rs/protocol/src/protocol.rs

@@ -32,6 +32,7 @@ use crate::mcp::RequestId;
 use crate::mcp::Resource as McpResource;
 use crate::mcp::ResourceTemplate as McpResourceTemplate;
 use crate::mcp::Tool as McpTool;
+use crate::memory_citation::MemoryCitation;
 use crate::message_history::HistoryEntry;
 use crate::models::BaseInstructions;
 use crate::models::ContentItem;
@@ -1996,6 +1997,8 @@ pub struct AgentMessageEvent {
     pub message: String,
     #[serde(default)]
     pub phase: Option<MessagePhase>,
+    #[serde(default)]
+    pub memory_citation: Option<MemoryCitation>,
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize, JsonSchema, TS)]