tabletype is 5th col

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

examples/experimental/sea_connector_test.py

@@ -194,7 +194,7 @@ def test_sea_result_set_json_array_inline():
                     log_metadata_results("columns", columns, logger)
 
                     # Extract column names
-                    column_names = extract_key_values(columns, "column_name") or extract_key_values(columns, "columnName") or extract_key_values(columns, "COLUMN_NAME")
+                    column_names = extract_key_values(columns, "col_name")
                     if column_names:
                         logger.info(f"Column names: {column_names}")
 

src/databricks/sql/backend/filters.py

@@ -153,7 +153,7 @@ def filter_tables_by_type(
             table_types if table_types and len(table_types) > 0 else DEFAULT_TABLE_TYPES
         )
 
-        # Table type is typically in the 4th column (index 3)
+        # Table type is the 6th column (index 5)
         return ResultSetFilter.filter_by_column_values(
-            result_set, 3, valid_types, case_sensitive=False
+            result_set, 5, valid_types, case_sensitive=False
         )

src/databricks/sql/backend/sea_backend.py

@@ -622,10 +622,9 @@ def get_tables(
         )
         assert result is not None, "execute_command returned None in synchronous mode"
 
-        # # Apply client-side filtering by table_types if specified
-        # from databricks.sql.backend.filters import ResultSetFilter
-
-        # result = ResultSetFilter.filter_tables_by_type(result, table_types)
+        # Apply client-side filtering by table_types if specified
+        from databricks.sql.backend.filters import ResultSetFilter
+        result = ResultSetFilter.filter_tables_by_type(result, table_types)
 
         return result