From 71bf1530408d74d0800731afb4c3a206ea34c501 Mon Sep 17 00:00:00 2001
From: ZedingZhang <me@zedingzhang.com>
Date: Fri, 23 Oct 2026 14:23:14 +0800
Subject: [PATCH] Python: Fix A2A dependency bounds test environment

---
 python/packages/a2a/pyproject.toml                       | 5 +++++
 .../dependencies/tests/test_dependency_bounds_runtime.py | 9 +++++++++
 python/uv.lock                                           | 8 ++++++++
 3 files changed, 22 insertions(+)

diff --git a/python/packages/a2a/pyproject.toml b/python/packages/a2a/pyproject.toml
index afb04ef66e6..3779355bdd3 100644
--- a/python/packages/a2a/pyproject.toml
+++ b/python/packages/a2a/pyproject.toml
@@ -28,6 +28,11 @@ dependencies = [
     "a2a-sdk>=1.0.0,<2",
 ]
 
+[dependency-groups]
+test = [
+    "agent-framework-orchestrations",
+]
+
 [tool.uv]
 prerelease = "if-necessary"
 environments = [
diff --git a/python/scripts/dependencies/tests/test_dependency_bounds_runtime.py b/python/scripts/dependencies/tests/test_dependency_bounds_runtime.py
index d67f3f3baf2..e7e57565faa 100644
--- a/python/scripts/dependencies/tests/test_dependency_bounds_runtime.py
+++ b/python/scripts/dependencies/tests/test_dependency_bounds_runtime.py
@@ -179,6 +179,15 @@ def test_test_mode_uses_dependency_pyright_when_available(tmp_path: Path) -> Non
     assert plans[0].typing_task == "dependency-pyright"
 
 
+def test_a2a_test_plan_includes_orchestrations() -> None:
+    workspace_root = Path(__file__).resolve().parents[3]
+
+    plans = _build_test_plans(workspace_root, "a2a")
+
+    assert len(plans) == 1
+    assert (workspace_root / "packages/orchestrations").resolve() in plans[0].internal_editables
+
+
 def test_dependency_pyright_uses_the_shared_runtime_environment() -> None:
     command = ["uv", "run"]
 
diff --git a/python/uv.lock b/python/uv.lock
index 84130377c9f..33ebc2383f6 100644
--- a/python/uv.lock
+++ b/python/uv.lock
@@ -183,12 +183,20 @@ dependencies = [
     { name = "agent-framework-core" },
 ]
 
+[package.dev-dependencies]
+test = [
+    { name = "agent-framework-orchestrations" },
+]
+
 [package.metadata]
 requires-dist = [
     { name = "a2a-sdk", specifier = ">=1.0.0,<2" },
     { name = "agent-framework-core", editable = "packages/core" },
 ]
 
+[package.metadata.requires-dev]
+test = [{ name = "agent-framework-orchestrations", editable = "packages/orchestrations" }]
+
 [[package]]
 name = "agent-framework-ag-ui"
 version = "1.7.0"