fix: backward compatible add_text() (#132)

Signed-off-by: Michele Dolfi <[email protected]>
DS4SD · Jan 21, 2025 · 7e45817 · 7e45817
1 parent 3e4bf7b
commit 7e45817
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/docling_core/types/doc/document.py b/docling_core/types/doc/document.py
@@ -1515,6 +1515,9 @@ def add_text(
         elif label in [DocItemLabel.SECTION_HEADER]:
             return self.add_heading(text=text, orig=orig, prov=prov, parent=parent)
 
+        elif label in [DocItemLabel.CODE]:
+            return self.add_code(text=text, orig=orig, prov=prov, parent=parent)
+
         else:
 
             if not parent: