Update extract_and_store_supabase.py

extract_and_store_supabase.py

@@ -116,14 +116,21 @@ def extract_structure_store_message(user_id:str,message_id:str , attachment_id:s
                                         "mention_text": mention_text,
                                         "normalizedValue": normalized_values
                                     })
-            document_entities['line_item_description'] = document_entities['line_item/description']
-            document_entities['line_item_quantity'] = document_entities['line_item/quantity']
-            document_entities['line_item_amount'] = document_entities['line_item/amount']
-            document_entities['line_item_unit_price'] = document_entities['line_item/unit_price']
-            document_entities.pop('line_item/description', None)
-            document_entities.pop('line_item/quantity', None)
-            document_entities.pop('line_item/amount', None)
-            document_entities.pop('line_item/unit_price', None)
+            if 'line_item/description' in document_entities:
+                document_entities['line_item_description'] = document_entities['line_item/description']
+                document_entities.pop('line_item/description', None)
+            
+            if 'line_item/quantity' in document_entities:
+                document_entities['line_item_quantity'] = document_entities['line_item/quantity']
+                document_entities.pop('line_item/quantity', None)
+            
+            if 'line_item/amount' in document_entities:
+                document_entities['line_item_amount'] = document_entities['line_item/amount']
+                document_entities.pop('line_item/amount', None)
+            
+            if 'line_item/unit_price' in document_entities:
+                document_entities['line_item_unit_price'] = document_entities['line_item/unit_price']
+                document_entities.pop('line_item/unit_price', None)
             document_entities['email'] = email
             document_entities['message_id'] = message_id
             print(document_entities)