mindeeapi.py aktualisieren

More api changes
DrRSatzteil · Jan 5, 2024 · 9428805 · 9428805
1 parent 6b7d442
commit 9428805
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mayanmindee/mindeeapi.py b/mayanmindee/mindeeapi.py
@@ -13,10 +13,11 @@ def ocr_standard(pdf_bytes, document_name, document_type):
 
 
 def ocr_custom(pdf_bytes, document_name, account_name, endpoint_name):
-    client = Client(api_key=os.getenv("MINDEE_API_KEY")).create_endpoint(
+    client = Client(api_key=os.getenv("MINDEE_API_KEY"))
+    custom_endpoint = client.create_endpoint(
         account_name=account_name,
         endpoint_name=endpoint_name,
     )
     doc = client.source_from_bytes(pdf_bytes, document_name)
-    parsed_doc = doc.parse(product.TypeCustomV1, endpoint_name=endpoint_name)
+    parsed_doc = client.parse(product.CustomV1, input_source=doc, endpoint=custom_endpoint)
     return parsed_doc