From 35d01df137ca1a366e4d2c01ca3e5b9f21d4e1bb Mon Sep 17 00:00:00 2001
From: ikorfale <82613188+ikorfale@users.noreply.github.com>
Date: Sat, 12 Sep 2026 18:06:01 +0300
Subject: [PATCH] docs: use correct package name for A2A extra

---
 docs/cookbook/serve-a2a.md | 2 +-
 docs/serving.md            | 2 +-
 tests/docs/test_all.py     | 7 +++++++
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/docs/cookbook/serve-a2a.md b/docs/cookbook/serve-a2a.md
index ef60493..4de1ed5 100644
--- a/docs/cookbook/serve-a2a.md
+++ b/docs/cookbook/serve-a2a.md
@@ -13,7 +13,7 @@ Note: because this tutorial relies upon advanced stdio/stderr communication usin
 tinyagent uses the python asyncio module to support async functionality. When running in Jupyter notebooks, this means we need to enable the use of nested event loops. We'll install tinyagent and enable this below using nest_asyncio.
 
 ```python
-%pip install 'tinyagent[a2a]' 'mcp-server-time' --quiet
+%pip install 'mozilla-ai-tinyagent[a2a]' 'mcp-server-time' --quiet
 
 import nest_asyncio
 
diff --git a/docs/serving.md b/docs/serving.md
index fb58dde..440df64 100644
--- a/docs/serving.md
+++ b/docs/serving.md
@@ -2,7 +2,7 @@
 
 `tinyagent` provides a simple way of serving agents from any of the supported frameworks using different protocols:
 
-- [Agent2Agent Protocol (A2A)](https://google.github.io/A2A/), via the [A2A Python SDK](https://github.com/google-a2a/a2a-python). In order to this protocol, you must install the 'a2a' extra: `pip install 'tinyagent[a2a]'`.
+- [Agent2Agent Protocol (A2A)](https://google.github.io/A2A/), via the [A2A Python SDK](https://github.com/google-a2a/a2a-python). In order to this protocol, you must install the 'a2a' extra: `pip install 'mozilla-ai-tinyagent[a2a]'`.
 
 - [Model Context Protocol (MCP)](https://modelcontextprotocol.io/specification/2025-03-26), via the [MCP Python SDK](https://github.com/modelcontextprotocol/python-sdk).
 
diff --git a/tests/docs/test_all.py b/tests/docs/test_all.py
index 0e98e47..9e739ed 100644
--- a/tests/docs/test_all.py
+++ b/tests/docs/test_all.py
@@ -14,6 +14,13 @@
 DOCS_DIR = pathlib.Path("docs")
 
 
+def test_install_commands_use_distribution_name() -> None:
+    for fpath in DOCS_DIR.glob("**/*"):
+        if fpath.suffix not in {".ipynb", ".md"}:
+            continue
+        assert "pip install 'tinyagent[" not in fpath.read_text(encoding="utf-8")
+
+
 @pytest.mark.parametrize(
     "fpath",
     [