Skip to content

Commit 3a824dc

Browse files
authored
refactor: reclassify DBT as client for tracking (#191)
1 parent acbb213 commit 3a824dc

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/firebolt/utils/usage_tracker.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,15 +50,15 @@ class ConnectorVersions(BaseModel):
5050
"",
5151
),
5252
("FireboltCLI", "create_connection", Path("firebolt_cli/utils.py"), "firebolt_cli"),
53-
]
54-
55-
DRIVER_MAP = [
5653
(
5754
"DBT",
5855
"open",
5956
Path("dbt/adapters/firebolt/connections.py"),
6057
"dbt.adapters.firebolt",
6158
),
59+
]
60+
61+
DRIVER_MAP = [
6262
("SQLAlchemy", "connect", Path("sqlalchemy/engine/default.py"), "firebolt_db"),
6363
]
6464

tests/unit/utils/test_usage_tracker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,12 +106,12 @@ def test_get_sdk_properties():
106106
),
107107
(
108108
[StackItem("open", "dbt/adapters/firebolt/connections.py")],
109-
DRIVER_MAP,
109+
CLIENT_MAP,
110110
{"DBT": "0.1.4"},
111111
),
112112
(
113113
[StackItem("open", "dbt/adapters/firebolt/connections.py")],
114-
CLIENT_MAP,
114+
DRIVER_MAP,
115115
{},
116116
),
117117
],

0 commit comments

Comments
 (0)