removed unused errors

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

Author: saishreeeee

src/databricks/sql/client.py

@@ -21,10 +21,8 @@
     InterfaceError,
     NotSupportedError,
     ProgrammingError,
-    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 (

src/databricks/sql/exc.py

@@ -126,46 +126,3 @@ class SessionAlreadyClosedError(RequestError):
 
 class CursorAlreadyClosedError(RequestError):
     """Thrown if CancelOperation receives a code 404. ThriftBackend should gracefully proceed as this is expected."""
-
-
-class ConnectionError(OperationalError):
-    """Thrown when connection to Databricks fails during initial setup"""
-
-    def __init__(
-        self,
-        message=None,
-        host_url=None,
-        http_path=None,
-        port=443,
-        user_agent=None,
-        original_exception=None,
-        **kwargs
-    ):
-        # Set up context for connection error telemetry
-        context = kwargs.get("context", {})
-        if host_url:
-            context.update(
-                {
-                    "host_url": host_url,
-                    "http_path": http_path or "",
-                    "port": port,
-                    "user_agent": user_agent,
-                    "original_exception": str(original_exception)
-                    if original_exception
-                    else None,
-                }
-            )
-
-        super().__init__(message=message, context=context, **kwargs)
-
-
-class AuthenticationError(ConnectionError):
-    """Thrown when authentication to Databricks fails"""
-
-    def __init__(self, message=None, auth_type=None, **kwargs):
-        context = kwargs.get("context", {})
-        if auth_type:
-            context["auth_type"] = auth_type
-        kwargs["context"] = context
-
-        super().__init__(message=message, **kwargs)

src/databricks/sql/telemetry/telemetry_client.py

@@ -36,7 +36,6 @@
 import uuid
 import locale
 from databricks.sql.telemetry.utils import BaseTelemetryClient
-import json
 
 logger = logging.getLogger(__name__)
 
@@ -450,21 +449,21 @@ def connection_failure_log(
     ):
         """Send error telemetry when connection creation fails, without requiring a session"""
 
-        UNAUTH_SESSION_ID = "Unauth_session_id"
+        UNAUTH_DUMMY_SESSION_ID = "unauth_session_id"
 
         TelemetryClientFactory.initialize_telemetry_client(
             telemetry_enabled=True,
-            session_id_hex=UNAUTH_SESSION_ID,
+            session_id_hex=UNAUTH_DUMMY_SESSION_ID,
             auth_provider=None,
             host_url=host_url,
         )
 
         telemetry_client = TelemetryClientFactory.get_telemetry_client(
-            UNAUTH_SESSION_ID
+            UNAUTH_DUMMY_SESSION_ID
         )
         telemetry_client._driver_connection_params = DriverConnectionParameters(
             http_path=http_path,
-            mode=DatabricksClientType.THRIFT,
+            mode=DatabricksClientType.THRIFT,  # TODO: Add SEA mode
             host_info=HostDetails(host_url=host_url, port=port),
         )
         telemetry_client._user_agent = user_agent