Apply renaming to DocItemLabel

Signed-off-by: Christoph Auer <[email protected]>
DS4SD · Sep 25, 2024 · f45ad08 · f45ad08
1 parent d527b9a
commit f45ad08
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 95 deletions.
diff --git a/deepsearch_glm/utils/doc_utils.py b/deepsearch_glm/utils/doc_utils.py
@@ -6,7 +6,7 @@
 from docling_core.types.experimental.document import DoclingDocument, FileInfo, BaseFigureData, BaseTableData, \
     TableCell, ProvenanceItem, PageItem
 
-from docling_core.types.experimental.labels import PageLabel
+from docling_core.types.experimental.labels import DocItemLabel
 
 
 def resolve_item(paths, obj):
@@ -129,7 +129,7 @@ def to_docling_document(doc_glm, update_name_label=False) -> DoclingDocument:
 
                     prov = ProvenanceItem(page_no=nelem["page"], charspan=tuple(nelem["span"]), bbox=BoundingBox.from_tuple(nelem["bbox"], origin=CoordOrigin.BOTTOMLEFT))
 
-                    caption_obj = doc.add_paragraph(label=PageLabel.CAPTION, text=text, prov=prov)
+                    caption_obj = doc.add_paragraph(label=DocItemLabel.CAPTION, text=text, prov=prov)
                     caption_refs.append(caption_obj.get_ref())
 
             figure = {
@@ -190,7 +190,7 @@ def to_docling_document(doc_glm, update_name_label=False) -> DoclingDocument:
                     prov = ProvenanceItem(page_no=pelem["page"], charspan=nelem["span"],
                                           bbox=BoundingBox.from_tuple(pelem["bbox"], origin=CoordOrigin.BOTTOMLEFT))
 
-                    caption_obj = doc.add_paragraph(label=PageLabel.CAPTION, text=text, prov=prov)
+                    caption_obj = doc.add_paragraph(label=DocItemLabel.CAPTION, text=text, prov=prov)
                     caption_refs.append(caption_obj.get_ref())
 
 

diff --git a/poetry.lock b/poetry.lock