Provider and lookup typing

msticpy/aiagents/config_utils.py

@@ -13,7 +13,6 @@
 from ..common.exceptions import MsticpyUserConfigError
 from ..common.pkg_config import get_config
 
-
 ConfigItem = Dict[str, Union[str, Callable]]
 ConfigList = List[ConfigItem]
 Config = Dict[str, Union[str, float, ConfigList]]

msticpy/aiagents/rag_agents.py

@@ -11,7 +11,6 @@
 """
 
 import sys
-
 from pathlib import Path
 from typing import List, Optional
 

msticpy/auth/azure_auth.py

@@ -21,6 +21,7 @@
     AzCredentials,
     AzureCloudConfig,
     AzureCredEnvNames,
+    ChainedTokenCredential,
     az_connect_core,
 )
 from .cred_wrapper import CredentialWrapper
@@ -99,7 +100,10 @@ def az_connect(
             az_cli_config.args.get("clientSecret") or ""
         )
     credentials = az_connect_core(
-        auth_methods=auth_methods, tenant_id=tenant_id, silent=silent, **kwargs
+        auth_methods=auth_methods,
+        tenant_id=tenant_id,
+        silent=silent,
+        **kwargs,
     )
     sub_client = SubscriptionClient(
         credential=credentials.modern,
@@ -174,7 +178,7 @@ def fallback_devicecode_creds(
     if not creds:
         raise CloudError("Could not obtain credentials.")
 
-    return AzCredentials(legacy_creds, creds)
+    return AzCredentials(legacy_creds, ChainedTokenCredential(creds))  # type: ignore[arg-type]
 
 
 def get_default_resource_name(resource_uri: str) -> str: