Merge pull request #285 from edenai/SD2-925-eagledoc-jsondecodeerror

Sd2 925 eagledoc jsondecodeerror
edenai · Dec 20, 2024 · fce78aa · fce78aa
2 parents c6bb767 + 18a87ec
commit fce78aa
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 13 deletions.
diff --git 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
@@ -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)