Merge branch 'main' into docs/add-new-example-mcp-without-llm-framework

Author: batprem

.github/actions/spelling/allow.txt

@@ -26,6 +26,7 @@ dunders
 genai
 gle
 inmemory
+kwarg
 langgraph
 lifecycles
 linting

.github/linters/.markdownlint.json

@@ -1,3 +1,4 @@
 releaseType: python
 handleGHRelease: true
 bumpMinorPreMajor: false
+bumpPatchForMinorPreMajor: true

.github/release-please.yml

@@ -62,6 +62,5 @@ jobs:
           VALIDATE_JAVASCRIPT_STANDARD: false
           VALIDATE_TYPESCRIPT_STANDARD: false
           VALIDATE_GIT_COMMITLINT: false
-          MARKDOWN_CONFIG_FILE: .markdownlint.json
           PYTHON_MYPY_CONFIG_FILE: .mypy.ini
-          FILTER_REGEX_EXCLUDE: "^examples/.*$|^CHANGELOG\\.md$"
+          FILTER_REGEX_INCLUDE: "^src/**"

.github/workflows/linter.yaml

@@ -0,0 +1,15 @@
+# Changelog
+
+## [0.2.3](https://github.com/google/a2a-python/compare/v0.2.2...v0.2.3) (2025-05-20)
+
+
+### Features
+
+* Add request context builder with referenceTasks ([#56](https://github.com/google/a2a-python/issues/56)) ([f20bfe7](https://github.com/google/a2a-python/commit/f20bfe74b8cc854c9c29720b2ea3859aff8f509e))
+
+## [0.2.2](https://github.com/google/a2a-python/compare/v0.2.1...v0.2.2) (2025-05-20)
+
+
+### Documentation
+
+* Write/Update Docstrings for Classes/Methods ([#59](https://github.com/google/a2a-python/issues/59)) ([9f773ef](https://github.com/google/a2a-python/commit/9f773eff4dddc4eec723d519d0050f21b9ccc042))

CHANGELOG.md

@@ -2,8 +2,20 @@
 
 ## Type generation from spec
 
-<!-- TODO replace spec.json with the public url so we always get the latest version-->
-
 ```bash
-uv run datamodel-codegen --input ./spec.json --input-file-type jsonschema --output ./src/a2a/types.py --target-python-version 3.10 --output-model-type pydantic_v2.BaseModel --disable-timestamp --use-schema-description --use-union-operator --use-field-description --use-default --use-default-kwarg --use-one-literal-as-default --class-name A2A --use-standard-collections
+uv run datamodel-codegen \
+  --url https://raw.githubusercontent.com/google/A2A/refs/heads/main/specification/json/a2a.json \
+  --input-file-type jsonschema \
+  --output ./src/a2a/types.py \
+  --target-python-version 3.10 \
+  --output-model-type pydantic_v2.BaseModel \
+  --disable-timestamp \
+  --use-schema-description \
+  --use-union-operator \
+  --use-field-description \
+  --use-default \
+  --use-default-kwarg \
+  --use-one-literal-as-default \
+  --class-name A2A \
+  --use-standard-collections
 ```

development.md

@@ -1,5 +1,5 @@
-from agent import CurrencyAgent # type: ignore[import-untyped]
-from typing_extensions import override
+from agent import CurrencyAgent  # type: ignore[import-untyped]
+
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events.event_queue import EventQueue
 from a2a.types import (
@@ -17,7 +17,6 @@ class CurrencyAgentExecutor(AgentExecutor):
     def __init__(self):
         self.agent = CurrencyAgent()
 
-    @override
     async def execute(
         self,
         context: RequestContext,
@@ -89,7 +88,6 @@ async def execute(
                     )
                 )
 
-    @override
     async def cancel(
         self, context: RequestContext, event_queue: EventQueue
     ) -> None:

examples/langgraph/agent_executor.py

@@ -2,6 +2,17 @@
 
 from a2a.server.agent_execution.agent_executor import AgentExecutor
 from a2a.server.agent_execution.context import RequestContext
+from a2a.server.agent_execution.request_context_builder import (
+    RequestContextBuilder,
+)
+from a2a.server.agent_execution.simple_request_context_builder import (
+    SimpleRequestContextBuilder,
+)
 
 
-__all__ = ['AgentExecutor', 'RequestContext']
+__all__ = [
+    'AgentExecutor',
+    'RequestContext',
+    'RequestContextBuilder',
+    'SimpleRequestContextBuilder',
+]

src/a2a/server/agent_execution/__init__.py

@@ -0,0 +1,18 @@
+from abc import ABC, abstractmethod
+
+from a2a.server.agent_execution import RequestContext
+from a2a.types import MessageSendParams, Task
+
+
+class RequestContextBuilder(ABC):
+    """Builds request context to be supplied to agent executor"""
+
+    @abstractmethod
+    async def build(
+        self,
+        params: MessageSendParams | None = None,
+        task_id: str | None = None,
+        context_id: str | None = None,
+        task: Task | None = None,
+    ) -> RequestContext:
+        pass

src/a2a/server/agent_execution/request_context_builder.py

@@ -0,0 +1,48 @@
+import asyncio
+
+from a2a.server.agent_execution import RequestContext, RequestContextBuilder
+from a2a.server.tasks import TaskStore
+from a2a.types import MessageSendParams, Task
+
+
+class SimpleRequestContextBuilder(RequestContextBuilder):
+    """Builds request context and populates referred tasks"""
+
+    def __init__(
+        self,
+        should_populate_referred_tasks: bool = False,
+        task_store: TaskStore | None = None,
+    ) -> None:
+        self._task_store = task_store
+        self._should_populate_referred_tasks = should_populate_referred_tasks
+
+    async def build(
+        self,
+        params: MessageSendParams | None = None,
+        task_id: str | None = None,
+        context_id: str | None = None,
+        task: Task | None = None,
+    ) -> RequestContext:
+        related_tasks: list[Task] | None = None
+
+        if (
+            self._task_store
+            and self._should_populate_referred_tasks
+            and params
+            and params.message.referenceTaskIds
+        ):
+            tasks = await asyncio.gather(
+                *[
+                    self._task_store.get(task_id)
+                    for task_id in params.message.referenceTaskIds
+                ]
+            )
+            related_tasks = [x for x in tasks if x is not None]
+
+        return RequestContext(
+            request=params,
+            task_id=task_id,
+            context_id=context_id,
+            task=task,
+            related_tasks=related_tasks,
+        )