This is an automated email from the ASF dual-hosted git repository.
Lee-W pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new d21f0687de3 Correct the common-ai toolset list and document the
shields extra (#71819)
d21f0687de3 is described below
commit d21f0687de3081e7446324c8473d8b6eccde663e
Author: Wei Lee <[email protected]>
AuthorDate: Fri Aug 21 11:05:40 2026 +0900
Correct the common-ai toolset list and document the shields extra (#71819)
---
providers/common/ai/docs/index.rst | 5 +++--
providers/common/ai/docs/toolsets.rst | 14 +++++++++++---
2 files changed, 14 insertions(+), 5 deletions(-)
diff --git a/providers/common/ai/docs/index.rst
b/providers/common/ai/docs/index.rst
index 2e95b070ab0..e2e21a3ba9c 100644
--- a/providers/common/ai/docs/index.rst
+++ b/providers/common/ai/docs/index.rst
@@ -122,8 +122,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 f14bac5cc53..a3530a7a7db 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
^^^^^^^^^^
