Skip to content

Commit 7de9f38

Browse files
committed
feat: add A2ABaseModel
Signed-off-by: Shingo OKAWA <shingo.okawa.g.h.c@gmail.com>
1 parent 4343be9 commit 7de9f38

3 files changed

Lines changed: 107 additions & 82 deletions

File tree

scripts/generate_types.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ uv run datamodel-codegen \
3232
--use-one-literal-as-default \
3333
--class-name A2A \
3434
--use-standard-collections \
35-
--use-subclass-enum
35+
--use-subclass-enum \
36+
--base-class a2a._base.A2ABaseModel
3637

3738
echo "Codegen finished successfully."

src/a2a/_base.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
from pydantic import BaseModel, ConfigDict
2+
3+
class A2ABaseModel(BaseModel):
4+
model_config = ConfigDict(
5+
populate_by_name=True,
6+
)

0 commit comments

Comments
 (0)