fix: DoclingDocument model validator should deal with any raw input (#419)

* fix: DoclingDocument model validator should deal with any raw input

Signed-off-by: Cesar Berrospi Ramis <[email protected]>

* docs: mention 3.14 as valid Python version for development

Signed-off-by: Cesar Berrospi Ramis <[email protected]>

---------

Signed-off-by: Cesar Berrospi Ramis <[email protected]>

Author: ceberam

README.md

@@ -1,7 +1,7 @@
 # Docling Core
 
 [![PyPI version](https://img.shields.io/pypi/v/docling-core)](https://pypi.org/project/docling-core/)
-![Python](https://img.shields.io/badge/python-3.9%20%7C%203.10%20%7C%20%203.11%20%7C%203.12%20%7C%203.13-blue)
+![Python](https://img.shields.io/badge/python-3.9%20%7C%203.10%20%7C%20%203.11%20%7C%203.12%20%7C%203.13%20%7C%203.14-blue)
 [![uv](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/uv/main/assets/badge/v0.json)](https://github.com/astral-sh/uv)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://pycqa.github.io/isort/)
@@ -21,7 +21,7 @@ pip install docling-core
 
 ### Development setup
 
-To develop for Docling Core, you need Python 3.9 / 3.10 / 3.11 / 3.12 / 3.13 and uv. You can then install from your local clone's root dir:
+To develop for Docling Core, you need Python 3.9 through 3.14 and the `uv` package. You can then install it from your local clone's root directory:
 ```bash
 uv sync --all-extras
 ```

docling_core/types/doc/document.py

@@ -2308,12 +2308,12 @@ class DoclingDocument(BaseModel):
 
     @model_validator(mode="before")
     @classmethod
-    def transform_to_content_layer(cls, data: dict) -> dict:
+    def transform_to_content_layer(cls, data: Any) -> Any:
         """transform_to_content_layer."""
         # Since version 1.1.0, all NodeItems carry content_layer property.
         # We must assign previous page_header and page_footer instances to furniture.
         # Note: model_validators which check on the version must use "before".
-        if "version" in data and data["version"] == "1.0.0":
+        if isinstance(data, dict) and data.get("version", "") == "1.0.0":
             for item in data.get("texts", []):
                 if "label" in item and item["label"] in [
                     DocItemLabel.PAGE_HEADER.value,

test/test_docling_doc.py

@@ -9,7 +9,7 @@
 import yaml
 from PIL import Image as PILImage
 from PIL import ImageDraw
-from pydantic import AnyUrl, ValidationError
+from pydantic import AnyUrl, BaseModel, ValidationError
 
 from docling_core.types.doc.base import BoundingBox, CoordOrigin, ImageRefMode, Size
 from docling_core.types.doc.document import (  # BoundingBox,
@@ -1156,6 +1156,14 @@ def test_upgrade_content_layer_from_1_0_0():
     assert doc.version == CURRENT_VERSION
     assert doc.texts[0].content_layer == ContentLayer.FURNITURE
 
+    # test that transform_to_content_layer model validator can handle any data type
+    class ContentOutput(BaseModel):
+        content: Union[str, DoclingDocument]
+
+    co = ContentOutput.model_validate_json('{"content": "Random string with version"}')
+    assert co
+    assert isinstance(co.content, str)
+
 
 def test_version_doc():