diff --git a/src/firebolt/utils/usage_tracker.py b/src/firebolt/utils/usage_tracker.py index 624ca7ef04f..54b9d413b8e 100644 --- a/src/firebolt/utils/usage_tracker.py +++ b/src/firebolt/utils/usage_tracker.py @@ -50,15 +50,15 @@ class ConnectorVersions(BaseModel): "", ), ("FireboltCLI", "create_connection", Path("firebolt_cli/utils.py"), "firebolt_cli"), -] - -DRIVER_MAP = [ ( "DBT", "open", Path("dbt/adapters/firebolt/connections.py"), "dbt.adapters.firebolt", ), +] + +DRIVER_MAP = [ ("SQLAlchemy", "connect", Path("sqlalchemy/engine/default.py"), "firebolt_db"), ] diff --git a/tests/unit/utils/test_usage_tracker.py b/tests/unit/utils/test_usage_tracker.py index 55eed561bf4..8ffa17e975c 100644 --- a/tests/unit/utils/test_usage_tracker.py +++ b/tests/unit/utils/test_usage_tracker.py @@ -106,12 +106,12 @@ def test_get_sdk_properties(): ), ( [StackItem("open", "dbt/adapters/firebolt/connections.py")], - DRIVER_MAP, + CLIENT_MAP, {"DBT": "0.1.4"}, ), ( [StackItem("open", "dbt/adapters/firebolt/connections.py")], - CLIENT_MAP, + DRIVER_MAP, {}, ), ],