Merge branch 'ext-links-sea' into merge-http-client

Author: varun-edachali-dbx

src/databricks/sql/backend/databricks_client.py

@@ -11,6 +11,8 @@
 from abc import ABC, abstractmethod
 from typing import Dict, Tuple, List, Optional, Any, Union, TYPE_CHECKING
 
+from databricks.sql.types import SSLOptions
+
 if TYPE_CHECKING:
     from databricks.sql.client import Cursor
 
@@ -25,6 +27,13 @@
 
 
 class DatabricksClient(ABC):
+    def __init__(self, ssl_options: SSLOptions, **kwargs):
+        self._use_arrow_native_complex_types = kwargs.get(
+            "_use_arrow_native_complex_types", True
+        )
+        self._max_download_threads = kwargs.get("max_download_threads", 10)
+        self._ssl_options = ssl_options
+
     # == Connection and Session Management ==
     @abstractmethod
     def open_session(

src/databricks/sql/backend/filters.py

@@ -9,12 +9,8 @@
     List,
     Optional,
     Any,
-    Dict,
     Callable,
-    TypeVar,
-    Generic,
     cast,
-    TYPE_CHECKING,
 )
 
 from databricks.sql.backend.types import ExecuteResponse, CommandId
@@ -29,16 +25,13 @@
 
 class ResultSetFilter:
     """
-    A general-purpose filter for result sets that can be applied to any backend.
-
-    This class provides methods to filter result sets based on various criteria,
-    similar to the client-side filtering in the JDBC connector.
+    A general-purpose filter for result sets.
     """
 
     @staticmethod
     def _filter_sea_result_set(
-        result_set: "SeaResultSet", filter_func: Callable[[List[Any]], bool]
-    ) -> "SeaResultSet":
+        result_set: SeaResultSet, filter_func: Callable[[List[Any]], bool]
+    ) -> SeaResultSet:
         """
         Filter a SEA result set using the provided filter function.
 
@@ -49,15 +42,13 @@ def _filter_sea_result_set(
         Returns:
             A filtered SEA result set
         """
+
         # Get all remaining rows
         all_rows = result_set.results.remaining_rows()
 
         # Filter rows
         filtered_rows = [row for row in all_rows if filter_func(row)]
 
-        # Import SeaResultSet here to avoid circular imports
-        from databricks.sql.result_set import SeaResultSet
-
         # Reuse the command_id from the original result set
         command_id = result_set.command_id
 
@@ -73,10 +64,13 @@ def _filter_sea_result_set(
         )
 
         # Create a new ResultData object with filtered data
+
         from databricks.sql.backend.sea.models.base import ResultData
 
         result_data = ResultData(data=filtered_rows, external_links=None)
 
+        from databricks.sql.result_set import SeaResultSet
+
         # Create a new SeaResultSet with the filtered data
         from databricks.sql.backend.sea.backend import SeaDatabricksClient
 
@@ -93,11 +87,11 @@ def _filter_sea_result_set(
 
     @staticmethod
     def filter_by_column_values(
-        result_set: "ResultSet",
+        result_set: ResultSet,
         column_index: int,
         allowed_values: List[str],
         case_sensitive: bool = False,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """
         Filter a result set by values in a specific column.
 
@@ -110,6 +104,7 @@ def filter_by_column_values(
         Returns:
             A filtered result set
         """
+
         # Convert to uppercase for case-insensitive comparison if needed
         if not case_sensitive:
             allowed_values = [v.upper() for v in allowed_values]
@@ -140,8 +135,8 @@ def filter_by_column_values(
 
     @staticmethod
     def filter_tables_by_type(
-        result_set: "ResultSet", table_types: Optional[List[str]] = None
-    ) -> "ResultSet":
+        result_set: ResultSet, table_types: Optional[List[str]] = None
+    ) -> ResultSet:
         """
         Filter a result set of tables by the specified table types.
 
@@ -156,6 +151,7 @@ def filter_tables_by_type(
         Returns:
             A filtered result set containing only tables of the specified types
         """
+
         # Default table types if none specified
         DEFAULT_TABLE_TYPES = ["TABLE", "VIEW", "SYSTEM TABLE"]
         valid_types = (