SIGN IN SIGN UP

feat(extraction): add Granite Vision 4.1 as alternative KVP extraction model (#3398)

Introduce a unified TransformersExtractionModel that supports multiple
prompt styles via an ExtractionPromptStyle enum. This replaces the
need for separate model classes per VLM.

- Add ExtractionPromptStyle enum (NUEXTRACT, GRANITE_VISION)
- Add prompt_utils.py with style-specific prompt builders
- Add TransformersExtractionModel with prompt-style dispatch
- Add GRANITE_VISION_4_1_TRANSFORMERS model spec
- Add extraction_prompt_style field to VlmExtractionPipelineOptions

Signed-off-by: Ben Wiesel <benwiesel@ibm.com>
Co-authored-by: Ben Wiesel <benwiesel@ibm.com>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
B
benvizel committed
b5f2e530e2d3f55aa26b6513836a55903abe32b3
Parent: 0c31706
Committed by GitHub <noreply@github.com> on 5/12/2026, 4:53:22 AM