Fix notion recursive search for blocks with children

onyx-dot-app · Nov 3, 2023 · 174f544 · 174f544 · vercel · Nov 3, 2023
1 parent 329824a
commit 174f544
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/backend/danswer/connectors/notion/connector.py b/backend/danswer/connectors/notion/connector.py
@@ -141,8 +141,15 @@ def _read_blocks(
                             text = rich_text["text"]["content"]
                             cur_result_text_arr.append(text)
 
-                if result["has_children"] and result_type == "child_page":
-                    child_pages.append(result_block_id)
+                if result["has_children"]:
+                    if result_type == "child_page":
+                        child_pages.append(result_block_id)
+                    else:
+                        subblock_result_lines, subblock_child_pages = self._read_blocks(
+                            result_block_id
+                        )
+                        result_lines.extend(subblock_result_lines)
+                        child_pages.extend(subblock_child_pages)
 
                 cur_result_text = "\n".join(cur_result_text_arr)
                 if cur_result_text: