add row_limit to Thrift backend

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

Author: varun-edachali-dbx

src/databricks/sql/backend/databricks_client.py

@@ -104,7 +104,7 @@ def execute_command(
             parameters: List of parameters to bind to the query
             async_op: Whether to execute the command asynchronously
             enforce_embedded_schema_correctness: Whether to enforce schema correctness
-            row_limit: Maximum number of rows to fetch overall. Only supported for SEA protocol.
+            row_limit: Maximum number of rows in the operation result. 
 
         Returns:
             If async_op is False, returns a ResultSet object containing the

src/databricks/sql/backend/thrift_backend.py

@@ -970,6 +970,7 @@ def execute_command(
             useArrowNativeTypes=spark_arrow_types,
             parameters=parameters,
             enforceEmbeddedSchemaCorrectness=enforce_embedded_schema_correctness,
+            resultRowLimit=row_limit,
         )
         resp = self.make_request(self._client.ExecuteStatement, req)