diff --git a/providers/common/ai/docs/index.rst b/providers/common/ai/docs/index.rst
index 4b0f82608b36d..2a80ebc81a6df 100644
--- a/providers/common/ai/docs/index.rst
+++ b/providers/common/ai/docs/index.rst
@@ -102,8 +102,9 @@ The provider's extras split into a few groups:
   ``pydantic-ai-slim`` optional dependency group; pydantic-ai supports more model providers
   than these four, each under its own extra name, so check the
   `pydantic-ai install docs <https://ai.pydantic.dev/install/#slim-install>`__ for the full list.
-* **Agent tooling** — ``mcp``, ``skills``, ``code-mode``: MCP servers, Agent Skills, and
-  code-mode tool execution.
+* **Agent tooling** — ``mcp``, ``skills``, ``code-mode``, ``shields``: MCP servers, Agent
+  Skills, code-mode tool execution, and shield capabilities (input/output guards, tool
+  guards, cost tracking).
 * **Document loading** — ``pdf``, ``docx``, ``avro``, ``parquet``: file formats for
   document pipelines.
 * **Retrieval / SQL** — ``sql``, ``common.sql``, ``langchain``, ``llamaindex``: RAG and
diff --git a/providers/common/ai/docs/toolsets.rst b/providers/common/ai/docs/toolsets.rst
index f14bac5cc539e..a3530a7a7dbbc 100644
--- a/providers/common/ai/docs/toolsets.rst
+++ b/providers/common/ai/docs/toolsets.rst
@@ -24,7 +24,8 @@ Airflow's 350+ provider hooks already have typed methods, rich docstrings,
 and managed credentials. Toolsets expose them as pydantic-ai tools so that
 LLM agents can call them during multi-turn reasoning.
 
-Four toolsets are included:
+Four toolsets are exported directly from the ``airflow.providers.common.ai.toolsets``
+package root:
 
 - :class:`~airflow.providers.common.ai.toolsets.hook.HookToolset` — generic
   adapter for any Airflow Hook.
@@ -37,7 +38,14 @@ Four toolsets are included:
   the agent a shell and a filesystem inside an isolated sandbox, off the
   Airflow worker. See :ref:`which boundary that is <sandbox-boundaries>`.
 
-All four implement pydantic-ai's
+Three more toolsets are documented later on this page. They are not re-exported from the
+package root, so import each of them from its own submodule::
+
+    from airflow.providers.common.ai.toolsets.datafusion import DataFusionToolset
+    from airflow.providers.common.ai.toolsets.logging import LoggingToolset
+    from airflow.providers.common.ai.toolsets.skills import AgentSkillsToolset
+
+All of the toolsets on this page implement pydantic-ai's
 `AbstractToolset <https://ai.pydantic.dev/toolsets/>`__ interface and can be
 passed to any pydantic-ai ``Agent``, including via
 :class:`~airflow.providers.common.ai.operators.agent.AgentOperator`.
@@ -842,7 +850,7 @@ Four rules for an implementation:
   when it might. The first fails the task for Airflow to retry; the second
   becomes a bounded prompt back to the model.
 - If you cannot enforce something the ``SandboxSpec`` asks for, **raise**. Never
-  provision a weaker sandbox than the DAG author asked for.
+  provision a weaker sandbox than the Dag author asked for.
 
 Parameters
 ^^^^^^^^^^