sqlite-graphrag 1.2.8

//! CLI argument types for the `enrich` subcommand.
//! Extracted from mod.rs (Wave C1).

use serde::{Deserialize, Serialize};
use std::path::PathBuf;

use super::DEFAULT_BODY_ENRICH_MAX_CHARS;
use super::DEFAULT_BODY_ENRICH_MIN_CHARS;
use super::DEFAULT_ENRICH_CIRCUIT_BREAKER_THRESHOLD;
use super::DEFAULT_ENRICH_GROUNDING_THRESHOLD;
use super::DEFAULT_ENRICH_MAX_ATTEMPTS;
use super::DEFAULT_ENRICH_PRESERVE_THRESHOLD;
use super::DEFAULT_ENRICH_RATE_LIMIT_BUFFER_SECS;
use super::DEFAULT_ENRICH_STALE_CLAIM_SECS;
use crate::constants::{DEFAULT_ENRICH_REST_CONCURRENCY, DEFAULT_OPENROUTER_CHAT_TIMEOUT_SECS};

// ---------------------------------------------------------------------------
// CLI args
// ---------------------------------------------------------------------------

/// Operation to perform in the `enrich` command.
#[derive(Debug, Clone, PartialEq, Eq, clap::ValueEnum, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub enum EnrichOperation {
    /// Add missing entity/relationship bindings to memories (fully implemented).
    /// memory-bindings LINKS each memory to the EXISTING entities extracted from
    /// its body — it does not invent a new graph, it only connects what is missing. Scans
    /// only UNBOUND memories (those with zero `memory_entities`).
    MemoryBindings,
    /// GAP-SG-24/26: additive augmentation — re-run binding extraction over
    /// memories that are ALREADY bound, filtered by `--names`/`--names-file`, to
    /// merge newly-discovered entities/relationships WITHOUT removing existing
    /// links. Requires a name filter (refuses to re-scan the whole namespace).
    AugmentBindings,
    /// Fill NULL/empty entity descriptions with LLM-generated summaries (fully implemented).
    EntityDescriptions,
    /// Expand short memory bodies into richer content (fully implemented, GAP-18).
    BodyEnrich,
    /// Rebuild missing memory embeddings without rewriting the memory body.
    ReEmbed,
    /// Calibrate relationship weights using LLM analysis (fully implemented; persists weight).
    WeightCalibrate,
    /// Reclassify relationship types using LLM judgment (fully implemented; persists relation).
    RelationReclassify,
    /// Connect isolated entities by suggesting and persisting new relationships (fully implemented).
    EntityConnect,
    /// Validate entity type assignments using LLM judgment (fully implemented; persists type).
    EntityTypeValidate,
    /// Enrich memory descriptions that are generic/auto-generated (fully implemented; persists description).
    DescriptionEnrich,
    /// Identify cross-domain bridges between disconnected subgraphs.
    /// Shares the O(k) pair scan + `entity_connect_seen` drain path with
    /// `entity-connect` (v1.1.04+ / v1.1.06); status backlog proxy remains 0.
    CrossDomainBridges,
    /// Classify memories into domain categories (fully implemented; persists metadata).
    DomainClassify,
    /// Audit the graph for quality issues (scan/report; does not mutate graph structure).
    GraphAudit,
    /// Synthesize deep-research findings into graph memories (fully implemented when bindings persist).
    DeepResearchSynth,
    /// Extract structured body from unstructured text (fully implemented; persists body).
    BodyExtract,
}

/// v1.1.1 (P2): which embedding table the `re-embed` operation backfills.
///
/// `memories` is the historical behaviour (and the default, so existing
/// invocations are unchanged); `entities` and `chunks` close the retroactive
/// coverage gap for `entity_embeddings` / `chunk_embeddings`; `all` runs the
/// three scans in one invocation.
#[derive(Debug, Clone, Copy, PartialEq, Eq, clap::ValueEnum, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub enum ReEmbedTarget {
    /// Memories without a live vector in `memory_embeddings` (default).
    Memories,
    /// Entities without a live vector in `entity_embeddings`.
    Entities,
    /// Chunks without a live vector in `chunk_embeddings`.
    Chunks,
    /// All three targets in a single run.
    All,
}

/// LLM provider for enrichment.
#[derive(Debug, Clone, PartialEq, Eq, clap::ValueEnum)]
pub enum EnrichMode {
    /// Use the OpenRouter chat-completions REST API (no local CLI; v1.0.95).
    #[value(name = "openrouter")]
    OpenRouter,
}

impl std::fmt::Display for EnrichMode {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            EnrichMode::OpenRouter => write!(f, "openrouter"),
        }
    }
}

/// Arguments for the `enrich` subcommand.
#[derive(clap::Args)]
#[command(
    about = "Enrich graph memories and entities using an LLM provider",
    after_long_help = "EXAMPLES:\n  \
    # Add missing entity bindings to all unbound memories\n  \
    sqlite-graphrag enrich --operation memory-bindings --mode openrouter \\\n    \
      --openrouter-model deepseek/deepseek-v4-flash:nitro\n\n  \
    # Fill entity descriptions (dry-run preview, no tokens spent)\n  \
    sqlite-graphrag enrich --operation entity-descriptions --dry-run --json\n\n  \
    # Expand short memory bodies (GAP-18)\n  \
    sqlite-graphrag enrich --operation body-enrich --min-output-chars 600\n\n  \
    # Rebuild only missing memory embeddings without rewriting bodies\n  \
    sqlite-graphrag enrich --operation re-embed --limit 100\n\n  \
    # Resume an interrupted body-enrich run\n  \
    sqlite-graphrag enrich --operation body-enrich --resume --json\n\n  \
    # Retry only failed items from a previous run\n  \
    sqlite-graphrag enrich --operation memory-bindings --retry-failed --json\n\n  \
    # Converge the whole backlog (internal scan+drain loop, no bash wrapper)\n  \
    sqlite-graphrag enrich --operation memory-bindings --mode openrouter \\\n    \
      --openrouter-model deepseek/deepseek-v4-flash:nitro --until-empty --max-runtime 600\n\n  \
    # Inspect / resurrect dead-letter items\n  \
    sqlite-graphrag enrich --operation memory-bindings --list-dead\n  \
    sqlite-graphrag enrich --operation memory-bindings --requeue-dead\n\n  \
    # Read-only status (no LLM, no singleton)\n  \
    sqlite-graphrag enrich --operation memory-bindings --status\n\n\
    OPERATIONS NOTE:\n  \
    memory-bindings LINKS each memory to the EXISTING entities extracted from its\n  \
    body — it does not invent a new graph, it connects what is missing. It scans\n  \
    only UNBOUND memories. To re-run extraction over ALREADY-bound memories and\n  \
    MERGE newly-found entities/relationships additively (without removing links),\n  \
    use --operation augment-bindings with --names/--names-file.\n\n\
    DEAD-LETTER SIDECAR (.enrich-queue.sqlite):\n  \
    A SQLite sidecar tracks each work item across runs. Schema (table `queue`):\n  \
    item_key (UNIQUE name/id), item_type (memory|entity), operation, memory_id,\n  \
    status (pending|processing|done|skipped|dead), attempt, error, error_class,\n  \
    next_retry_at (backoff cooldown). --until-empty loops scan→drain internally\n  \
    until eligible items are exhausted; transient failures (incl. malformed/non-\n  \
    JSON LLM output, GAP-SG-09) reschedule with backoff until --max-attempts, then\n  \
    land in status='dead'. Use --status to see the queue, --list-dead to inspect\n  \
    the sink, --requeue-dead to retry it, and --ignore-backoff to skip cooldowns.\n  \
    --names/--names-file also remedy a cooldown by targeting a specific subset.\n\n\
    EXIT CODES:\n  \
    0  success\n  \
    1  validation error (bad args, binary not found)\n  \
    14 I/O error"
)]
#[derive(Clone)]
pub struct EnrichArgs {
    /// Enrichment operation to run. Required for write operations; optional for
    /// the read-only queue inspectors (`--status` / `--list-dead` /
    /// `--requeue-dead`), where it defaults to `memory-bindings` when omitted
    /// (GAP-SG-31).
    #[arg(
        long,
        short = 'o',
        value_enum,
        value_name = "OPERATION",
        // GAP-CLI-DRY-01: dry-run is read-only (no LLM); still needs --operation.
        // R-AN-01: --print-schema exits before any work.
        required_unless_present_any = ["status", "list_dead", "requeue_dead", "list_skipped", "requeue_skipped", "prune_dead_orphans", "prune_dead_entity_orphans", "print_schema"]
    )]
    pub operation: Option<EnrichOperation>,

    /// LLM provider to use. Required for write operations; not needed for the
    /// read-only queue inspectors (`--status` / `--list-dead` /
    /// `--requeue-dead` / `--list-skipped` / `--requeue-skipped`), which never
    /// call the LLM (GAP-SG-31).
    ///
    /// GAP-CLI-DRY-01 (v1.1.8): also optional for `--dry-run` (preview only).
    ///
    /// When omitted on a path that allows it, the resolved provider is
    /// `openrouter`. That default is deliberate: it is a REST call and never
    /// spawns a headless CLI (GAP-HEADLESS-DEFAULT).
    #[arg(
        long,
        value_enum,
        required_unless_present_any = ["status", "list_dead", "requeue_dead", "list_skipped", "requeue_skipped", "prune_dead_orphans", "prune_dead_entity_orphans", "dry_run", "print_schema"]
    )]
    pub mode: Option<EnrichMode>,

    /// Maximum number of items to process in this run. Omit for all.
    #[arg(long, value_name = "N", value_parser = crate::parsers::parse_list_limit_range)]
    pub limit: Option<usize>,

    /// GAP-SG-185 / v1.2.4: keyset page size for the enrich SCAN phase.
    ///
    /// SQL row buffers and in-process candidate pages are this wide; production
    /// enqueue walks pages without retaining the full eligible key list. The
    /// sidecar queue still stores one row per eligible item. Precedence: this
    /// flag > XDG `enrich.scan_page_size` > 512. Accepted range: 1..=4096
    /// (clamped).
    #[arg(long, value_name = "N")]
    pub scan_page_size: Option<usize>,

    /// v1.1.1 (P2): embedding table backfilled by `--operation re-embed`.
    /// `memories` (default) preserves the historical behaviour; `entities`
    /// and `chunks` rebuild `entity_embeddings` / `chunk_embeddings`; `all`
    /// covers the three tables in one run. The scan also selects rows whose
    /// stored `dim` diverges from the configured `--embedding-dim` (P10),
    /// so legacy-dimension vectors are regenerated, not only missing ones.
    /// Ignored (rejected) for every other operation.
    #[arg(long, value_enum, value_name = "TARGET", default_value_t = ReEmbedTarget::Memories)]
    pub target: ReEmbedTarget,

    /// Preview items without calling the LLM (zero tokens consumed).
    #[arg(long)]
    pub dry_run: bool,

    /// Namespace to operate on. Default: global.
    #[arg(long)]
    pub namespace: Option<String>,

    // -- Provider flags (OpenRouter, v1.0.95) --
    /// OpenRouter text model to use (REQUIRED with --mode openrouter; no default).
    #[arg(long, value_name = "MODEL")]
    pub openrouter_model: Option<String>,

    /// OpenRouter API key. Prefer `config add-key --provider openrouter --from-stdin`,
    /// which stores it at rest under XDG; this flag overrides that store for a
    /// single invocation and is visible in the process table.
    #[arg(long, value_name = "KEY")]
    pub openrouter_api_key: Option<String>,

    /// Timeout per item in seconds when using OpenRouter. Default: 600.
    ///
    /// GAP-SG-17: raised from 300 to 600 because dense bodies (close to the
    /// ~32K-token context ceiling of the configured model) routinely take
    /// longer than five minutes to generate via `deepseek-v4-flash:nitro`.
    /// Raise it further for very large corpora; lower it for short snippets.
    ///
    // `--openrouter-timeout` USED to be declared here. It is now a GLOBAL
    // argument on `Cli` (v1.2.3), so redeclaring it on this struct would give
    // clap two arguments with the same id and abort at startup. The flag still
    // accepts being written at the subcommand position — `enrich
    // --openrouter-timeout 600` is unchanged — and the value is read through
    // `openrouter_chat_timeout_secs()` below.
    /// Optional OpenRouter base URL override (reserved; defaults to the public API).
    #[arg(long, value_name = "URL")]
    pub openrouter_base_url: Option<String>,

    // -- Cost controls --
    /// Abort when cumulative cost exceeds this USD budget (API key only; ignored for OAuth).
    #[arg(long, value_name = "USD")]
    pub max_cost_usd: Option<f64>,

    // -- Queue controls --
    /// Resume a previously interrupted run (skip already-done items).
    #[arg(long)]
    pub resume: bool,

    /// Retry only items that failed in a previous run.
    #[arg(long)]
    pub retry_failed: bool,

    /// v1.1.2 (Bug 4): reset `processing` rows whose `claimed_at` is older than
    /// `--stale-claim-secs` (default 1800s = 30 min) back to `pending`, then exit.
    /// Recover rows orphaned by a kill -9 mid-enrich without a full re-scan.
    #[arg(long)]
    pub reset_stale_claims: bool,

    /// v1.1.2 (Bug 4): age threshold (seconds) for `--reset-stale-claims` and
    /// the run-startup stale sweep. Default 1800 (30 min).
    #[arg(long, value_name = "SECONDS", default_value_t = DEFAULT_ENRICH_STALE_CLAIM_SECS)]
    pub stale_claim_secs: u64,

    /// GAP-ENRICH-BACKLOG-CONVERGE: loop scan→drain internally until the queue
    /// empties of eligible items or --max-runtime elapses; removes the need for
    /// an external bash retry loop.
    #[arg(long)]
    pub until_empty: bool,

    /// GAP-ENRICH-BACKLOG-CONVERGE: wall-clock ceiling in seconds for
    /// --until-empty. Defaults to 3600 when omitted.
    #[arg(long, value_name = "SECONDS")]
    pub max_runtime: Option<u64>,

    /// GAP-ENRICH-BACKLOG-CONVERGE: attempts per item before it becomes a
    /// dead-letter (status='dead'). Range 1..=20. Default 8.
    ///
    /// GAP-SG-21: the default was raised from 5 to 8 because GAP-SG-09 now
    /// reclassifies malformed / non-JSON LLM output as TRANSIENT (retryable)
    /// rather than a permanent HardFailure. A flaky structured-output model
    /// (e.g. deepseek-v4-flash:nitro) may emit several bad generations in a row
    /// even after JSON repair (GAP-SG-10) recovers most of them; the extra
    /// attempts give the backlog room to converge before an item is parked in
    /// the dead-letter sink. Permanent faults (ProviderError, NotFound) still
    /// dead-letter on the first attempt regardless of this value.
    #[arg(long, value_name = "N", default_value_t = DEFAULT_ENRICH_MAX_ATTEMPTS, value_parser = clap::value_parser!(u32).range(1..=20))]
    pub max_attempts: u32,

    /// GAP-ENRICH-BACKLOG-CONVERGE: read-only mode — report queue and backlog
    /// counts without calling the LLM or acquiring the singleton.
    ///
    /// Field semantics (v1.1.03 clarification):
    /// - `scan_backlog` = candidates a fresh scan WOULD select from the database
    ///   (REAL pending work, same WHERE predicate as the scanners).
    /// - `queue_pending` = a COMPUTED COUNT over the sidecar queue, NOT a physical
    ///   queue of rows to process — it stays non-zero even after a clean drain.
    /// - `eligible_now == 0` WITH `queue_pending > 0` means COOLDOWN (rate-limit
    ///   backoff), NOT a deadlock: items are parked on `next_retry_at`.
    /// - `eligible_now > 0` stuck against `state: "draining"` IS a deadlock —
    ///   stale `processing` claims hold the state. Run `--reset-stale-claims`
    ///   to clear processing claims older than the threshold.
    #[arg(long)]
    pub status: bool,

    /// GAP-SG-23: list every dead-letter item (status='dead') for the current
    /// operation with its error_class, attempt count and last error message.
    /// Read-only — no LLM, no singleton. Use it to inspect what `--requeue-dead`
    /// would resurrect before running it.
    #[arg(long)]
    pub list_dead: bool,

    /// GAP-SG-11/14: resurrect dead-letter items — move every `status='dead'`
    /// row back to `pending`, zeroing `attempt`, `next_retry_at`, `error` and
    /// `error_class`. Distinct from `--retry-failed`, which only resets the
    /// legacy `status='failed'` rows; dead-letter rows are the terminal sink of
    /// the v1.0.96 converge loop and are never re-selected without this flag.
    /// No LLM call or singleton is taken — it only rewrites queue statuses.
    #[arg(long)]
    pub requeue_dead: bool,

    /// GAP-SG-96 / G-PR-4: list every `status='skipped'` row for the operation
    /// (preservation_failed / veto sink). Read-only — no LLM, no singleton.
    #[arg(long)]
    pub list_skipped: bool,

    /// GAP-SG-96 / G-PR-3/4: resurrect skipped items — move every
    /// `status='skipped'` row back to `pending`, zeroing attempt/backoff so a
    /// lower grounding threshold or corpus fix can re-process them without SQL.
    #[arg(long)]
    pub requeue_skipped: bool,

    /// GAP-SG-66: prune ORPHAN dead-letter rows — remove every `status='dead'`
    /// memory row whose `item_key` (the memory name) no longer exists in the
    /// main DB for this namespace. These are terminal "not found" failures that
    /// `--requeue-dead` can never recover (re-processing re-fails the same way),
    /// so they inflate `queue_dead` forever. Read-only on the main DB; deletes
    /// only confirmed-orphan rows from the queue sidecar. Entity-keyed dead rows
    /// are left untouched. No LLM, no singleton — like `--list-dead`.
    #[arg(long)]
    pub prune_dead_orphans: bool,

    /// v1.1.2: prune dead ENTITY orphan rows — remove every `status='dead'`
    /// `item_type='entity'` row from the queue sidecar. Distinct from
    /// `--prune-dead-orphans` (memory-keyed, consults the main DB): entity dead
    /// rows are terminal artifacts of re-extraction and have no recovery path,
    /// so no main-DB check is needed. Required because the v1.1.1 re-embed bug
    /// left 14680 entity-keyed dead-letter rows that the memory-scoped pruner
    /// cannot reach. No LLM, no singleton — like `--list-dead`.
    #[arg(long, conflicts_with = "prune_dead_orphans")]
    pub prune_dead_entity_orphans: bool,

    /// GAP-SG-16: ignore the per-item backoff cooldown (`next_retry_at`) when
    /// selecting candidates, so items waiting on exponential backoff are
    /// processed immediately. Use to drain a backlog whose cooldown windows are
    /// long but the provider has recovered. Without it, `--status` reports such
    /// items under `waiting` and they are skipped until their `next_retry_at`.
    #[arg(long)]
    pub ignore_backoff: bool,

    /// GAP-SG-28: read-only `body-extract` — extract entities/relationships into
    /// the graph WITHOUT rewriting (or truncating) the memory body. The default
    /// `body-extract` restructures the stored body in place; with this flag the
    /// body is left untouched and only graph bindings are persisted (additive,
    /// via the same upsert path as `memory-bindings`). Ignored for every other
    /// operation.
    #[arg(long)]
    pub body_extract_graph_only: bool,

    /// GAP-ENRICH-BACKLOG-CONVERGE: REST concurrency for `--mode openrouter`
    /// (clamp 1..=16). This is the ONLY flag that controls drain fan-out in
    /// OpenRouter mode; `--llm-parallelism` is inert there.
    #[arg(
        long,
        value_name = "N",
        default_value_t = DEFAULT_ENRICH_REST_CONCURRENCY,
        value_parser = clap::value_parser!(u32).range(
            crate::constants::MIN_ENRICH_REST_CONCURRENCY as i64
                ..=crate::constants::MAX_ENRICH_REST_CONCURRENCY as i64
        )
    )]
    pub rest_concurrency: u32,

    // -- body-enrich specific flags (GAP-18) --
    /// Minimum output character count for body-enrich. Default: 500.
    #[arg(long, value_name = "CHARS", default_value_t = DEFAULT_BODY_ENRICH_MIN_CHARS)]
    pub min_output_chars: usize,

    /// Maximum output character count for body-enrich. Default: 2000.
    #[arg(long, value_name = "CHARS", default_value_t = DEFAULT_BODY_ENRICH_MAX_CHARS)]
    pub max_output_chars: usize,

    /// Accepted as a no-op: no line reads this field.
    ///
    /// The name promises an LLM judge that verifies the enriched body preserves
    /// every fact of the original, and that judge was never wired. A sweep for
    /// `preserve_check` across `src/` returns the declaration and one struct
    /// literal that SETS it — no reader. It is doubly inert, because
    /// `default_value_t = true` on a `bool` also means the flag cannot change
    /// its own value.
    ///
    /// Kept accepted rather than removed so an existing invocation does not
    /// start failing with exit 2, and declared here rather than described,
    /// because a doc comment promising an effect no line produces is how this
    /// class survives review (GAP-SG-302).
    #[arg(
        long,
        default_value_t = true,
        help = "No-op: the preservation judge is not wired; the value is never read"
    )]
    pub preserve_check: bool,

    /// Path to a custom prompt template file for body-enrich.
    #[arg(long, value_name = "PATH")]
    pub prompt_template: Option<PathBuf>,

    // GAP-SG-204. The help text used to end with "It applies only to the
    // subprocess modes", which reads as a live capability sitting behind a mode
    // the operator merely has to select. v1.2.0 deleted those backends and
    // `EnrichMode` has had a single variant ever since, so no such selection
    // exists. A flag that does nothing is a smaller problem than a flag that
    // documents a way to make it work.
    //
    // The rationale lives in `//` and not in `///` on purpose: everything above
    // the `#[arg]` is rendered into `--help`, and an operator reading the flag
    // needs the rule, not the history. Release archaeology belongs in
    // `gaps.md` and the CHANGELOG.
    /// ALWAYS INERT; accepted only so existing scripts keep parsing.
    ///
    /// Use `--rest-concurrency` to size the drain fan-out. Passing this emits a
    /// warning and changes nothing.
    #[arg(long, value_name = "N", value_parser = clap::value_parser!(u32).range(1..=32))]
    pub llm_parallelism: Option<u32>,

    // -- Output / infra --
    /// Emit NDJSON output. Always true; flag accepted for compatibility.
    #[arg(long)]
    pub json: bool,

    /// Database path override.
    #[arg(long)]
    pub db: Option<String>,

    /// G30: poll for the job singleton every second for up to N seconds
    /// when another invocation holds the lock. Default: 0 (fail fast).
    #[arg(long, value_name = "SECONDS")]
    pub wait_job_singleton: Option<u64>,

    /// G30: force acquisition of the singleton lock by removing a stale
    /// lock file from a previously crashed invocation. Use only when you
    /// are certain no other `enrich`/`ingest` is running.
    #[arg(long, default_value_t = false)]
    pub force_job_singleton: bool,

    /// Select a subset of names to enrich instead of the full candidate set.
    /// Comma-separated, e.g. `--names a,b,c`. Semantics depend on the
    /// operation (GAP-CLI-NAMES-01): entity-keyed ops (`entity-descriptions`,
    /// `entity-connect`, …) treat values as **entity** names; memory-keyed
    /// ops (`memory-bindings`, `body-enrich`, …) treat values as **memory**
    /// names. Prefer `--entity-names` / `--memory-names` for an explicit
    /// namespace. Empty when omitted (processes all candidates).
    ///
    /// GAP-SG-18: also a cooldown remedy — when `--status` shows items under
    /// `waiting` (parked on `next_retry_at` backoff), pass the exact names here
    /// to re-enqueue and process just that subset on the next run instead of
    /// waiting for every cooldown to elapse. REQUIRED for `--operation
    /// augment-bindings`, which refuses to re-scan the whole namespace.
    #[arg(long, value_name = "NAMES", value_delimiter = ',')]
    pub names: Vec<String>,

    /// G37: read the subset of memory names from a file (one per line).
    /// Lines starting with `#` and empty lines are ignored. Combined with
    /// `--names` (union) when both are set.
    #[arg(long, value_name = "PATH")]
    pub names_file: Option<PathBuf>,

    /// G35: probe the LLM provider with a 1-turn ping before processing
    /// the batch. Aborts with a clear error if the rate-limit window is
    /// closed (avoids burning N turns only to fail on item 1).
    #[arg(long, default_value_t = false)]
    pub preflight_check: bool,

    /// G35: number of seconds before the OAuth rate-limit reset at which
    /// the preflight probe should refuse to start. Default 300 (5 min).
    #[arg(long, value_name = "SECONDS", default_value_t = DEFAULT_ENRICH_RATE_LIMIT_BUFFER_SECS)]
    pub rate_limit_buffer: u64,

    /// G28-D: refuse to start when the 1-minute load average exceeds
    /// `2 × ncpus` (or XDG `system.max_load_per_ncpu` if set).
    /// Set to false to skip the check on contended runners.
    ///
    /// The key is `system.` and not `enrich.`: this help named a key that the
    /// registry never declared, so the documented `config set` answered exit 1
    /// with a did-you-mean pointing at the real one.
    ///
    /// "Set to false" above was unreachable until v1.2.8: `default_value_t =
    /// true` on a `bool` gives `ArgAction::SetTrue` over a default that is
    /// already `true`, so this flag alone cannot turn the check off.
    /// `--no-max-load-check` is what does, following the `--auto-describe` pair
    /// in `ingest` (GAP-SG-302).
    #[arg(long, default_value_t = true, overrides_with = "no_max_load_check")]
    pub max_load_check: bool,
    /// Skip the load-average refusal on a contended runner.
    #[arg(
        long = "no-max-load-check",
        default_value_t = false,
        help = "Start even when the 1-minute load average is above the ceiling"
    )]
    pub no_max_load_check: bool,

    /// G28-D: when the system is saturated, abort the job after this
    /// many consecutive HardFailure outcomes. Default 5.
    #[arg(long, value_name = "N", default_value_t = DEFAULT_ENRICH_CIRCUIT_BREAKER_THRESHOLD)]
    pub circuit_breaker_threshold: u32,

    /// G29 Step 4: minimum trigram-Jaccard similarity between the
    /// original body and the LLM-rewritten body for the rewrite to be
    /// accepted. Scores below the threshold are rejected and emitted as
    /// `EnrichItemResult::PreservationFailed`. Default 0.7 (per the G29
    /// gap specification). Ignored when `--operation` is not
    /// `body-enrich`.
    #[arg(long, value_name = "FLOAT", default_value_t = DEFAULT_ENRICH_PRESERVE_THRESHOLD)]
    pub preserve_threshold: f64,

    /// GAP-CLI-ED-03: minimum grounding coverage of an entity description
    /// against linked memory bodies. Uses trigram coverage (`|A∩B|/|A|`),
    /// not symmetric Jaccard, because descriptions are short. Only applies to
    /// `entity-descriptions`.
    ///
    /// Precedence: this flag, then XDG
    /// `enrich.entity_description.grounding_threshold`, then the compiled
    /// `DEFAULT_ENRICH_GROUNDING_THRESHOLD`. Read through the accessor of the
    /// same name, never as a bare field.
    ///
    /// The constant is deliberately NOT an intra-doc link here: this field is
    /// `pub` while the constant is `pub(crate)`, and `[lints.rustdoc]` denies
    /// `private_intra_doc_links` since GAP-SG-211.
    ///
    /// Zero is a LITERAL zero — it accepts every candidate. It used to mean
    /// "use the compiled default", and that sentinel was the defect: a
    /// `default_value_t` here meant clap always supplied a value, so the branch
    /// reading the compiled constant only ran when an operator typed `0`
    /// explicitly. Raising the constant therefore changed nothing in
    /// production, and the XDG key had no reader at all.
    #[arg(long, value_name = "FLOAT")]
    pub entity_description_grounding_threshold: Option<f64>,

    /// GAP-CLI-ED-06 / CAPA-B: re-scan entities whose description is empty OR
    /// matches high-precision low-quality compound markers (e.g. "is a software
    /// component", "is a configuration file" — not bare domain phrases).
    /// Once per invocation, reopens matching `skipped`/`done` queue rows for
    /// those scan keys to `pending` (does not reopen `dead`; use
    /// `--requeue-dead`). Default false (write-once for non-empty descriptions).
    #[arg(long, default_value_t = false)]
    pub force_redescribe: bool,

    /// Wave 2 / GAP-CLI-OBS-04: sample N entities with descriptions and report
    /// `quality_pct` / `scan_backlog_low_grounding_est` on `--status`.
    /// `scan_backlog_low_grounding_est` is a **sample-based estimate only** —
    /// it is not a drain backlog and is not processed by `--force-redescribe`.
    /// Precedence: this flag > XDG `enrich.entity_description.quality_sample`
    /// > default 50. Set 0 to disable sampling.
    #[arg(long, value_name = "N", value_parser = crate::parsers::parse_quality_sample_range)]
    pub quality_sample: Option<usize>,

    /// GAP-CLI-NAMES-02: explicit entity name filter (entity-descriptions,
    /// entity-connect, …). Alias of `--names` for entity-keyed ops.
    #[arg(long, value_name = "NAMES", value_delimiter = ',')]
    pub entity_names: Vec<String>,

    /// Restrict an entity-keyed operation to entities carrying this type label.
    ///
    /// v1.2.8: `entity-type-validate` scanned the whole namespace at random,
    /// with no way to aim it. That is the operation which can propose a better
    /// label for the entities the closed vocabulary collapsed into `concept`,
    /// and without a filter the only way to reach them was to pay for every
    /// entity in the graph. `--entity-type concept` aims it at exactly the
    /// bucket that needs revisiting.
    ///
    /// Free text, matched literally against the stored label, because the
    /// vocabulary is open since V017 and a closed value set here would refuse
    /// the very labels this flag exists to look for.
    #[arg(long, value_name = "LABEL")]
    pub entity_type: Option<String>,

    /// GAP-SG-283: entity type labels this project accepts, comma separated.
    ///
    /// `remember --strict-entity-types` and `link --strict-relations` already
    /// let a project declare its vocabulary on the hand-written write channels.
    /// `enrich` declared nothing, and it is the channel that writes type labels
    /// in VOLUME: `entity-type-validate` persists whatever the model returns.
    ///
    /// Precedence: this flag > XDG `enrich.entity_type.allowed_types` > the
    /// canonical vocabulary compiled into the binary. An empty declaration
    /// falls through to the next layer rather than refusing every label.
    ///
    /// Labels are shape-normalised on the way in, so `Issue-Tracker` and
    /// `issue_tracker` declare the same member. What to do with a label outside
    /// the set is `--on-unknown-type`.
    #[arg(long, value_name = "LIST", value_delimiter = ',')]
    pub allowed_types: Vec<String>,

    /// GAP-SG-283: policy for a validated entity type outside `--allowed-types`.
    ///
    /// `keep` (the default) stores the label as written, which is byte-for-byte
    /// the v1.2.8 behaviour — an existing caller passing no flag is unaffected.
    /// `fallback` stores the nearest accepted label and preserves the raw one in
    /// the entity description, so the rewrite has a declared inverse. `strict`
    /// refuses the item with exit 1, mirroring `remember --strict-entity-types`.
    ///
    /// Precedence: this flag > XDG `enrich.entity_type.on_unknown_type` >
    /// `keep`. Applies to `--operation entity-type-validate`; every other
    /// operation ignores it.
    ///
    /// Parsed against the same list the policy module parses, so the flag and
    /// the reader cannot accept different sets.
    #[arg(
        long,
        value_name = "POLICY",
        value_parser = clap::builder::PossibleValuesParser::new(super::events::UNKNOWN_TYPE_POLICIES)
    )]
    pub on_unknown_type: Option<String>,

    /// GAP-CLI-NAMES-02: explicit memory name filter (memory-bindings,
    /// body-enrich, …). Alias of `--names` for memory-keyed ops.
    #[arg(long, value_name = "NAMES", value_delimiter = ',')]
    pub memory_names: Vec<String>,

    /// GAP-CLI-EC-03: limit entity-connect pair scan to the subgraph of
    /// entities linked to this memory name.
    #[arg(long, value_name = "NAME")]
    pub anchor_memory: Option<String>,

    /// GAP-CLI-ED-05: optional domain label for entity-descriptions
    /// (`auto` = no hint, `none` = force neutral, or free-form e.g. `fiscal`).
    /// Precedence: this flag > XDG `enrich.entity_description.domain` > `auto`.
    #[arg(long, value_name = "LABEL", default_value = "auto")]
    pub entity_description_domain: String,

    /// GAP-CLI-PRIO-05: cooperative yield every N processed items so hot-set
    /// entity-descriptions can preempt long entity-connect drains. 0 disables.
    /// XDG key: `enrich.yield_every_n_items` (default 10).
    #[arg(long, value_name = "N")]
    pub yield_every_n_items: Option<usize>,

    /// GAP-CLI-OBS-02 / PRIO-04: run gate ops in order
    /// memory-bindings → entity-descriptions (before entity-connect).
    #[arg(long, default_value_t = false)]
    pub ops_gate: bool,

    /// Emit the JSON Schema for `enrich --status` stdout and exit 0 without
    /// opening the database or calling the LLM (agent-native R-AN-01).
    #[arg(
        long,
        default_value_t = false,
        help = "Print JSON Schema for enrich --status output and exit"
    )]
    pub print_schema: bool,
}

impl EnrichArgs {
    /// GAP-SG-31: resolved enrichment operation.
    ///
    /// `operation` is `Option` so the read-only queue inspectors
    /// (`--status` / `--list-dead` / `--requeue-dead`) can run without it.
    /// Write paths always carry a value (enforced by
    /// `required_unless_present_any` at parse time); the read-only paths fall
    /// back to `memory-bindings`, the most common queue, when it is omitted.
    pub(crate) fn operation(&self) -> EnrichOperation {
        self.operation
            .clone()
            .unwrap_or(EnrichOperation::MemoryBindings)
    }

    /// GAP-SG-31: resolved LLM provider.
    ///
    /// `mode` is `Option` because clap does not require it for the read-only
    /// inspectors, nor for `--dry-run` since GAP-CLI-DRY-01 (v1.1.8). Every
    /// WRITE path still carries a value, enforced by `required_unless_present_any`
    /// at parse time — omitting `--mode` on a write run exits 2.
    ///
    /// The fallback is [`EnrichMode::OpenRouter`], and that choice is load-bearing
    /// rather than arbitrary: GAP-HEADLESS-DEFAULT requires that an omitted
    /// `--mode` never reach a provider that spawns a headless CLI, and OpenRouter
    /// is a REST call that spawns nothing. Do not change it to a subprocess mode.
    ///
    /// Note this doc previously claimed the fallback was "only ever observed by
    /// read-only code". That stopped being true in v1.1.8 when `--dry-run` joined
    /// the exemption list; dry-run observes it too, and is safe for a different
    /// reason — binary resolution is skipped before the mode is inspected.
    pub(crate) fn mode(&self) -> EnrichMode {
        self.mode.clone().unwrap_or(EnrichMode::OpenRouter)
    }

    /// Effective chat-completion budget, in the documented precedence:
    /// `--openrouter-timeout` > XDG `llm.openrouter_timeout_secs` >
    /// [`DEFAULT_OPENROUTER_CHAT_TIMEOUT_SECS`].
    ///
    /// Reads through `runtime_config` rather than off this struct because the
    /// flag is GLOBAL since v1.2.3 and therefore no longer lives here. The
    /// middle layer is the part that used to be unreachable: with the value
    /// read straight off the enum variant, the XDG key could never win, so an
    /// operator who set it saw no effect and no diagnostic.
    pub(crate) fn openrouter_chat_timeout_secs(&self) -> u64 {
        crate::runtime_config::openrouter_chat_timeout_secs(DEFAULT_OPENROUTER_CHAT_TIMEOUT_SECS)
    }

    /// GAP-SG-185: resolved keyset page size for the SCAN phase.
    ///
    /// Precedence: `--scan-page-size` > XDG `enrich.scan_page_size` > default 512.
    pub(crate) fn scan_page_size(&self) -> usize {
        crate::runtime_config::enrich_scan_page_size(self.scan_page_size)
    }

    /// G-PR-7: resolved minimum grounding coverage for entity descriptions.
    ///
    /// Precedence: `--entity-description-grounding-threshold` > XDG
    /// `enrich.entity_description.grounding_threshold` >
    /// [`DEFAULT_ENRICH_GROUNDING_THRESHOLD`]. This is the ONLY reader of that
    /// XDG key; the key was listed as a dead configuration channel until this
    /// accessor existed.
    pub(crate) fn entity_description_grounding_threshold(&self) -> f64 {
        crate::runtime_config::resolve_f64(
            self.entity_description_grounding_threshold,
            "enrich.entity_description.grounding_threshold",
            DEFAULT_ENRICH_GROUNDING_THRESHOLD,
        )
    }
}