diff --git a/backend/utils/elements.py b/backend/utils/elements.py index 9b3fcef..0361dd3 100644 --- a/backend/utils/elements.py +++ b/backend/utils/elements.py @@ -323,7 +323,8 @@ def _create_text_element(doc: Document, element_class: type[Text]) -> Element: Returns: Element: Text element created from the Document object. """ - match doc.metadata["source"]: + source = doc.metadata.get("source", "content") + match source: case "content": element = element_class( type=doc.metadata["type"], @@ -357,7 +358,8 @@ def _create_image_element(doc: Document, element_class: type[Image]) -> Element: Returns: Element: Image element created from the Document object. """ - match doc.metadata["source"]: + source = doc.metadata.get("source", "content") + match source: case "content": element = element_class( type=doc.metadata["type"],