retry errors

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/client.py

@@ -24,6 +24,7 @@
     AuthenticationError,
     ConnectionError,
 )
+from urllib3.exceptions import MaxRetryError
 from databricks.sql.thrift_api.TCLIService import ttypes
 from databricks.sql.thrift_backend import ThriftBackend
 from databricks.sql.utils import (
@@ -307,6 +308,8 @@ def read(self) -> Optional[OAuthToken]:
             self._open_session_resp = self.thrift_backend.open_session(
                 session_configuration, catalog, schema
             )
+        except (RequestError, MaxRetryError) as e:
+            raise
         except Exception as e:
             raise ConnectionError(
                 message=f"Failed to establish connection: {str(e)}",