File tree Expand file tree Collapse file tree 4 files changed +16
-5
lines changed Expand file tree Collapse file tree 4 files changed +16
-5
lines changed Original file line number Diff line number Diff line change 58
58
--use-default-kwarg \
59
59
--use-one-literal-as-default \
60
60
--class-name A2A \
61
- --use-standard-collections
61
+ --use-standard-collections \
62
+ --use-subclass-enum
62
63
echo "Codegen finished."
63
64
64
65
- name : Create Pull Request with Updates
Original file line number Diff line number Diff line change @@ -17,5 +17,6 @@ uv run datamodel-codegen \
17
17
--use-default-kwarg \
18
18
--use-one-literal-as-default \
19
19
--class-name A2A \
20
- --use-standard-collections
20
+ --use-standard-collections \
21
+ --use-subclass-enum
21
22
```
Original file line number Diff line number Diff line change @@ -13,7 +13,7 @@ class A2A(RootModel[Any]):
13
13
root : Any
14
14
15
15
16
- class In (Enum ):
16
+ class In (str , Enum ):
17
17
"""
18
18
The location of the API key. Valid values are "query", "header", or "cookie".
19
19
"""
@@ -484,7 +484,7 @@ class JSONRPCSuccessResponse(BaseModel):
484
484
"""
485
485
486
486
487
- class Role (Enum ):
487
+ class Role (str , Enum ):
488
488
"""
489
489
Message sender's role
490
490
"""
@@ -731,7 +731,7 @@ class TaskResubscriptionRequest(BaseModel):
731
731
"""
732
732
733
733
734
- class TaskState (Enum ):
734
+ class TaskState (str , Enum ):
735
735
"""
736
736
Represents the possible states of a Task.
737
737
"""
Original file line number Diff line number Diff line change @@ -1487,3 +1487,12 @@ def test_a2a_error_validation_and_serialization() -> None:
1487
1487
invalid_data : dict [str , Any ] = {'code' : - 99999 , 'message' : 'Unknown error' }
1488
1488
with pytest .raises (ValidationError ):
1489
1489
A2AError .model_validate (invalid_data )
1490
+
1491
+
1492
+ def test_subclass_enums () -> None :
1493
+ """validate subtype enum types"""
1494
+ assert "cookie" == In .cookie
1495
+
1496
+ assert "user" == Role .user
1497
+
1498
+ assert "working" == TaskState .working
You can’t perform that action at this time.
0 commit comments