diff --git a/providers/influxdb/src/airflow/providers/influxdb/hooks/influxdb.py b/providers/influxdb/src/airflow/providers/influxdb/hooks/influxdb.py
index 7f039eefdde8d..c5ba2a9845949 100644
--- a/providers/influxdb/src/airflow/providers/influxdb/hooks/influxdb.py
+++ b/providers/influxdb/src/airflow/providers/influxdb/hooks/influxdb.py
@@ -31,10 +31,7 @@
 from influxdb_client.client.write.point import Point
 from influxdb_client.client.write_api import SYNCHRONOUS
 
-try:
-    from airflow.sdk import BaseHook
-except ImportError:
-    from airflow.hooks.base import BaseHook  # type: ignore[attr-defined,no-redef]
+from airflow.providers.influxdb.version_compat import BaseHook
 
 if TYPE_CHECKING:
     import pandas as pd
diff --git a/providers/influxdb/src/airflow/providers/influxdb/version_compat.py b/providers/influxdb/src/airflow/providers/influxdb/version_compat.py
index ab72155414178..c98ecb412aa57 100644
--- a/providers/influxdb/src/airflow/providers/influxdb/version_compat.py
+++ b/providers/influxdb/src/airflow/providers/influxdb/version_compat.py
@@ -28,10 +28,16 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]:
 
 
 AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0)
+AIRFLOW_V_3_1_PLUS: bool = get_base_airflow_version_tuple() >= (3, 1, 0)
+
+if AIRFLOW_V_3_1_PLUS:
+    from airflow.sdk import BaseHook
+else:
+    from airflow.hooks.base import BaseHook  # type: ignore[attr-defined,no-redef]
 
 if AIRFLOW_V_3_0_PLUS:
     from airflow.sdk import BaseOperator
 else:
     from airflow.models import BaseOperator  # type: ignore[no-redef]
 
-__all__ = ["AIRFLOW_V_3_0_PLUS", "BaseOperator"]
+__all__ = ["AIRFLOW_V_3_0_PLUS", "AIRFLOW_V_3_1_PLUS", "BaseHook", "BaseOperator"]