feat: Backup to first section head if title not found

microsoft · Jun 7, 2024 · 3fdb73a · 3fdb73a
1 parent ff5ff4f
commit 3fdb73a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/function_app.py b/function_app.py
@@ -87,8 +87,15 @@ async def raw_to_extracted(input: BlobClientTrigger) -> None:
             paragraph
             for paragraph in doc_result.paragraphs or []
             if paragraph.role == ParagraphRole.TITLE
+        ),  # First, title
+        next(
+            (
+                paragraph
+                for paragraph in doc_result.paragraphs or []
+                if paragraph.role == ParagraphRole.SECTION_HEADING
+            ),  # Second, section heading
+            None,  # Third, nothing
         ),
-        None,
     )
     raw_text_model = RawTextModel(
         content=doc_result.content,