fix: Fix KeyError in isd_to_elements (#876)

Author: awalker4

CHANGELOG.md

@@ -5,6 +5,7 @@
 ### Features
 
 ### Fixes
+* Fix KeyError when `isd_to_elements` doesn't find a type
 
 ### BREAKING CHANGES
 

test_unstructured/staging/test_base_staging.py

@@ -46,6 +46,7 @@ def test_isd_to_elements():
         {"text": "Blurb2", "type": "Title"},
         {"text": "Blurb3", "type": "ListItem"},
         {"text": "Blurb4", "type": "BulletedText"},
+        {"text": "No Type"},
     ]
 
     elements = base.isd_to_elements(isd)

unstructured/staging/base.py

@@ -76,7 +76,7 @@ def isd_to_elements(isd: List[Dict[str, Any]]) -> List[Element]:
         if _metadata_dict is not None:
             metadata = ElementMetadata.from_dict(_metadata_dict)
 
-        if item["type"] in TYPE_TO_TEXT_ELEMENT_MAP:
+        if item.get("type") in TYPE_TO_TEXT_ELEMENT_MAP:
             _text_class = TYPE_TO_TEXT_ELEMENT_MAP[item["type"]]
             elements.append(
                 _text_class(
@@ -85,7 +85,7 @@ def isd_to_elements(isd: List[Dict[str, Any]]) -> List[Element]:
                     metadata=metadata,
                 ),
             )
-        elif item["type"] == "CheckBox":
+        elif item.get("type") == "CheckBox":
             elements.append(
                 CheckBox(
                     checked=item["checked"],