worker.py aktualisieren

Use classification fields for mappings and tags if needed
DrRSatzteil · Jan 11, 2024 · 460f3a7 · 460f3a7
1 parent 8d36960
commit 460f3a7
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/mayanmindee/worker.py b/mayanmindee/worker.py
@@ -268,7 +268,10 @@ def process_custom(document_id: int, document_type: str, synchronous: False) ->
         )
 
     for field_name, metadata_mappings in required_fields.items():
-        result = parsed_doc.document.inference.prediction.fields[field_name].contents_string()
+        try:
+            result = parsed_doc.document.inference.prediction.fields[field_name].contents_string()
+        except:
+            parsed_doc.document.inference.prediction.classifications[field_name].contents_string()
         for metadata_mapping in metadata_mappings:
             if "postprocess" in metadata_mapping[1]:
                 result = post_processing(
@@ -292,7 +295,10 @@ def process_custom(document_id: int, document_type: str, synchronous: False) ->
         )
 
     for field_name, tag_mappings in required_fields.items():
-        result = parsed_doc.document.inference.prediction.fields[field_name].contents_string()
+        try:
+            result = parsed_doc.document.inference.prediction.fields[field_name].contents_string()
+        except:
+            parsed_doc.document.inference.prediction.classifications[field_name].contents_string()
         if result:
             for tag_mapping in tag_mappings:
                 if is_similar(