Skip to content

Commit 443dc37

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 d4a47ca commit 443dc37

19 files changed

Lines changed: 66 additions & 39 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 & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ integrations:
124124
- /docs/apache-airflow-providers-snowflake/operators/snowflake.rst
125125
- /docs/apache-airflow-providers-snowflake/operators/snowpark.rst
126126
- /docs/apache-airflow-providers-snowflake/operators/snowpark_containers.rst
127-
- /docs/apache-airflow-providers-snowflake/operators/snowflake_cortex_agent.rst
127+
- /docs/apache-airflow-providers-snowflake/operators/cortex_agent.rst
128128
logo: /docs/integration-logos/Snowflake.png
129129
tags: [service]
130130

@@ -134,7 +134,7 @@ operators:
134134
- airflow.providers.snowflake.operators.snowflake
135135
- airflow.providers.snowflake.operators.snowpark
136136
- airflow.providers.snowflake.operators.snowpark_containers
137-
- airflow.providers.snowflake.operators.snowflake_cortex_agent
137+
- airflow.providers.snowflake.operators.cortex_agent
138138

139139
task-decorators:
140140
- class-name: airflow.providers.snowflake.decorators.snowpark.snowpark_task
@@ -158,8 +158,8 @@ hooks:
158158
- integration-name: Snowflake
159159
python-modules:
160160
- airflow.providers.snowflake.hooks.snowflake
161-
- airflow.providers.snowflake.hooks.snowflake_sql_api
162-
- airflow.providers.snowflake.hooks.snowflake_cortex_agent
161+
- airflow.providers.snowflake.hooks.sql_api
162+
- airflow.providers.snowflake.hooks.cortex_agent
163163

164164
transfers:
165165
- source-integration-name: Amazon Simple Storage Service (S3)

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

Lines changed: 4 additions & 4 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,8 +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",
81+
"airflow.providers.snowflake.hooks.sql_api",
82+
"airflow.providers.snowflake.hooks.cortex_agent",
8383
],
8484
}
8585
],

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

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,19 @@
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 airflow.utils.deprecation_tools import add_deprecated_classes
19+
20+
add_deprecated_classes(
21+
{
22+
"snowflake_cortex_agent": {
23+
"SnowflakeCortexAgentHook": (
24+
"airflow.providers.snowflake.hooks.cortex_agent.SnowflakeCortexAgentHook"
25+
),
26+
},
27+
"snowflake_sql_api": {
28+
"SnowflakeSqlApiHook": ("airflow.providers.snowflake.hooks.sql_api.SnowflakeSqlApiHook"),
29+
},
30+
},
31+
__name__,
32+
)

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: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,16 @@
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 airflow.utils.deprecation_tools import add_deprecated_classes
19+
20+
add_deprecated_classes(
21+
{
22+
"snowflake_cortex_agent": {
23+
"SnowflakeCortexAgentOperator": (
24+
"airflow.providers.snowflake.operators.cortex_agent.SnowflakeCortexAgentOperator"
25+
),
26+
},
27+
},
28+
__name__,
29+
)

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)