Revert "remove un-necessary filters changes"

This reverts commit 5e75fb5.

Author: varun-edachali-dbx

src/databricks/sql/backend/filters.py

@@ -9,27 +9,36 @@
     List,
     Optional,
     Any,
+    Dict,
     Callable,
+    TypeVar,
+    Generic,
     cast,
+    TYPE_CHECKING,
 )
 
+from databricks.sql.backend.types import ExecuteResponse, CommandId
+from databricks.sql.backend.sea.models.base import ResultData
 from databricks.sql.backend.sea.backend import SeaDatabricksClient
-from databricks.sql.backend.types import ExecuteResponse
 
-from databricks.sql.result_set import ResultSet, SeaResultSet
+if TYPE_CHECKING:
+    from databricks.sql.result_set import ResultSet, SeaResultSet
 
 logger = logging.getLogger(__name__)
 
 
 class ResultSetFilter:
     """
-    A general-purpose filter for result sets.
+    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.
     """
 
     @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.
 
@@ -40,13 +49,15 @@ 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
 
@@ -62,13 +73,10 @@ 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
         filtered_result_set = SeaResultSet(
             connection=result_set.connection,
@@ -83,11 +91,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.
 
@@ -100,7 +108,6 @@ 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]
@@ -131,8 +138,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.
 
@@ -147,7 +154,6 @@ 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 = (