Use ModelFactory for resolve last state model

Signed-off-by: Matheus Cruz <matheuscruz.dev@gmail.com>

Author: mcruzdev

experimental/fluent/func/src/main/java/io/serverlessworkflow/fluent/func/dsl/EnrichWithModelBiFunction.java

@@ -20,8 +20,8 @@
 /**
  * A function that enriches a typed value by combining it with the root workflow input.
  *
- * <p>This is useful when you need to merge a task output or state value with the original
- * workflow input, where the value is typed and the root input is a {@link WorkflowModel}.
+ * <p>This is useful when you need to merge a task output or state value with the original workflow
+ * input, where the value is typed and the root input is a {@link WorkflowModel}.
  *
  * @param <T> The type of the input value to be enriched
  * @param <R> The type of the enriched result

experimental/fluent/func/src/main/java/io/serverlessworkflow/fluent/func/dsl/FuncDSL.java

@@ -309,7 +309,18 @@ public static <T, R> JavaContextFunction<T, R> enrich(
     return (lastState, workflowContext) -> {
       Objects.requireNonNull(lastStateClass, "lastStateClass must not be null");
       final WorkflowModel rootInput = workflowContext.instanceData().input();
-      final T typedLastState = lastStateClass.cast(lastState);
+      final WorkflowModel lastStateModel =
+          workflowContext.definition().application().modelFactory().fromAny(lastState);
+      final T typedLastState =
+          lastStateModel
+              .as(lastStateClass)
+              .orElseThrow(
+                  () ->
+                      new IllegalArgumentException(
+                          "Cannot convert lastState of type "
+                              + (lastState != null ? lastState.getClass().getName() : "null")
+                              + " to "
+                              + lastStateClass.getName()));
       return fn.apply(typedLastState, rootInput);
     };
   }