extract status from resp instead of additional expensive call

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

src/databricks/sql/backend/thrift_backend.py

@@ -876,7 +876,9 @@ def get_execution_result(
         is_staging_operation = t_result_set_metadata_resp.isStagingOperation
         is_direct_results = resp.hasMoreRows
 
-        status = self.get_query_state(command_id)
+        status = CommandState.from_thrift_state(resp.status)
+        if status is None:
+            raise ValueError(f"Unknown command state: {resp.status}")
 
         execute_response = ExecuteResponse(
             command_id=command_id,