fix(doctags): fix code export (#181)

Signed-off-by: Panos Vagenas <[email protected]>

Author: vagenas

docling_core/types/doc/document.py

@@ -800,7 +800,7 @@ def export_to_document_tokens(
         :param add_content: bool:  (Default value = True)
 
         """
-        body = f"<{self.label.value}{new_line}"
+        body = f"<{self.label.value}>{new_line}"
 
         if add_location:
             body += self.get_location_tokens(
@@ -813,7 +813,7 @@ def export_to_document_tokens(
         if add_content and self.text is not None:
             body += f"<_{self.code_language.value}_>{self.text}{new_line}"
 
-        body += f"</{self.label.value}\n"
+        body += f"</{self.label.value}>\n"
 
         return body
 

test/data/doc/constructed_doc.dt

@@ -30,7 +30,7 @@ Affiliation 2</text>
 <list_item>item 2 of neighboring list</list_item>
 <unordered_list><list_item>item 1 of sub list</list_item>
 <paragraph>Here a code snippet:</paragraph>
-<code<_unknown_><p>Hello world</p></code
+<code><_unknown_><p>Hello world</p></code>
 <paragraph>(to be displayed inline)</paragraph>
 </unordered_list>
 <paragraph>Here a formula:</paragraph>
@@ -39,7 +39,7 @@ Affiliation 2</text>
 </unordered_list>
 </unordered_list>
 <paragraph>Here a code block:</paragraph>
-<code<_unknown_>print("Hello world")</code
+<code><_unknown_>print("Hello world")</code>
 <paragraph>Here a formula block:</paragraph>
 <formula>E=mc^2</formula>
 <paragraph>The end.</paragraph>

test/data/doc/constructed_doc.dt.gt

@@ -30,7 +30,7 @@ Affiliation 2</text>
 <list_item>item 2 of neighboring list</list_item>
 <unordered_list><list_item>item 1 of sub list</list_item>
 <paragraph>Here a code snippet:</paragraph>
-<code<_unknown_><p>Hello world</p></code
+<code><_unknown_><p>Hello world</p></code>
 <paragraph>(to be displayed inline)</paragraph>
 </unordered_list>
 <paragraph>Here a formula:</paragraph>
@@ -39,7 +39,7 @@ Affiliation 2</text>
 </unordered_list>
 </unordered_list>
 <paragraph>Here a code block:</paragraph>
-<code<_unknown_>print("Hello world")</code
+<code><_unknown_>print("Hello world")</code>
 <paragraph>Here a formula block:</paragraph>
 <formula>E=mc^2</formula>
 <paragraph>The end.</paragraph>

test/data/doc/constructed_document.yaml.dt

@@ -30,7 +30,7 @@ Affiliation 2</text>
 <list_item>item 2 of neighboring list</list_item>
 <unordered_list><list_item>item 1 of sub list</list_item>
 <paragraph>Here a code snippet:</paragraph>
-<code<_unknown_><p>Hello world</p></code
+<code><_unknown_><p>Hello world</p></code>
 <paragraph>(to be displayed inline)</paragraph>
 </unordered_list>
 <paragraph>Here a formula:</paragraph>
@@ -39,7 +39,7 @@ Affiliation 2</text>
 </unordered_list>
 </unordered_list>
 <paragraph>Here a code block:</paragraph>
-<code<_unknown_>print("Hello world")</code
+<code><_unknown_>print("Hello world")</code>
 <paragraph>Here a formula block:</paragraph>
 <formula>E=mc^2</formula>
 <paragraph>The end.</paragraph>