Fix handle_flight_error for ClientError case

Author: FlorentinD

graphdatascience/query_runner/gds_arrow_client.py

@@ -865,12 +865,8 @@ def _decode_pyarrow_array(array: Array) -> Array:
 
     @staticmethod
     def handle_flight_error(e: Exception) -> None:
-        if (
-            isinstance(e, flight.FlightServerError)
-            or isinstance(e, flight.FlightInternalError)
-            or isinstance(e, ClientError)
-        ):
-            original_message = e.args[0]
+        if isinstance(e, flight.FlightServerError | flight.FlightInternalError | ClientError):
+            original_message = e.args[0] if len(e.args) > 0 else e.message
             improved_message = original_message.replace(
                 "Flight RPC failed with message: org.apache.arrow.flight.FlightRuntimeException: ", ""
             )