diff --git a/src/a2a/_base.py b/src/a2a/_base.py index 6c50734c..76912c3f 100644 --- a/src/a2a/_base.py +++ b/src/a2a/_base.py @@ -1,3 +1,5 @@ +from typing import Any + from pydantic import BaseModel, ConfigDict from pydantic.alias_generators import to_camel @@ -31,8 +33,16 @@ class A2ABaseModel(BaseModel): model_config = ConfigDict( # SEE: https://docs.pydantic.dev/latest/api/config/#pydantic.config.ConfigDict.populate_by_name - validate_by_name=True, - validate_by_alias=True, - serialize_by_alias=True, + populate_by_name=True, alias_generator=to_camel_custom, ) + + def model_dump( + self, + **kwargs + ) -> dict[str, Any]: + + if 'by_alias' not in kwargs: + kwargs['by_alias'] = True + + return super().model_dump(**kwargs)