Move out some exceptions to TaskSDK

As part of the ongoing server/client separation,
this commit moves out some exceptions to the Task sdk.

The AirflowException usage in task sdk has been changed to
RuntimeError.
Also, the TaskNotFound exception is still duplicated due to
the many usages in core API and SDK.

Author: ephraimbuddy

.pre-commit-config.yaml

@@ -1656,10 +1656,12 @@ repos:
           ^airflow-core/src/airflow/api_fastapi/core_api/services/public/connections\.py$|
           ^airflow-core/src/airflow/api_fastapi/core_api/datamodels/hitl\.py$|
           ^airflow-core/src/airflow/api_fastapi/core_api/datamodels/variables\.py$|
+          ^airflow-core/src/airflow/api_fastapi/core_api/routes/public/dag_run\.py$|
           ^airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid\.py$|
           ^airflow-core/src/airflow/api_fastapi/core_api/routes/ui/structure\.py$|
           ^airflow-core/src/airflow/api_fastapi/core_api/services/ui/connections\.py$|
           ^airflow-core/src/airflow/api_fastapi/core_api/services/ui/grid\.py$|
+          ^airflow-core/src/airflow/api_fastapi/execution_api/routes/connections\.py$|
           ^airflow-core/src/airflow/api_fastapi/execution_api/routes/hitl\.py$|
           ^airflow-core/src/airflow/api_fastapi/execution_api/routes/task_instances\.py$|
           ^airflow-core/src/airflow/api_fastapi/logging/decorators\.py$|

airflow-core/src/airflow/api_fastapi/core_api/routes/public/dag_run.py

@@ -77,9 +77,9 @@
 )
 from airflow.api_fastapi.core_api.services.public.dag_run import DagRunWaiter
 from airflow.api_fastapi.logging.decorators import action_logging
-from airflow.exceptions import ParamValidationError
 from airflow.listeners.listener import get_listener_manager
 from airflow.models import DagModel, DagRun
+from airflow.sdk.exceptions import ParamValidationError
 from airflow.utils.state import DagRunState
 from airflow.utils.types import DagRunTriggeredByType, DagRunType
 

airflow-core/src/airflow/api_fastapi/execution_api/routes/connections.py

@@ -23,8 +23,8 @@
 
 from airflow.api_fastapi.execution_api.datamodels.connection import ConnectionResponse
 from airflow.api_fastapi.execution_api.deps import JWTBearerDep
-from airflow.exceptions import AirflowNotFoundException
 from airflow.models.connection import Connection
+from airflow.sdk.exceptions import AirflowNotFoundException
 
 
 async def has_connection_access(

airflow-core/src/airflow/cli/commands/connection_command.py

@@ -32,9 +32,9 @@
 from airflow.cli.simple_table import AirflowConsole
 from airflow.cli.utils import is_stdout, print_export_output
 from airflow.configuration import conf
-from airflow.exceptions import AirflowNotFoundException
 from airflow.models import Connection
 from airflow.providers_manager import ProvidersManager
+from airflow.sdk.exceptions import AirflowNotFoundException
 from airflow.secrets.local_filesystem import load_connections_dict
 from airflow.utils import cli as cli_utils, helpers, yaml
 from airflow.utils.cli import suppress_logs_and_warning

airflow-core/src/airflow/example_dags/example_skip_dag.py

@@ -24,9 +24,9 @@
 
 import pendulum
 
-from airflow.exceptions import AirflowSkipException
 from airflow.providers.standard.operators.empty import EmptyOperator
 from airflow.sdk import DAG, BaseOperator
+from airflow.sdk.exceptions import AirflowSkipException
 from airflow.utils.trigger_rule import TriggerRule
 
 if TYPE_CHECKING: