Enhance SEA HTTP Client

src/databricks/sql/auth/retry.py

@@ -4,7 +4,7 @@
 import typing
 from importlib.metadata import version
 from enum import Enum
-from typing import List, Optional, Tuple, Union
+from typing import Any, List, Optional, Tuple, Union
 
 import urllib3
 
@@ -123,15 +123,15 @@ def __init__(
             _total: int = urllib3_kwargs.pop("total")
             _attempts_remaining = _total
 
-        _urllib_kwargs_we_care_about = dict(
+        _urllib_kwargs_we_care_about: dict[str, Any] = dict(
             total=_attempts_remaining,
             respect_retry_after_header=True,
             backoff_factor=self.delay_min,
             allowed_methods=["POST"],
             status_forcelist=[429, 503, *self.force_dangerous_codes],
         )
-
-        urllib3_kwargs.update(**_urllib_kwargs_we_care_about)
+        _urllib_kwargs_we_care_about.update(**urllib3_kwargs)
+        urllib3_kwargs = _urllib_kwargs_we_care_about
 
         super().__init__(
             **urllib3_kwargs,

src/databricks/sql/backend/sea/backend.py

@@ -49,7 +49,7 @@
 
 
 def _filter_session_configuration(
-    session_configuration: Optional[Dict[str, str]]
+    session_configuration: Optional[Dict[str, Any]]
 ) -> Optional[Dict[str, str]]:
     if not session_configuration:
         return None
@@ -59,7 +59,7 @@ def _filter_session_configuration(
 
     for key, value in session_configuration.items():
         if key.upper() in ALLOWED_SESSION_CONF_TO_DEFAULT_VALUES_MAP:
-            filtered_session_configuration[key.lower()] = value
+            filtered_session_configuration[key.lower()] = str(value)
         else:
             ignored_configs.add(key)
 
@@ -183,7 +183,7 @@ def max_download_threads(self) -> int:
 
     def open_session(
         self,
-        session_configuration: Optional[Dict[str, str]],
+        session_configuration: Optional[Dict[str, Any]],
         catalog: Optional[str],
         schema: Optional[str],
     ) -> SessionId: