Fix client request error checks

Author: trevorbonas

tools/python/liveanalytics_influxdb3_migration_plugin/liveanalytics_migration_client/liveanalytics_influxdb3_migration_client.py

@@ -32,16 +32,18 @@ def __init__(
         db_name: str,
         s3_bucket_name: str,
         resume_migration: bool = False,
-        timeout_seconds: int = 120,
+        timeout_seconds: int = 1_200,
         region: str = "us-west-2",
     ) -> None:
         """
         Initialize
 
         Args:
             db_name (str): Timestream for LiveAnalytics database name.
-            s3_bucket (str): S3 bucket name.
+            s3_bucket (str) S3 bucket name.
             resume_migration (bool): Whether to resume an existing migration, skipping unload operations.
+            timeout_seconds (int): The number of seconds to wait for each migration request.
+            region (str): The AWS Region to use.
 
         Returns:
             None
@@ -432,7 +434,7 @@ def bulk_invoke_http_trigger(self, metadata):
                 )
                 trigger_invocation_response.raise_for_status()
                 response_body = trigger_invocation_response.json()
-                if response_body["status"] == "error":
+                if response_body["status"] != 200 and response_body["status"] != 202:
                     raise RuntimeError(
                         f"Migrating {s3_key} failed: {response_body['message']}"
                     )
@@ -455,8 +457,7 @@ def bulk_invoke_http_trigger(self, metadata):
                     f"Final verification failed: {final_invocation_response.json()['message']}"
                 )
         except Exception as e:
-            self.error(f"HTTP invocation failed: {e}")
-            self.error("View processing engine logs for more information")
+            self.error(f"HTTP invocation failed: {e}. View processing engine logs for more information")
             sys.exit(1)
 
     def get_num_completed_and_total_parquet_files(self):

tools/python/liveanalytics_influxdb3_migration_plugin/liveanalytics_migration_plugin/liveanalytics_migration_plugin.py

@@ -515,7 +515,7 @@ def transform_row_to_lp(influxdb3_local, row, table_name, field_types):
         influxdb3_local (InfluxDB client): Logging and ingestion client
         row (string): Row in parquet file
         table_name (string): Table name
-        field_types(array): Array of field types (only includes double and int64)
+        field_types (list[tuple[str, Any]]): Array of field types, which only includes double, float, and int64.
 
     Returns:
         LineBuilder: LineBuilder object ready for writing to InfluxDB.
@@ -545,7 +545,7 @@ def transform_row_to_lp(influxdb3_local, row, table_name, field_types):
         elif parse_dimensions and isinstance(
             val, (str, numpy.bool_, pandas.StringDtype().type)
         ):
-            builder.tag(col, val)
+            builder.tag(col_name, val)
         elif not parse_dimensions and (pandas.isna(val) or val is None):
             influxdb3_local.info(
                 f"Skipping field value with nulled or missing value for column {col}"