mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-01 15:02:21 +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>
This commit is contained in:
Christoph Auer 2024-12-03 13:58:27 +01:00
parent db70916f57
commit 65fa584a1a
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
docling/models/ds_glm_model.py
View File

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