AlejandroMorgante opened a new pull request, #70933: URL: https://github.com/apache/airflow/pull/70933
Add `QueryAgentEngineOperator` for direct synchronous Agent Engine invocations through the public `ReasoningEngineExecutionService` GAPIC. This complements `RunQueryJobOperator` when a Google Cloud Storage-backed asynchronous job is not needed. The hook configures the regional endpoint and forwards retry, timeout, metadata, input, and class method options. The operator returns the JSON-serializable Agent Engine output directly, without first fetching the runtime or creating a query job. The new path is covered by hook and operator unit tests, provider documentation, and the existing Agent Engine system-test example. related: #68479 Testing: - `uv run --project providers/google pytest providers/google/tests/unit/google/cloud/hooks/vertex_ai/test_agent_engine.py providers/google/tests/unit/google/cloud/operators/vertex_ai/test_agent_engine.py -q` (`43 passed`) - `breeze testing providers-tests --test-type "Providers[google]"` (`4972 passed, 59 skipped`) - `breeze run mypy providers/google/src/airflow/providers/google/cloud/hooks/vertex_ai/agent_engine.py providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/agent_engine.py` - `prek run --from-ref upstream/main --stage pre-commit` - `prek run --from-ref upstream/main --stage manual` - `breeze build-docs google --one-pass-only` (documentation and spellcheck) --- ##### Was generative AI tooling used to co-author this PR? - [X] Yes — Codex (GPT-5) Generated-by: Codex (GPT-5) following [the guidelines](https://github.com/apache/airflow/blob/main/contributing-docs/05_pull_requests.rst#gen-ai-assisted-contributions) -- 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]
