Merge branch 'sea-migration' into fetch-json-inline

Author: varun-edachali-dbx

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

@@ -354,7 +354,7 @@ def _results_message_to_execute_response(
             description=description,
             has_been_closed_server_side=False,
             lz4_compressed=lz4_compressed,
-            is_staging_operation=False,
+            is_staging_operation=response.manifest.is_volume_operation,
             arrow_schema_bytes=None,
             result_format=response.manifest.format,
         )

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

@@ -92,4 +92,4 @@ class ResultManifest:
     truncated: bool = False
     chunks: Optional[List[ChunkInfo]] = None
     result_compression: Optional[str] = None
-    is_volume_operation: Optional[bool] = None
+    is_volume_operation: bool = False

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

@@ -65,7 +65,7 @@ def _parse_manifest(data: Dict[str, Any]) -> ResultManifest:
         truncated=manifest_data.get("truncated", False),
         chunks=chunks,
         result_compression=manifest_data.get("result_compression"),
-        is_volume_operation=manifest_data.get("is_volume_operation"),
+        is_volume_operation=manifest_data.get("is_volume_operation", False),
     )