Use custom camelCase alias generator

Author: holtskinner

scripts/generate_types.sh

@@ -35,7 +35,8 @@ uv run datamodel-codegen \
   --use-subclass-enum \
   --base-class a2a._base.A2ABaseModel \
   --field-constraints \
-  --snake-case-field
+  --snake-case-field \
+  --no-alias
 
 echo "Formatting generated file with ruff..."
 uv run ruff format "$GENERATED_FILE"

src/a2a/_base.py

@@ -1,4 +1,21 @@
 from pydantic import BaseModel, ConfigDict
+from pydantic.alias_generators import to_camel
+
+
+def to_camel_custom(snake: str) -> str:
+    """Convert a snake_case string to camelCase.
+
+    Args:
+        snake: The string to convert.
+
+    Returns:
+        The converted camelCase string.
+    """
+    # First, remove any trailing underscores. This is common for names that
+    # conflict with Python keywords, like 'in_' or 'from_'.
+    if snake.endswith('_'):
+        snake = snake.rstrip('_')
+    return to_camel(snake)
 
 
 class A2ABaseModel(BaseModel):
@@ -13,4 +30,5 @@ class A2ABaseModel(BaseModel):
         validate_by_name=True,
         validate_by_alias=True,
         serialize_by_alias=True,
+        alias_generator=to_camel_custom,
     )