fix create element

artefactory · Mar 14, 2024 · a153ef5 · a153ef5
1 parent 380f9c4
commit a153ef5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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"],