Update extract_and_store_supabase.py

extract_and_store_supabase.py

@@ -70,7 +70,7 @@ 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:
-                        if entity_type in allowed_entities:
+                        if ent.get('type') in allowed_entities:
                             mention_text = ent.get('mentionText')
                             normalised_values = ent.get('normalizedValue') if 'normalizedValue' in ent else None
                             document_entities[ent.get('type')] = {"mention_text":mention_text,"normalizedValue":normalised_values}