Merge remote-tracking branch 'origin/main' into fix/gh-609

Author: ishymko

.github/workflows/linter.yaml

@@ -23,7 +23,7 @@ jobs:
         run: |
           echo "$HOME/.cargo/bin" >> $GITHUB_PATH
       - name: Install dependencies
-        run: uv sync --locked --dev
+        run: uv sync --locked
 
       - name: Run Ruff Linter
         id: ruff-lint

.github/workflows/unit-tests.yml

@@ -53,7 +53,7 @@ jobs:
         run: |
           echo "$HOME/.cargo/bin" >> $GITHUB_PATH
       - name: Install dependencies
-        run: uv sync --locked --dev --extra all
+        run: uv sync --locked
       - name: Run tests and check coverage
         run: uv run pytest --cov=a2a --cov-report term --cov-fail-under=88
       - name: Show coverage summary in log

README.md

@@ -34,6 +34,18 @@
 
 ---
 
+## 🧩 Compatibility
+
+This SDK implements the A2A Protocol Specification [`v0.3.0`](https://a2a-protocol.org/v0.3.0/specification).
+
+| Transport | Client | Server |
+| :--- | :---: | :---: |
+| **JSON-RPC** | ✅ | ✅ |
+| **HTTP+JSON/REST** | ✅ | ✅ |
+| **GRPC** | ✅ | ✅ |
+
+---
+
 ## 🚀 Getting Started
 
 ### Prerequisites

pyproject.toml

@@ -89,7 +89,6 @@ style = "pep440"
 dev = [
   "datamodel-code-generator>=0.30.0",
   "mypy>=1.15.0",
-  "PyJWT>=2.0.0",
   "pytest>=8.3.5",
   "pytest-asyncio>=0.26.0",
   "pytest-cov>=6.1.1",
@@ -101,15 +100,13 @@ dev = [
   "types-protobuf",
   "types-requests",
   "pre-commit",
-  "fastapi>=0.115.2",
-  "sse-starlette",
-  "starlette",
   "pyupgrade",
   "autoflake",
   "no_implicit_optional",
   "trio",
   "uvicorn>=0.35.0",
   "pytest-timeout>=2.4.0",
+  "a2a-sdk[all]",
 ]
 
 [[tool.uv.index]]
@@ -118,6 +115,9 @@ url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
 explicit = true
 
+[tool.uv.sources]
+a2a-sdk = { workspace = true }
+
 [tool.mypy]
 plugins = ["pydantic.mypy"]
 exclude = ["src/a2a/grpc/"]

scripts/run_integration_tests.sh scripts/run_db_tests.shscripts/run_integration_tests.sh renamed to scripts/run_db_tests.sh

@@ -96,7 +96,7 @@ trap cleanup EXIT
 echo "Running integration tests..."
 cd "$PROJECT_ROOT"
 
-uv run --extra all pytest -v \
+uv run pytest -v \
     tests/server/tasks/test_database_task_store.py \
     tests/server/tasks/test_database_push_notification_config_store.py \
     "${PYTEST_ARGS[@]}"

src/a2a/client/errors.py

@@ -21,6 +21,14 @@ def __init__(self, status_code: int, message: str):
         self.message = message
         super().__init__(f'HTTP Error {status_code}: {message}')
 
+    def __repr__(self) -> str:
+        """Returns an unambiguous representation showing structured attributes."""
+        return (
+            f'{self.__class__.__name__}('
+            f'status_code={self.status_code!r}, '
+            f'message={self.message!r})'
+        )
+
 
 class A2AClientJSONError(A2AClientError):
     """Client exception for JSON errors during response parsing or validation."""
@@ -34,6 +42,10 @@ def __init__(self, message: str):
         self.message = message
         super().__init__(f'JSON Error: {message}')
 
+    def __repr__(self) -> str:
+        """Returns an unambiguous representation showing structured attributes."""
+        return f'{self.__class__.__name__}(message={self.message!r})'
+
 
 class A2AClientTimeoutError(A2AClientError):
     """Client exception for timeout errors during a request."""
@@ -47,6 +59,10 @@ def __init__(self, message: str):
         self.message = message
         super().__init__(f'Timeout Error: {message}')
 
+    def __repr__(self) -> str:
+        """Returns an unambiguous representation showing structured attributes."""
+        return f'{self.__class__.__name__}(message={self.message!r})'
+
 
 class A2AClientInvalidArgsError(A2AClientError):
     """Client exception for invalid arguments passed to a method."""
@@ -60,6 +76,10 @@ def __init__(self, message: str):
         self.message = message
         super().__init__(f'Invalid arguments error: {message}')
 
+    def __repr__(self) -> str:
+        """Returns an unambiguous representation showing structured attributes."""
+        return f'{self.__class__.__name__}(message={self.message!r})'
+
 
 class A2AClientInvalidStateError(A2AClientError):
     """Client exception for an invalid client state."""
@@ -73,6 +93,10 @@ def __init__(self, message: str):
         self.message = message
         super().__init__(f'Invalid state error: {message}')
 
+    def __repr__(self) -> str:
+        """Returns an unambiguous representation showing structured attributes."""
+        return f'{self.__class__.__name__}(message={self.message!r})'
+
 
 class A2AClientJSONRPCError(A2AClientError):
     """Client exception for JSON-RPC errors returned by the server."""
@@ -85,3 +109,7 @@ def __init__(self, error: JSONRPCErrorResponse):
         """
         self.error = error.error
         super().__init__(f'JSON-RPC Error {error.error}')
+
+    def __repr__(self) -> str:
+        """Returns an unambiguous representation showing the JSON-RPC error object."""
+        return f'{self.__class__.__name__}({self.error!r})'

src/a2a/client/transports/base.py

@@ -1,5 +1,8 @@
 from abc import ABC, abstractmethod
 from collections.abc import AsyncGenerator, Callable
+from types import TracebackType
+
+from typing_extensions import Self
 
 from a2a.client.middleware import ClientCallContext
 from a2a.types import (
@@ -19,6 +22,19 @@
 class ClientTransport(ABC):
     """Abstract base class for a client transport."""
 
+    async def __aenter__(self) -> Self:
+        """Enters the async context manager, returning the transport itself."""
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> None:
+        """Exits the async context manager, ensuring close() is called."""
+        await self.close()
+
     @abstractmethod
     async def send_message(
         self,

src/a2a/client/transports/grpc.py

@@ -64,10 +64,11 @@ def _get_grpc_metadata(
         extensions: list[str] | None = None,
     ) -> list[tuple[str, str]] | None:
         """Creates gRPC metadata for extensions."""
-        if extensions is not None:
-            return [(HTTP_EXTENSION_HEADER, ','.join(extensions))]
-        if self.extensions is not None:
-            return [(HTTP_EXTENSION_HEADER, ','.join(self.extensions))]
+        extensions_to_use = extensions or self.extensions
+        if extensions_to_use:
+            return [
+                (HTTP_EXTENSION_HEADER.lower(), ','.join(extensions_to_use))
+            ]
         return None
 
     @classmethod

src/a2a/server/apps/jsonrpc/fastapi_app.py

@@ -1,6 +1,6 @@
 import logging
 
-from collections.abc import Callable
+from collections.abc import Awaitable, Callable
 from typing import TYPE_CHECKING, Any
 
 
@@ -72,9 +72,10 @@ def __init__(  # noqa: PLR0913
         http_handler: RequestHandler,
         extended_agent_card: AgentCard | None = None,
         context_builder: CallContextBuilder | None = None,
-        card_modifier: Callable[[AgentCard], AgentCard] | None = None,
+        card_modifier: Callable[[AgentCard], Awaitable[AgentCard] | AgentCard]
+        | None = None,
         extended_card_modifier: Callable[
-            [AgentCard, ServerCallContext], AgentCard
+            [AgentCard, ServerCallContext], Awaitable[AgentCard] | AgentCard
         ]
         | None = None,
         max_content_length: int | None = 10 * 1024 * 1024,  # 10MB

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

@@ -4,7 +4,7 @@
 import traceback
 
 from abc import ABC, abstractmethod
-from collections.abc import AsyncGenerator, Callable
+from collections.abc import AsyncGenerator, Awaitable, Callable
 from typing import TYPE_CHECKING, Any
 
 from pydantic import ValidationError
@@ -51,6 +51,7 @@
     PREV_AGENT_CARD_WELL_KNOWN_PATH,
 )
 from a2a.utils.errors import MethodNotImplementedError
+from a2a.utils.helpers import maybe_await
 
 
 logger = logging.getLogger(__name__)
@@ -178,9 +179,10 @@ def __init__(  # noqa: PLR0913
         http_handler: RequestHandler,
         extended_agent_card: AgentCard | None = None,
         context_builder: CallContextBuilder | None = None,
-        card_modifier: Callable[[AgentCard], AgentCard] | None = None,
+        card_modifier: Callable[[AgentCard], Awaitable[AgentCard] | AgentCard]
+        | None = None,
         extended_card_modifier: Callable[
-            [AgentCard, ServerCallContext], AgentCard
+            [AgentCard, ServerCallContext], Awaitable[AgentCard] | AgentCard
         ]
         | None = None,
         max_content_length: int | None = 10 * 1024 * 1024,  # 10MB
@@ -576,7 +578,7 @@ async def _handle_get_agent_card(self, request: Request) -> JSONResponse:
 
         card_to_serve = self.agent_card
         if self.card_modifier:
-            card_to_serve = self.card_modifier(card_to_serve)
+            card_to_serve = await maybe_await(self.card_modifier(card_to_serve))
 
         return JSONResponse(
             card_to_serve.model_dump(
@@ -605,7 +607,9 @@ async def _handle_get_authenticated_extended_agent_card(
             context = self._context_builder.build(request)
             # If no base extended card is provided, pass the public card to the modifier
             base_card = card_to_serve if card_to_serve else self.agent_card
-            card_to_serve = self.extended_card_modifier(base_card, context)
+            card_to_serve = await maybe_await(
+                self.extended_card_modifier(base_card, context)
+            )
 
         if card_to_serve:
             return JSONResponse(