diff --git a/airflow/providers/apache/pinot/hooks/pinot.py b/airflow/providers/apache/pinot/hooks/pinot.py index 0246a5d3a0fee..0fe11a04e4b82 100644 --- a/airflow/providers/apache/pinot/hooks/pinot.py +++ b/airflow/providers/apache/pinot/hooks/pinot.py @@ -56,6 +56,11 @@ class PinotAdminHook(BaseHook): "Exception" is in the output message. """ + conn_name_attr = "conn_id" + default_conn_name = "pinot_admin_default" + conn_type = "pinot_admin" + hook_name = "Pinot Admin" + def __init__( self, conn_id: str = "pinot_admin_default", @@ -258,6 +263,8 @@ class PinotDbApiHook(DbApiHook): conn_name_attr = "pinot_broker_conn_id" default_conn_name = "pinot_broker_default" + conn_type = "pinot" + hook_name = "Pinot Broker" supports_autocommit = False def get_conn(self) -> Any: