Merge branch 'main' into snake-case-field

Author: holtskinner

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

@@ -47,8 +47,8 @@ jobs:
           token: ${{ secrets.A2A_BOT_PAT }}
           committer: a2a-bot <a2a-bot@google.com>
           author: a2a-bot <a2a-bot@google.com>
-          commit-message: 'feat: Update A2A types from specification 🤖'
-          title: 'feat: Update A2A types from specification 🤖'
+          commit-message: 'feat(spec): Update A2A types from specification 🤖'
+          title: 'feat(spec): Update A2A types from specification 🤖'
           body: |
             This PR updates `src/a2a/types.py` based on the latest `specification/json/a2a.json` from [a2aproject/A2A](https://github.com/a2aproject/A2A/commit/${{ github.event.client_payload.sha }}).
           branch: auto-update-a2a-types-${{ github.event.client_payload.sha }}

.ruff.toml

@@ -143,7 +143,6 @@ inline-quotes = "single"
 
 [format]
 exclude = [
-    "types.py",
     "src/a2a/grpc/**",
 ]
 docstring-code-format = true

scripts/generate_types.sh

@@ -36,4 +36,7 @@ uv run datamodel-codegen \
   --base-class a2a._base.A2ABaseModel \
   --snake-case-field
 
+echo "Formatting generated file with ruff..."
+uv run ruff format "$GENERATED_FILE"
+
 echo "Codegen finished successfully."

src/a2a/types.py

@@ -133,7 +133,9 @@ class AgentSkill(A2ABaseModel):
     """
     Supported media types for output.
     """
-    tags: list[str] = Field(..., examples=[['cooking', 'customer support', 'billing']])
+    tags: list[str] = Field(
+        ..., examples=[['cooking', 'customer support', 'billing']]
+    )
     """
     Set of tagwords describing classes of capabilities for this specific skill.
     """
@@ -1285,7 +1287,9 @@ class Artifact(A2ABaseModel):
 
 
 class DeleteTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse
     """
@@ -1294,7 +1298,9 @@ class DeleteTaskPushNotificationConfigResponse(
 
 
 class GetTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
     """
@@ -1303,7 +1309,9 @@ class GetTaskPushNotificationConfigResponse(
 
 
 class ListTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse
     """
@@ -1458,7 +1466,9 @@ class SendStreamingMessageRequest(A2ABaseModel):
 
 
 class SetTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
     """
@@ -1511,7 +1521,9 @@ class TaskStatus(A2ABaseModel):
     Additional status updates for client
     """
     state: TaskState
-    timestamp: str | None = Field(default=None, examples=['2023-10-27T10:00:00Z'])
+    timestamp: str | None = Field(
+        default=None, examples=['2023-10-27T10:00:00Z']
+    )
     """
     ISO 8601 datetime string when the status was recorded.
     """
@@ -1777,7 +1789,9 @@ class SendStreamingMessageSuccessResponse(A2ABaseModel):
     """
 
 
-class CancelTaskResponse(RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]):
+class CancelTaskResponse(
+    RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]
+):
     root: JSONRPCErrorResponse | CancelTaskSuccessResponse
     """
     JSON-RPC response for the 'tasks/cancel' method.
@@ -1820,7 +1834,9 @@ class JSONRPCResponse(
     """
 
 
-class SendMessageResponse(RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]):
+class SendMessageResponse(
+    RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]
+):
     root: JSONRPCErrorResponse | SendMessageSuccessResponse
     """
     JSON-RPC response model for the 'message/send' method.