SameerMesiah97 opened a new pull request, #70101: URL: https://github.com/apache/airflow/pull/70101
**Description** This change extends `SnowflakeCortexAgentHook` with support for managing Snowflake Cortex Agent Objects. The hook now exposes methods to describe, list and delete Cortex Agents using the Snowflake Cortex Agent REST API. To support these endpoints, the internal request helper has been updated to accept optional query parameters while continuing to centralize authentication, request execution and error handling. **Rationale** The existing hook supports executing Cortex Agents through `run_agent()`, but the Snowflake Cortex Agent REST API also provides endpoints for managing Cortex Agent Objects. Exposing these operations through the hook allows users to inspect and manage agents without needing to invoke the REST API directly. Adding query parameter support to the shared request helper also provides a reusable foundation for additional Cortex Agent management operations in future enhancements. **Tests** Added unit tests verifying that: * `describe_agent()` sends the expected request and returns the API response. * `list_agents()` correctly forwards optional query parameters and returns the list of agents. * `delete_agent()` issues the expected DELETE request, including both values of the `if_exists` parameter. * Existing `run_agent()` tests continue to validate the shared request helper after the addition of query parameter support. -- 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]
