remove redundant params (byte limit, catalog, schema) in exec command

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

Author: varun-edachali-dbx

examples/experimental/sea_connector_test.py

@@ -105,7 +105,7 @@ def test_sea_result_set_json_array_inline():
         )
 
         # Create cursor
-        cursor = connection.cursor(arraysize=0, buffer_size_bytes=0)
+        cursor = connection.cursor()
 
         # Execute a query that returns a small result set (will use INLINE disposition)
         logger.info("Executing query: SELECT * FROM range(1, 10) AS id")

src/databricks/sql/backend/models/requests.py

@@ -21,19 +21,16 @@ class StatementParameter:
 class ExecuteStatementRequest:
     """Request to execute a SQL statement."""
 
-    warehouse_id: str
-    statement: str
     session_id: str
+    statement: str
+    warehouse_id: str
     disposition: str = "EXTERNAL_LINKS"
     format: str = "JSON_ARRAY"
+    result_compression: Optional[str] = None
+    parameters: Optional[List[StatementParameter]] = None
     wait_timeout: str = "10s"
     on_wait_timeout: str = "CONTINUE"
     row_limit: Optional[int] = None
-    byte_limit: Optional[int] = None
-    parameters: Optional[List[StatementParameter]] = None
-    catalog: Optional[str] = None
-    schema: Optional[str] = None
-    result_compression: Optional[str] = None
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert the request to a dictionary for JSON serialization."""
@@ -50,15 +47,6 @@ def to_dict(self) -> Dict[str, Any]:
         if self.row_limit is not None and self.row_limit > 0:
             result["row_limit"] = self.row_limit
 
-        if self.byte_limit is not None and self.byte_limit > 0:
-            result["byte_limit"] = self.byte_limit
-
-        if self.catalog:
-            result["catalog"] = self.catalog
-
-        if self.schema:
-            result["schema"] = self.schema
-
         if self.result_compression:
             result["result_compression"] = self.result_compression
 

src/databricks/sql/backend/sea_backend.py

@@ -353,8 +353,7 @@ def execute_command(
             format=format,
             wait_timeout="0s" if async_op else "10s",
             on_wait_timeout="CONTINUE",
-            row_limit=max_rows if max_rows > 0 else None,
-            byte_limit=max_bytes if max_bytes > 0 else None,
+            row_limit=max_rows,
             parameters=sea_parameters if sea_parameters else None,
             result_compression=result_compression,
         )