diff --git a/firebolt_provider/__init__.py b/firebolt_provider/__init__.py index 458f7c1..737cfb9 100644 --- a/firebolt_provider/__init__.py +++ b/firebolt_provider/__init__.py @@ -9,5 +9,11 @@ def get_provider_info() -> Dict[str, Any]: "name": "Firebolt Provider", "description": "A Firebolt provider for Apache Airflow.", "hook-class-names": ["firebolt_provider.hooks.firebolt.FireboltHook"], + "connection-types": [ + { + "connection-type": "firebolt", + "hook-class-name": "firebolt_provider.hooks.firebolt.FireboltHook", + } + ], "extra-links": ["firebolt_provider.operators.firebolt.RegistryLink"], } diff --git a/firebolt_provider/hooks/firebolt.py b/firebolt_provider/hooks/firebolt.py index c74f581..dd90da0 100644 --- a/firebolt_provider/hooks/firebolt.py +++ b/firebolt_provider/hooks/firebolt.py @@ -17,6 +17,7 @@ # under the License. # +import logging from collections import namedtuple from typing import Any, Dict, List, Optional, Tuple, Union @@ -40,6 +41,10 @@ # Airflow 2.0 path for the base class from airflow.hooks.dbapi import DbApiHook +# Reduce noise from httpx logger +httpx_logger = logging.getLogger("httpx") +httpx_logger.setLevel(logging.WARNING) + class FireboltHook(DbApiHook): """