Small fix

Signed-off-by: Christoph Auer <[email protected]>
DS4SD · Sep 25, 2024 · b7b35a7 · b7b35a7
1 parent f45ad08
commit b7b35a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepsearch_glm/utils/doc_utils.py b/deepsearch_glm/utils/doc_utils.py
@@ -270,7 +270,7 @@ def to_docling_document(doc_glm, update_name_label=False) -> DoclingDocument:
             prov = ProvenanceItem(page_no=pelem["page"], charspan=(0, len(text)),
                                   bbox=BoundingBox.from_tuple(pelem["bbox"], origin=CoordOrigin.BOTTOMLEFT))
 
-            doc.add_paragraph(label=PageLabel(name_label), text=text, prov=prov)
+            doc.add_paragraph(label=DocItemLabel(name_label), text=text, prov=prov)
 
         else:
             pitem = {