fegger
7e706793fa
- Add AGENTS.md and project-specific Zed skills (odoo-ocr-pipeline, odoo-xml-import, local-vlm-client). - Implement Pydantic schemas for documents, invoices, review results, and VLM responses. - Add unified BaseVLMClient with Ollama implementation and llama.cpp stub. - Build pipeline stages: loader, classifier, digital_pdf/scanned_print/handwritten/mixed_unknown branches, extractor, reviewer, xml_builder. - Add CLI entry point with sidecar JSON and confidence-gated XML output. - Include prompts for classifier, OCR, extraction, and review models. - Add tests with FakeVLMClient; pytest, ruff, and mypy all pass.
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
"""Tests for the structured extraction stage."""
|
|
|
|
|
|
import pytest
|
|
from PIL import Image
|
|
|
|
from odoo_ocr.config import Settings
|
|
from odoo_ocr.pipeline.extractor import extract_invoice
|
|
from odoo_ocr.schemas import ExtractedInvoice
|
|
from tests.conftest import FakeVLMClient
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_extract_invoice() -> None:
|
|
client = FakeVLMClient(
|
|
responses={
|
|
"invoice data extraction": {
|
|
"vendor_name": "Acme",
|
|
"vendor_address": None,
|
|
"vendor_vat": None,
|
|
"invoice_number": "INV-1",
|
|
"invoice_date": "2024-01-01",
|
|
"due_date": None,
|
|
"currency": "EUR",
|
|
"payment_terms": None,
|
|
"line_items": [
|
|
{
|
|
"description": "Widget",
|
|
"quantity": 2.0,
|
|
"unit_price": 50.0,
|
|
"total_price": 100.0,
|
|
"tax_rate": 0.0,
|
|
}
|
|
],
|
|
"subtotal": 100.0,
|
|
"tax_total": 0.0,
|
|
"total": 100.0,
|
|
"iban": None,
|
|
"raw_ocr_text": "",
|
|
}
|
|
}
|
|
)
|
|
result = await extract_invoice("raw text", [Image.new("RGB", (10, 10))], client, Settings())
|
|
assert isinstance(result, ExtractedInvoice)
|
|
assert result.vendor_name == "Acme"
|
|
assert result.line_items_sum() == 100
|