mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-27 20:44:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass nested clusters through GLM as payload

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2024-12-03 13:58:27 +01:00
parent 4dcc738b6d
commit 0240ae2930
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docling/models/ds_glm_model.py
View File

@ -215,17 +215,21 @@ class GlmModel:
) )
], ],
obj_type=layout_label_to_ds_type.get(element.label), obj_type=layout_label_to_ds_type.get(element.label),
payload=TypeAdapter(List[Cluster]).dump_python( payload={
"children": TypeAdapter(List[Cluster]).dump_python(
element.cluster.children element.cluster.children
), # hack to channel child clusters through GLM )
}, # hack to channel child clusters through GLM
) )
) )
elif isinstance(element, ContainerElement): elif isinstance(element, ContainerElement):
main_text.append( main_text.append(
BaseText( BaseText(
payload=TypeAdapter(List[Cluster]).dump_python( payload={
"children": TypeAdapter(List[Cluster]).dump_python(
element.cluster.children element.cluster.children
), # hack to channel child clusters through GLM )
}, # hack to channel child clusters through GLM
obj_type=layout_label_to_ds_type.get(element.label), obj_type=layout_label_to_ds_type.get(element.label),
name=element.label, name=element.label,
prov=[ prov=[