Fix test to match docling output changes

Also sets docling-core minimum version to 2.18.0 for deterministic test results. https://github.com/DS4SD/docling/blob/e1436a8b0574e6bb2bb89bd65e98221e418d7142/CHANGELOG.md#v2210---2025-02-10
run-llama · Feb 15, 2025 · ef742d3 · ef742d3
1 parent f434efb
commit ef742d3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/llama-index-integrations/node_parser/llama-index-node-parser-docling/pyproject.toml b/llama-index-integrations/node_parser/llama-index-node-parser-docling/pyproject.toml
@@ -30,12 +30,12 @@ license = "MIT"
 name = "llama-index-node-parser-docling"
 packages = [{include = "llama_index/"}]
 readme = "README.md"
-version = "0.3.0"
+version = "0.3.1"
 
 [tool.poetry.dependencies]
 python = "^3.10"
 llama-index-core = "^0.12.0"
-docling-core = "^2.2.0"
+docling-core = "^2.18.0"
 
 [tool.poetry.group.dev]
 

diff --git a/...ntegrations/node_parser/llama-index-node-parser-docling/tests/test_node_parser_docling.py b/...ntegrations/node_parser/llama-index-node-parser-docling/tests/test_node_parser_docling.py
@@ -34,6 +34,7 @@
                 "version": "1.0.0",
                 "doc_items": [
                     {
+                        "content_layer": "body",
                         "self_ref": "#/texts/0",
                         "parent": {"$ref": "#/body"},
                         "children": [],
@@ -75,6 +76,7 @@
                         "version": "1.0.0",
                         "doc_items": [
                             {
+                                "content_layer": "body",
                                 "self_ref": "#/texts/1",
                                 "parent": {"$ref": "#/body"},
                                 "children": [],
@@ -88,7 +90,7 @@
                             "filename": "sample.html",
                         },
                     },
-                    "hash": "0a8df027ead9e42831f12f8aa680afe5138436ecd58c32a6289212bc4d0a644a",
+                    "hash": "a6c3f2701d8f99dfe60b8fcfa3602c83412e92a234c747ea9b9554bf1894d484",
                     "class_name": "RelatedNodeInfo",
                 },
             },
@@ -110,6 +112,7 @@
                 "version": "1.0.0",
                 "doc_items": [
                     {
+                        "content_layer": "body",
                         "self_ref": "#/texts/1",
                         "parent": {"$ref": "#/body"},
                         "children": [],
@@ -151,6 +154,7 @@
                         "version": "1.0.0",
                         "doc_items": [
                             {
+                                "content_layer": "body",
                                 "self_ref": "#/texts/0",
                                 "parent": {"$ref": "#/body"},
                                 "children": [],
@@ -164,7 +168,7 @@
                             "filename": "sample.html",
                         },
                     },
-                    "hash": "fbfaa945f53349cff0ee00b81a8d3926ca76874fdaf3eac7888f41c5f6a74f0c",
+                    "hash": "0cc445f97c4273fde805655dea71ac576eca74a8210b8589d43fd96a8f79100a",
                     "class_name": "RelatedNodeInfo",
                 },
             },
@@ -192,6 +196,7 @@
                 "version": "1.0.0",
                 "doc_items": [
                     {
+                        "content_layer": "body",
                         "self_ref": "#/texts/0",
                         "parent": {"$ref": "#/body"},
                         "children": [],
@@ -244,6 +249,7 @@
                 "version": "1.0.0",
                 "doc_items": [
                     {
+                        "content_layer": "body",
                         "self_ref": "#/texts/1",
                         "parent": {"$ref": "#/body"},
                         "children": [],