fix: generate StrEnum types for enums (#134)

Author: kthota-g

.github/workflows/update-a2a-types.yml

@@ -58,7 +58,8 @@ jobs:
             --use-default-kwarg \
             --use-one-literal-as-default \
             --class-name A2A \
-            --use-standard-collections
+            --use-standard-collections \
+            --use-subclass-enum
           echo "Codegen finished."
 
       - name: Create Pull Request with Updates

development.md

@@ -17,5 +17,6 @@ uv run datamodel-codegen \
   --use-default-kwarg \
   --use-one-literal-as-default \
   --class-name A2A \
-  --use-standard-collections
+  --use-standard-collections \
+  --use-subclass-enum
 ```

src/a2a/types.py

@@ -13,7 +13,7 @@ class A2A(RootModel[Any]):
     root: Any
 
 
-class In(Enum):
+class In(str, Enum):
     """
     The location of the API key. Valid values are "query", "header", or "cookie".
     """
@@ -484,7 +484,7 @@ class JSONRPCSuccessResponse(BaseModel):
     """
 
 
-class Role(Enum):
+class Role(str, Enum):
     """
     Message sender's role
     """
@@ -731,7 +731,7 @@ class TaskResubscriptionRequest(BaseModel):
     """
 
 
-class TaskState(Enum):
+class TaskState(str, Enum):
     """
     Represents the possible states of a Task.
     """

tests/test_types.py

@@ -1487,3 +1487,12 @@ def test_a2a_error_validation_and_serialization() -> None:
     invalid_data: dict[str, Any] = {'code': -99999, 'message': 'Unknown error'}
     with pytest.raises(ValidationError):
         A2AError.model_validate(invalid_data)
+
+
+def test_subclass_enums() -> None:
+    """validate subtype enum types"""
+    assert "cookie" == In.cookie
+
+    assert "user" == Role.user
+
+    assert "working" == TaskState.working