gemini_memory_rs/

engine.rs

//! The memory engine facade.
//!
//! [`MemoryEngine`] owns everything that outlives a conversation — the
//! repository, the compiled index, the event log, the extractor seams — and
//! [`MemorySession`] owns everything that does not: the candidate ledger, the
//! overlay, the turn counter, and the prepared snapshot the current turn is
//! being answered from.
//!
//! The split is the architecture in miniature. A session may end abruptly at
//! any moment; nothing it holds is authoritative until reconciliation writes it
//! through the engine.

use chrono::Utc;
use parking_lot::RwLock;
use std::sync::Arc;
use std::time::Duration;

use crate::bm25::{IndexedMemory, MemoryIndex};
use crate::core::{
    MemoryError, MemoryEvent, MemoryEventLog, MemoryRuntimeConfig, MemoryStatus, MutationIntent,
    SessionEventWriter, SessionId, TurnId, UserId,
};
use crate::ingestion::{
    BoundedObservationExtractor, CadenceTracker, InMemorySessionLedger, LedgerOutcome,
    MemoryObservationExtractor, ObservationExtractionContext, RuleBasedObservationExtractor,
    ScheduledWork, SessionLedger, SessionMemoryOverlay,
};
use crate::okf::{MemoryRepository, OkfRepository};
use crate::reconcile::{MemoryCommitter, ReconciliationReport, consolidate};
use crate::retrieval::{
    DeterministicPlanExtractor, DeterministicPlanner, IndexHandle, KnownEntities,
    LocalMemoryRetriever, MemoryRetriever, PreparedMemorySnapshot, RetrievalBudget,
    RetrievalPlanExtractor, RetrievalRequest, SemanticFallback, context_for, fuse_snapshots,
};
use crate::transcript::GenerationGuard;

/// How long a model-backed retrieval plan may take before the rule-based plan
/// is used instead.
pub const PLAN_EXTRACTION_TIMEOUT_MS: u64 = 4_000;

/// Everything that outlives a conversation.
pub struct MemoryEngine {
    user: UserId,
    config: MemoryRuntimeConfig,
    repository: Arc<dyn MemoryRepository>,
    canonical: Arc<IndexHandle>,
    planner: Arc<RwLock<Arc<DeterministicPlanner>>>,
    events: Arc<dyn MemoryEventLog>,
    plan_extractor: Arc<RwLock<Arc<dyn RetrievalPlanExtractor>>>,
    /// Whether the caller installed their own plan extractor.
    caller_supplied_extractor: Arc<std::sync::atomic::AtomicBool>,
    observation_extractor: Arc<dyn MemoryObservationExtractor>,
    /// The optional paraphrase-tolerant backend, shared by every session.
    semantic: Option<Arc<dyn SemanticFallback>>,
}

impl MemoryEngine {
    /// An engine backed entirely by in-process storage.
    ///
    /// Suitable for tests, single-node deployments and local development; swap
    /// the repository and event log for durable ones in production.
    pub fn in_memory(user: UserId) -> Self {
        Self::new(
            user,
            Arc::new(OkfRepository::in_memory()),
            Arc::new(crate::core::InMemoryEventLog::new()),
            MemoryRuntimeConfig::default(),
        )
    }

    /// An engine over the given repository and event log.
    pub fn new(
        user: UserId,
        repository: Arc<dyn MemoryRepository>,
        events: Arc<dyn MemoryEventLog>,
        config: MemoryRuntimeConfig,
    ) -> Self {
        let planner = Arc::new(DeterministicPlanner::new());
        Self {
            user,
            config,
            repository,
            canonical: Arc::new(IndexHandle::new()),
            planner: Arc::new(RwLock::new(planner.clone())),
            events,
            plan_extractor: Arc::new(RwLock::new(Arc::new(DeterministicPlanExtractor::new(
                planner,
            )))),
            caller_supplied_extractor: Arc::new(std::sync::atomic::AtomicBool::new(false)),
            observation_extractor: Arc::new(RuleBasedObservationExtractor::new()),
            semantic: None,
        }
    }

    /// Attach a paraphrase-tolerant retrieval backend.
    ///
    /// Every session opened afterwards consults it. Without this the seam was
    /// unreachable in practice: `LocalMemoryRetriever::with_semantic_fallback`
    /// existed, but [`MemoryEngine::begin_session`] built the retriever itself
    /// and never called it, so no application could install one.
    ///
    /// The backend is asked on both the speculative and the tool path, under
    /// the deadlines in [`crate::core::RetrievalConfig`]. A backend slower than
    /// `immediate_semantic_timeout_ms` simply times out on the tool path and
    /// still contributes through speculation.
    pub fn with_semantic_fallback(mut self, fallback: Arc<dyn SemanticFallback>) -> Self {
        self.semantic = Some(fallback);
        self
    }

    /// Use a model-backed retrieval-plan extractor.
    ///
    /// The extractor is wrapped in a deadline that falls back to the rule-based
    /// plan, so an unavailable model degrades retrieval quality rather than
    /// stalling the pipeline.
    pub fn with_plan_extractor(self, extractor: Arc<dyn RetrievalPlanExtractor>) -> Self {
        // Measured: a constrained-decode plan call is ~2s on gemini-2.5-flash.
        // The previous 500ms bound meant the model's plan was *never* used —
        // every query silently fell back to the rule-based planner, which is
        // English-only, so a Hinglish question retrieved nothing at all.
        // Planning is speculative and off the response path; a generous bound
        // costs nothing a user can perceive.
        *self.plan_extractor.write() = Arc::new(crate::retrieval::BoundedPlanExtractor::new(
            extractor,
            Duration::from_millis(PLAN_EXTRACTION_TIMEOUT_MS),
        ));
        self.caller_supplied_extractor
            .store(true, std::sync::atomic::Ordering::Release);
        self
    }

    /// Adopt a refreshed planner, and rebuild the default extractor around it.
    ///
    /// The default extractor *owns* a planner rather than reading the shared
    /// one, so refreshing only the shared handle would leave it planning
    /// against a stale entity table — and a query justified solely by a known
    /// entity would wrongly skip memory. A caller-supplied extractor is left
    /// alone; it is not ours to replace.
    fn install_planner(&self, planner: Arc<DeterministicPlanner>) {
        *self.planner.write() = planner.clone();
        if !self
            .caller_supplied_extractor
            .load(std::sync::atomic::Ordering::Acquire)
        {
            *self.plan_extractor.write() = Arc::new(DeterministicPlanExtractor::new(planner));
        }
    }

    /// Use a model-backed observation extractor, under the configured deadline.
    pub fn with_observation_extractor(
        mut self,
        extractor: Arc<dyn MemoryObservationExtractor>,
    ) -> Self {
        self.observation_extractor = Arc::new(BoundedObservationExtractor::new(
            extractor,
            Duration::from_millis(self.config.ingestion.extraction_soft_timeout_ms),
        ));
        self
    }

    /// The user whose memory this engine serves.
    pub fn user(&self) -> &UserId {
        &self.user
    }

    /// The canonical repository.
    pub fn repository(&self) -> &Arc<dyn MemoryRepository> {
        &self.repository
    }

    /// The event log.
    pub fn events(&self) -> &Arc<dyn MemoryEventLog> {
        &self.events
    }

    /// Runtime configuration.
    pub fn config(&self) -> &MemoryRuntimeConfig {
        &self.config
    }

    /// Compile the retrieval index from the canonical corpus.
    ///
    /// The index is derived and disposable; this rebuilds it from scratch, which
    /// is also the recovery path after any index corruption or schema change.
    pub async fn compile_index(&self) -> Result<u64, MemoryError> {
        let records = self.repository.all(&self.user).await?;
        let index = MemoryIndex::build(
            records
                .iter()
                .filter(|m| m.status == MemoryStatus::Active)
                .map(IndexedMemory::from_canonical),
        );

        let known = KnownEntities::from_index(&index);
        self.canonical.replace(index);
        self.install_planner(Arc::new(DeterministicPlanner::with_entities(known)));

        // The semantic backend is brought along, for the same reason
        // `MemorySession::recompile_canonical` brings it along: this function's
        // whole job is to make the retrieval layer agree with the repository,
        // and a semantic index left behind is a retrieval layer that does not.
        //
        // Without this the two compile paths disagreed. A session finishing a
        // turn reconciled the backend; an engine compiling its index did not —
        // so the startup sequence this crate documents (`restore` a persisted
        // index, then `compile_index`) produced a lexical index in step with
        // the corpus and a semantic one that could be missing every record
        // added since the vectors were last written, until some later session
        // happened to reconcile.
        //
        // Failure is swallowed for the reason spelled out on the session path:
        // the durable record and the lexical index are already correct here, so
        // failing the compile because a vector store was unreachable would turn
        // a degraded semantic layer into a failed refresh. `semantic_ranking`
        // drops ids that no longer resolve, so a backend left behind loses
        // facts rather than serving wrong ones.
        let revision = self.canonical.revision();
        if let Some(semantic) = &self.semantic {
            let active: Vec<(crate::core::MemoryId, String)> = records
                .iter()
                .filter(|m| m.status == MemoryStatus::Active)
                .map(|m| (m.id.clone(), crate::retrieval::embedding_text(m)))
                .collect();
            let _ = semantic.reconcile(&active, revision).await;
        }

        let writer = SessionEventWriter::new(
            self.events.clone(),
            self.user.clone(),
            SessionId::new("index"),
        );
        let _ = writer
            .append(None, MemoryEvent::IndexRevisionPublished { revision })
            .await;
        Ok(revision)
    }

    /// Begin a logical conversation.
    pub fn begin_session(&self, session_id: SessionId) -> MemorySession {
        let overlay_handle = Arc::new(IndexHandle::new());
        let mut retriever = LocalMemoryRetriever::new(
            self.canonical.clone(),
            overlay_handle.clone(),
            self.config.retrieval.clone(),
        );
        if let Some(semantic) = &self.semantic {
            retriever = retriever.with_semantic_fallback(semantic.clone());
        }
        let retriever = Arc::new(retriever);

        MemorySession {
            memory_map: RwLock::new((u64::MAX, String::new())),
            user: self.user.clone(),
            session_id: session_id.clone(),
            config: self.config.clone(),
            repository: self.repository.clone(),
            ledger: Arc::new(InMemorySessionLedger::new(
                session_id.clone(),
                self.config.ingestion.clone(),
            )),
            overlay: RwLock::new(SessionMemoryOverlay::new()),
            overlay_handle,
            retriever,
            planner: self.planner.clone(),
            plan_extractor: self.plan_extractor.clone(),
            caller_supplied_extractor: self.caller_supplied_extractor.clone(),
            observation_extractor: self.observation_extractor.clone(),
            generation: GenerationGuard::new(),
            current_turn: RwLock::new(TurnId::ZERO),
            prepared: RwLock::new(PreparedMemorySnapshot::empty(TurnId::ZERO)),
            active: RwLock::new(PreparedMemorySnapshot::empty(TurnId::ZERO)),
            cadence: RwLock::new(CadenceTracker::new(&self.config, Utc::now())),
            events: SessionEventWriter::new(self.events.clone(), self.user.clone(), session_id),
            canonical: self.canonical.clone(),
        }
    }

    /// Run a promotion sweep over staged patterns.
    pub async fn promote_patterns(&self) -> Result<usize, MemoryError> {
        let records = self.repository.all(&self.user).await?;
        let outcomes =
            crate::reconcile::sweep(&records, &self.config.pattern_promotion, Utc::now());
        let promoted = outcomes
            .iter()
            .filter(|o| matches!(o, crate::reconcile::PromotionOutcome::Promote(_)))
            .count();
        if outcomes.is_empty() {
            return Ok(0);
        }
        crate::reconcile::commit_promotions(
            &self.repository,
            &self.user,
            outcomes,
            &format!("promotion-{}", Utc::now().timestamp()),
        )
        .await?;
        self.compile_index().await?;
        Ok(promoted)
    }
}

/// Everything scoped to one logical conversation.
pub struct MemorySession {
    user: UserId,
    session_id: SessionId,
    config: MemoryRuntimeConfig,
    repository: Arc<dyn MemoryRepository>,
    ledger: Arc<InMemorySessionLedger>,
    overlay: RwLock<SessionMemoryOverlay>,
    overlay_handle: Arc<IndexHandle>,
    retriever: Arc<LocalMemoryRetriever>,
    planner: Arc<RwLock<Arc<DeterministicPlanner>>>,
    plan_extractor: Arc<RwLock<Arc<dyn RetrievalPlanExtractor>>>,
    caller_supplied_extractor: Arc<std::sync::atomic::AtomicBool>,
    observation_extractor: Arc<dyn MemoryObservationExtractor>,
    generation: GenerationGuard,
    /// The turn currently in flight.
    ///
    /// Distinct from the active snapshot's `source_turn_id`, which names the
    /// turn the snapshot was *prepared from* — one turn behind, and zero before
    /// any preparation has happened. Stamping a memory command with that would
    /// corrupt its provenance and the last-seen ordering that picks between
    /// competing session candidates.
    current_turn: RwLock<TurnId>,
    prepared: RwLock<PreparedMemorySnapshot>,
    active: RwLock<PreparedMemorySnapshot>,
    cadence: RwLock<CadenceTracker>,
    events: SessionEventWriter,
    canonical: Arc<IndexHandle>,
    /// The memory map, and the index revision it was built from.
    ///
    /// Cached because the map is read once per turn by the instruction
    /// amendment, and rebuilding it is a pass over every record — at 16,000
    /// records that is real work to repeat for an answer that only changes when
    /// the corpus does.
    memory_map: RwLock<(u64, String)>,
}

impl MemorySession {
    /// The logical conversation this session represents.
    pub fn session_id(&self) -> &SessionId {
        &self.session_id
    }

    /// The candidate ledger.
    pub fn ledger(&self) -> &Arc<InMemorySessionLedger> {
        &self.ledger
    }

    /// The retriever serving this session.
    pub fn retriever(&self) -> &Arc<LocalMemoryRetriever> {
        &self.retriever
    }

    /// The vocabulary a model needs in order to fill `recall_context`'s
    /// `about` and `attribute` fields.
    ///
    /// Put this in the system instruction. Without it a model names the right
    /// predicate 2% of the time, which is below the 8% at which filtering
    /// starts paying for itself; with it, 69%. It is a few hundred tokens and
    /// bounded by the user's vocabulary rather than by how much they have
    /// accumulated — 282 tokens at 16,000 records. See
    /// [`crate::retrieval::vocabulary`].
    ///
    /// Recomputed only when the canonical index has moved, so calling this
    /// every turn is cheap.
    pub fn memory_map(&self) -> String {
        let revision = self.canonical.revision();
        {
            let cached = self.memory_map.read();
            if cached.0 == revision {
                return cached.1.clone();
            }
        }
        let fresh = crate::retrieval::vocabulary::memory_map_from_index(
            &self.canonical.read(),
            crate::retrieval::MEMORY_MAP_LIMIT,
        );
        *self.memory_map.write() = (revision, fresh.clone());
        fresh
    }

    /// The runtime configuration this session was opened with.
    pub fn config(&self) -> &MemoryRuntimeConfig {
        &self.config
    }

    /// Freeze the prepared snapshot for a turn that is starting.
    ///
    /// Everything the model is answered with for this turn comes from the
    /// snapshot taken here, so a retrieval landing mid-response cannot change
    /// what B appears to remember halfway through a sentence.
    pub fn begin_turn(&self, turn_id: TurnId) -> u64 {
        let prepared = self.prepared.read().clone();
        *self.active.write() = prepared;
        *self.current_turn.write() = turn_id;
        self.cadence.write().touch(Utc::now());
        self.generation.advance()
    }

    /// The turn currently in flight.
    pub fn current_turn(&self) -> TurnId {
        *self.current_turn.read()
    }

    /// The snapshot the current turn is being answered from.
    pub fn active_snapshot(&self) -> PreparedMemorySnapshot {
        self.active.read().clone()
    }

    /// The most recently prepared snapshot, whether or not a turn is using it.
    pub fn prepared_snapshot(&self) -> PreparedMemorySnapshot {
        self.prepared.read().clone()
    }

    /// The generation guard, for cancelling stale speculative work.
    pub fn generation(&self) -> &GenerationGuard {
        &self.generation
    }

    /// Prepare context speculatively from a transcript.
    ///
    /// Publishes the resulting snapshot only if the conversation has not moved
    /// on since the work started.
    pub async fn prepare(
        &self,
        turn_id: TurnId,
        transcript: &str,
    ) -> Result<PreparedMemorySnapshot, MemoryError> {
        let generation = self.generation.current();
        let now = Utc::now();
        let planner = self.planner.read().clone();
        let context = context_for(&planner, transcript, turn_id, generation, now);
        let extractor = self.plan_extractor.read().clone();
        let plan = extractor.extract(context).await?;

        let snapshot = self
            .retriever
            .prepare(RetrievalRequest { plan, now })
            .await?;

        if self.generation.is_current(generation) {
            *self.prepared.write() = snapshot.clone();
        }
        Ok(snapshot)
    }

    /// Serve a `recall_context` tool call.
    ///
    /// Reads the frozen snapshot when it plainly covers the query, and
    /// otherwise runs a live local search and **fuses the snapshot into it**.
    /// The search is bounded and never reaches the network.
    ///
    /// The fusion is the part worth explaining. `satisfies` decides the fast
    /// path by word overlap between the question and the prepared statements —
    /// which is the right test for "can I skip the search entirely" and the
    /// wrong one for "is this snapshot any good". Measured against snapshots
    /// that *already contained the answer*, it refused 65 of 93 paraphrased
    /// questions, and refused hardest exactly where speculation is most
    /// valuable: 0 of 6 asked in-situ, 1 of 20 needing a step of inference.
    /// Every one of those refusals threw away a correct answer and replaced it
    /// with a lexical search that could not find one.
    ///
    /// So a refusal no longer discards the snapshot; it demotes it to one
    /// ranking among two. That also gives the speculative path somewhere to
    /// deliver: it runs with a 100 ms semantic budget where this path has 10,
    /// so a remote backend's results reach the model here or nowhere.
    pub async fn recall(&self, query: &str, turn_id: TurnId) -> serde_json::Value {
        let now = Utc::now();
        let active = self.active_snapshot();
        if active.satisfies(query, now) {
            return active.to_tool_payload();
        }
        let budget = RetrievalBudget::interactive_with(&self.config.retrieval);
        match self
            .retriever
            .retrieve_immediate(query, turn_id, budget)
            .await
        {
            Ok(live) => fuse_snapshots(
                &live,
                &active,
                self.config.retrieval.max_memories,
                self.config.retrieval.max_tokens,
                now,
            )
            .to_tool_payload(),
            // Memory failure degrades to whatever was already prepared, and to
            // "nothing found" if that is empty. It never fails a turn.
            Err(_) if !active.is_empty() => active.to_tool_payload(),
            Err(_) => PreparedMemorySnapshot::empty(turn_id).to_tool_payload(),
        }
    }

    /// Serve a `recall_context` tool call restricted to a scope or narrowed by
    /// the caller's `about`/`attribute` hints.
    ///
    /// An unrestricted, unhinted recall may be answered from the frozen
    /// snapshot. Anything else runs a live local search, because the snapshot
    /// was prepared speculatively and knows neither the restriction nor the
    /// hints — serving it would answer a different question quickly.
    ///
    /// The hints only ever reorder. See
    /// [`RetrievalPlan::subject_hint`](crate::retrieval::RetrievalPlan::subject_hint)
    /// for why they must not do more than that.
    pub async fn recall_scoped(
        &self,
        query: &str,
        turn_id: TurnId,
        scope: crate::runtime::tools::RecallScope,
        about: Option<String>,
        attribute: Option<String>,
    ) -> serde_json::Value {
        let kinds = scope.kinds();
        let hinted = about.as_ref().is_some_and(|v| !v.trim().is_empty())
            || attribute.as_ref().is_some_and(|v| !v.trim().is_empty());
        if kinds.is_empty() && !hinted {
            return self.recall(query, turn_id).await;
        }
        match self
            .retriever
            .retrieve_scoped(
                query,
                turn_id,
                // The session's configuration, not the default one. A
                // deployment that sets `immediate_semantic_timeout_ms` to zero
                // means it on every recall, and a scoped or hinted question is
                // not a different kind of recall — it is the same recall with a
                // narrower plan. The unscoped path above already reads the
                // config; these two disagreeing meant a caller could disable
                // semantic work on the tool path and still have it run,
                // whenever the model happened to fill in `about` or a scope.
                RetrievalBudget::interactive_with(&self.config.retrieval),
                kinds,
                about,
                attribute,
            )
            .await
        {
            Ok(snapshot) => snapshot.to_tool_payload(),
            Err(_) => PreparedMemorySnapshot::empty(turn_id).to_tool_payload(),
        }
    }

    /// Record a finalized user turn: durable evidence, then extraction.
    ///
    /// The transcript event is appended before extraction runs, so a crash
    /// between the two loses an extraction that can be retried rather than the
    /// evidence itself.
    pub async fn observe_final_transcript(
        &self,
        turn_id: TurnId,
        transcript: &str,
    ) -> Result<Vec<LedgerOutcome>, MemoryError> {
        self.events
            .append(
                Some(turn_id),
                MemoryEvent::FinalTranscriptRecorded {
                    text: transcript.to_string(),
                },
            )
            .await?;

        let observations = match self
            .observation_extractor
            .extract(
                ObservationExtractionContext::user_turn(
                    transcript,
                    self.session_id.clone(),
                    turn_id,
                    Utc::now(),
                )
                .with_known_predicates(self.known_predicates()),
            )
            .await
        {
            Ok(observations) => observations,
            Err(error) => {
                // Degrade — a failed extraction must never fail the turn — but
                // record it. Silently returning nothing makes a broken
                // extractor indistinguishable from a quiet conversation.
                let _ = self
                    .events
                    .append(
                        Some(turn_id),
                        MemoryEvent::ExtractionFailed {
                            stage: "observation".to_string(),
                            reason: error.to_string(),
                        },
                    )
                    .await;
                Vec::new()
            }
        };

        let mut outcomes = Vec::new();
        for observation in observations {
            let fingerprint = observation.fingerprint();
            if let Some(intent) = observation.mutation_intent {
                self.events
                    .append(
                        Some(turn_id),
                        MemoryEvent::ExplicitMutationRequested {
                            intent,
                            statement: observation.canonical_statement.clone(),
                        },
                    )
                    .await?;
            }
            let outcome = self.ledger.append_observation(observation).await?;
            if let LedgerOutcome::Rejected(reason) = &outcome {
                // The real fingerprint, so the audit trail names the fact that
                // was refused rather than a placeholder.
                let _ = self
                    .events
                    .append(
                        Some(turn_id),
                        MemoryEvent::ObservationRejected {
                            fingerprint: fingerprint.clone(),
                            reason: *reason,
                        },
                    )
                    .await;
            }
            outcomes.push(outcome);
        }

        self.refresh_overlay().await;
        Ok(outcomes)
    }

    /// Complete a turn and run whatever the cadence says is now due.
    pub async fn on_turn_complete(
        &self,
        turn_id: TurnId,
    ) -> Result<Vec<ScheduledWork>, MemoryError> {
        let due = self.cadence.write().on_turn_complete(Utc::now());
        for work in &due {
            match work {
                ScheduledWork::MicroReconcile => {
                    self.ledger.micro_reconcile();
                    self.refresh_overlay().await;
                }
                ScheduledWork::Checkpoint => {
                    self.ledger.micro_reconcile();
                    self.refresh_overlay().await;
                    let turns = self.cadence.read().total_turns();
                    self.events
                        .append(turn_id.into(), MemoryEvent::SessionCheckpointed { turns })
                        .await?;
                }
                ScheduledWork::SealSession => {}
            }
        }
        Ok(due)
    }

    /// Whether the session has been idle long enough to seal.
    pub fn is_idle(&self) -> bool {
        self.cadence.read().is_idle(Utc::now())
    }

    /// Seal the session and reconcile its evidence into canonical memory.
    ///
    /// Idempotent by session id: reconciling twice commits once.
    pub async fn finish(&self) -> Result<ReconciliationReport, MemoryError> {
        self.ledger.micro_reconcile();
        let sealed = self.ledger.seal().await?;
        self.cadence.write().seal();
        self.events
            .append(
                None,
                MemoryEvent::SessionSealed {
                    candidate_count: sealed.candidates.len(),
                },
            )
            .await?;

        let output = consolidate(&sealed);
        let committer =
            MemoryCommitter::new(self.repository.clone()).with_events(self.events.clone());
        let report = committer
            .reconcile(&self.user, output, self.session_id.as_str())
            .await?;

        if !report.is_empty() {
            self.recompile_canonical().await?;
        }
        self.overlay.write().clear();
        self.overlay_handle.replace(MemoryIndex::new());
        self.retriever.invalidate_cache();
        Ok(report)
    }

    /// Apply an explicit memory command from the user.
    ///
    /// Explicit intent takes effect in the conversation immediately and commits
    /// durably afterwards. The durable event is appended before the overlay is
    /// touched, so the engine never tells a user their correction was recorded
    /// when it was not.
    pub async fn apply_explicit_command(
        &self,
        intent: MutationIntent,
        statement: &str,
        turn_id: TurnId,
    ) -> Result<serde_json::Value, MemoryError> {
        self.events
            .append(
                Some(turn_id),
                MemoryEvent::ExplicitMutationRequested {
                    intent,
                    statement: statement.to_string(),
                },
            )
            .await?;

        if intent == MutationIntent::List {
            return Ok(serde_json::json!({
                "status": "accepted",
                "operation": "list",
                "facts": self.known_statements(),
            }));
        }

        let observation = explicit_observation(
            intent,
            statement,
            self.resolve_target_predicate(intent, statement),
            self.session_id.clone(),
            turn_id,
            Utc::now(),
        );
        let outcome = self.ledger.append_observation(observation).await?;
        self.refresh_overlay().await;

        let accepted = !matches!(outcome, LedgerOutcome::Rejected(_));
        Ok(serde_json::json!({
            "status": if accepted { "accepted" } else { "refused" },
            "operation": operation_label(intent),
            "effective_in_session": accepted,
            "durable_commit": "pending",
        }))
    }

    /// The predicate the corpus already uses for the fact a command is about.
    ///
    /// An explicit "remember that…" or "correct that…" arrives as a sentence,
    /// not as a triple, and the fact it concerns already has a name in the
    /// corpus. Finding that name is what makes the command land on the record
    /// it means: reconciliation matches on `subject|predicate`, and so does the
    /// in-session suppression that hides a durable fact the user has just
    /// contradicted. A command filed under a predicate of its own supersedes
    /// nothing and hides nothing — it becomes a second, contradicting record.
    ///
    /// This is the same trick `known_predicates` plays for the extraction
    /// model, applied to the path that never asks a model anything: the
    /// vocabulary comes from the corpus rather than from a constant.
    ///
    /// Deliberately conservative. A weak match is worse than no match, because
    /// hijacking the wrong window suppresses a fact the user never mentioned, so
    /// the hit has to clear the same score floor the assembler uses before its
    /// predicate is adopted.
    fn resolve_target_predicate(
        &self,
        intent: MutationIntent,
        statement: &str,
    ) -> Option<crate::core::CanonicalPredicate> {
        use crate::bm25::Query;

        // A deletion names what to remove, not a fact to restate, and it is
        // routed by its own `memory_removal` predicate.
        if matches!(
            intent,
            MutationIntent::Forget | MutationIntent::Delete | MutationIntent::List
        ) {
            return None;
        }

        let query = Query::new(statement)
            .with_boost_only(crate::retrieval::non_topical_terms(statement))
            .with_limit(1);
        let canonical = self.canonical.read();
        let hit = canonical.search(&query, Utc::now()).into_iter().next()?;
        if hit.score < self.config.retrieval.minimum_candidate_score {
            return None;
        }
        canonical.get(&hit.id).map(|doc| doc.predicate.clone())
    }

    /// The predicate names this user's corpus already uses.
    ///
    /// Offered to the extraction model so a correction lands on the predicate
    /// it is correcting. Reconciliation matches on subject and predicate; a
    /// model free to name each fact afresh writes `dietary_preference` in one
    /// session and `dietary_identity` in the next, and "actually I'm
    /// pescatarian now" becomes a second active record rather than superseding
    /// the first. This is the entity table's trick applied to predicates: the
    /// vocabulary comes from the corpus.
    ///
    /// Bounded, because it goes in a prompt. Session candidates come first —
    /// a correction usually chases something said minutes ago.
    pub fn known_predicates(&self) -> Vec<String> {
        const LIMIT: usize = 60;
        let mut out: Vec<String> = Vec::new();
        let mut push = |predicate: &str| {
            if out.len() < LIMIT && !predicate.is_empty() && !out.iter().any(|p| p == predicate) {
                out.push(predicate.to_string());
            }
        };
        for candidate in self.ledger.usable_candidates() {
            push(candidate.predicate.as_str());
        }
        let now = Utc::now();
        for doc in self.canonical.read().documents() {
            if doc.is_retrievable(now) {
                push(doc.predicate.as_str());
            }
        }
        out
    }

    /// The predicate/value pairs memory can currently assert.
    ///
    /// Session facts shadow canonical ones: something the user said this
    /// conversation is a better answer than something recalled from months ago.
    pub fn known_values(&self) -> Vec<(crate::core::CanonicalPredicate, serde_json::Value)> {
        let mut out: Vec<(crate::core::CanonicalPredicate, serde_json::Value)> = Vec::new();
        let mut push = |predicate: crate::core::CanonicalPredicate, value: serde_json::Value| {
            if !out.iter().any(|(p, _)| p == &predicate) {
                out.push((predicate, value));
            }
        };

        for candidate in self.ledger.usable_candidates() {
            if candidate.mutation_intent == Some(MutationIntent::List) {
                continue;
            }
            push(
                candidate.predicate.clone(),
                serde_json::Value::String(candidate.value.display()),
            );
        }
        let now = Utc::now();
        for doc in self.canonical.read().documents() {
            if doc.is_retrievable(now) {
                push(
                    doc.predicate.clone(),
                    serde_json::Value::String(doc.value.clone()),
                );
            }
        }
        out
    }

    /// Every statement currently retrievable, canonical and provisional.
    pub fn known_statements(&self) -> Vec<String> {
        let mut statements: Vec<String> = self
            .canonical
            .read()
            .documents()
            .map(|d| d.statement.clone())
            .collect();
        statements.extend(
            self.ledger
                .usable_candidates()
                .iter()
                .map(|c| c.canonical_statement.clone()),
        );
        statements.sort();
        statements.dedup();
        statements
    }

    /// Everything the user has explicitly asked to be remembered this session.
    pub fn pending_explicit_commands(&self) -> Vec<MutationIntent> {
        self.ledger
            .usable_candidates()
            .iter()
            .filter_map(|c| c.mutation_intent)
            .collect()
    }

    async fn refresh_overlay(&self) {
        let candidates = self.ledger.usable_candidates();

        // Anything the user stated outright this session hides the durable
        // record it contradicts for the rest of the conversation. Reconciliation
        // will make that permanent; until then the overlay is the truth.
        let suppressed: std::collections::HashSet<String> = candidates
            .iter()
            .filter(|c| {
                c.explicitness.is_explicit() && c.mutation_intent != Some(MutationIntent::List)
            })
            .map(|c| c.subject_predicate().to_string())
            .collect();
        self.retriever.suppress_windows(suppressed);

        let revision = {
            let mut overlay = self.overlay.write();
            overlay.rebuild(&self.user, &self.session_id, &candidates, Utc::now());
            self.overlay_handle.replace(clone_index(overlay.index()));
            overlay.revision()
        };
        self.retriever.invalidate_cache();
        let _ = self
            .events
            .append(None, MemoryEvent::SessionOverlayUpdated { revision })
            .await;
    }

    async fn recompile_canonical(&self) -> Result<(), MemoryError> {
        let records = self.repository.all(&self.user).await?;
        self.canonical.replace(MemoryIndex::build(
            records
                .iter()
                .filter(|m| m.status == MemoryStatus::Active)
                .map(IndexedMemory::from_canonical),
        ));

        // The semantic backend is recompiled too, and for the same reason the
        // lexical index is: reconciliation has just decided what is true, and a
        // retriever holding the previous answer is a retriever that has stopped
        // agreeing with memory.
        //
        // It is passed the whole active set rather than a diff — see
        // `SemanticFallback::reconcile` — and a well-behaved backend embeds only
        // what it does not already hold, so the cost of a correction is one
        // embedding rather than one per record.
        //
        // A failure here is logged rather than propagated. The lexical index and
        // the durable record are already correct at this point; refusing to
        // finish a recompile because a vector store was unreachable would turn a
        // degraded semantic layer into a failed commit, and the degradation is
        // the safe direction — `semantic_ranking` drops ids that no longer
        // resolve, so a stale backend loses facts rather than serving wrong ones.
        if let Some(semantic) = self.retriever.semantic() {
            let active: Vec<(crate::core::MemoryId, String)> = records
                .iter()
                .filter(|m| m.status == MemoryStatus::Active)
                .map(|m| (m.id.clone(), crate::retrieval::embedding_text(m)))
                .collect();
            // Failure is swallowed on purpose. By this point the durable record
            // and the lexical index are already correct, so refusing to finish
            // the recompile because a vector store was unreachable would turn a
            // degraded semantic layer into a failed commit. The degradation is
            // also the safe direction: `semantic_ranking` drops ids that no
            // longer resolve, so a backend left behind loses facts rather than
            // serving wrong ones.
            let _ = semantic.reconcile(&active, self.canonical.revision()).await;
        }
        let planner = Arc::new(DeterministicPlanner::with_entities(
            KnownEntities::from_index(&self.canonical.read()),
        ));
        *self.planner.write() = planner.clone();
        if !self
            .caller_supplied_extractor
            .load(std::sync::atomic::Ordering::Acquire)
        {
            *self.plan_extractor.write() = Arc::new(DeterministicPlanExtractor::new(planner));
        }
        Ok(())
    }
}

/// Build the observation behind an explicit memory command.
///
/// Explicit commands are the one path where the engine trusts a statement
/// wholesale: the user said it about themselves, on purpose, to be remembered.
fn explicit_observation(
    intent: MutationIntent,
    statement: &str,
    target: Option<crate::core::CanonicalPredicate>,
    session_id: SessionId,
    turn_id: TurnId,
    now: chrono::DateTime<Utc>,
) -> crate::core::MemoryObservation {
    use crate::core::{
        CanonicalPredicate, EntityRef, Explicitness, MemoryKind, MemoryValue, ObservationId,
        ProposedPersistence, SensitivityClass, SpeakerAttribution, TemporalScope,
        TranscriptEvidence,
    };

    // A correction has to land on the predicate of the fact it corrects.
    //
    // Reconciliation matches on `subject|predicate`, and so does the in-session
    // suppression that hides a durable record the user has just contradicted.
    // Naming the new fact `preference` regardless of what it is about — which
    // is what this did — means the window suppressed (`user|preference`) is
    // never the window the record lives in (`user|beverage_preference`), so the
    // correction is accepted, reported as effective, and the next recall serves
    // the old value alongside the new one. `target` is the predicate the corpus
    // already uses for this fact, resolved before the observation is built.
    let predicate = match intent {
        MutationIntent::Forget | MutationIntent::Delete => "memory_removal".to_string(),
        MutationIntent::List => "memory_listing".to_string(),
        _ => target
            .map(|p| p.as_str().to_string())
            .unwrap_or_else(|| "preference".to_string()),
    };

    crate::core::MemoryObservation {
        observation_id: ObservationId::generate(),
        session_id,
        turn_id,
        subject: EntityRef::user(),
        predicate: CanonicalPredicate::new(&predicate),
        value: MemoryValue::Text(statement.to_string()),
        canonical_statement: statement.to_string(),
        kind: MemoryKind::Preference,
        explicitness: Explicitness::ExplicitCommand,
        confidence: 1.0,
        persistence: ProposedPersistence::Durable,
        temporal_scope: TemporalScope::Persistent,
        valid_from: Some(now),
        expected_expiry: None,
        transcript_evidence: TranscriptEvidence::new(statement),
        speaker_attribution: SpeakerAttribution::User,
        sensitivity: SensitivityClass::Normal,
        mutation_intent: Some(intent),
        search_terms: Vec::new(),
    }
}

fn operation_label(intent: MutationIntent) -> &'static str {
    match intent {
        MutationIntent::Remember => "remember",
        MutationIntent::Correct => "correct",
        MutationIntent::Forget => "forget",
        MutationIntent::Delete => "delete",
        MutationIntent::List => "list",
    }
}

/// Copy an index so the overlay and the retriever's handle stay independent.
///
/// The overlay owns the authoritative projection of the ledger; the handle is
/// what the retriever reads without ever blocking on a rebuild.
fn clone_index(source: &MemoryIndex) -> MemoryIndex {
    MemoryIndex::build(source.documents().cloned())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::core::InMemoryEventLog;

    fn engine() -> MemoryEngine {
        MemoryEngine::in_memory(UserId::new("usr_1"))
    }

    /// One active canonical record, for seeding a repository directly.
    fn seed_record(id: &str) -> crate::core::CanonicalMemory {
        use crate::core::{
            CanonicalPredicate, EntityRef, Explicitness, MemoryId, MemoryKind, MemorySource,
            MemoryValue, RetrievalMetadata, TemporalMetadata, TemporalScope,
        };
        crate::core::CanonicalMemory {
            id: MemoryId::new(id),
            owner: UserId::new("usr_1"),
            kind: MemoryKind::Preference,
            predicate: CanonicalPredicate::new("dietary_identity"),
            status: MemoryStatus::Active,
            confidence: 0.9,
            subject: EntityRef::named("user"),
            value: MemoryValue::Text("pescatarian".into()),
            statement: "The user is pescatarian.".into(),
            evidence_summary: "stated".into(),
            source: MemorySource::from_explicitness(
                Explicitness::ExplicitStatement,
                SessionId::new("ses_seed"),
                TurnId(1),
            ),
            temporal: TemporalMetadata::created_at(Utc::now()),
            retrieval: RetrievalMetadata {
                subject: crate::core::normalize_token("user"),
                ..Default::default()
            },
            temporal_scope: TemporalScope::Persistent,
            qualifier: None,
            evidence: Default::default(),
            privacy: Default::default(),
            supersedes: Vec::new(),
            superseded_by: None,
        }
    }

    /// A backend that records what it was asked to reconcile against.
    #[derive(Default)]
    struct RecordingSemantic {
        seen: parking_lot::Mutex<Vec<Vec<String>>>,
    }

    #[async_trait::async_trait]
    impl SemanticFallback for RecordingSemantic {
        async fn search(
            &self,
            _query: &str,
            _limit: usize,
        ) -> Result<Vec<crate::core::MemoryId>, MemoryError> {
            Ok(Vec::new())
        }
        async fn reconcile(
            &self,
            active: &[(crate::core::MemoryId, String)],
            _revision: u64,
        ) -> Result<(), MemoryError> {
            self.seen
                .lock()
                .push(active.iter().map(|(id, _)| id.to_string()).collect());
            Ok(())
        }
    }

    /// `compile_index` must bring the semantic backend along.
    ///
    /// Both compile paths exist to make retrieval agree with the repository,
    /// and only one of them used to reconcile. That left the documented startup
    /// sequence — restore a persisted index, then `compile_index` — with a
    /// lexical index in step and a semantic one that could be missing every
    /// record written since the vectors were last saved, until some later
    /// session happened to seal.
    #[tokio::test]
    async fn compiling_the_index_reconciles_the_semantic_backend() {
        let recorder = Arc::new(RecordingSemantic::default());
        let engine =
            MemoryEngine::in_memory(UserId::new("usr_1")).with_semantic_fallback(recorder.clone());

        // Seed the repository directly rather than through a session. That is
        // the situation this path exists for — a process starting against a
        // corpus it did not just write, restoring a persisted index and
        // compiling — and going through a session would let the session's own
        // reconcile do the work and prove nothing about this one.
        engine
            .repository()
            .commit(
                crate::okf::MemoryTransaction::new(engine.user().clone(), "tx-seed")
                    .put(seed_record("mem_seed")),
            )
            .await
            .expect("seed commit");

        let before = recorder.seen.lock().len();
        engine.compile_index().await.expect("compile");
        let seen = recorder.seen.lock();
        assert!(
            seen.len() > before,
            "compile_index did not reconcile the semantic backend at all"
        );
        assert!(
            !seen.last().expect("a reconcile happened").is_empty(),
            "compile_index reconciled against an empty corpus: {seen:?}"
        );
    }

    /// A scoped or hinted recall must honour the session's own deadlines.
    ///
    /// `recall` read the configured budget and `recall_scoped` built a default
    /// one, so a deployment that set `immediate_semantic_timeout_ms` to zero to
    /// keep semantic work off the tool path still got the default 10 ms
    /// whenever the model filled in a scope or a hint — which, now that the
    /// memory map tells it what the hint values are, is most of the time.
    ///
    /// Asserted by whether the backend was *consulted*, not by comparing two
    /// budget constructors: the bug was which constructor the call site used,
    /// and a test that calls the right one itself cannot see that.
    #[tokio::test]
    async fn a_zero_semantic_deadline_is_honoured_on_the_scoped_path_too() {
        #[derive(Default)]
        struct CountingSemantic {
            searches: std::sync::atomic::AtomicUsize,
        }
        #[async_trait::async_trait]
        impl SemanticFallback for CountingSemantic {
            async fn search(
                &self,
                _query: &str,
                _limit: usize,
            ) -> Result<Vec<crate::core::MemoryId>, MemoryError> {
                self.searches
                    .fetch_add(1, std::sync::atomic::Ordering::SeqCst);
                Ok(Vec::new())
            }
        }

        let mut config = MemoryRuntimeConfig::default();
        // The deployment's decision: no semantic work on the tool path.
        config.retrieval.immediate_semantic_timeout_ms = 0;

        let backend = Arc::new(CountingSemantic::default());
        let engine = MemoryEngine::new(
            UserId::new("usr_1"),
            Arc::new(OkfRepository::in_memory()),
            Arc::new(InMemoryEventLog::new()),
            config,
        )
        .with_semantic_fallback(backend.clone());
        engine
            .repository()
            .commit(
                crate::okf::MemoryTransaction::new(engine.user().clone(), "tx-seed")
                    .put(seed_record("mem_seed")),
            )
            .await
            .expect("seed commit");
        engine.compile_index().await.expect("compile");

        let session = engine.begin_session(SessionId::new("ses_1"));
        session.begin_turn(TurnId(1));
        // A hinted recall — the shape `recall_context` produces once the model
        // can name a filter value.
        let _ = session
            .recall_scoped(
                "what does the user eat",
                TurnId(1),
                crate::runtime::tools::RecallScope::All,
                Some("user".to_string()),
                None,
            )
            .await;

        assert_eq!(
            backend.searches.load(std::sync::atomic::Ordering::SeqCst),
            0,
            "the semantic backend was consulted despite a zero deadline — the \
             scoped path built a default budget instead of the configured one"
        );
    }

    #[tokio::test]
    async fn a_fact_stated_this_session_is_recalled_in_a_later_turn() {
        let engine = engine();
        let session = engine.begin_session(SessionId::new("ses_1"));

        session.begin_turn(TurnId(1));
        session
            .observe_final_transcript(TurnId(1), "I am pescatarian")
            .await
            .unwrap();
        session.on_turn_complete(TurnId(1)).await.unwrap();

        session.begin_turn(TurnId(2));
        let snapshot = session
            .prepare(
                TurnId(2),
                "what do you remember about my dietary preferences",
            )
            .await
            .unwrap();
        assert!(
            snapshot
                .facts
                .iter()
                .any(|f| f.statement.contains("pescatarian")),
            "prepared: {:?}",
            snapshot.facts
        );
    }

    #[tokio::test]
    async fn a_fact_survives_into_the_next_session_after_reconciliation() {
        let engine = engine();

        let first = engine.begin_session(SessionId::new("ses_1"));
        first.begin_turn(TurnId(1));
        first
            .observe_final_transcript(TurnId(1), "I am pescatarian")
            .await
            .unwrap();
        let report = first.finish().await.unwrap();
        assert_eq!(report.creates, 1);

        engine.compile_index().await.unwrap();

        let second = engine.begin_session(SessionId::new("ses_2"));
        second.begin_turn(TurnId(1));
        let snapshot = second
            .prepare(
                TurnId(1),
                "what do you remember about my dietary preferences",
            )
            .await
            .unwrap();
        assert!(
            snapshot
                .facts
                .iter()
                .any(|f| f.statement.contains("pescatarian"))
        );
    }

    #[tokio::test]
    async fn a_correction_stated_mid_session_takes_effect_on_the_next_turn() {
        let engine = engine();
        let session = engine.begin_session(SessionId::new("ses_1"));

        session.begin_turn(TurnId(1));
        session
            .observe_final_transcript(TurnId(1), "I am vegetarian")
            .await
            .unwrap();
        session.on_turn_complete(TurnId(1)).await.unwrap();

        session.begin_turn(TurnId(2));
        session
            .observe_final_transcript(TurnId(2), "actually I am pescatarian")
            .await
            .unwrap();
        session.on_turn_complete(TurnId(2)).await.unwrap();

        session.begin_turn(TurnId(3));
        let payload = session
            .recall("diet vegetarian pescatarian", TurnId(3))
            .await;
        let rendered = payload.to_string();
        assert!(rendered.contains("pescatarian"), "got {rendered}");
    }

    #[tokio::test]
    async fn a_correction_hides_the_durable_fact_it_contradicts_immediately() {
        let engine = engine();

        // Monday: the user is vegetarian, and it is reconciled and indexed.
        let monday = engine.begin_session(SessionId::new("ses_monday"));
        monday.begin_turn(TurnId(1));
        monday
            .observe_final_transcript(TurnId(1), "I am vegetarian")
            .await
            .unwrap();
        monday.finish().await.unwrap();
        engine.compile_index().await.unwrap();

        // Thursday: the user corrects it. The old fact must stop being
        // retrieved now, not after the session reconciles.
        let thursday = engine.begin_session(SessionId::new("ses_thursday"));
        thursday.begin_turn(TurnId(1));
        thursday
            .observe_final_transcript(TurnId(1), "actually I am pescatarian")
            .await
            .unwrap();
        thursday.on_turn_complete(TurnId(1)).await.unwrap();

        thursday.begin_turn(TurnId(2));
        let snapshot = thursday
            .prepare(
                TurnId(2),
                "what do you remember about my dietary preferences",
            )
            .await
            .unwrap();

        let statements: Vec<&str> = snapshot
            .facts
            .iter()
            .map(|f| f.statement.as_str())
            .collect();
        assert!(
            statements.iter().any(|s| s.contains("pescatarian")),
            "the correction was not recalled: {statements:?}"
        );
        assert!(
            !statements.iter().any(|s| s.contains("vegetarian")),
            "the corrected-away fact was still recalled: {statements:?}"
        );
    }

    #[tokio::test]
    async fn asking_what_is_remembered_does_not_itself_become_a_memory() {
        let engine = engine();
        let session = engine.begin_session(SessionId::new("ses_1"));
        session.begin_turn(TurnId(1));
        session
            .observe_final_transcript(TurnId(1), "I am pescatarian")
            .await
            .unwrap();
        session.begin_turn(TurnId(2));
        session
            .observe_final_transcript(TurnId(2), "what do you remember about me")
            .await
            .unwrap();

        let snapshot = session
            .prepare(
                TurnId(3),
                "what do you remember about my dietary preferences",
            )
            .await
            .unwrap();
        assert!(
            !snapshot
                .facts
                .iter()
                .any(|f| f.statement.contains("asked what is remembered")),
            "a question about memory was recalled as a memory: {:?}",
            snapshot.facts
        );
    }

    #[tokio::test]
    async fn a_generic_question_gets_no_memory_and_no_search() {
        let engine = engine();
        let session = engine.begin_session(SessionId::new("ses_1"));
        session.begin_turn(TurnId(1));
        let snapshot = session
            .prepare(TurnId(1), "what is the capital of France")
            .await
            .unwrap();
        assert!(snapshot.is_empty());
        assert_eq!(
            session.recall("capital of France", TurnId(1)).await["status"],
            "not_found"
        );
    }

    #[tokio::test]
    async fn the_active_snapshot_does_not_change_mid_turn() {
        let engine = engine();
        let session = engine.begin_session(SessionId::new("ses_1"));

        session.begin_turn(TurnId(1));
        session
            .observe_final_transcript(TurnId(1), "I am pescatarian")
            .await
            .unwrap();
        session.begin_turn(TurnId(2));
        session.prepare(TurnId(2), "what do I eat").await.unwrap();

        let during_turn = session.active_snapshot();
        // A newer preparation lands while turn 2 is still in flight.
        session
            .observe_final_transcript(TurnId(3), "I am allergic to nuts")
            .await
            .unwrap();
        session
            .prepare(TurnId(3), "what am I allergic to")
            .await
            .unwrap();

        assert_eq!(
            session.active_snapshot().snapshot_id,
            during_turn.snapshot_id,
            "the in-flight turn keeps its frozen snapshot"
        );
    }

    #[tokio::test]
    async fn bystander_speech_never_reaches_the_ledger() {
        let engine = engine();
        let session = engine.begin_session(SessionId::new("ses_1"));
        session.begin_turn(TurnId(1));
        session
            .observe_final_transcript(TurnId(1), "the weather is nice today")
            .await
            .unwrap();
        assert!(session.ledger().is_empty());
    }

    #[tokio::test]
    async fn reconciling_the_same_session_twice_writes_once() {
        let engine = engine();
        let session = engine.begin_session(SessionId::new("ses_1"));
        session.begin_turn(TurnId(1));
        session
            .observe_final_transcript(TurnId(1), "I am pescatarian")
            .await
            .unwrap();

        session.finish().await.unwrap();
        session.finish().await.unwrap();

        let stored = engine.repository().all(engine.user()).await.unwrap();
        assert_eq!(stored.len(), 1);
    }

    #[tokio::test]
    async fn the_event_log_records_the_turn_before_the_extraction() {
        let log = Arc::new(InMemoryEventLog::new());
        let engine = MemoryEngine::new(
            UserId::new("usr_1"),
            Arc::new(OkfRepository::in_memory()),
            log.clone(),
            MemoryRuntimeConfig::default(),
        );
        let session = engine.begin_session(SessionId::new("ses_1"));
        session
            .observe_final_transcript(TurnId(1), "I am pescatarian")
            .await
            .unwrap();

        let entries = log.entries();
        assert_eq!(entries[0].payload.label(), "final_transcript_recorded");
    }

    #[tokio::test]
    async fn a_staged_pattern_is_promoted_once_it_spans_sessions_and_days() {
        let engine = engine();

        // Two sessions, each stating the routine outright.
        for (idx, session_id) in ["ses_1", "ses_2"].iter().enumerate() {
            let session = engine.begin_session(SessionId::new(*session_id));
            session.begin_turn(TurnId(1));
            session
                .observe_final_transcript(TurnId(1), "I always go to the gym before work")
                .await
                .unwrap();
            session.finish().await.unwrap();
            let _ = idx;
        }

        let stored = engine.repository().all(engine.user()).await.unwrap();
        let routine = stored
            .iter()
            .find(|m| m.predicate.as_str() == "exercise_routine")
            .expect("the routine was recorded");
        assert_eq!(routine.evidence.distinct_sessions, 2);
    }
}