From 67832e7325d233cd6f3471072e58d7e8d3648b07 Mon Sep 17 00:00:00 2001
From: Sid Chatterjee <sid@cloudflare.com>
Date: Thu, 10 Oct 2024 19:33:18 +0100
Subject: [PATCH] Make callback the last arg for Workflow do

---
 types/defines/rpc.d.ts | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/types/defines/rpc.d.ts b/types/defines/rpc.d.ts
index 284a940b87f..7e1bdbda247 100644
--- a/types/defines/rpc.d.ts
+++ b/types/defines/rpc.d.ts
@@ -225,11 +225,22 @@ declare module "cloudflare:workers" {
   };
 
   export type WorkflowStep = {
-    do: <T extends Rpc.Serializable>(
-      name: string,
-      callback: () => Promise<T>,
-      config?: WorkflowStepConfig
-    ) => Promise<T>;
+    do:
+      | (<T extends Rpc.Serializable>(
+          name: string,
+          callback: () => Promise<T>
+        ) => Promise<T>)
+      | (<T extends Rpc.Serializable>(
+          name: string,
+          config: WorkflowStepConfig,
+          callback: () => Promise<T>
+        ) => Promise<T>)
+      | (<T extends Rpc.Serializable>(
+          name: string,
+          config: WorkflowStepConfig | (() => Promise<T>),
+          callback?: () => Promise<T>
+        ) => Promise<T>);
+
     sleep: (name: string, duration: WorkflowSleepDuration) => Promise<void>;
   };