Set State["PreviousStateGuid"] in StateHistory

lib/floe/workflow.rb

@@ -161,13 +161,20 @@ def current_state
     private
 
     def step_next
-      if context.next_state
-        # if rerunning due to an error (and we are using Retry)
-        if context.state_name == context.next_state && context.failed? && context.state.key?("Retrier")
-          restore_values = context.state.slice("RetryCount", "Input", "Retrier")
-        end
-        context.state = {"Name" => context.next_state, "Input" => context.output}.merge!(restore_values || {})
+      context.state_history << context.state
+
+      return unless context.next_state
+
+      next_state = {"Name" => context.next_state, "PreviousStateGuid" => context.state["Guid"]}
+
+      # if rerunning due to an error (and we are using Retry)
+      if context.state_name == context.next_state && context.failed? && context.state.key?("Retrier")
+        next_state.merge!(context.state.slice("RetryCount", "Input", "Retrier"))
+      else
+        next_state["Input"] = context.output
       end
+
+      context.state = next_state
     end
   end
 end

lib/floe/workflow/state.rb

@@ -73,8 +73,6 @@ def finish
         level = context.output&.[]("Error") ? :error : :info
         logger.public_send(level, "Running state: [#{long_name}] with input [#{context.input}]...Complete #{context.next_state ? "- next state [#{context.next_state}]" : "workflow -"} output: [#{context.output}]")
 
-        context.state_history << context.state
-
         0
       end