Save new request body when retrying the request to save progress (#1732)

Fixes OPS-3194.

Author: MarceloRGonc

packages/engine/src/lib/services/progress.service.ts

@@ -107,9 +107,15 @@ const sendUpdateRunRequest = async (flowRunId: FlowRunId): Promise<void> => {
       } as BodyAccessKeyRequest,
       {
         retries: MAX_RETRIES,
-        retryCondition: (error: AxiosError) => {
+        retryCondition: async (error: AxiosError) => {
           throwIfExecutionTimeExceeded();
-          return isRetryableError(error);
+
+          if (!isRetryableError(error)) {
+            return false;
+          }
+
+          await updateRequestBody(request, error);
+          return true;
         },
         retryDelay: (retryCount: number) => (retryCount + 1) * 200, // 200ms, 400ms, 600ms
       },
@@ -129,6 +135,22 @@ const sendUpdateRunRequest = async (flowRunId: FlowRunId): Promise<void> => {
   logger.debug(`Progress update request for ${flowRunId} took ${duration}ms`);
 };
 
+async function updateRequestBody(
+  request: UpdateRunProgressRequest,
+  error: AxiosError,
+): Promise<void> {
+  try {
+    if (error.config) {
+      const newBodyAccessKey = await saveRequestBody(request);
+      error.config.data = JSON.stringify({
+        bodyAccessKey: newBodyAccessKey,
+      });
+    }
+  } catch (e) {
+    logger.info('Failed to update progress request body.', e);
+  }
+}
+
 type UpdateStepProgressParams = {
   engineConstants: EngineConstants;
   flowExecutorContext: FlowExecutorContext;