diff --git a/edenai_apis/apis/base64/base64_api.py b/edenai_apis/apis/base64/base64_api.py index 2cefe53c..96883993 100644 --- a/edenai_apis/apis/base64/base64_api.py +++ b/edenai_apis/apis/base64/base64_api.py @@ -406,7 +406,7 @@ def ocr__data_extraction( except ValueError: bbox = BoundingBox.unknown() - if key := value.get(key): + if key := value.get("key"): items.append( ItemDataExtraction( key=key, diff --git a/edenai_apis/apis/eagledoc/eagledoc_api.py b/edenai_apis/apis/eagledoc/eagledoc_api.py index 121622d1..e6eebb3b 100644 --- a/edenai_apis/apis/eagledoc/eagledoc_api.py +++ b/edenai_apis/apis/eagledoc/eagledoc_api.py @@ -4,20 +4,20 @@ import requests +from edenai_apis.apis.eagledoc.eagledoc_ocr_normalizer import ( + eagledoc_financial_parser, + eagledoc_invoice_parser, + eagledoc_receipt_parser, +) from edenai_apis.features import OcrInterface, ProviderInterface +from edenai_apis.features.ocr.financial_parser.financial_parser_dataclass import ( + FinancialParserDataClass, +) from edenai_apis.features.ocr.invoice_parser import InvoiceParserDataClass from edenai_apis.features.ocr.receipt_parser import ReceiptParserDataClass from edenai_apis.loaders.loaders import ProviderDataEnum, load_provider from edenai_apis.utils.exception import ProviderException from edenai_apis.utils.types import ResponseType -from edenai_apis.features.ocr.financial_parser.financial_parser_dataclass import ( - FinancialParserDataClass, -) -from edenai_apis.apis.eagledoc.eagledoc_ocr_normalizer import ( - eagledoc_invoice_parser, - eagledoc_receipt_parser, - eagledoc_financial_parser, -) class EagledocApi(ProviderInterface, OcrInterface): @@ -50,10 +50,8 @@ def _make_post_request(self, file: BufferedReader, endpoint: str = ""): try: original_response = response.json() - except JSONDecodeError: - raise ProviderException( - message="Internal Server Error", code=500 - ) from JSONDecodeError + except JSONDecodeError as exc: + raise ProviderException(message="Internal Server Error", code=500) from exc if response.status_code != 200: raise ProviderException(message=response.json(), code=response.status_code)