diff --git a/src/core/extraction/__init__.py b/src/core/extraction/__init__.py index 8cbd4ae..166edc6 100644 --- a/src/core/extraction/__init__.py +++ b/src/core/extraction/__init__.py @@ -11,6 +11,7 @@ from .extractor import ( ) from .llm_extractor import ( CACHE_DIR_NAME, + _image_to_base64, build_extraction_user_message, extract_document, extract_normal_info, @@ -25,6 +26,7 @@ from .llm_extractor import ( __all__ = [ "CACHE_DIR_NAME", + "_image_to_base64", "build_extraction_user_message", "extract_document", "extract_normal_info", diff --git a/src/core/matching/__init__.py b/src/core/matching/__init__.py index 1f83b01..67599c0 100644 --- a/src/core/matching/__init__.py +++ b/src/core/matching/__init__.py @@ -3,6 +3,28 @@ 提供发票与支付记录的金额匹配功能。 """ -from .matcher import match_invoices_to_cards +from .matcher import ( + _build_invoice_summary, + _build_payment_records, + _invoices_to_records, + _match, + _match_by_filename, + _match_one_to_many, + _match_one_to_one, + _relative_tolerance, + _safe_float, + match_invoices_to_cards, +) -__all__ = ["match_invoices_to_cards"] +__all__ = [ + "_build_invoice_summary", + "_build_payment_records", + "_invoices_to_records", + "_match", + "_match_by_filename", + "_match_one_to_many", + "_match_one_to_one", + "_relative_tolerance", + "_safe_float", + "match_invoices_to_cards", +] diff --git a/src/infra/documents/__init__.py b/src/infra/documents/__init__.py index 1eb5675..4598a09 100644 --- a/src/infra/documents/__init__.py +++ b/src/infra/documents/__init__.py @@ -10,6 +10,8 @@ from .consumable import ( ) from .invoice import ( CACHE_DIR_NAME, + INVOICE_LEVEL_COLUMNS, + PAYMENT_RECORD_COLUMNS, classify_invoice_batch, load_csv, load_invoice_csv, @@ -20,6 +22,8 @@ from .invoice import ( __all__ = [ "CACHE_DIR_NAME", + "INVOICE_LEVEL_COLUMNS", + "PAYMENT_RECORD_COLUMNS", "classify_invoice_batch", "load_csv", "load_invoice_csv", diff --git a/tests/test_extractor.py b/tests/test_extractor.py index f5eaa02..6283323 100644 --- a/tests/test_extractor.py +++ b/tests/test_extractor.py @@ -129,7 +129,7 @@ class TestExtractInvoices: } ] - monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match) + monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match) records, apps, groups = extract_invoices(str(tmp_path)) @@ -194,7 +194,7 @@ class TestExtractInvoices: }, ] - monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match) + monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match) records, apps, groups = extract_invoices(str(tmp_path)) @@ -241,7 +241,7 @@ class TestExtractInvoices: } ] - monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match) + monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match) records, apps, groups = extract_invoices(str(tmp_path)) @@ -283,7 +283,7 @@ class TestExtractInvoices: } ] - monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match) + monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match) records, apps, groups = extract_invoices(str(tmp_path)) @@ -324,7 +324,7 @@ class TestExtractInvoices: } ] - monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match) + monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match) records, apps, groups = extract_invoices(str(tmp_path))