fix type issues

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

Author: varun-edachali-dbx

src/databricks/sql/backend/sea/backend.py

@@ -3,7 +3,7 @@
 import logging
 import time
 import re
-from typing import Dict, Tuple, List, Optional, Union, TYPE_CHECKING, Set
+from typing import Any, Dict, Tuple, List, Optional, Union, TYPE_CHECKING, Set
 
 from databricks.sql.backend.sea.models.base import ExternalLink, ResultManifest
 from databricks.sql.backend.sea.utils.constants import (

src/databricks/sql/backend/sea/queue.py

@@ -32,7 +32,7 @@
 class SeaResultSetQueueFactory(ABC):
     @staticmethod
     def build_queue(
-        sea_result_data: ResultData,
+        result_data: ResultData,
         manifest: ResultManifest,
         statement_id: str,
         ssl_options: Optional[SSLOptions] = None,
@@ -45,7 +45,7 @@ def build_queue(
         Factory method to build a result set queue for SEA backend.
 
         Args:
-            sea_result_data (ResultData): Result data from SEA response
+            result_data (ResultData): Result data from SEA response
             manifest (ResultManifest): Manifest from SEA response
             statement_id (str): Statement ID for the query
             description (List[List[Any]]): Column descriptions
@@ -59,7 +59,7 @@ def build_queue(
 
         if manifest.format == ResultFormat.JSON_ARRAY.value:
             # INLINE disposition with JSON_ARRAY format
-            return JsonQueue(sea_result_data.data)
+            return JsonQueue(result_data.data)
         elif manifest.format == ResultFormat.ARROW_STREAM.value:
             # EXTERNAL_LINKS disposition
             if not max_download_threads:
@@ -74,11 +74,13 @@ def build_queue(
                 raise ValueError(
                     "SEA client is required for EXTERNAL_LINKS disposition"
                 )
-            if not manifest:
-                raise ValueError("Manifest is required for EXTERNAL_LINKS disposition")
+            if not result_data.external_links:
+                raise ValueError(
+                    "External links are required for EXTERNAL_LINKS disposition"
+                )
 
             return SeaCloudFetchQueue(
-                initial_links=sea_result_data.external_links,
+                initial_links=result_data.external_links,
                 max_download_threads=max_download_threads,
                 ssl_options=ssl_options,
                 sea_client=sea_client,
@@ -125,7 +127,7 @@ def __init__(
         statement_id: str,
         total_chunk_count: int,
         lz4_compressed: bool = False,
-        description: Optional[List[Tuple]] = None,
+        description: List[Tuple] = [],
     ):
         """
         Initialize the SEA CloudFetchQueue.

src/databricks/sql/utils.py

@@ -369,7 +369,7 @@ def __init__(
         start_row_offset: int = 0,
         result_links: Optional[List[TSparkArrowResultLink]] = None,
         lz4_compressed: bool = True,
-        description: Optional[List[Tuple]] = None,
+        description: List[Tuple] = [],
     ):
         """
         Initialize the Thrift CloudFetchQueue.