nit

Author: jprakash-db

src/databricks/sql/auth/common.py

@@ -1,4 +1,5 @@
 from enum import Enum
+from typing import Optional
 
 
 class AuthType(Enum):
@@ -7,7 +8,7 @@ class AuthType(Enum):
     AZURE_SP_M2M = "azure-sp-m2m"
 
 
-def get_effective_azure_login_app_id(hostname: str) -> str:
+def get_effective_azure_login_app_id(hostname: Optional[str]) -> str:
     """
     Get the effective Azure login app ID for a given hostname.
     This function determines the appropriate Azure login app ID based on the hostname.

src/databricks/sql/auth/oauth.py

@@ -6,7 +6,7 @@
 import webbrowser
 from datetime import datetime, timezone
 from http.server import HTTPServer
-from typing import List
+from typing import List, Optional
 
 import oauthlib.oauth2
 import requests
@@ -317,9 +317,9 @@ class ClientCredentialsTokenSource(RefreshableTokenSource):
 
     def __init__(
         self,
-        token_url: str,
-        oauth_client_id: str,
-        oauth_client_secret: str,
+        token_url: Optional[str],
+        oauth_client_id: Optional[str],
+        oauth_client_secret: Optional[str],
         extra_params: dict = {},
     ):
         self.oauth_client_id = oauth_client_id