Merge branch 'main' into jules_wip_14959778467806306980

Author: holtskinner

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

@@ -62,19 +62,33 @@ jobs:
             --use-subclass-enum
           echo "Codegen finished."
 
+      - name: Install Buf
+        uses: bufbuild/buf-setup-action@v1
+
+      - name: Run buf generate
+        run: |
+          set -euo pipefail # Exit immediately if a command exits with a non-zero status
+
+          echo "Running buf generate..."
+          buf generate
+          uv run scripts/grpc_gen_post_processor.py
+          echo "Buf generate finished."
+
       - name: Create Pull Request with Updates
         uses: peter-evans/create-pull-request@v6
         with:
           token: ${{ secrets.A2A_BOT_PAT }}
           committer: "a2a-bot <[email protected]>"
           author: "a2a-bot <[email protected]>"
-          commit-message: "chore: Update A2A types from specification 🤖"
-          title: "chore: Update A2A types from specification 🤖"
+          commit-message: "feat: Update A2A types from specification 🤖"
+          title: "feat: Update A2A types from specification 🤖"
           body: |
             This PR updates `src/a2a/types.py` based on the latest `specification/json/a2a.json` from [google-a2a/A2A](https://github.com/google-a2a/A2A/commit/${{ github.event.client_payload.sha }}).
           branch: "auto-update-a2a-types-${{ github.event.client_payload.sha }}"
           base: main
           labels: |
             automated
             dependencies
-          add-paths: ${{ steps.vars.outputs.GENERATED_FILE }}
+          add-paths: |
+            ${{ steps.vars.outputs.GENERATED_FILE }}
+            src/a2a/grpc/

buf.gen.yaml

@@ -19,12 +19,12 @@ managed:
 plugins:
   # Generate python protobuf related code
   # Generates *_pb2.py files, one for each .proto
-  - remote: buf.build/protocolbuffers/python:v29.3
+  - remote: buf.build/protocolbuffers/python
     out: src/a2a/grpc
   # Generate python service code.
   # Generates *_pb2_grpc.py
   - remote: buf.build/grpc/python
     out: src/a2a/grpc
   # Generates *_pb2.pyi files.
-  - remote: buf.build/protocolbuffers/pyi:v29.3
+  - remote: buf.build/protocolbuffers/pyi
     out: src/a2a/grpc

src/a2a/grpc/a2a_pb2.py

@@ -1,3 +1,5 @@
+import datetime
+
 from google.api import annotations_pb2 as _annotations_pb2
 from google.api import client_pb2 as _client_pb2
 from google.api import field_behavior_pb2 as _field_behavior_pb2
@@ -7,7 +9,8 @@ from google.protobuf.internal import containers as _containers
 from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
-from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
@@ -77,7 +80,7 @@ class TaskStatus(_message.Message):
     state: TaskState
     update: Message
     timestamp: _timestamp_pb2.Timestamp
-    def __init__(self, state: _Optional[_Union[TaskState, str]] = ..., update: _Optional[_Union[Message, _Mapping]] = ..., timestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+    def __init__(self, state: _Optional[_Union[TaskState, str]] = ..., update: _Optional[_Union[Message, _Mapping]] = ..., timestamp: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
 
 class Part(_message.Message):
     __slots__ = ("text", "file", "data")