Function diff_db_cmd

explainshell/manager.py:932–974

Diff fresh extraction against the database.

(
    ctx: click.Context,
    mode: str,
    files: tuple[str, ...],
    batch: int | None,
    debug: bool,
)

@click.argument("files", nargs=-1, required=True)
@click.pass_context
def diff_db_cmd(
    ctx: click.Context,
    mode: str,
    files: tuple[str, ...],
    batch: int | None,
    debug: bool,
) -> None:
    """Diff fresh extraction against the database."""
    try:
        parsed_mode, model = _parse_mode(mode)
    except ValueError as e:
        raise click.UsageError(str(e))

    if batch is not None:
        if batch < 1:
            raise click.UsageError("--batch must be >= 1")
        if not model:
            raise click.UsageError(
                "--batch requires a model (e.g. llm:gemini/<model>, llm:openai/<model>, or llm:azure/<deployment>)"
            )
        if not model.startswith(_BATCH_MODEL_PREFIXES):
            raise click.UsageError(
                "--batch only supports gemini/, openai/, and azure/ models"
            )

    db_path = _require_db(ctx)
    try:
        gz_files = util.collect_gz_files(list(files))
    except ValueError as e:
        raise click.UsageError(str(e))
    if not gz_files:
        raise click.UsageError("No .gz files found.")

    run_dir: str = _ensure_run_dir(ctx)
    s = store.Store.create(db_path)
    t0 = time.monotonic()
    batch_result = _run_diff_db(
        gz_files, parsed_mode, model, run_dir, debug, s, batch_size=batch
    )
    elapsed = time.monotonic() - t0
    rc = _log_summary(batch_result, 0, elapsed)
    if rc != 0:
        sys.exit(rc)


@diff.command("extractors")