diff --git a/tasks/INDEX.md b/tasks/INDEX.md index 723f065..3ee249e 100644 --- a/tasks/INDEX.md +++ b/tasks/INDEX.md @@ -65,10 +65,11 @@ P0~P5 는 직렬. P6~P9 는 P5 이후 병렬 가능. - P5 — [p5/](p5/) — 2 components - [p5-1 golden-fixture-runner](p5/p5-1-golden-fixture-runner.md) - [p5-2 metrics-compare](p5/p5-2-metrics-compare.md) -- P6 — [p6/](p6/) — 3 components +- P6 — [p6/](p6/) — 4 components - [p6-1 image-extractor-exif](p6/p6-1-image-extractor-exif.md) - [p6-2 ocr-adapter](p6/p6-2-ocr-adapter.md) - [p6-3 caption-adapter](p6/p6-3-caption-adapter.md) + - [p6-4 image-ingest-wiring](p6/p6-4-image-ingest-wiring.md) - P7 — [p7/](p7/) — 2 components - [p7-1 pdf-text-extractor](p7/p7-1-pdf-text-extractor.md) - [p7-2 pdf-page-chunker](p7/p7-2-pdf-page-chunker.md) diff --git a/tasks/p6/p6-4-image-ingest-wiring.md b/tasks/p6/p6-4-image-ingest-wiring.md new file mode 100644 index 0000000..716a2fb --- /dev/null +++ b/tasks/p6/p6-4-image-ingest-wiring.md @@ -0,0 +1,259 @@ +--- +phase: P6 +component: kebab-app (image ingest dispatch + chunking) +task_id: p6-4 +title: "Wire ImageExtractor + OCR + caption into kebab-app::ingest end-to-end" +status: planned +depends_on: [p6-1, p6-2, p6-3, p1-6, p3-5] +unblocks: [] +contract_source: ../../docs/superpowers/specs/2026-04-27-kebab-final-form-design.md +contract_sections: [§3.4 ImageRefBlock, §6.1 ingest pipeline, §7.2 Extractor/Chunker traits, §9.1 image extraction policy] +--- + +# p6-4 — Image ingest wiring (kebab-app + chunker) + +## Goal + +Make `kebab ingest` end-to-end functional for image assets. The library-level pipeline (`ImageExtractor`, `OllamaVisionOcr`, `apply_caption`) is complete and tested in isolation — this task connects the wires from `kebab-source-fs` (which already classifies `.png` / `.jpg` / `.webp` / `.gif` / `.tiff` as `MediaType::Image(_)`) through `kebab-app::ingest_with_config` to `kebab-store-sqlite` + `kebab-store-vector`, so a user running `kebab ingest` against a workspace containing diagrams / screenshots / camera photos sees those assets indexed and searchable. + +## Why now / why this size + +P6-1 / P6-2 / P6-3 each shipped a focused library and a passing test suite, but all three deliver value only after `kebab-app::ingest` learns how to dispatch on `MediaType::Image(_)`. The wiring is small (one new dispatch arm, one chunking branch, one LM-construction call site) but materially user-facing — without it, the entire P6 phase is invisible from the CLI. Pulling this into its own task keeps the P6-1/2/3 specs frozen as written while letting the integration evolve under its own contract. + +## Allowed dependencies + +`kebab-app` 의 현재 Cargo.toml 그대로의 surface — 본 task 는 그 위에 `kebab-parse-image` 한 줄을 추가합니다. + +- `kebab-core` +- `kebab-config` +- `kebab-source-fs` +- `kebab-parse-md`, `kebab-parse-types` +- `kebab-normalize` +- `kebab-chunk` (image-document branch in `md-heading-v1` — this task extends it) +- `kebab-store-sqlite`, `kebab-store-vector` +- `kebab-search` +- `kebab-embed`, `kebab-embed-local` +- `kebab-llm`, `kebab-llm-local` (constructs `OllamaLanguageModel` for caption) +- `kebab-rag` +- **`kebab-parse-image` (NEW — added by this task)** +- `anyhow`, `serde_json`, `tracing` + +## Forbidden dependencies + +- `kebab-tui`, `kebab-desktop` (P9 미시작 — UI crate 가 ingest 를 호출하면 layering 위반) +- `kebab-eval` (cycle 위험 — eval 이 ingest 를 호출하므로 그 반대는 금지) +- 본 task 가 신설하는 자체 image extractor / OCR / caption 비즈니스 로직 — 모두 `kebab-parse-image` 에 이미 존재. `kebab-app` 안에 image-specific 로직 추가 금지 (얇은 dispatch + glue 만 허용). + +## Inputs + +| input | type | source | +|-------|------|--------| +| workspace assets | `RawAsset` stream | `kebab-source-fs::SourceConnector::scan` (already classifies image types) | +| image bytes | `&[u8]` | filesystem read in `kebab-app` | +| `Config` | `kebab_config::Config` | CLI `--config` flag → `Config::load` | +| OCR config | `config.image.ocr` | P6-2 | +| Caption config | `config.image.caption` | P6-3 | + +## Outputs + +| output | type | downstream | +|--------|------|------------| +| `CanonicalDocument` per image | written via `DocumentStore::put_document` | `kebab-store-sqlite` | +| One synthesized chunk per image | `Vec` (length 1) | `kebab-store-sqlite::put_chunks` + `kebab-store-vector::upsert` | +| Updated `IngestReport` counters | `scanned / new / updated / skipped / errors` | wire output (`ingest_report.v1`) | + +## Public surface + +No new public types. The wiring exists inside `kebab-app::ingest_with_config` (and its private helpers). `kebab-chunk` gains one image-only branch in `md_heading_v1`: + +```rust +// crates/kebab-chunk/src/md_heading_v1.rs (additions only — sketch) +fn chunk(&self, doc: &CanonicalDocument, policy: &ChunkPolicy) -> Result> { + if is_image_only_document(doc) { + return Ok(vec![image_chunk(doc, policy)?]); + } + // ... existing markdown heading logic untouched ... +} + +/// Returns true iff `doc` is an image-only document (single `ImageRef` +/// block). P6-1's `ImageExtractor` already guarantees this shape today +/// — the predicate exists as a defensive guard against (a) a future +/// task that introduces multi-block image documents, and (b) accidental +/// route-through of a `[Block::Heading, Block::ImageRef, ...]` shape +/// that would look image-ish but should still flow through the +/// markdown chunker. +fn is_image_only_document(doc: &CanonicalDocument) -> bool { + doc.blocks.len() == 1 + && matches!(doc.blocks.first(), Some(Block::ImageRef(_))) +} + +fn image_chunk(doc: &CanonicalDocument, policy: &ChunkPolicy) -> Result { + let block = match &doc.blocks[0] { + Block::ImageRef(b) => b, + _ => unreachable!("guarded by is_image_only_document"), + }; + let text = compose_image_chunk_text(block); + // chunk_id derived from doc_id + chunker_version + [block_id] + policy_hash + // (existing §4.2 recipe — no new ID kind). + Ok(Chunk { /* ... */ }) +} +``` + +`compose_image_chunk_text` is the canonical place where the (β) plain-concatenation policy lives: + +```rust +fn compose_image_chunk_text(block: &ImageRefBlock) -> String { + let alt = if block.alt.is_empty() { + // alt should never be empty at this stage because P6-1 falls + // back to the filename when it is, but the chunker stays + // defensive — an empty first line would degrade lexical + // search hits on filenames. + block.src.rsplit('/').next().unwrap_or("[image]").to_string() + } else { + block.alt.clone() + }; + let ocr = block.ocr.as_ref().map(|o| o.joined.as_str()).unwrap_or(""); + let cap = block.caption.as_ref().map(|c| c.text.as_str()).unwrap_or(""); + [alt, ocr.to_string(), cap.to_string()] + .into_iter() + .filter(|s| !s.is_empty()) + .collect::>() + .join("\n\n") +} +``` + +## Behavior contract + +### ingest dispatch (kebab-app) + +- `kebab-app::ingest_with_config` reads the asset stream from `kebab-source-fs`. For each `RawAsset`: + - `MediaType::Markdown` → existing markdown extractor path (unchanged). + - `MediaType::Image(_)` → new image branch (this task). + - `MediaType::Pdf | MediaType::Audio(_) | MediaType::Other(_)` → `skipped += 1` (existing behaviour). +- Image branch: + 1. Read bytes via `kebab-source-fs` (existing helper). + 2. Build a `kebab_core::ExtractContext { asset, workspace_root, config: &ExtractConfig::default() }`. + 3. Call `ImageExtractor::new().extract(&ctx, &bytes)`. Failure (`Err(_)`) → `errors += 1`, log, continue to next asset (do not abort the whole ingest). + 4. Take a mutable borrow of `doc.blocks[0]` (must be `Block::ImageRef`); take a mutable borrow of `doc.provenance.events`. + 5. If `config.image.ocr.enabled`: + - Build `OllamaVisionOcr::new(&cfg)`. + - Call `apply_ocr(&engine, &bytes, block, lang_hint, &mut events)`. + - Failure → log a `tracing::warn!`, append `ProvenanceKind::Warning` event with the error message via the helper, continue. + - `block.ocr` stays `None` on failure (P6-2 contract). + 6. If `config.image.caption.enabled`: + - Build the LM **once per ingest session** (not per asset) — see "LM construction" below. + - Call `apply_caption(&*llm, &bytes, block, lang_hint, &cfg, &mut events)`. + - Failure → same lenient policy as OCR: warn, log, continue. `block.caption` stays `None`. + 7. Pass the (possibly partially-populated) `CanonicalDocument` to the existing chunker → embedder → store path, identical to markdown. + +### Parallelism + +- The image branch shares the existing markdown worker pool — one asset dispatch unit (markdown OR image) per worker — so `config.indexing.max_parallel_extractors` keeps its current meaning. The current `kebab-app` ingest is sequential per-asset (single worker irrespective of the knob value); image branch adds zero new concurrency. A future P+ task may parallelise both branches, at which point the OCR / caption HTTP calls naturally become the throughput ceiling (one in-flight request per worker — `reqwest::blocking::Client` is shared but each call blocks its worker thread until response). +- Implication for sizing: a 5000-asset ingest with OCR enabled runs as roughly `5000 × (per-asset OCR latency)` end-to-end. With `gemma4:e4b` at ~3-5s per call this is the 4-7 hour range the brainstorming flagged. Books-as-PDF route bypasses this entirely. + +### Lang hint + +- `lang_hint: Option<&Lang>` passed to `apply_ocr` / `apply_caption` reads from `doc.lang` (set to `Lang("und")` by P6-1). +- `kebab-parse-image` already special-cases `"und"` so the prompt does not embed a misleading hint. + +### LM / OCR engine construction + +Both the caption LM and the OCR adapter wrap `reqwest::blocking::Client`, whose internal `Arc` makes a single instance cheap to share across all assets. Both are constructed **once per ingest invocation**, before the asset loop, gated on the matching `enabled` flag. + +- **Caption** — when `config.image.caption.enabled = true`, build `OllamaLanguageModel::new(&cfg)` once. Stored as `Box` (or trait object behind `&`) and passed to every `apply_caption` call. +- **OCR** — when `config.image.ocr.enabled = true`, build `OllamaVisionOcr::new(&cfg)` once. Same `Arc`-share property; passed by `&` to every `apply_ocr` call. +- Endpoints — `OllamaVisionOcr::new` already falls back to `models.llm.endpoint` when `image.ocr.endpoint` is `None`, so a single host typically serves both LLM and OCR. The two adapters can therefore share an Ollama host or run against separate hosts independently. +- **Construction failure** (e.g. invalid endpoint string, empty model id) → ingest aborts with the constructor's error before any asset is scanned. Never silently disables OCR / caption. +- Per-asset cost — only the HTTP call to Ollama. Adapter struct is reused. + +### Chunking (kebab-chunk md-heading-v1) + +- The chunker gains an image-only branch as sketched above. The branch: + - Returns exactly one `Chunk` per image document. + - `chunk.text` = (β) plain concatenation of `[alt, ocr_joined, caption_text]` joined by `\n\n`, dropping empty parts. + - `chunk.block_ids = vec![block.common.block_id.clone()]`. + - `chunk.heading_path = vec![]` (image documents have no heading hierarchy). + - `chunk.source_spans = vec![block.common.source_span.clone()]` — `Vec` per `kebab_core::Chunk` definition; image branch contributes one element holding the `Region { x, y, w, h }` from P6-1. + - `chunk.token_estimate` follows the existing `md-heading-v1` token-count convention (whitespace-segmented words clamped to `policy.target_tokens`). + - `chunk.policy_hash` is the existing `ChunkPolicy` hex digest the chunker already computes for markdown; image branch reuses the same value to keep policy edits invalidating image chunks alongside markdown chunks. +- Determinism: the existing `id_for_chunk(doc_id, chunker_version, &[block_id], policy_hash)` recipe applies unchanged. +- Oversized text: an image whose `ocr.joined` exceeds `policy.target_tokens` produces an oversized chunk. Acceptable in v1 — the user-facing scenario (diagrams / screenshots / camera photos) typically yields ≤ 1 page of OCR. Books are routed through P7 PDF instead (see "Out of scope"). A `tracing::warn!` fires when this happens so a future P+ task can quantify how often the boundary is hit. + +### `enabled = false` for both + +- When `config.image.ocr.enabled = false` AND `config.image.caption.enabled = false`, the image is still extracted, stored, and chunked — the user gets EXIF + dimensions + filename indexed, with empty OCR / caption. +- The synthesized chunk text falls back to just the filename. Lexical search on filenames still works; vector search produces a best-effort embedding from a one-line input. + +### Failure semantics summary + +| failure | counter | doc stored? | provenance | +|---|---|---|---| +| `ImageExtractor::extract` Err (decode fail / unrecognised) | `errors+=1` | no | n/a | +| `MediaType::Image(_)` but `supports()` returns false (won't happen with current trait — defensive) | `skipped+=1` | no | n/a | +| `apply_ocr` Err | unchanged | yes (block.ocr = None) | `ProvenanceKind::Warning`, agent `kb-app` | +| `apply_caption` Err | unchanged | yes (block.caption = None) | `ProvenanceKind::Warning`, agent `kb-app` | +| HEIC / RAW → `MediaType::Other(_)` | `skipped+=1` (existing) | no | n/a | + +## Storage / wire effects + +- `kebab-store-sqlite::documents` table gains rows whose `parser_version = "image-meta-v1"`. +- `kebab-store-sqlite::blocks` table gains rows of `block_kind = "imageref"`. +- `kebab-store-sqlite::chunks` table gains rows whose `chunker_version = "md-heading-v1"` AND whose `block_ids` reference a single `imageref` block. +- `kebab-store-vector::chunk_embeddings__` gains one vector per image chunk. +- `IngestReport` (wire `ingest_report.v1`) counters update naturally: `scanned` includes images, `new` / `updated` track image docs, `errors` counts decode failures, `skipped` counts unsupported formats. +- No new wire schemas or `kebab-core` types. + +## Test plan + +| kind | description | fixture / data | +|------|-------------|----------------| +| integration | TempDir KB + 1 PNG (`hello-world.png` with text) + `image.ocr.enabled = true` + `image.caption.enabled = false` (mock LM unused) → `kebab-app::ingest` produces 1 doc + 1 chunk; chunk text contains the filename + OCR text | `kebab-app/tests/image_pipeline.rs` (new), wiremock for OCR Ollama call | +| integration | Same fixture but `caption.enabled = true` with mock LM returning `"a red square with text"` → chunk text contains alt + OCR + caption joined by `\n\n` | wiremock for Ollama (both `/api/generate` calls) | +| integration | Determinism: ingest the same PNG twice → identical `doc_id`, `chunk_id` (P1 idempotency contract holds) | inline | +| integration | OCR Ollama returns 503 → asset still indexed; `block.ocr = None`; provenance has Warning event; `errors` counter NOT incremented; ingest returns Ok | wiremock | +| integration | Caption Ollama returns 503 → asset still indexed; `block.caption = None`; provenance Warning; `errors` not incremented | wiremock | +| integration | `image.ocr.enabled = false` AND `image.caption.enabled = false` → image still indexed; chunk text = filename only | inline | +| integration | Hybrid search across mixed corpus (1 markdown + 1 PNG) returns image chunk for an OCR-text query | inline (real `multilingual-e5-small` embedding) | +| integration | `kebab inspect doc ` returns the image `CanonicalDocument` with `block.ocr` / `block.caption` populated | inline | +| unit | `kebab-chunk::md_heading_v1::is_image_only_document` returns true for `[ImageRef]`, false for `[Heading, ImageRef]` (image embedded in markdown — currently a P+ case but the predicate must not misfire) | unit | +| unit | `compose_image_chunk_text` drops empty parts: alt-only, alt+ocr, alt+caption, alt+ocr+caption all formatted correctly | unit | +| smoke | Update `docs/SMOKE.md` so the runbook ingests at least one image fixture and verifies search-by-OCR-text works | docs change | + +The opt-in real-Ollama integration test from P6-2 / P6-3 stays inside `kebab-parse-image`; this task's integration tests use wiremock so `cargo test --workspace` stays hermetic. + +## Definition of Done + +### Spec PR (this PR — `spec/p6-4-image-ingest-wiring`) + +- [x] `tasks/p6/p6-4-image-ingest-wiring.md` 작성 + self-review (placeholder / 모순 / 모호성 / scope) +- [x] `tasks/INDEX.md` "P6 — 4 components" 반영 +- [x] PR 본문에 design §3.4, §6.1, §9.1 링크 +- [x] brainstorming 의 모든 결정 (옵션 A 청킹 / β 청크 텍스트 / Lenient 실패 정책 / LM 1회 빌드 / 책 P7 이관 / P6-5 미시작) 본문 반영 + +### Implementation PR (follow-up — `feat/p6-4-image-ingest-wiring`) + +- [ ] `cargo check --workspace` passes +- [ ] `cargo test --workspace --no-fail-fast -j 1` passes (all new integration tests green) +- [ ] `cargo clippy --workspace --all-targets -- -D warnings` passes +- [ ] `kebab ingest` against a TempDir KB containing 1 markdown + 1 PNG produces `scanned 2 / new 2 / errors 0` +- [ ] `kebab search --mode lexical ""` returns the image chunk +- [ ] `kebab inspect doc ` shows non-empty `block.ocr` / `block.caption` +- [ ] `docs/SMOKE.md` includes an image-fixture step + +## Out of scope + +- **Books / scanned PDFs** — routed through P7 PDF pipeline. P7's PDF text extractor handles text-embed PDFs natively; scanned PDFs are P7's internal concern (page render → OCR call to the same `kebab-parse-image::OllamaVisionOcr` adapter). +- **Image-region chunker** — current `OllamaVisionOcr` returns a single full-image region, so per-region chunking has no signal. When a region-aware engine (Tesseract / Apple Vision sidecar) lands, a separate `image-region-chunker` task can split on `OcrText.regions[]`. +- **Long-OCR splitting** — image documents whose OCR exceeds `target_tokens` produce oversized chunks. Acceptable for diagrams / screenshots / photos (the v1 user scenario per the brainstorming with the user). Books deliberately use the PDF path instead. +- **Retry mechanism for partial OCR / caption failures** — current escape hatch is "delete the asset, re-ingest". A `kebab ingest --retry-image-analysis` flag is a P+ enhancement once operational data shows it's needed. +- **Parallelism beyond `config.indexing.max_parallel_extractors`** — the existing knob applies. Per-image OCR-bound parallelism (multiple in-flight Ollama calls) is a P+ scale-hardening task that has been explicitly de-scoped because the user routes books through PDF instead of image. +- **Progress reporting** (`kebab ingest --progress`) — same reasoning as parallelism. +- **W3C Media Fragments citation form** (`path#xywh=...`) — `Citation` already carries `Region` source span; fragment URI rendering can be a wire-only follow-up when downstream UI needs it. +- **Wire schema bump** — no new wire types; `ingest_report.v1` counters absorb image events naturally. + +## Risks / notes + +- **OCR / caption failures are silent in `IngestReport` counters.** The user only sees them via `--debug` traces or `kebab inspect doc ` (Provenance Warning events). This is the intentional Lenient policy from the brainstorming; flag in the PR description so users know how to detect partial failures. A future spec extension could introduce a `image_ocr_failed` / `image_caption_failed` counter alongside `errors`. +- **LM endpoint validation runs once per ingest.** A misconfigured `models.llm.endpoint` aborts ingest before any asset is processed. This is correct fail-fast behaviour but means a single broken endpoint takes the whole ingest down — even markdown assets that don't need the LM. The user fix is `image.caption.enabled = false` or correcting the endpoint. +- **`unreachable!` in the chunker branch is guarded by `is_image_only_document`.** If a future task introduces multi-block image documents (e.g. embedded markdown caption alongside the image), the predicate must change in lockstep. Keep both functions in the same module so the invariant is local. +- **Determinism stress.** The existing markdown path's `kb-normalize::build_canonical_document` shares one `now_utc()` reading across the per-document Provenance events. The image path needs the same: P6-1 already does `let now = OffsetDateTime::now_utc();` once and reuses it for both events; this task's wiring must not introduce a second `now()` call between extract and apply_ocr/caption that would break per-document timestamp parity.