Feature/add aggregate support merge

.gitignore

@@ -137,3 +137,4 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+.aider*

setup.py

@@ -13,7 +13,7 @@
 ]
 EXTRAS = {
     "dev": [
-        "black>=24.10.0",
+        "black>=24.8.0",
         "mypy>=1.14.1",
         "pytest>=8.3.4",
         "python-dotenv>=1.0.1",

sqlalchemy_kusto/dbapi.py

@@ -1,12 +1,12 @@
 from collections import namedtuple
 from typing import Any
 
-from azure.identity import WorkloadIdentityCredential
 from azure.kusto.data import (
     ClientRequestProperties,
     KustoClient,
     KustoConnectionStringBuilder,
 )
+from azure.identity import DefaultAzureCredential
 from azure.kusto.data._models import KustoResultColumn
 from azure.kusto.data.exceptions import KustoAuthenticationError, KustoServiceError
 
@@ -39,12 +39,12 @@ def connect(
     cluster: str,
     database: str,
     msi: bool = False,
-    workload_identity: bool = False,
     user_msi: str | None = None,
+    workload_identity: bool = False,
     azure_ad_client_id: str | None = None,
     azure_ad_client_secret: str | None = None,
     azure_ad_tenant_id: str | None = None,
-):
+):  # pylint: disable=too-many-positional-arguments
     """Return a connection to the database."""
     return Connection(
         cluster,
@@ -71,7 +71,7 @@ def __init__(
         azure_ad_client_id: str | None = None,
         azure_ad_client_secret: str | None = None,
         azure_ad_tenant_id: str | None = None,
-    ):
+    ):  # pylint: disable=too-many-positional-arguments
         self.closed = False
         self.cursors: list[Cursor] = []
         kcsb = None
@@ -85,9 +85,8 @@ def __init__(
                 authority_id=azure_ad_tenant_id,
             )
         elif workload_identity:
-            # Workload Identity
             kcsb = KustoConnectionStringBuilder.with_azure_token_credential(
-                cluster, WorkloadIdentityCredential()
+                cluster, DefaultAzureCredential()
             )
         elif msi:
             # Managed Service Identity (MSI)