Apply suggestions from code review

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Author: ricardozanini

experimental/fluent/func/src/main/java/io/serverlessworkflow/fluent/func/FuncTaskItemListBuilder.java

@@ -217,13 +217,12 @@ public FuncTaskItemListBuilder tryCatch(
     return this.addTaskItem(new TaskItem(name, new Task().withTryTask(tryTaskBuilder.build())));
   }
 
-  public FuncTaskItemListBuilder wait(
-      Consumer<io.serverlessworkflow.fluent.spec.WaitTaskBuilder> itemsConfigurer) {
+  public FuncTaskItemListBuilder wait(Consumer<WaitTaskBuilder> itemsConfigurer) {
     return wait(null, itemsConfigurer);
   }
 
   public FuncTaskItemListBuilder wait(String name, Consumer<WaitTaskBuilder> itemsConfigurer) {
-    name = this.defaultNameAndRequireConfig(name, itemsConfigurer, "wait");
+    name = this.defaultNameAndRequireConfig(name, itemsConfigurer, TYPE_WAIT);
     final WaitTaskBuilder waitTaskBuilder = new WaitTaskBuilder();
     itemsConfigurer.accept(waitTaskBuilder);
     return this.addTaskItem(new TaskItem(name, new Task().withWaitTask(waitTaskBuilder.build())));

impl/test/src/test/java/io/serverlessworkflow/impl/test/WaitExecutorTest.java

@@ -97,11 +97,29 @@ void testWaitWithDurationInlineComposite() {
     assertThat(elapsed).isGreaterThanOrEqualTo(1500); // 1 second + 500 milliseconds
   }
 
-  // ========== DurationLiteral Tests (via Duration.parse) ==========
+  // ========== DurationLiteral Tests (TimeoutAfter.durationLiteral) ==========
 
   @Test
   void testWaitWithDurationLiteralISO8601Seconds() {
     Workflow workflow =
+        WorkflowBuilder.workflow("wait-literal-seconds", "test", "0.1.0")
+            .tasks(
+                list ->
+                    list.wait(
+                        w ->
+                            w.build()
+                                .setWait(
+                                    new io.serverlessworkflow.api.types.TimeoutAfter()
+                                        .withDurationLiteral("PT1S"))))
+            .build();
+
+    long startTime = System.currentTimeMillis();
+    WorkflowModel model = appl.workflowDefinition(workflow).instance(Map.of()).start().join();
+    long elapsed = System.currentTimeMillis() - startTime;
+
+    assertThat(model).isNotNull();
+    assertThat(elapsed).isGreaterThanOrEqualTo(1000);
+  }
         WorkflowBuilder.workflow("wait-literal-seconds", "test", "0.1.0")
             .tasks(DSL.wait(Duration.parse("PT1S")))
             .build();
@@ -119,7 +137,14 @@ void testWaitWithDurationLiteralISO8601Composite() {
     // PT1.5S = 1 second 500 milliseconds (keep test fast)
     Workflow workflow =
         WorkflowBuilder.workflow("wait-literal-composite", "test", "0.1.0")
-            .tasks(DSL.wait(Duration.parse("PT1.5S")))
+            .tasks(
+                list ->
+                    list.wait(
+                        w ->
+                            w.build()
+                                .setWait(
+                                    new io.serverlessworkflow.api.types.TimeoutAfter()
+                                        .withDurationLiteral("PT1.5S"))))
             .build();
 
     long startTime = System.currentTimeMillis();
@@ -135,7 +160,14 @@ void testWaitWithDurationLiteralISO8601Milliseconds() {
     // PT0.1S = 100 milliseconds
     Workflow workflow =
         WorkflowBuilder.workflow("wait-literal-millis", "test", "0.1.0")
-            .tasks(DSL.wait(Duration.parse("PT0.1S")))
+            .tasks(
+                list ->
+                    list.wait(
+                        w ->
+                            w.build()
+                                .setWait(
+                                    new io.serverlessworkflow.api.types.TimeoutAfter()
+                                        .withDurationLiteral("PT0.1S"))))
             .build();
 
     long startTime = System.currentTimeMillis();