Merge remote-tracking branch 'upstream/main' into poc/agent-catalog

Signed-off-by: Shingo OKAWA <[email protected]>

Author: ognis1205

.coveragerc

@@ -5,6 +5,7 @@ omit =
     */site-packages/*
     */__init__.py
     */noxfile.py*
+    src/a2a/grpc/*
 
 [report]
 exclude_lines =
@@ -15,4 +16,4 @@ exclude_lines =
     if TYPE_CHECKING
     @abstractmethod
     pass
-    raise ImportError
+    raise ImportError

.github/actions/spelling/allow.txt

@@ -31,6 +31,7 @@ coro
 datamodel
 dunders
 genai
+getkwargs
 gle
 httpapi
 ietf
@@ -40,14 +41,24 @@ langgraph
 lifecycles
 linkset
 linting
+lstrips
+mockurl
 oauthoidc
+<<<<<<< HEAD
 ognis
+=======
+oidc
+>>>>>>> upstream/main
 opensource
 poc
 protoc
 pyi
 pyversions
+resub
 socio
 sse
 tagwords
+taskupdate
+testuuid
+typeerror
 vulnz

.github/workflows/unit-tests.yml

@@ -40,12 +40,8 @@ jobs:
       - name: Install dependencies
         run: uv sync --dev
 
-      - name: Run tests
-        run: uv run pytest
+      - name: Run tests and check coverage
+        run: uv run pytest --cov=a2a --cov-report=xml --cov-fail-under=85
 
-      - name: Upload coverage report
-        uses: actions/upload-artifact@v4
-        with:
-          name: coverage-report-${{ matrix.python-version }}
-          path: coverage.xml
-          if-no-files-found: ignore
+      - name: Show coverage summary in log
+        run: uv run coverage report

.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/

CHANGELOG.md

@@ -1,5 +1,25 @@
 # Changelog
 
+## [0.2.8](https://github.com/google-a2a/a2a-python/compare/v0.2.7...v0.2.8) (2025-06-12)
+
+
+### Features
+
+* Add HTTP Headers to ServerCallContext for Improved Handler Access ([#182](https://github.com/google-a2a/a2a-python/issues/182)) ([d5e5f5f](https://github.com/google-a2a/a2a-python/commit/d5e5f5f7e7a3cab7de13cff545a874fc58d85e46))
+* Update A2A types from specification 🤖 ([#191](https://github.com/google-a2a/a2a-python/issues/191)) ([174230b](https://github.com/google-a2a/a2a-python/commit/174230bf6dfb6bf287d233a101b98cc4c79cad19))
+
+
+### Bug Fixes
+
+* Add `protobuf==6.31.1` to dependencies ([#189](https://github.com/google-a2a/a2a-python/issues/189)) ([ae1c31c](https://github.com/google-a2a/a2a-python/commit/ae1c31c1da47f6965c02e0564dc7d3791dd03e2c)), closes [#185](https://github.com/google-a2a/a2a-python/issues/185)
+
+## [0.2.7](https://github.com/google-a2a/a2a-python/compare/v0.2.6...v0.2.7) (2025-06-11)
+
+
+### Features
+
+* Update A2A types from specification 🤖 ([#179](https://github.com/google-a2a/a2a-python/issues/179)) ([3ef4240](https://github.com/google-a2a/a2a-python/commit/3ef42405f6096281fe90b1df399731bd009bde12))
+
 ## [0.2.6](https://github.com/google-a2a/a2a-python/compare/v0.2.5...v0.2.6) (2025-06-09)
 
 

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

pyproject.toml

@@ -20,6 +20,7 @@ dependencies = [
     "grpcio>=1.60",
     "grpcio-tools>=1.60",
     "grpcio_reflection>=1.7.0",
+    "protobuf==6.31.1",
 ]
 
 classifiers = [

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")

src/a2a/grpc/a2a_pb2.pyi

@@ -16,7 +16,6 @@ class A2AServiceStub(object):
     - TaskPushNotification are a resource whose parent is a task. They have get,
     list and create methods.
     - AgentCard is a static resource with only a get method.
-    Of particular note of deviation from JSONRPC approach the request metadata
     fields are not present as they don't comply with AIP rules, and the
     optional history_length on the get task method is not present as it also
     violates AIP-127 and AIP-131.
@@ -86,7 +85,6 @@ class A2AServiceServicer(object):
     - TaskPushNotification are a resource whose parent is a task. They have get,
     list and create methods.
     - AgentCard is a static resource with only a get method.
-    Of particular note of deviation from JSONRPC approach the request metadata
     fields are not present as they don't comply with AIP rules, and the
     optional history_length on the get task method is not present as it also
     violates AIP-127 and AIP-131.
@@ -228,7 +226,6 @@ class A2AService(object):
     - TaskPushNotification are a resource whose parent is a task. They have get,
     list and create methods.
     - AgentCard is a static resource with only a get method.
-    Of particular note of deviation from JSONRPC approach the request metadata
     fields are not present as they don't comply with AIP rules, and the
     optional history_length on the get task method is not present as it also
     violates AIP-127 and AIP-131.