feat: memories forgetting (#12900)

Add diff based memory forgetting

codex-rs/state/src/model/memories.rs

@@ -21,6 +21,21 @@ pub struct Stage1Output {
     pub generated_at: DateTime<Utc>,
 }
 
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct Stage1OutputRef {
+    pub thread_id: ThreadId,
+    pub source_updated_at: DateTime<Utc>,
+    pub rollout_slug: Option<String>,
+}
+
+#[derive(Debug, Clone, PartialEq, Eq, Default)]
+pub struct Phase2InputSelection {
+    pub selected: Vec<Stage1Output>,
+    pub previous_selected: Vec<Stage1Output>,
+    pub retained_thread_ids: Vec<ThreadId>,
+    pub removed: Vec<Stage1OutputRef>,
+}
+
 #[derive(Debug)]
 pub(crate) struct Stage1OutputRow {
     thread_id: String,
@@ -70,6 +85,18 @@ fn epoch_seconds_to_datetime(secs: i64) -> Result<DateTime<Utc>> {
         .ok_or_else(|| anyhow::anyhow!("invalid unix timestamp: {secs}"))
 }
 
+pub(crate) fn stage1_output_ref_from_parts(
+    thread_id: String,
+    source_updated_at: i64,
+    rollout_slug: Option<String>,
+) -> Result<Stage1OutputRef> {
+    Ok(Stage1OutputRef {
+        thread_id: ThreadId::try_from(thread_id)?,
+        source_updated_at: epoch_seconds_to_datetime(source_updated_at)?,
+        rollout_slug,
+    })
+}
+
 /// Result of trying to claim a stage-1 memory extraction job.
 #[derive(Debug, Clone, PartialEq, Eq)]
 pub enum Stage1JobClaimOutcome {

codex-rs/state/src/model/mod.rs

@@ -16,10 +16,12 @@ pub use backfill_state::BackfillStatus;
 pub use log::LogEntry;
 pub use log::LogQuery;
 pub use log::LogRow;
+pub use memories::Phase2InputSelection;
 pub use memories::Phase2JobClaimOutcome;
 pub use memories::Stage1JobClaim;
 pub use memories::Stage1JobClaimOutcome;
 pub use memories::Stage1Output;
+pub use memories::Stage1OutputRef;
 pub use memories::Stage1StartupClaimParams;
 pub use thread_metadata::Anchor;
 pub use thread_metadata::BackfillStats;
@@ -32,6 +34,7 @@ pub use thread_metadata::ThreadsPage;
 pub(crate) use agent_job::AgentJobItemRow;
 pub(crate) use agent_job::AgentJobRow;
 pub(crate) use memories::Stage1OutputRow;
+pub(crate) use memories::stage1_output_ref_from_parts;
 pub(crate) use thread_metadata::ThreadRow;
 pub(crate) use thread_metadata::anchor_from_item;
 pub(crate) use thread_metadata::datetime_to_epoch_seconds;