gopidesupavan commented on code in PR #69887:
URL: https://github.com/apache/airflow/pull/69887#discussion_r3587955292
##########
providers/common/ai/src/airflow/providers/common/ai/example_dags/example_agent_capabilities.py:
##########
@@ -117,3 +122,32 @@ def example_agent_capabilities_composed():
# [END howto_operator_agent_capabilities_composed]
example_agent_capabilities_composed()
+
+
+# ---------------------------------------------------------------------------
+# 4. Input guardrails: reject unsafe input before the agent run starts
+# ---------------------------------------------------------------------------
+
+
+# [START howto_operator_agent_capabilities_input_guard]
+if InputGuard is not None:
Review Comment:
it seems... the providers example dags are importing now in core tests and
optional deps not installed..
```
FAILED
airflow-core/tests/unit/always/test_example_dags.py::test_should_be_importable[providers/common/ai/src/airflow/providers/common/ai/example_dags/example_agent_capabilities.py]
- AssertionError:
import_errors={'/opt/airflow/providers/common/ai/src/airflow/providers/common/ai/example_dags/example_agent_capabilities.py':
'Traceback (most recent call last):\n File "<frozen importlib._bootstrap>",
line 241, in _call_with_frames_removed\n File
"/opt/airflow/providers/common/ai/src/airflow/providers/common/ai/example_dags/example_agent_capabilities.py",
line 29, in <module>\n from pydantic_ai_shields import
InputGuard\nModuleNotFoundError: No module named \'pydantic_ai_shields\'\n'}
assert 1 == 0
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
