fix(): workflows concurrency (#13645)

packages/core/workflows-sdk/src/utils/composer/transform.ts

@@ -167,7 +167,10 @@ export function transform(
   const ret = {
     __id: uniqId,
     __type: OrchestrationUtils.SymbolWorkflowStepTransformer,
-    __temporary_storage_key: null as { key: string } | null,
+  } as WorkflowData & {
+    __id: string
+    __type: string
+    __temporary_storage_key: { key: string } | null
   }
 
   const returnFn = async function (
@@ -176,6 +179,7 @@ export function transform(
   ): Promise<any> {
     if ("transaction" in transactionContext) {
       const temporaryDataKey = `${transactionContext.transaction.modelId}_${transactionContext.transaction.transactionId}_${uniqId}`
+
       ret.__temporary_storage_key ??= { key: temporaryDataKey }
 
       if (
@@ -199,15 +203,14 @@ export function transform(
       const fn = functions[i]
       const arg = i === 0 ? stepValue : finalResult
 
-      finalResult = await fn.apply(fn, [arg, transactionContext])
+      finalResult = fn.apply(fn, [arg, transactionContext])
+      if (finalResult instanceof Promise) {
+        finalResult = await finalResult
+      }
     }
 
     if ("transaction" in transactionContext) {
       const temporaryDataKey = ret.__temporary_storage_key!
-      if (!temporaryDataKey) {
-        return finalResult
-      }
-
       transactionContext.transaction.setTemporaryData(
         temporaryDataKey,
         finalResult
@@ -217,10 +220,11 @@ export function transform(
     return finalResult
   }
 
-  const proxyfiedRet = proxify<WorkflowData & { __resolver: any }>(
-    ret as unknown as WorkflowData
-  )
+  const proxyfiedRet = proxify<
+    WorkflowData & { __resolver: any; __temporary_storage_key: string | null }
+  >(ret as unknown as WorkflowData)
   proxyfiedRet.__resolver = returnFn as any
+  proxyfiedRet.__temporary_storage_key = null as string | null
 
   return proxyfiedRet
 }