Skip to content
Snippets Groups Projects
Unverified Commit 5c53f417 authored by haarisedhi102's avatar haarisedhi102 Committed by GitHub
Browse files

remove space from MarkdownNodeParser Header metadata (#11982)

parent d651dbb8
No related branches found
No related tags found
No related merge requests found
...@@ -99,11 +99,11 @@ class MarkdownNodeParser(NodeParser): ...@@ -99,11 +99,11 @@ class MarkdownNodeParser(NodeParser):
updated_headers = {} updated_headers = {}
for i in range(1, new_header_level): for i in range(1, new_header_level):
key = f"Header {i}" key = f"Header_{i}"
if key in headers_metadata: if key in headers_metadata:
updated_headers[key] = headers_metadata[key] updated_headers[key] = headers_metadata[key]
updated_headers[f"Header {new_header_level}"] = new_header updated_headers[f"Header_{new_header_level}"] = new_header
return updated_headers return updated_headers
def _build_node_from_split( def _build_node_from_split(
......
...@@ -19,8 +19,8 @@ Header 2 content ...@@ -19,8 +19,8 @@ Header 2 content
] ]
) )
assert len(splits) == 2 assert len(splits) == 2
assert splits[0].metadata == {"Header 1": "Main Header"} assert splits[0].metadata == {"Header_1": "Main Header"}
assert splits[1].metadata == {"Header 1": "Header 2"} assert splits[1].metadata == {"Header_1": "Header 2"}
assert splits[0].text == "Main Header\n\nHeader 1 content" assert splits[0].text == "Main Header\n\nHeader 1 content"
assert splits[1].text == "Header 2\nHeader 2 content" assert splits[1].text == "Header 2\nHeader 2 content"
...@@ -80,11 +80,11 @@ Content ...@@ -80,11 +80,11 @@ Content
] ]
) )
assert len(splits) == 4 assert len(splits) == 4
assert splits[0].metadata == {"Header 1": "Main Header"} assert splits[0].metadata == {"Header_1": "Main Header"}
assert splits[1].metadata == {"Header 1": "Main Header", "Header 2": "Sub-header"} assert splits[1].metadata == {"Header_1": "Main Header", "Header_2": "Sub-header"}
assert splits[2].metadata == { assert splits[2].metadata == {
"Header 1": "Main Header", "Header_1": "Main Header",
"Header 2": "Sub-header", "Header_2": "Sub-header",
"Header 3": "Sub-sub header", "Header_3": "Sub-sub header",
} }
assert splits[3].metadata == {"Header 1": "New title"} assert splits[3].metadata == {"Header_1": "New title"}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment