Update extract_and_store_supabase.py

extract_and_store_supabase.py

@@ -82,12 +82,12 @@ def extract_structure_store_message(user_id:str,message_id:str , attachment_id:s
             if entities is not None:
                 for ent in entities:
                     if ent.get('type') is not None:
-                        entity_type = ent.get('type')
+                        entity_type = ent.get('type') or ""
             
                         # Check if the entity type is in the allowed list
                         if entity_type in allowed_entities:
-                            mention_text = ent.get('mentionText')
-                            normalized_values = ent.get('normalizedValue') if 'normalizedValue' in ent else None
+                            mention_text = ent.get('mentionText') or ""
+                            normalized_values = ent.get('normalizedValue') or ""
             
                             # Initialize a list for the entity type if not already present
                             if entity_type not in document_entities:
@@ -102,10 +102,10 @@ def extract_structure_store_message(user_id:str,message_id:str , attachment_id:s
                         # Handling 'line_item' and its properties (line_item/description, line_item/quantity, etc.)
                         if entity_type == 'line_item' and 'properties' in ent:
                             for prop in ent['properties']:
-                                prop_type = prop.get('type')
+                                prop_type = prop.get('type') or ""
                                 if prop_type in allowed_entities:
-                                    mention_text = prop.get('mentionText')
-                                    normalized_values = prop.get('normalizedValue') if 'normalizedValue' in prop else None
+                                    mention_text = prop.get('mentionText') or ""
+                                    normalized_values = prop.get('normalizedValue') or ""
             
                                     # Initialize a list for the property type if not already present
                                     if prop_type not in document_entities: