SameerMesiah97 opened a new pull request, #69939: URL: https://github.com/apache/airflow/pull/69939
**Description** This change introduces a new `SnowflakeCortexAgentOperator` for executing Snowflake Cortex Agents from Airflow DAGs. The operator delegates to the existing `SnowflakeCortexAgentHook`, providing a declarative interface for invoking Cortex Agents with support for Airflow templating and XCom. **Rationale** `SnowflakeCortexAgentHook` already exposes the Snowflake Cortex Agent Run API, but users currently need to invoke it from custom Python tasks. This operator promotes that functionality to a first-class Airflow task, making Cortex Agent execution consistent with other operators in the Snowflake provider and enabling it to be composed naturally within DAGs. **Tests** Added unit tests verifying that: * `SnowflakeCortexAgentOperator` correctly delegates execution to `SnowflakeCortexAgentHook.run_agent`. * The response returned by the hook is propagated by the operator. **Example DAG** Added a new example DAG, `example_snowflake_cortex_agent`, demonstrating how to execute a Snowflake Cortex Agent using `SnowflakeCortexAgentOperator`. **Documentation** Added documentation for `SnowflakeCortexAgentOperator`, including its supported parameters and a usage example. **Backwards Compatibility** This is a new operator and does not modify existing functionality. No breaking changes are introduced. -- 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]
