data-science-document-ai 1.61.2__py3-none-any.whl → 1.61.3__py3-none-any.whl

{data_science_document_ai-1.61.2.dist-info → data_science_document_ai-1.61.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: data-science-document-ai
-Version: 1.61.2
+Version: 1.61.3
 Summary: "Document AI repo for data science"
 Author: Naomi Nguyen
 Author-email: naomi.nguyen@forto.com

{data_science_document_ai-1.61.2.dist-info → data_science_document_ai-1.61.3.dist-info}/RECORD

@@ -10,7 +10,7 @@ src/pdf_processing.py,sha256=Fx-Glb9niEUU3WUCrBZ02ZYV-E2vWoUM0ifN7-0A1Q4,19961
 src/postprocessing/common.py,sha256=uFaJYpctS4vr-0Z3InRyfRZcEar0UWpcTxdB_TDCJ5E,26671
 src/postprocessing/postprocess_booking_confirmation.py,sha256=nK32eDiBNbauyQz0oCa9eraysku8aqzrcoRFoWVumDU,4827
 src/postprocessing/postprocess_commercial_invoice.py,sha256=3I8ijluTZcOs_sMnFZxfkAPle0UFQ239EMuvZfDZVPg,1028
-src/postprocessing/postprocess_partner_invoice.py,sha256=WuaTQK5D09dV_QNrh29ZoKX9IvQn2Ub-WnAMyRjCsvI,14240
+src/postprocessing/postprocess_partner_invoice.py,sha256=ijBh8LRN23HWdotqMw6uf0ro1IaiL_obBiiikMwXYXk,14768
 src/prompts/library/arrivalNotice/other/placeholders.json,sha256=1vzly1amgyKt3jr2JJQbb24kNZsnI289iduvoUo5dJU,3061
 src/prompts/library/arrivalNotice/other/prompt.txt,sha256=QNuU-BvMA8VbdupVNapad4O3WmCotH5cKNxImRMbKDk,2906
 src/prompts/library/bookingConfirmation/evergreen/placeholders.json,sha256=49mmxxjExniMqLkT37zd5q8ILCLPGsfugKOMkR926kk,5854
@@ -55,6 +55,6 @@ src/prompts/prompt_library.py,sha256=VJWHeXN-s501C2GiidIIvQQuZdU6T1R27hE2dKBiI40
 src/setup.py,sha256=prJNY3N2qu14ttfCYnvRtpCqdHQeSW34zXwmuAGWX_M,7472
 src/tms.py,sha256=UXbIo1QE--hIX6NZi5Qyp2R_CP338syrY9pCTPrfgnE,1741
 src/utils.py,sha256=Ow5_Jals88o8mbZ1BoHfZpHZoCfig_UQb5aalH-mpWE,17278
-data_science_document_ai-1.61.2.dist-info/METADATA,sha256=eEkafeJHhdHNOAKuEAmx-3dmS33GHvACepPhKgL0XH4,2058
-data_science_document_ai-1.61.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-data_science_document_ai-1.61.2.dist-info/RECORD,,
+data_science_document_ai-1.61.3.dist-info/METADATA,sha256=Lq6ZHTHHCJJg5zOaI7awo2Uf4Mx1ZC0FgG7XmXKaETk,2058
+data_science_document_ai-1.61.3.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+data_science_document_ai-1.61.3.dist-info/RECORD,,

src/postprocessing/postprocess_partner_invoice.py

@@ -119,16 +119,22 @@ def post_process_bundeskasse(aggregated_data):
         if is_forto_recipient(line_item):
             is_recipient_forto = True
 
-    update_recipient_and_vendor(aggregated_data, is_recipient_forto)
+    update_aggregated_data_fields(aggregated_data, is_recipient_forto)
 
 
-def update_recipient_and_vendor(aggregated_data, is_recipient_forto):
-    """Update the recipient and vendor information in the aggregated data."""
+def update_aggregated_data_fields(aggregated_data, is_recipient_forto):
+    """Update the recipient, vendor and deferredDutyPayer information in the aggregated data."""
     # Check if the "recipientName" and "recipientAddress" keys exist
     keys_to_init = ["recipientName", "recipientAddress", "vendorName", "vendorAddress"]
     for key in keys_to_init:
         aggregated_data.setdefault(key, {"formattedValue": "", "documentValue": ""})
 
+    # Update the vendor details always to Bundeskasse Trier
+    aggregated_data["vendorName"]["formattedValue"] = "Bundeskasse Trier"
+    aggregated_data["vendorAddress"][
+        "formattedValue"
+    ] = "Dasbachstraße 15, 54292 Trier, Germany"
+
     if is_recipient_forto:
         # Update the aggregated data with the recipient information
         aggregated_data["recipientName"][
@@ -138,11 +144,17 @@ def update_recipient_and_vendor(aggregated_data, is_recipient_forto):
             "formattedValue"
         ] = "Schönhauser Allee 9, 10119 Berlin, Germany"
 
-    # Update the vendor details always to Bundeskasse Trier
-    aggregated_data["vendorName"]["formattedValue"] = "Bundeskasse Trier"
-    aggregated_data["vendorAddress"][
-        "formattedValue"
-    ] = "Dasbachstraße 15, 54292 Trier, Germany"
+        # Update the defferDutyPayer to Forto Logistics SE & Co KG 'DE789147263644738 - Fort'
+        lineitems = aggregated_data.get("lineItem", [])
+
+        for lineitem in lineitems:
+            if "deferredDutyPayer" in lineitem:
+                lineitem["deferredDutyPayer"] = {
+                    "documentValue": lineitem.get("deferredDutyPayer", {}).get(
+                        "documentValue"
+                    ),
+                    "formattedValue": "DE789147263644738",
+                }
 
 
 def select_unique_bank_account(bank_account):