mirror of
https://github.com/docling-project/docling-core.git
synced 2026-05-17 13:10:44 +00:00
Cesar Berrospi Ramis
9134ffee93
Signed-off-by: Cesar Berrospi Ramis <75900930+ceberam@users.noreply.github.com>
28 lines
874 B
Python
28 lines
874 B
Python
#
|
|
# Copyright IBM Corp. 2024 - 2024
|
|
# SPDX-License-Identifier: MIT
|
|
#
|
|
|
|
"""Define utility models and types related to field aliases."""
|
|
from pydantic import BaseModel, ConfigDict
|
|
|
|
|
|
class AliasModel(BaseModel):
|
|
"""Model for alias fields to ensure instantiation and serialization by alias."""
|
|
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
def model_dump(self, **kwargs) -> dict:
|
|
"""Generate a dictionary representation of the model using field aliases."""
|
|
if "by_alias" not in kwargs:
|
|
kwargs = {**kwargs, "by_alias": True}
|
|
|
|
return super().model_dump(**kwargs)
|
|
|
|
def model_dump_json(self, **kwargs) -> str:
|
|
"""Generate a JSON representation of the model using field aliases."""
|
|
if "by_alias" not in kwargs:
|
|
kwargs = {**kwargs, "by_alias": True}
|
|
|
|
return super().model_dump_json(**kwargs)
|