fixes #37

Author: jph00

CHANGELOG.md

@@ -2,6 +2,14 @@
 
 <!-- do not remove -->
 
+## 0.3.1
+
+
+### Bugs Squashed
+
+- Add edge case tests and fix them ([#36](https://github.com/AnswerDotAI/toolslm/issues/36))
+
+
 ## 0.3.0
 
 ### Breaking changes

toolslm/md_hier.py

@@ -47,9 +47,9 @@ def clean_heading(text): return re.sub(r'[.]+', '', text).strip()  # Only remove
         result[key] = h['content']
     return dict2obj(result)
 
-def create_heading_dict(text):
+def create_heading_dict(text, rm_fenced=True):
     "Create a nested dictionary structure from markdown headings."
-    text = re.sub(r'```[\s\S]*?```', '', text)
+    if rm_fenced: text = re.sub(r'```[\s\S]*?```', '', text)
     headings = re.findall(r'^#+.*', text, flags=re.MULTILINE)
     result = {}
     stack = [result]