khive_runtime/

reference_resolution.rs

//! `resolve_reference`: the Layer-0 deterministic reference resolver from the
//! "unified-verb" draft ADR (Slice 1 — resolver + ring).
//!
//! Turns a natural-language reference into an id through four ordered
//! stages, never guessing among close candidates:
//!
//! 1. **Id-string passthrough.** A ref that already looks like a UUID or an
//!    8+ hex-char prefix resolves through the existing by-ID path
//!    (`KhiveRuntime::resolve_by_id` / `resolve_prefix_unfiltered`) instead of
//!    being treated as free text — it must not error just because it arrived
//!    through `resolve_reference` rather than `get`. Scoped to entity ids
//!    only, identically for the full-UUID and prefix forms (matching the
//!    ring's entity-only contract); a note/edge/event id-string is
//!    `NotFound` here, not an error — a caller resolving those uses `get`.
//! 2. **Recently-referenced ring.** An exact (case-insensitive) or substring
//!    match against this actor's ring (`reference_ring::ReferenceRing`).
//! 3. **Exact-name storage lookup.** A deterministic, case-sensitive match
//!    against `entities.name` in the caller's namespace (`deleted_at IS
//!    NULL`) — covers any entity that already exists but was never
//!    created/get/updated/deleted/merged/linked by this actor in this
//!    session, so stage 2's ring never saw it (#849).
//! 4. **Hybrid-search fallback.** `KhiveRuntime::hybrid_search` over the
//!    caller's namespace, ranked by RRF score.
//!
//! A single candidate clearing the stage's confidence bar resolves; multiple
//! viable candidates or none never silently pick — they return `Ambiguous`
//! or `NotFound` for the caller to disambiguate.

use std::str::FromStr;

use uuid::Uuid;

use khive_storage::types::PageRequest;
use khive_storage::EntityFilter;

use crate::error::{RuntimeError, RuntimeResult};
use crate::operations::Resolved;
use crate::reference_ring::ReferenceRing;
use crate::runtime::{KhiveRuntime, NamespaceToken};

/// A candidate id surfaced when a reference did not resolve outright.
#[derive(Clone, Debug, PartialEq)]
pub struct ReferenceCandidate {
    pub id: Uuid,
    pub name: Option<String>,
    pub score: f64,
}

/// Outcome of `resolve_reference`. Never a silent pick among close
/// candidates: `Ambiguous` always lists what it found instead of guessing.
#[derive(Clone, Debug, PartialEq)]
pub enum ReferenceResolution {
    Resolved { id: Uuid, confidence: f64 },
    Ambiguous { candidates: Vec<ReferenceCandidate> },
    NotFound,
}

/// Ring-match confidence for an exact (case-insensitive) name match.
const RING_EXACT_CONFIDENCE: f64 = 0.95;
/// Ring-match confidence for a substring match (either direction).
const RING_SUBSTRING_CONFIDENCE: f64 = 0.7;
/// A single ring candidate auto-resolves only at or above this bar; below
/// it the candidate is still surfaced as `Ambiguous` rather than silently
/// accepted or dropped. Ring scores are fixed constants on a 0..1 scale, so
/// a fixed bar is meaningful here: the search stage below needs a
/// different rule (`SEARCH_RESOLVED_CONFIDENCE`) because RRF scores aren't
/// on that scale.
const RING_AUTO_RESOLVE_CONFIDENCE: f64 = 0.7;
/// Confidence for a stage-3 exact-name storage match: a deterministic,
/// case-sensitive equality on `entities.name` — stronger evidence than the
/// ring's case-insensitive session cache (`RING_EXACT_CONFIDENCE`), so it
/// sits above both ring bands, but still below the absolute certainty of an
/// id-string passthrough (1.0), which the caller supplied directly rather
/// than by name.
const EXACT_NAME_CONFIDENCE: f64 = 0.98;
/// Hybrid-search fallback: the top hit auto-resolves over a runner-up only
/// when it leads by at least this ratio — RRF scores are not on a fixed
/// 0..1 confidence scale, so a fixed absolute bar can't express "decisively
/// best" the way it can for the ring. Below the margin, every hit above the
/// score floor is surfaced as a candidate instead.
const SEARCH_MARGIN_RATIO: f64 = 2.0;
/// Vector hits below this raw cosine-similarity score are removed before RRF
/// fusion. Lexical hits remain admissible because RRF magnitude encodes rank,
/// not textual relevance, and genuine partial-name matches score below this floor.
///
/// This is a raw cosine value in `[-1.0, 1.0]`, matching the canonical
/// vector-store score produced by `khive-db` (`1 - cosine_distance`).
const SEARCH_VECTOR_SIMILARITY_FLOOR: f64 = 0.3;
/// Confidence reported on a search-stage `Resolved` outcome: not the raw
/// RRF score, which lives on a much smaller scale (`sum 1/(k + rank)`, e.g.
/// ~0.016-0.033) and would never clear a 0..1 confidence bar. Fixed below
/// both ring bands so callers can tell "the ring recognized this" from
/// "search picked this out" by confidence alone; the raw RRF value is still
/// preserved in `ReferenceCandidate.score` for `Ambiguous` listings.
const SEARCH_RESOLVED_CONFIDENCE: f64 = 0.6;
/// Floor on the retrieval depth stage 4 asks `hybrid_search` for, independent
/// of the caller's requested `limit`.
///
/// `KhiveRuntime::hybrid_search` truncates its returned hit list to exactly
/// the `limit` it is called with (its final step is `fused.truncate(limit as
/// usize)`), so that `limit` doubles as both "how deep to search" and "how
/// many hits to hand back" — for `resolve`, whose caller-facing default
/// `limit` is 5 (`khive_pack_kg::handlers::resolve::DEFAULT_LIMIT`), a
/// caller asking for a short candidate list was silently also asking for a
/// shallow search. A genuine canonical-name match ranked, say, 6th-to-10th —
/// exactly where a qualified natural-language ref (a project prefix ahead of
/// a short canonical name) lands when it beats the exact-name stage above
/// but only partially matches the text leg and is corroborated by the vector
/// leg — was truncated out of the stage-4 candidate set entirely even though
/// the same query against a wider `limit` (e.g. the `search` verb's own
/// default of 10) surfaces it as the top hit (#908). Retrieving at this
/// floor decouples "how deep to search" from "how many candidates the caller
/// asked for". The full pool is retained for the decisiveness check, while an
/// `Ambiguous` payload is truncated back to the caller's `limit` before it is
/// returned.
const STAGE4_MIN_SEARCH_LIMIT: u32 = 20;

/// Resolve one natural-language reference for `token`'s actor.
///
/// `limit` bounds the hybrid-search fallback candidate count (Layer-0 stage
/// 4); it has no effect on the id-string or ring stages, which are always
/// exact-or-nothing / small in-memory scans. `entity_kind`, if set, restricts
/// stage 3 to that entity kind (e.g. `"concept"`); the id-string and ring
/// stages are kind-agnostic by construction (a ring entry or an explicit id
/// is not filtered by kind).
pub async fn resolve_reference(
    runtime: &KhiveRuntime,
    ring: &ReferenceRing,
    token: &NamespaceToken,
    nl_ref: &str,
    limit: u32,
    entity_kind: Option<&str>,
) -> RuntimeResult<ReferenceResolution> {
    resolve_reference_with_entity_type(runtime, ring, token, nl_ref, limit, entity_kind, None).await
}

/// Resolve with an optional canonical entity subtype in the exact-name and
/// search stages. ID and ring matches retain their existing kind-agnostic
/// behavior, so a subtype filter cannot turn an explicit ID into a fuzzy hit.
#[allow(clippy::too_many_arguments)]
pub async fn resolve_reference_with_entity_type(
    runtime: &KhiveRuntime,
    ring: &ReferenceRing,
    token: &NamespaceToken,
    nl_ref: &str,
    limit: u32,
    entity_kind: Option<&str>,
    entity_type: Option<&str>,
) -> RuntimeResult<ReferenceResolution> {
    let trimmed = nl_ref.trim();
    if trimmed.is_empty() {
        return Ok(ReferenceResolution::NotFound);
    }

    // Stage 1: id-string passthrough (UUID / 8+ hex prefix) via the existing
    // by-ID path. A ref shaped like an id but absent from storage is
    // NotFound, not a fallthrough to ring/search: the caller named a
    // specific id, so a miss there is the true answer. Scoped to entity ids
    // only (both full-UUID and prefix forms) to match the ring's entity-only
    // contract (`reference_ring::substrate_admits_as_entity`); a non-entity
    // id-string is `NotFound` here: callers needing those already have `get`.
    if let Ok(uuid) = Uuid::from_str(trimmed) {
        return match runtime.resolve_by_id(token, uuid).await? {
            Some(Resolved::Entity(_)) => Ok(ReferenceResolution::Resolved {
                id: uuid,
                confidence: 1.0,
            }),
            Some(_) | None => Ok(ReferenceResolution::NotFound),
        };
    }
    if is_hex_prefix(trimmed) {
        return match runtime.resolve_prefix_unfiltered(trimmed).await {
            Ok(Some(uuid)) => match runtime.resolve_by_id(token, uuid).await? {
                Some(Resolved::Entity(_)) => Ok(ReferenceResolution::Resolved {
                    id: uuid,
                    confidence: 1.0,
                }),
                Some(_) | None => Ok(ReferenceResolution::NotFound),
            },
            Ok(None) => Ok(ReferenceResolution::NotFound),
            Err(RuntimeError::AmbiguousPrefix { matches, .. }) => {
                let mut entity_matches = Vec::with_capacity(matches.len());
                for id in matches {
                    if matches!(
                        runtime.resolve_by_id(token, id).await?,
                        Some(Resolved::Entity(_))
                    ) {
                        entity_matches.push(id);
                    }
                }
                match entity_matches.len() {
                    0 => Ok(ReferenceResolution::NotFound),
                    1 => Ok(ReferenceResolution::Resolved {
                        id: entity_matches[0],
                        confidence: 1.0,
                    }),
                    _ => Ok(ReferenceResolution::Ambiguous {
                        candidates: entity_matches
                            .into_iter()
                            .map(|id| ReferenceCandidate {
                                id,
                                name: None,
                                score: 1.0,
                            })
                            .collect(),
                    }),
                }
            }
            Err(e) => Err(e),
        };
    }

    // Stage 2: recently-referenced ring.
    let actor = token.actor();
    let actor_key = format!("{}:{}", actor.kind, actor.id);
    let ring_entries = ring.snapshot(token.namespace().as_str(), &actor_key);
    let needle = trimmed.to_ascii_lowercase();

    let exact: Vec<ReferenceCandidate> = ring_entries
        .iter()
        .filter(|e| {
            e.name
                .as_deref()
                .is_some_and(|n| n.to_ascii_lowercase() == needle)
        })
        .map(|e| ReferenceCandidate {
            id: e.id,
            name: e.name.clone(),
            score: RING_EXACT_CONFIDENCE,
        })
        .collect();
    if let Some(resolution) = resolve_from_candidates(exact) {
        return Ok(resolution);
    }

    let substring: Vec<ReferenceCandidate> = ring_entries
        .iter()
        .filter(|e| {
            e.name.as_deref().is_some_and(|n| {
                let n_lower = n.to_ascii_lowercase();
                n_lower.contains(&needle) || needle.contains(&n_lower)
            })
        })
        .map(|e| ReferenceCandidate {
            id: e.id,
            name: e.name.clone(),
            score: RING_SUBSTRING_CONFIDENCE,
        })
        .collect();
    if let Some(resolution) = resolve_from_candidates(substring) {
        return Ok(resolution);
    }

    // Stage 3: exact-name storage lookup (#849) — a deterministic,
    // case-sensitive match against `entities.name` in the caller's
    // namespace, run before the hybrid-search fallback so an existing exact
    // name always resolves regardless of FTS ranking, RRF score, or whether
    // this actor's session ever referenced the entity (the ring's blind
    // spot). Single match resolves; multiple exact matches are `Ambiguous`;
    // none falls through to hybrid search unchanged.
    if let Some(resolution) =
        exact_name_match(runtime, token, trimmed, entity_kind, entity_type).await?
    {
        return Ok(resolution);
    }

    // Stage 4: hybrid-search fallback over the namespace. Search deeper than
    // the caller's requested `limit` (see `STAGE4_MIN_SEARCH_LIMIT`) so a
    // genuine match ranked just outside a small `limit` isn't truncated out
    // of the pool before it's even ranked against the alternatives; the
    // caller's `limit` still bounds how many candidates get rendered below.
    let candidate_limit = limit.max(1);
    let search_limit = candidate_limit.max(STAGE4_MIN_SEARCH_LIMIT);
    let hits = runtime
        .hybrid_search_with_vector_similarity_floor(
            token,
            trimmed,
            None,
            search_limit,
            entity_kind,
            entity_type,
            &[],
            None,
            SEARCH_VECTOR_SIMILARITY_FLOOR,
        )
        .await?;
    let candidates: Vec<ReferenceCandidate> = hits
        .into_iter()
        .map(|h| ReferenceCandidate {
            id: h.entity_id,
            name: h.title,
            score: h.score.to_f64(),
        })
        .collect();

    match candidates.len() {
        0 => Ok(ReferenceResolution::NotFound),
        // A lone hit is presence-decisive: there is no competing candidate to
        // be ambiguous against, regardless of its raw RRF magnitude (see
        // `SEARCH_RESOLVED_CONFIDENCE`).
        1 => Ok(ReferenceResolution::Resolved {
            id: candidates[0].id,
            confidence: SEARCH_RESOLVED_CONFIDENCE,
        }),
        _ => {
            let top_score = candidates[0].score;
            let second_score = candidates[1].score;
            let decisive =
                second_score <= f64::EPSILON || top_score / second_score >= SEARCH_MARGIN_RATIO;
            if decisive {
                Ok(ReferenceResolution::Resolved {
                    id: candidates[0].id,
                    confidence: SEARCH_RESOLVED_CONFIDENCE,
                })
            } else {
                // Non-exact ambiguity: bound the payload to the caller's
                // `limit`. Any deterministic identity (an exact canonical
                // name) was already resolved by stage 3 above, so nothing
                // withheld here is "the" answer — outside exact matches there
                // is no oracle for a single canonical candidate. Raising
                // `limit` surfaces deeper ranks (#970; resolve() contract).
                let mut candidates = candidates;
                candidates.truncate(candidate_limit as usize);
                Ok(ReferenceResolution::Ambiguous { candidates })
            }
        }
    }
}

/// Apply the shared "single-above-bar resolves, multiple is ambiguous"
/// contract to a candidate set already known to be an exact or substring
/// ring match. Returns `None` when `candidates` is empty — the caller falls
/// through to the next resolution stage instead of reporting `NotFound`
/// prematurely.
fn resolve_from_candidates(candidates: Vec<ReferenceCandidate>) -> Option<ReferenceResolution> {
    match candidates.len() {
        0 => None,
        1 => {
            let top = &candidates[0];
            Some(if top.score >= RING_AUTO_RESOLVE_CONFIDENCE {
                ReferenceResolution::Resolved {
                    id: top.id,
                    confidence: top.score,
                }
            } else {
                ReferenceResolution::Ambiguous { candidates }
            })
        }
        _ => Some(ReferenceResolution::Ambiguous { candidates }),
    }
}

/// Stage 3 of `resolve_reference` (#849): a deterministic, case-sensitive
/// exact match against `entities.name`, scoped to `token.namespace()` (the
/// same single-namespace default the rest of this pipeline and the sibling
/// by-name lookup in `khive-pack-kg`'s `resolve_name_async` use) and to
/// `entity_kind` when the caller filtered by one. `query_entities` already
/// excludes soft-deleted rows (`deleted_at IS NULL` is baked into every
/// query — see `khive-db::stores::entity::build_entity_where`), so no
/// separate filter is needed here. Returns `None` (fall through to the next
/// stage) when nothing matches; `Some(Resolved)` on a single hit; and
/// `Some(Ambiguous)` when the name is not unique.
async fn exact_name_match(
    runtime: &KhiveRuntime,
    token: &NamespaceToken,
    name: &str,
    entity_kind: Option<&str>,
    entity_type: Option<&str>,
) -> RuntimeResult<Option<ReferenceResolution>> {
    let mut filter = EntityFilter {
        name_exact: Some(name.to_string()),
        kinds: entity_kind.map(|k| vec![k.to_string()]).unwrap_or_default(),
        ..EntityFilter::default()
    };
    if let (Some(kind), Some(entity_type)) = (entity_kind, entity_type) {
        filter
            .entity_types_by_kind
            .insert(kind.to_string(), vec![entity_type.to_string()]);
    }
    // A storage-level `name = ?` predicate (not `name_prefix` + in-memory
    // filter) so a namespace with many newer case variants of `name` can
    // never page the exact target out from under a `created_at DESC` sort
    // (#849, #852) — every row this query returns already equals `name`.
    // The page is small (10, not 1), but the zero/one/many decision is made
    // from `page.total` (the storage-computed COUNT(*) under the same
    // predicate), never from `page.items.len()` — with 11+ byte-identical
    // exact names the fetched page is still only 10 rows, and deciding from
    // its length alone would under-report cardinality. `Ambiguous.candidates`
    // is a bounded sample of up to 10 of the `page.total` matches, not the
    // complete set — the variant carries no total field, so callers must not
    // assume `candidates.len() == page.total` (#852).
    let page = runtime
        .entities(token)?
        .query_entities(
            token.namespace().as_str(),
            filter,
            PageRequest {
                offset: 0,
                limit: 10,
            },
        )
        .await
        .map_err(RuntimeError::Storage)?;

    let total = page.total.unwrap_or(page.items.len() as u64);

    let exact: Vec<ReferenceCandidate> = page
        .items
        .into_iter()
        .map(|e| ReferenceCandidate {
            id: e.id,
            name: Some(e.name),
            score: EXACT_NAME_CONFIDENCE,
        })
        .collect();

    Ok(match total {
        0 => None,
        1 => exact
            .into_iter()
            .next()
            .map(|top| ReferenceResolution::Resolved {
                id: top.id,
                confidence: EXACT_NAME_CONFIDENCE,
            }),
        _ => Some(ReferenceResolution::Ambiguous { candidates: exact }),
    })
}

fn is_hex_prefix(s: &str) -> bool {
    s.len() >= 8 && s.chars().all(|c| c.is_ascii_hexdigit())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::config::{NamespaceToken as TokenCtor, RuntimeConfig};
    use crate::embedder_registry::EmbedderProvider;
    use crate::retrieval::SearchSource;
    use khive_gate::ActorRef;
    use khive_types::{namespace::Namespace, SubstrateKind};
    use lattice_embed::{EmbeddingModel, EmbeddingService};
    use std::sync::Arc;

    struct ConstantEmbeddingService {
        dimensions: usize,
    }

    #[async_trait::async_trait]
    impl EmbeddingService for ConstantEmbeddingService {
        async fn embed(
            &self,
            texts: &[String],
            _model: EmbeddingModel,
        ) -> Result<Vec<Vec<f32>>, lattice_embed::EmbedError> {
            Ok(texts.iter().map(|_| vec![1.0; self.dimensions]).collect())
        }

        fn supports_model(&self, _model: EmbeddingModel) -> bool {
            true
        }

        fn name(&self) -> &'static str {
            "resolve-test-constant-embedding"
        }
    }

    struct ConstantEmbedderProvider {
        name: String,
        dimensions: usize,
    }

    #[async_trait::async_trait]
    impl EmbedderProvider for ConstantEmbedderProvider {
        fn name(&self) -> &str {
            &self.name
        }

        fn dimensions(&self) -> usize {
            self.dimensions
        }

        async fn build(&self) -> RuntimeResult<Arc<dyn EmbeddingService>> {
            Ok(Arc::new(ConstantEmbeddingService {
                dimensions: self.dimensions,
            }))
        }
    }

    fn runtime_with_constant_embeddings() -> KhiveRuntime {
        let model = EmbeddingModel::AllMiniLmL6V2;
        let runtime = KhiveRuntime::new(RuntimeConfig {
            db_path: None,
            embedding_model: Some(model),
            packs: vec!["kg".to_string()],
            ..RuntimeConfig::no_embeddings()
        })
        .expect("in-memory runtime");
        runtime.register_embedder(ConstantEmbedderProvider {
            name: model.to_string(),
            dimensions: model.dimensions(),
        });
        runtime
    }

    fn actor_token(actor_id: &str) -> NamespaceToken {
        TokenCtor::mint_authorized(Namespace::local(), ActorRef::new("agent", actor_id))
    }

    #[tokio::test]
    async fn id_string_passthrough_resolves_full_uuid() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        let entity = rt
            .create_entity(
                &token,
                "concept",
                None,
                "PassthroughTarget",
                None,
                None,
                vec![],
            )
            .await
            .expect("create entity");

        let resolution = resolve_reference(&rt, &ring, &token, &entity.id.to_string(), 5, None)
            .await
            .expect("resolve_reference");
        assert_eq!(
            resolution,
            ReferenceResolution::Resolved {
                id: entity.id,
                confidence: 1.0
            }
        );
    }

    #[tokio::test]
    async fn id_string_passthrough_never_errors_on_a_miss() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        let missing = Uuid::new_v4();
        let resolution = resolve_reference(&rt, &ring, &token, &missing.to_string(), 5, None)
            .await
            .expect("must not error, only report NotFound");
        assert_eq!(resolution, ReferenceResolution::NotFound);
    }

    #[tokio::test]
    async fn ring_exact_match_resolves_without_search() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();
        let actor = token.actor();
        let actor_key = format!("{}:{}", actor.kind, actor.id);

        let id = Uuid::new_v4();
        ring.admit(
            token.namespace().as_str(),
            &actor_key,
            id,
            Some("the old record".to_string()),
        );

        let resolution = resolve_reference(&rt, &ring, &token, "the old record", 5, None)
            .await
            .expect("resolve_reference");
        assert_eq!(
            resolution,
            ReferenceResolution::Resolved {
                id,
                confidence: RING_EXACT_CONFIDENCE
            }
        );
    }

    #[tokio::test]
    async fn ring_ambiguous_on_multiple_exact_matches() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();
        let actor = token.actor();
        let actor_key = format!("{}:{}", actor.kind, actor.id);

        let id_a = Uuid::new_v4();
        let id_b = Uuid::new_v4();
        ring.admit(
            token.namespace().as_str(),
            &actor_key,
            id_a,
            Some("duplicate name".to_string()),
        );
        ring.admit(
            token.namespace().as_str(),
            &actor_key,
            id_b,
            Some("duplicate name".to_string()),
        );

        let resolution = resolve_reference(&rt, &ring, &token, "duplicate name", 5, None)
            .await
            .expect("resolve_reference");
        match resolution {
            ReferenceResolution::Ambiguous { candidates } => {
                assert_eq!(candidates.len(), 2);
            }
            other => panic!("expected Ambiguous, got {other:?}"),
        }
    }

    #[tokio::test]
    async fn no_ring_entry_and_no_search_hit_is_not_found() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        let resolution =
            resolve_reference(&rt, &ring, &token, "nothing matches this at all", 5, None)
                .await
                .expect("resolve_reference");
        assert_eq!(resolution, ReferenceResolution::NotFound);
    }

    #[tokio::test]
    async fn actor_isolation_blocks_cross_actor_ring_reads() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token_a = actor_token("actor-a");
        let token_b = actor_token("actor-b");
        let ring = ReferenceRing::new();
        let actor_a = token_a.actor();
        let actor_key_a = format!("{}:{}", actor_a.kind, actor_a.id);

        let id = Uuid::new_v4();
        ring.admit(
            token_a.namespace().as_str(),
            &actor_key_a,
            id,
            Some("shared-namespace-name".to_string()),
        );

        // actor-b, same namespace, must NOT resolve via actor-a's ring entry.
        let resolution = resolve_reference(&rt, &ring, &token_b, "shared-namespace-name", 5, None)
            .await
            .expect("resolve_reference");
        assert_eq!(resolution, ReferenceResolution::NotFound);
    }

    // Regression for #849/#852: the stage-3 exact-name lookup used to filter
    // `name_prefix` (`LIKE 'RoLoRA%'`) in memory, and `query_entities` ranks
    // a `name_prefix` page by `CASE WHEN LOWER(name) = prefix THEN 0 ELSE 1
    // END, created_at DESC`. Case-insensitive variants of the target name
    // tie for priority 0 with the true exact match, so 100+ *newer*
    // lowercase variants can fill the `LIMIT 100` page and page the older,
    // case-exact target out entirely — the stage then falls through to
    // hybrid search instead of resolving deterministically. The fix issues a
    // storage-level `name = ?` (binary) predicate instead, so decoys that
    // merely match case-insensitively never enter the result set at all.
    #[tokio::test]
    async fn exact_name_stage_survives_many_newer_case_variant_decoys() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        let target = rt
            .create_entity(&token, "concept", None, "RoLoRA", None, None, vec![])
            .await
            .expect("create target entity");

        // Case variants of the same name, not suffixed variants: SQLite's
        // `LIKE` is case-insensitive for ASCII, so a `rolora`-named decoy
        // still matches the `LIKE 'RoLoRA%'` pattern the buggy `name_prefix`
        // stage used, and the exact-match-ranking `CASE WHEN LOWER(name) =
        // ...` ties every one of these decoys with the true target at
        // priority 0 — leaving `created_at DESC` as the only tiebreak.
        let decoy_cases = ["rolora", "ROLORA", "RoLoRa", "roLORA"];
        for i in 0..120 {
            rt.create_entity(
                &token,
                "concept",
                None,
                decoy_cases[i % decoy_cases.len()],
                None,
                None,
                vec![],
            )
            .await
            .expect("create decoy entity");
        }

        let resolution = resolve_reference(&rt, &ring, &token, "RoLoRA", 5, None)
            .await
            .expect("resolve_reference");
        assert_eq!(
            resolution,
            ReferenceResolution::Resolved {
                id: target.id,
                confidence: EXACT_NAME_CONFIDENCE,
            }
        );
    }

    /// Issue #852: the zero/one/many decision must come from the
    /// storage-computed `page.total` (a full `COUNT(*)` under the exact-name
    /// predicate), not from `page.items.len()`, which the stage's own
    /// `LIMIT 10` caps regardless of true cardinality. 11 byte-identical
    /// exact names exceed that page limit, so the fetched page can only ever
    /// carry 10 rows — `Ambiguous` must still fire (storage says 11 total,
    /// not the truncated 10), and the returned `candidates` are a bounded
    /// sample of the match set, not its entirety. That truncation is an
    /// intentional, documented contract of this stage, not a bug: this test
    /// pins both halves so a future change can't silently drop one.
    #[tokio::test]
    async fn exact_name_ambiguous_decision_uses_storage_total_not_page_len() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        for _ in 0..11 {
            rt.create_entity(&token, "concept", None, "DupeExactName", None, None, vec![])
                .await
                .expect("create duplicate-named entity");
        }

        let resolution = resolve_reference(&rt, &ring, &token, "DupeExactName", 5, None)
            .await
            .expect("resolve_reference");

        match resolution {
            ReferenceResolution::Ambiguous { candidates } => {
                assert_eq!(
                    candidates.len(),
                    10,
                    "candidate set is a bounded 10-row sample of the 11 storage matches, \
                     not the complete set"
                );
            }
            other => panic!("expected Ambiguous driven by storage total (11), got {other:?}"),
        }
    }

    // Regression for #908 / #970: an exact canonical-name ref resolves to a
    // single id through the stage-3 exact-name lookup, which short-circuits
    // BEFORE the stage-4 hybrid fallback and its `limit` bound. Many other
    // entities are strong hybrid matches for the same term — enough that
    // hybrid alone would rank them close together and return `Ambiguous` — yet
    // the exact name resolves deterministically to its owner. The returned
    // `EXACT_NAME_CONFIDENCE` (not the stage-4 `SEARCH_RESOLVED_CONFIDENCE`)
    // proves the result came from stage 3, regardless of the target's hybrid
    // rank. An exact name is an identity, not a ranked candidate.
    #[tokio::test]
    async fn exact_name_resolves_ahead_of_competing_hybrid_matches() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        let target = rt
            .create_entity(
                &token,
                "concept",
                None,
                "ADR-040",
                Some("khive ADR-040"),
                None,
                vec![],
            )
            .await
            .expect("create target entity");

        // Competitors that also match "ADR-040" strongly in hybrid search but
        // do NOT carry it as an exact name; only the target owns the exact
        // name, so only the target satisfies the stage-3 lookup.
        for i in 0..12 {
            rt.create_entity(
                &token,
                "concept",
                None,
                &format!("ADR-040 companion {i}"),
                Some("ADR-040 ADR-040 ADR-040 ADR-040 ADR-040"),
                None,
                vec![],
            )
            .await
            .expect("create competing entity");
        }

        // Even at limit = 1 — the tightest bound — the exact name resolves to
        // the target, at the stage-3 confidence, never a ranked hybrid pick.
        let resolution = resolve_reference(&rt, &ring, &token, "ADR-040", 1, None)
            .await
            .expect("resolve_reference");
        assert_eq!(
            resolution,
            ReferenceResolution::Resolved {
                id: target.id,
                confidence: EXACT_NAME_CONFIDENCE,
            }
        );
    }

    // The complement of the exact-name contract: a NON-exact ref (no entity
    // carries it as an exact name) that stays ambiguous returns a bounded
    // sample capped at the caller's `limit`. Outside exact matches there is no
    // oracle for a single "canonical" candidate, so the bound is the
    // intentional, documented resolve() contract (raise `limit` to surface
    // deeper ranks) — not a withheld identity. Any deterministic identity
    // would have resolved upstream in stage 3.
    #[tokio::test]
    async fn fallback_stage_bounds_non_exact_payload_to_limit() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        // Twelve near-equal matches, each with a distinct name and the same
        // descriptive text — none is the "right" answer, so the result is a
        // genuinely bounded ambiguous sample, not a suppressed target.
        for i in 0..12 {
            rt.create_entity(
                &token,
                "concept",
                None,
                &format!("Retrieval Fusion Note {i}"),
                Some("khive retrieval fusion ranking note"),
                None,
                vec![],
            )
            .await
            .expect("create entity");
        }

        let resolution = resolve_reference(
            &rt,
            &ring,
            &token,
            "khive retrieval fusion ranking",
            5,
            None,
        )
        .await
        .expect("resolve_reference");

        match resolution {
            ReferenceResolution::Ambiguous { candidates } => {
                assert_eq!(
                    candidates.len(),
                    5,
                    "a non-exact ref's ambiguity payload is bounded to the caller's limit"
                );
            }
            other => panic!("expected a bounded Ambiguous sample, got {other:?}"),
        }
    }

    // Regression for #908: genuine ambiguity — two entities with the exact
    // same canonical name — must still return `Ambiguous`, never a silent
    // pick, after widening stage 4's retrieval floor.
    #[tokio::test]
    async fn fallback_stage_still_reports_ambiguous_on_genuine_tie() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        let a = rt
            .create_entity(
                &token,
                "concept",
                None,
                "Twin Record",
                Some("khive Twin Record document"),
                None,
                vec![],
            )
            .await
            .expect("create entity a");
        let b = rt
            .create_entity(
                &token,
                "concept",
                None,
                "Twin Record",
                Some("khive Twin Record document"),
                None,
                vec![],
            )
            .await
            .expect("create entity b");

        // Exact byte-identical names hit stage 3 (exact-name storage
        // lookup), not stage 4 — but the same "must not silently pick"
        // contract applies at both stages, and stage 3 is a cheaper,
        // deterministic way to pin it.
        let resolution = resolve_reference(&rt, &ring, &token, "Twin Record", 5, None)
            .await
            .expect("resolve_reference");

        match resolution {
            ReferenceResolution::Ambiguous { candidates } => {
                let ids: std::collections::HashSet<Uuid> =
                    candidates.iter().map(|c| c.id).collect();
                assert!(ids.contains(&a.id) && ids.contains(&b.id));
            }
            other => panic!("expected Ambiguous on a genuine name tie, got {other:?}"),
        }
    }

    #[tokio::test]
    async fn fallback_stage_resolves_high_similarity_semantic_only_candidate() {
        let rt = runtime_with_constant_embeddings();
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        let entity = rt
            .create_entity(&token, "concept", None, "Canine", None, None, vec![])
            .await
            .expect("create semantic match");

        let raw_hits = rt
            .vector_search(
                &token,
                None,
                Some("domestic dog"),
                5,
                Some(SubstrateKind::Entity),
            )
            .await
            .expect("vector search");
        assert_eq!(raw_hits.len(), 1);
        assert_eq!(raw_hits[0].subject_id, entity.id);
        // Identical constant embeddings have canonical cosine score 1.0,
        // well above the raw-cosine floor of 0.3.
        assert!((raw_hits[0].score.to_f64() - 1.0).abs() < 1e-6);

        let hits = rt
            .hybrid_search(&token, "domestic dog", None, 5, None, None, &[], None)
            .await
            .expect("hybrid search");
        assert_eq!(hits.len(), 1);
        assert_eq!(hits[0].source, SearchSource::Vector);

        let resolution = resolve_reference(&rt, &ring, &token, "domestic dog", 5, None)
            .await
            .expect("resolve_reference");

        assert_eq!(
            resolution,
            ReferenceResolution::Resolved {
                id: entity.id,
                confidence: SEARCH_RESOLVED_CONFIDENCE,
            }
        );
    }

    #[tokio::test]
    async fn fallback_stage_resolves_moderate_similarity_semantic_only_candidate() {
        // A canonical cosine score of 0.5 clears the raw-cosine floor of 0.3.
        // This guards against reintroducing the legacy `(1 + cos) / 2` floor
        // conversion, which would incorrectly raise the comparison floor to
        // 0.65 after vector-store scores moved to the canonical cosine scale.
        let rt = runtime_with_constant_embeddings();
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();
        let dimensions = EmbeddingModel::AllMiniLmL6V2.dimensions();

        let entity = rt
            .create_entity(
                &token,
                "concept",
                None,
                "Moderately Similar Candidate",
                None,
                None,
                vec![],
            )
            .await
            .expect("create moderate-similarity entity");
        let vectors = rt.vectors(&token).expect("vector store");
        vectors
            .delete(entity.id)
            .await
            .expect("delete generated vector");
        let aligned = dimensions * 3 / 4;
        let mut moderate_vector = vec![1.0f32; aligned];
        moderate_vector.extend(vec![-1.0f32; dimensions - aligned]);
        vectors
            .insert(
                entity.id,
                SubstrateKind::Entity,
                token.namespace().as_str(),
                "entity.body",
                vec![moderate_vector],
            )
            .await
            .expect("insert moderate-similarity vector");

        let query = "totally-nonexistent-moderate-zzz";
        let raw_hits = rt
            .vector_search(&token, None, Some(query), 5, Some(SubstrateKind::Entity))
            .await
            .expect("vector search");
        assert_eq!(raw_hits.len(), 1);
        assert_eq!(raw_hits[0].subject_id, entity.id);
        assert!((raw_hits[0].score.to_f64() - 0.5).abs() < 1e-6);

        let resolution = resolve_reference(&rt, &ring, &token, query, 5, None)
            .await
            .expect("resolve_reference");

        assert_eq!(
            resolution,
            ReferenceResolution::Resolved {
                id: entity.id,
                confidence: SEARCH_RESOLVED_CONFIDENCE,
            }
        );
    }

    #[tokio::test]
    async fn fallback_stage_drops_orthogonal_semantic_only_candidate() {
        // An orthogonal vector has canonical cosine score 0.0, below the
        // raw-cosine floor of 0.3. A sole semantic candidate at this
        // similarity must be dropped, not resolved.
        let rt = runtime_with_constant_embeddings();
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();
        let dimensions = EmbeddingModel::AllMiniLmL6V2.dimensions();

        let entity = rt
            .create_entity(
                &token,
                "concept",
                None,
                "Unrelated Orthogonal",
                None,
                None,
                vec![],
            )
            .await
            .expect("create unrelated entity");
        let vectors = rt.vectors(&token).expect("vector store");
        vectors
            .delete(entity.id)
            .await
            .expect("delete generated vector");
        let mut orthogonal_vector = vec![1.0f32; dimensions / 2];
        orthogonal_vector.extend(vec![-1.0f32; dimensions - dimensions / 2]);
        vectors
            .insert(
                entity.id,
                SubstrateKind::Entity,
                token.namespace().as_str(),
                "entity.body",
                vec![orthogonal_vector],
            )
            .await
            .expect("insert orthogonal vector");

        let raw_hits = rt
            .vector_search(
                &token,
                None,
                Some("totally-nonexistent-orthogonal-zzz"),
                5,
                Some(SubstrateKind::Entity),
            )
            .await
            .expect("vector search");
        assert_eq!(raw_hits.len(), 1);
        assert!((raw_hits[0].score.to_f64() - 0.0).abs() < 1e-6);

        let resolution = resolve_reference(
            &rt,
            &ring,
            &token,
            "totally-nonexistent-orthogonal-zzz",
            5,
            None,
        )
        .await
        .expect("resolve_reference");

        assert_eq!(resolution, ReferenceResolution::NotFound);
    }

    #[tokio::test]
    async fn fallback_stage_drops_low_similarity_semantic_only_candidates() {
        let rt = runtime_with_constant_embeddings();
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();
        let dimensions = EmbeddingModel::AllMiniLmL6V2.dimensions();

        let entity = rt
            .create_entity(
                &token,
                "concept",
                None,
                "Unrelated Alpha",
                None,
                None,
                vec![],
            )
            .await
            .expect("create unrelated entity");
        let vectors = rt.vectors(&token).expect("vector store");
        vectors
            .delete(entity.id)
            .await
            .expect("delete generated vector");
        // A quarter of dimensions aligned, three-quarters opposed against the
        // constant all-ones query embedding: canonical cosine score
        // 2*0.25 - 1 == -0.5. This is below the floor without being the
        // maximally-opposite case, so it exercises a genuine below-floor
        // mismatch rather than the degenerate cosine -1 extreme.
        let quarter = dimensions / 4;
        let mut mismatched_vector = vec![1.0f32; quarter];
        mismatched_vector.extend(vec![-1.0f32; dimensions - quarter]);
        vectors
            .insert(
                entity.id,
                SubstrateKind::Entity,
                token.namespace().as_str(),
                "entity.body",
                vec![mismatched_vector],
            )
            .await
            .expect("insert mismatched vector");

        let raw_hits = rt
            .vector_search(
                &token,
                None,
                Some("totally-nonexistent-zzz"),
                5,
                Some(SubstrateKind::Entity),
            )
            .await
            .expect("vector search");
        assert_eq!(raw_hits.len(), 1);
        assert!((raw_hits[0].score.to_f64() + 0.5).abs() < 1e-6);

        let resolution = resolve_reference(&rt, &ring, &token, "totally-nonexistent-zzz", 5, None)
            .await
            .expect("resolve_reference");

        assert_eq!(resolution, ReferenceResolution::NotFound);
    }

    // Regression for #908: garbage input that matches nothing must still be
    // `NotFound` after widening stage 4's retrieval floor — the widened pool
    // must not turn "nothing relevant exists" into a spurious pick.
    #[tokio::test]
    async fn fallback_stage_still_not_found_on_garbage() {
        let rt = KhiveRuntime::memory().expect("in-memory runtime");
        let token = actor_token("resolver-test");
        let ring = ReferenceRing::new();

        rt.create_entity(
            &token,
            "concept",
            None,
            "ADR-040",
            Some("khive ADR-040"),
            None,
            vec![],
        )
        .await
        .expect("create unrelated entity");

        let resolution = resolve_reference(
            &rt,
            &ring,
            &token,
            "zzqxw completely unrelated garbage nonsense",
            5,
            None,
        )
        .await
        .expect("resolve_reference");
        assert_eq!(resolution, ReferenceResolution::NotFound);
    }
}