Skip to content

Commit 059587a

Browse files
author
Sameer Mesiah
committed
Remove redundant Snowflake prefixes from provider modules
Rename Cortex Agent and SQL API modules and update their references. Preserve existing import paths through deprecated aliases.
1 parent bf2de26 commit 059587a

19 files changed

Lines changed: 68 additions & 41 deletions

File tree

providers/snowflake/docs/operators/snowflake_cortex_agent.rst renamed to providers/snowflake/docs/operators/cortex_agent.rst

File renamed without changes.

‎providers/snowflake/provider.yaml‎

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ integrations:
125125
- /docs/apache-airflow-providers-snowflake/operators/snowflake.rst
126126
- /docs/apache-airflow-providers-snowflake/operators/snowpark.rst
127127
- /docs/apache-airflow-providers-snowflake/operators/snowpark_containers.rst
128-
- /docs/apache-airflow-providers-snowflake/operators/snowflake_cortex_agent.rst
128+
- /docs/apache-airflow-providers-snowflake/operators/cortex_agent.rst
129129
logo: /docs/integration-logos/Snowflake.png
130130
tags: [service]
131131

@@ -135,7 +135,7 @@ operators:
135135
- airflow.providers.snowflake.operators.snowflake
136136
- airflow.providers.snowflake.operators.snowpark
137137
- airflow.providers.snowflake.operators.snowpark_containers
138-
- airflow.providers.snowflake.operators.snowflake_cortex_agent
138+
- airflow.providers.snowflake.operators.cortex_agent
139139

140140
task-decorators:
141141
- class-name: airflow.providers.snowflake.decorators.snowpark.snowpark_task
@@ -159,9 +159,8 @@ hooks:
159159
- integration-name: Snowflake
160160
python-modules:
161161
- airflow.providers.snowflake.hooks.snowflake
162-
- airflow.providers.snowflake.hooks.snowflake_sql_api
163-
- airflow.providers.snowflake.hooks.snowflake_cortex_agent
164-
- airflow.providers.snowflake.hooks.cortex_model
162+
- airflow.providers.snowflake.hooks.sql_api
163+
- airflow.providers.snowflake.hooks.cortex_agent
165164

166165
transfers:
167166
- source-integration-name: Amazon Simple Storage Service (S3)

‎providers/snowflake/src/airflow/providers/snowflake/get_provider_info.py‎

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def get_provider_info():
3434
"/docs/apache-airflow-providers-snowflake/operators/snowflake.rst",
3535
"/docs/apache-airflow-providers-snowflake/operators/snowpark.rst",
3636
"/docs/apache-airflow-providers-snowflake/operators/snowpark_containers.rst",
37-
"/docs/apache-airflow-providers-snowflake/operators/snowflake_cortex_agent.rst",
37+
"/docs/apache-airflow-providers-snowflake/operators/cortex_agent.rst",
3838
],
3939
"logo": "/docs/integration-logos/Snowflake.png",
4040
"tags": ["service"],
@@ -47,7 +47,7 @@ def get_provider_info():
4747
"airflow.providers.snowflake.operators.snowflake",
4848
"airflow.providers.snowflake.operators.snowpark",
4949
"airflow.providers.snowflake.operators.snowpark_containers",
50-
"airflow.providers.snowflake.operators.snowflake_cortex_agent",
50+
"airflow.providers.snowflake.operators.cortex_agent",
5151
],
5252
}
5353
],
@@ -78,9 +78,8 @@ def get_provider_info():
7878
"integration-name": "Snowflake",
7979
"python-modules": [
8080
"airflow.providers.snowflake.hooks.snowflake",
81-
"airflow.providers.snowflake.hooks.snowflake_sql_api",
82-
"airflow.providers.snowflake.hooks.snowflake_cortex_agent",
83-
"airflow.providers.snowflake.hooks.cortex_model",
81+
"airflow.providers.snowflake.hooks.sql_api",
82+
"airflow.providers.snowflake.hooks.cortex_agent",
8483
],
8584
}
8685
],

‎providers/snowflake/src/airflow/providers/snowflake/hooks/__init__.py‎

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,20 @@
1414
# KIND, either express or implied. See the License for the
1515
# specific language governing permissions and limitations
1616
# under the License.
17+
18+
from __future__ import annotations
19+
20+
from airflow.utils.deprecation_tools import add_deprecated_classes
21+
22+
__deprecated_classes = {
23+
"snowflake_cortex_agent": {
24+
"SnowflakeCortexAgentHook": (
25+
"airflow.providers.snowflake.hooks.cortex_agent.SnowflakeCortexAgentHook"
26+
),
27+
},
28+
"snowflake_sql_api": {
29+
"SnowflakeSqlApiHook": ("airflow.providers.snowflake.hooks.sql_api.SnowflakeSqlApiHook"),
30+
},
31+
}
32+
33+
add_deprecated_classes(__deprecated_classes, __name__)

providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_cortex_agent.py renamed to providers/snowflake/src/airflow/providers/snowflake/hooks/cortex_agent.py

File renamed without changes.

providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_sql_api.py renamed to providers/snowflake/src/airflow/providers/snowflake/hooks/sql_api.py

File renamed without changes.

‎providers/snowflake/src/airflow/providers/snowflake/operators/__init__.py‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,17 @@
1414
# KIND, either express or implied. See the License for the
1515
# specific language governing permissions and limitations
1616
# under the License.
17+
18+
from __future__ import annotations
19+
20+
from airflow.utils.deprecation_tools import add_deprecated_classes
21+
22+
__deprecated_classes = {
23+
"snowflake_cortex_agent": {
24+
"SnowflakeCortexAgentOperator": (
25+
"airflow.providers.snowflake.operators.cortex_agent.SnowflakeCortexAgentOperator"
26+
),
27+
},
28+
}
29+
30+
add_deprecated_classes(__deprecated_classes, __name__)

providers/snowflake/src/airflow/providers/snowflake/operators/snowflake_cortex_agent.py renamed to providers/snowflake/src/airflow/providers/snowflake/operators/cortex_agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from typing import TYPE_CHECKING, Any
2424

2525
from airflow.providers.common.compat.sdk import BaseOperator
26-
from airflow.providers.snowflake.hooks.snowflake_cortex_agent import SnowflakeCortexAgentHook
26+
from airflow.providers.snowflake.hooks.cortex_agent import SnowflakeCortexAgentHook
2727

2828
if TYPE_CHECKING:
2929
from airflow.providers.common.compat.sdk import Context

‎providers/snowflake/src/airflow/providers/snowflake/operators/snowflake.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
SQLIntervalCheckOperator,
3434
SQLValueCheckOperator,
3535
)
36-
from airflow.providers.snowflake.hooks.snowflake_sql_api import SnowflakeSqlApiHook
36+
from airflow.providers.snowflake.hooks.sql_api import SnowflakeSqlApiHook
3737
from airflow.providers.snowflake.triggers.snowflake_trigger import SnowflakeSqlApiTrigger
3838

3939
_DURABLE_UNSET = object()

‎providers/snowflake/src/airflow/providers/snowflake/triggers/snowflake_trigger.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
from asgiref.sync import sync_to_async
2424

25-
from airflow.providers.snowflake.hooks.snowflake_sql_api import SnowflakeSqlApiHook
25+
from airflow.providers.snowflake.hooks.sql_api import SnowflakeSqlApiHook
2626
from airflow.triggers.base import BaseTrigger, TriggerEvent
2727

2828
if TYPE_CHECKING:

0 commit comments

Comments
 (0)