This is an automated email from the ASF dual-hosted git repository.
Lee-W pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
from 874bc6c215b TS SDK: Add multi-way branching with dag.switch (#74070)
add cdcbf12a3f1 Support PAT and key-pair auth in the Snowflake Cortex
Agent hook (#73932)
No new revisions were added by this update.
Summary of changes:
providers/common/ai/docs/model_providers.rst | 15 +-
.../docs/connections}/index.rst | 3 +-
.../docs/connections/pydantic_ai_snowflake.rst | 132 +++++++
providers/snowflake/docs/connections/snowflake.rst | 10 +-
providers/snowflake/docs/index.rst | 14 +-
.../docs/operators/snowflake_cortex_agent.rst | 19 +
providers/snowflake/provider.yaml | 40 +++
providers/snowflake/pyproject.toml | 4 +
.../providers/snowflake/get_provider_info.py | 33 +-
.../providers/snowflake/hooks/cortex_model.py | 210 +++++++++++
.../airflow/providers/snowflake/hooks/snowflake.py | 10 +
.../snowflake/hooks/snowflake_cortex_agent.py | 74 ++--
.../providers/snowflake/hooks/snowflake_sql_api.py | 63 +---
.../providers/snowflake/utils/_rest_auth.py | 186 ++++++++++
.../snowflake/utils/sql_api_generate_jwt.py | 2 +-
.../unit/snowflake/hooks/test_cortex_model.py | 397 +++++++++++++++++++++
.../tests/unit/snowflake/hooks/test_snowflake.py | 30 ++
.../snowflake/hooks/test_snowflake_cortex_agent.py | 219 +++++++-----
.../unit/snowflake/hooks/test_snowflake_sql_api.py | 129 ++++++-
.../tests/unit/snowflake/utils/test__rest_auth.py | 273 ++++++++++++++
uv.lock | 8 +-
21 files changed, 1680 insertions(+), 191 deletions(-)
copy providers/{opensearch/docs/operators =>
snowflake/docs/connections}/index.rst (97%)
create mode 100644
providers/snowflake/docs/connections/pydantic_ai_snowflake.rst
create mode 100644
providers/snowflake/src/airflow/providers/snowflake/hooks/cortex_model.py
create mode 100644
providers/snowflake/src/airflow/providers/snowflake/utils/_rest_auth.py
create mode 100644
providers/snowflake/tests/unit/snowflake/hooks/test_cortex_model.py
create mode 100644
providers/snowflake/tests/unit/snowflake/utils/test__rest_auth.py
