diff --git a/providers/pgvector/pyproject.toml b/providers/pgvector/pyproject.toml index e5608717c424d..232ea4858d17c 100644 --- a/providers/pgvector/pyproject.toml +++ b/providers/pgvector/pyproject.toml @@ -58,6 +58,7 @@ requires-python = ">=3.10" # After you modify the dependencies, and rebuild your Breeze CI image with ``breeze ci-image build`` dependencies = [ "apache-airflow>=2.10.0", + "apache-airflow-providers-common-compat>=1.7.4", # + TODO: bump to next version "apache-airflow-providers-postgres>=5.7.1", "pgvector>=0.3.1", ] diff --git a/providers/pgvector/tests/system/pgvector/example_pgvector.py b/providers/pgvector/tests/system/pgvector/example_pgvector.py index 763d9b30edfc2..5e53d4d18ae8d 100644 --- a/providers/pgvector/tests/system/pgvector/example_pgvector.py +++ b/providers/pgvector/tests/system/pgvector/example_pgvector.py @@ -18,11 +18,9 @@ import pendulum -try: - from airflow.sdk import dag, task -except ImportError: - # Airflow 2 path - from airflow.decorators import dag, task # type: ignore[attr-defined,no-redef] +# This example uses common.compat for Airflow 2.x/3.x compatibility. +# If you only need Airflow 3+, you can use: from airflow.sdk import dag, task +from airflow.providers.common.compat.sdk import dag, task from airflow.providers.pgvector.hooks.pgvector import PgVectorHook from airflow.providers.pgvector.operators.pgvector import PgVectorIngestOperator diff --git a/providers/pgvector/tests/system/pgvector/example_pgvector_openai.py b/providers/pgvector/tests/system/pgvector/example_pgvector_openai.py index 92099865e4846..27cf13161f1dc 100644 --- a/providers/pgvector/tests/system/pgvector/example_pgvector_openai.py +++ b/providers/pgvector/tests/system/pgvector/example_pgvector_openai.py @@ -18,11 +18,9 @@ import pendulum -try: - from airflow.sdk import dag, task, teardown -except ImportError: - # Airflow 2 path - from airflow.decorators import dag, task, teardown # type: ignore[attr-defined,no-redef] +# This example uses common.compat for Airflow 2.x/3.x compatibility. +# If you only need Airflow 3+, you can use: from airflow.sdk import dag, task, teardown +from airflow.providers.common.compat.sdk import dag, task, teardown from airflow.providers.openai.operators.openai import OpenAIEmbeddingOperator from airflow.providers.pgvector.operators.pgvector import PgVectorIngestOperator