breaking: implement workflow hooks (first iteration) (#8346)

packages/core/core-flows/src/api-key/workflows/create-api-keys.ts

@@ -1,5 +1,9 @@
 import { ApiKeyDTO, CreateApiKeyDTO } from "@medusajs/types"
-import { createWorkflow, WorkflowData } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createApiKeysStep } from "../steps"
 
 type WorkflowInput = { api_keys: CreateApiKeyDTO[] }
@@ -7,7 +11,7 @@ type WorkflowInput = { api_keys: CreateApiKeyDTO[] }
 export const createApiKeysWorkflowId = "create-api-keys"
 export const createApiKeysWorkflow = createWorkflow(
   createApiKeysWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<ApiKeyDTO[]> => {
-    return createApiKeysStep(input)
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<ApiKeyDTO[]> => {
+    return new WorkflowResponse(createApiKeysStep(input))
   }
 )

packages/core/core-flows/src/api-key/workflows/revoke-api-keys.ts

@@ -3,7 +3,11 @@ import {
   FilterableApiKeyProps,
   RevokeApiKeyDTO,
 } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { revokeApiKeysStep } from "../steps"
 
 type RevokeApiKeysStepInput = {
@@ -16,7 +20,7 @@ type WorkflowInput = RevokeApiKeysStepInput
 export const revokeApiKeysWorkflowId = "revoke-api-keys"
 export const revokeApiKeysWorkflow = createWorkflow(
   revokeApiKeysWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<ApiKeyDTO[]> => {
-    return revokeApiKeysStep(input)
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<ApiKeyDTO[]> => {
+    return new WorkflowResponse(revokeApiKeysStep(input))
   }
 )

packages/core/core-flows/src/api-key/workflows/update-api-keys.ts

@@ -3,7 +3,11 @@ import {
   FilterableApiKeyProps,
   UpdateApiKeyDTO,
 } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateApiKeysStep } from "../steps"
 
 type UpdateApiKeysStepInput = {
@@ -16,7 +20,7 @@ type WorkflowInput = UpdateApiKeysStepInput
 export const updateApiKeysWorkflowId = "update-api-keys"
 export const updateApiKeysWorkflow = createWorkflow(
   updateApiKeysWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<ApiKeyDTO[]> => {
-    return updateApiKeysStep(input)
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<ApiKeyDTO[]> => {
+    return new WorkflowResponse(updateApiKeysStep(input))
   }
 )

packages/core/core-flows/src/common/workflows/batch-links.ts

@@ -2,6 +2,7 @@ import { LinkDefinition } from "@medusajs/modules-sdk"
 import { BatchWorkflowInput } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
 } from "@medusajs/workflows-sdk"
@@ -23,10 +24,10 @@ export const batchLinksWorkflow = createWorkflow(
       dismissRemoteLinkStep(input.delete || [])
     )
 
-    return {
+    return new WorkflowResponse({
       created,
       updated,
       deleted,
-    }
+    })
   }
 )

packages/core/core-flows/src/common/workflows/create-links.ts

@@ -1,11 +1,15 @@
 import { LinkDefinition } from "@medusajs/modules-sdk"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createRemoteLinkStep } from "../steps/create-remote-links"
 
 export const createLinksWorkflowId = "create-link"
 export const createLinksWorkflow = createWorkflow(
   createLinksWorkflowId,
   (input: WorkflowData<LinkDefinition[]>) => {
-    return createRemoteLinkStep(input)
+    return new WorkflowResponse(createRemoteLinkStep(input))
   }
 )

packages/core/core-flows/src/common/workflows/dismiss-links.ts

@@ -1,11 +1,15 @@
 import { LinkDefinition } from "@medusajs/modules-sdk"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { dismissRemoteLinkStep } from "../steps/dismiss-remote-links"
 
 export const dismissLinksWorkflowId = "dismiss-link"
 export const dismissLinksWorkflow = createWorkflow(
   dismissLinksWorkflowId,
   (input: WorkflowData<LinkDefinition[]>) => {
-    return dismissRemoteLinkStep(input)
+    return new WorkflowResponse(dismissRemoteLinkStep(input))
   }
 )

packages/core/core-flows/src/common/workflows/update-links.ts

@@ -1,11 +1,15 @@
 import { LinkDefinition } from "@medusajs/modules-sdk"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateRemoteLinksStep } from "../steps/update-remote-links"
 
 export const updateLinksWorkflowId = "update-link"
 export const updateLinksWorkflow = createWorkflow(
   updateLinksWorkflowId,
   (input: WorkflowData<LinkDefinition[]>) => {
-    return updateRemoteLinksStep(input)
+    return new WorkflowResponse(updateRemoteLinksStep(input))
   }
 )

packages/core/core-flows/src/customer-group/workflows/create-customer-groups.ts

@@ -1,5 +1,9 @@
 import { CustomerGroupDTO, CreateCustomerGroupDTO } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createCustomerGroupsStep } from "../steps"
 
 type WorkflowInput = { customersData: CreateCustomerGroupDTO[] }
@@ -7,7 +11,9 @@ type WorkflowInput = { customersData: CreateCustomerGroupDTO[] }
 export const createCustomerGroupsWorkflowId = "create-customer-groups"
 export const createCustomerGroupsWorkflow = createWorkflow(
   createCustomerGroupsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<CustomerGroupDTO[]> => {
-    return createCustomerGroupsStep(input.customersData)
+  (
+    input: WorkflowData<WorkflowInput>
+  ): WorkflowResponse<CustomerGroupDTO[]> => {
+    return new WorkflowResponse(createCustomerGroupsStep(input.customersData))
   }
 )

packages/core/core-flows/src/customer-group/workflows/update-customer-groups.ts

@@ -3,7 +3,11 @@ import {
   FilterableCustomerGroupProps,
   CustomerGroupUpdatableFields,
 } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateCustomerGroupsStep } from "../steps"
 
 type WorkflowInput = {
@@ -14,7 +18,9 @@ type WorkflowInput = {
 export const updateCustomerGroupsWorkflowId = "update-customer-groups"
 export const updateCustomerGroupsWorkflow = createWorkflow(
   updateCustomerGroupsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<CustomerGroupDTO[]> => {
-    return updateCustomerGroupsStep(input)
+  (
+    input: WorkflowData<WorkflowInput>
+  ): WorkflowResponse<CustomerGroupDTO[]> => {
+    return new WorkflowResponse(updateCustomerGroupsStep(input))
   }
 )

packages/core/core-flows/src/customer/workflows/create-addresses.ts

@@ -1,6 +1,7 @@
 import { CreateCustomerAddressDTO, CustomerAddressDTO } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -16,7 +17,9 @@ type WorkflowInput = { addresses: CreateCustomerAddressDTO[] }
 export const createCustomerAddressesWorkflowId = "create-customer-addresses"
 export const createCustomerAddressesWorkflow = createWorkflow(
   createCustomerAddressesWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<CustomerAddressDTO[]> => {
+  (
+    input: WorkflowData<WorkflowInput>
+  ): WorkflowResponse<CustomerAddressDTO[]> => {
     const unsetInput = transform(input, (data) => ({
       create: data.addresses,
     }))
@@ -26,6 +29,6 @@ export const createCustomerAddressesWorkflow = createWorkflow(
       maybeUnsetDefaultBillingAddressesStep(unsetInput)
     )
 
-    return createCustomerAddressesStep(input.addresses)
+    return new WorkflowResponse(createCustomerAddressesStep(input.addresses))
   }
 )

packages/core/core-flows/src/customer/workflows/create-customer-account.ts

@@ -1,5 +1,9 @@
 import { CreateCustomerDTO, CustomerDTO } from "@medusajs/types"
-import { createWorkflow, WorkflowData } from "@medusajs/workflows-sdk"
+import {
+  createWorkflow,
+  WorkflowData,
+  WorkflowResponse,
+} from "@medusajs/workflows-sdk"
 import { createCustomersStep } from "../steps"
 import { transform } from "@medusajs/workflows-sdk"
 import { setAuthAppMetadataStep } from "../../auth"
@@ -12,7 +16,7 @@ type WorkflowInput = {
 export const createCustomerAccountWorkflowId = "create-customer-account"
 export const createCustomerAccountWorkflow = createWorkflow(
   createCustomerAccountWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<CustomerDTO> => {
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<CustomerDTO> => {
     const customers = createCustomersStep([input.customersData])
 
     const customer = transform(
@@ -26,6 +30,6 @@ export const createCustomerAccountWorkflow = createWorkflow(
       value: customer.id,
     })
 
-    return customer
+    return new WorkflowResponse(customer)
   }
 )

packages/core/core-flows/src/customer/workflows/create-customers.ts

@@ -1,5 +1,9 @@
 import { CustomerDTO, CreateCustomerDTO } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createCustomersStep } from "../steps"
 
 type WorkflowInput = { customersData: CreateCustomerDTO[] }
@@ -7,7 +11,7 @@ type WorkflowInput = { customersData: CreateCustomerDTO[] }
 export const createCustomersWorkflowId = "create-customers"
 export const createCustomersWorkflow = createWorkflow(
   createCustomersWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<CustomerDTO[]> => {
-    return createCustomersStep(input.customersData)
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<CustomerDTO[]> => {
+    return new WorkflowResponse(createCustomersStep(input.customersData))
   }
 )

packages/core/core-flows/src/customer/workflows/update-addresses.ts

@@ -5,6 +5,7 @@ import {
 } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -23,7 +24,9 @@ type WorkflowInput = {
 export const updateCustomerAddressesWorkflowId = "update-customer-addresses"
 export const updateCustomerAddressesWorkflow = createWorkflow(
   updateCustomerAddressesWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<CustomerAddressDTO[]> => {
+  (
+    input: WorkflowData<WorkflowInput>
+  ): WorkflowResponse<CustomerAddressDTO[]> => {
     const unsetInput = transform(input, (data) => ({
       update: data,
     }))
@@ -33,6 +36,6 @@ export const updateCustomerAddressesWorkflow = createWorkflow(
       maybeUnsetDefaultBillingAddressesStep(unsetInput)
     )
 
-    return updateCustomerAddressesStep(input)
+    return new WorkflowResponse(updateCustomerAddressesStep(input))
   }
 )

packages/core/core-flows/src/customer/workflows/update-customers.ts

@@ -3,7 +3,11 @@ import {
   CustomerUpdatableFields,
   FilterableCustomerProps,
 } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateCustomersStep } from "../steps"
 
 type UpdateCustomersStepInput = {
@@ -16,7 +20,7 @@ type WorkflowInput = UpdateCustomersStepInput
 export const updateCustomersWorkflowId = "update-customers"
 export const updateCustomersWorkflow = createWorkflow(
   updateCustomersWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<CustomerDTO[]> => {
-    return updateCustomersStep(input)
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<CustomerDTO[]> => {
+    return new WorkflowResponse(updateCustomersStep(input))
   }
 )

packages/core/core-flows/src/defaults/workflows/create-defaults.ts

@@ -1,11 +1,11 @@
-import { createWorkflow } from "@medusajs/workflows-sdk"
+import { createWorkflow, WorkflowResponse } from "@medusajs/workflows-sdk"
 import { createDefaultSalesChannelStep } from "../../sales-channel"
 import { createDefaultStoreStep } from "../steps/create-default-store"
 
 export const createDefaultsWorkflowID = "create-defaults"
 export const createDefaultsWorkflow = createWorkflow(
   createDefaultsWorkflowID,
-  (input) => {
+  () => {
     const salesChannel = createDefaultSalesChannelStep({
       data: {
         name: "Default Sales Channel",
@@ -18,6 +18,6 @@ export const createDefaultsWorkflow = createWorkflow(
       },
     })
 
-    return store
+    return new WorkflowResponse(store)
   }
 )

packages/core/core-flows/src/definition/cart/workflows/add-to-cart.ts

@@ -4,6 +4,7 @@ import {
 } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -118,6 +119,6 @@ export const addToCartWorkflow = createWorkflow(
       refreshPaymentCollectionForCartStep({ cart_id: input.cart.id })
     )
 
-    return items
+    return new WorkflowResponse(items)
   }
 )

packages/core/core-flows/src/definition/cart/workflows/complete-cart.ts

@@ -2,6 +2,7 @@ import { OrderDTO } from "@medusajs/types"
 import { Modules, OrderEvents } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -20,7 +21,7 @@ import { confirmVariantInventoryWorkflow } from "./confirm-variant-inventory"
 export const completeCartWorkflowId = "complete-cart"
 export const completeCartWorkflow = createWorkflow(
   completeCartWorkflowId,
-  (input: WorkflowData<any>): WorkflowData<OrderDTO> => {
+  (input: WorkflowData<any>): WorkflowResponse<OrderDTO> => {
     const cart = useRemoteQueryStep({
       entry_point: "cart",
       fields: completeCartFields,
@@ -94,6 +95,6 @@ export const completeCartWorkflow = createWorkflow(
 
     emitEventStep({ eventName: OrderEvents.PLACED, data: { id: order.id } })
 
-    return order
+    return new WorkflowResponse(order)
   }
 )

packages/core/core-flows/src/definition/cart/workflows/confirm-variant-inventory.ts

@@ -2,6 +2,7 @@ import { ConfirmVariantInventoryWorkflowInputDTO } from "@medusajs/types"
 import { MedusaError, deepFlatMap } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   transform,
 } from "@medusajs/workflows-sdk"
@@ -24,7 +25,7 @@ export const confirmVariantInventoryWorkflow = createWorkflow(
   confirmVariantInventoryWorkflowId,
   (
     input: WorkflowData<ConfirmVariantInventoryWorkflowInputDTO>
-  ): WorkflowData<Output> => {
+  ): WorkflowResponse<Output> => {
     const confirmInventoryInput = transform({ input }, (data) => {
       const productVariantInventoryItems = new Map<string, any>()
       const stockLocationIds = new Set<string>()
@@ -111,6 +112,6 @@ export const confirmVariantInventoryWorkflow = createWorkflow(
 
     confirmInventoryStep(confirmInventoryInput)
 
-    return confirmInventoryInput
+    return new WorkflowResponse(confirmInventoryInput)
   }
 )

packages/core/core-flows/src/definition/cart/workflows/create-carts.ts

@@ -2,6 +2,7 @@ import { CartDTO, CreateCartWorkflowInputDTO } from "@medusajs/types"
 import { MedusaError } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -28,7 +29,9 @@ import { refreshPaymentCollectionForCartStep } from "./refresh-payment-collectio
 export const createCartWorkflowId = "create-cart"
 export const createCartWorkflow = createWorkflow(
   createCartWorkflowId,
-  (input: WorkflowData<CreateCartWorkflowInputDTO>): WorkflowData<CartDTO> => {
+  (
+    input: WorkflowData<CreateCartWorkflowInputDTO>
+  ): WorkflowResponse<CartDTO> => {
     const variantIds = transform({ input }, (data) => {
       return (data.input.items ?? []).map((i) => i.variant_id)
     })
@@ -153,6 +156,6 @@ export const createCartWorkflow = createWorkflow(
       })
     )
 
-    return cart
+    return new WorkflowResponse(cart)
   }
 )

packages/core/core-flows/src/definition/cart/workflows/list-shipping-options-for-cart.ts

@@ -4,6 +4,7 @@ import {
   createWorkflow,
   transform,
   WorkflowData,
+  WorkflowResponse,
 } from "@medusajs/workflows-sdk"
 import { useRemoteQueryStep } from "../../../common/steps/use-remote-query"
 
@@ -108,6 +109,6 @@ export const listShippingOptionsForCartWorkflow = createWorkflow(
       }
     )
 
-    return shippingOptionsWithPrice
+    return new WorkflowResponse(shippingOptionsWithPrice)
   }
 )

packages/core/core-flows/src/definition/cart/workflows/update-line-item-in-cart.ts

@@ -1,6 +1,7 @@
 import { UpdateLineItemInCartWorkflowInputDTO } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -97,6 +98,6 @@ export const updateLineItemInCartWorkflow = createWorkflow(
 
     const updatedItem = transform({ result }, (data) => data.result?.[0])
 
-    return updatedItem
+    return new WorkflowResponse(updatedItem)
   }
 )

packages/core/core-flows/src/file/workflows/upload-files.ts

@@ -1,5 +1,9 @@
 import { FileDTO } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { uploadFilesStep } from "../steps"
 
 type WorkflowInput = {
@@ -14,7 +18,7 @@ type WorkflowInput = {
 export const uploadFilesWorkflowId = "upload-files"
 export const uploadFilesWorkflow = createWorkflow(
   uploadFilesWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<FileDTO[]> => {
-    return uploadFilesStep(input)
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<FileDTO[]> => {
+    return new WorkflowResponse(uploadFilesStep(input))
   }
 )

packages/core/core-flows/src/fulfillment/workflows/batch-shipping-option-rules.ts

@@ -10,6 +10,7 @@ import {
   parallelize,
   transform,
   WorkflowData,
+  WorkflowResponse,
 } from "@medusajs/workflows-sdk"
 import {
   createShippingOptionRulesStep,
@@ -27,7 +28,7 @@ export const batchShippingOptionRulesWorkflow = createWorkflow(
         UpdateShippingOptionRuleDTO
       >
     >
-  ): WorkflowData<BatchWorkflowOutput<ShippingOptionRuleDTO>> => {
+  ): WorkflowResponse<BatchWorkflowOutput<ShippingOptionRuleDTO>> => {
     const actionInputs = transform({ input }, (data) => {
       const { create, update, delete: del } = data.input
       return {
@@ -43,6 +44,8 @@ export const batchShippingOptionRulesWorkflow = createWorkflow(
       deleteShippingOptionRulesStep(actionInputs.deleteInput)
     )
 
-    return transform({ created, deleted, updated }, (data) => data)
+    return new WorkflowResponse(
+      transform({ created, deleted, updated }, (data) => data)
+    )
   }
 )

packages/core/core-flows/src/fulfillment/workflows/create-fulfillment.ts

@@ -1,5 +1,9 @@
 import { FulfillmentDTO, FulfillmentWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createFulfillmentStep } from "../steps"
 
 export const createFulfillmentWorkflowId = "create-fulfillment-workflow"
@@ -7,9 +11,7 @@ export const createFulfillmentWorkflow = createWorkflow(
   createFulfillmentWorkflowId,
   (
     input: WorkflowData<FulfillmentWorkflow.CreateFulfillmentWorkflowInput>
-  ): WorkflowData<FulfillmentDTO> => {
-    const fulfillment = createFulfillmentStep(input)
-
-    return fulfillment
+  ): WorkflowResponse<FulfillmentDTO> => {
+    return new WorkflowResponse(createFulfillmentStep(input))
   }
 )

packages/core/core-flows/src/fulfillment/workflows/create-return-fulfillment.ts

@@ -1,5 +1,9 @@
 import { FulfillmentDTO, FulfillmentWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createReturnFulfillmentStep } from "../steps"
 
 export const createReturnFulfillmentWorkflowId =
@@ -8,9 +12,7 @@ export const createReturnFulfillmentWorkflow = createWorkflow(
   createReturnFulfillmentWorkflowId,
   (
     input: WorkflowData<FulfillmentWorkflow.CreateFulfillmentWorkflowInput>
-  ): WorkflowData<FulfillmentDTO> => {
-    const fulfillment = createReturnFulfillmentStep(input)
-
-    return fulfillment
+  ): WorkflowResponse<FulfillmentDTO> => {
+    return new WorkflowResponse(createReturnFulfillmentStep(input))
   }
 )

packages/core/core-flows/src/fulfillment/workflows/create-service-zones.ts

@@ -1,5 +1,9 @@
-import { FulfillmentWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import { FulfillmentWorkflow, ServiceZoneDTO } from "@medusajs/types"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createServiceZonesStep } from "../steps"
 
 export const createServiceZonesWorkflowId = "create-service-zones-workflow"
@@ -7,9 +11,7 @@ export const createServiceZonesWorkflow = createWorkflow(
   createServiceZonesWorkflowId,
   (
     input: WorkflowData<FulfillmentWorkflow.CreateServiceZonesWorkflowInput>
-  ): WorkflowData => {
-    const serviceZones = createServiceZonesStep(input.data)
-
-    return serviceZones
+  ): WorkflowResponse<ServiceZoneDTO[]> => {
+    return new WorkflowResponse(createServiceZonesStep(input.data))
   }
 )

packages/core/core-flows/src/fulfillment/workflows/create-shipment.ts

@@ -1,6 +1,7 @@
 import { FulfillmentWorkflow } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   transform,
 } from "@medusajs/workflows-sdk"
@@ -18,8 +19,10 @@ export const createShipmentWorkflow = createWorkflow(
       shipped_at: new Date(),
     }))
 
-    return updateFulfillmentWorkflow.runAsStep({
-      input: update,
-    })
+    return new WorkflowResponse(
+      updateFulfillmentWorkflow.runAsStep({
+        input: update,
+      })
+    )
   }
 )

packages/core/core-flows/src/fulfillment/workflows/create-shipping-options.ts

@@ -3,6 +3,7 @@ import {
   createWorkflow,
   transform,
   WorkflowData,
+  WorkflowResponse,
 } from "@medusajs/workflows-sdk"
 import {
   createShippingOptionsPriceSetsStep,
@@ -19,7 +20,7 @@ export const createShippingOptionsWorkflow = createWorkflow(
     input: WorkflowData<
       FulfillmentWorkflow.CreateShippingOptionsWorkflowInput[]
     >
-  ): WorkflowData<FulfillmentWorkflow.CreateShippingOptionsWorkflowOutput> => {
+  ): WorkflowResponse<FulfillmentWorkflow.CreateShippingOptionsWorkflowOutput> => {
     validateFulfillmentProvidersStep(input)
 
     const data = transform(input, (data) => {
@@ -81,7 +82,6 @@ export const createShippingOptionsWorkflow = createWorkflow(
     )
 
     setShippingOptionsPriceSetsStep(normalizedLinkData)
-
-    return createdShippingOptions
+    return new WorkflowResponse(createdShippingOptions)
   }
 )

packages/core/core-flows/src/fulfillment/workflows/create-shipping-profiles.ts

@@ -1,5 +1,9 @@
 import { FulfillmentWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createShippingProfilesStep } from "../steps"
 
 export const createShippingProfilesWorkflowId =
@@ -8,9 +12,7 @@ export const createShippingProfilesWorkflow = createWorkflow(
   createShippingProfilesWorkflowId,
   (
     input: WorkflowData<FulfillmentWorkflow.CreateShippingProfilesWorkflowInput>
-  ): WorkflowData<FulfillmentWorkflow.CreateShippingProfilesWorkflowOutput> => {
-    const shippingProfiles = createShippingProfilesStep(input.data)
-
-    return shippingProfiles
+  ): WorkflowResponse<FulfillmentWorkflow.CreateShippingProfilesWorkflowOutput> => {
+    return new WorkflowResponse(createShippingProfilesStep(input.data))
   }
 )

packages/core/core-flows/src/fulfillment/workflows/update-fulfillment.ts

@@ -1,11 +1,15 @@
 import { FulfillmentWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateFulfillmentStep } from "../steps"
 
 export const updateFulfillmentWorkflowId = "update-fulfillment-workflow"
 export const updateFulfillmentWorkflow = createWorkflow(
   updateFulfillmentWorkflowId,
   (input: WorkflowData<FulfillmentWorkflow.UpdateFulfillmentWorkflowInput>) => {
-    return updateFulfillmentStep(input)
+    return new WorkflowResponse(updateFulfillmentStep(input))
   }
 )

packages/core/core-flows/src/fulfillment/workflows/update-service-zones.ts

@@ -1,5 +1,9 @@
-import { FulfillmentWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import { FulfillmentWorkflow, ServiceZoneDTO } from "@medusajs/types"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateServiceZonesStep } from "../steps/update-service-zones"
 
 export const updateServiceZonesWorkflowId = "update-service-zones-workflow"
@@ -7,9 +11,7 @@ export const updateServiceZonesWorkflow = createWorkflow(
   updateServiceZonesWorkflowId,
   (
     input: WorkflowData<FulfillmentWorkflow.UpdateServiceZonesWorkflowInput>
-  ): WorkflowData => {
-    const serviceZones = updateServiceZonesStep(input)
-
-    return serviceZones
+  ): WorkflowResponse<ServiceZoneDTO[]> => {
+    return new WorkflowResponse(updateServiceZonesStep(input))
   }
 )

packages/core/core-flows/src/fulfillment/workflows/update-shipping-options.ts

@@ -3,6 +3,7 @@ import {
   createWorkflow,
   transform,
   WorkflowData,
+  WorkflowResponse,
 } from "@medusajs/workflows-sdk"
 import {
   setShippingOptionsPricesStep,
@@ -18,7 +19,7 @@ export const updateShippingOptionsWorkflow = createWorkflow(
     input: WorkflowData<
       FulfillmentWorkflow.UpdateShippingOptionsWorkflowInput[]
     >
-  ): WorkflowData<FulfillmentWorkflow.UpdateShippingOptionsWorkflowOutput> => {
+  ): WorkflowResponse<FulfillmentWorkflow.UpdateShippingOptionsWorkflowOutput> => {
     validateFulfillmentProvidersStep(input)
 
     const data = transform(input, (data) => {
@@ -66,6 +67,6 @@ export const updateShippingOptionsWorkflow = createWorkflow(
       normalizedShippingOptionsPrices.shippingOptionsPrices
     )
 
-    return updatedShippingOptions
+    return new WorkflowResponse(updatedShippingOptions)
   }
 )

packages/core/core-flows/src/fulfillment/workflows/update-shipping-profiles.ts

@@ -1,5 +1,9 @@
 import { FulfillmentWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateShippingProfilesStep } from "../steps/update-shipping-profiles"
 
 export const updateShippingProfilesWorkflowId =
@@ -8,9 +12,7 @@ export const updateShippingProfilesWorkflow = createWorkflow(
   updateShippingProfilesWorkflowId,
   (
     input: WorkflowData<FulfillmentWorkflow.UpdateShippingProfilesWorkflowInput>
-  ): WorkflowData<FulfillmentWorkflow.CreateShippingProfilesWorkflowOutput> => {
-    const shippingProfiles = updateShippingProfilesStep(input)
-
-    return shippingProfiles
+  ): WorkflowResponse<FulfillmentWorkflow.CreateShippingProfilesWorkflowOutput> => {
+    return new WorkflowResponse(updateShippingProfilesStep(input))
   }
 )

packages/core/core-flows/src/inventory/workflows/bulk-create-delete-levels.ts

@@ -1,5 +1,9 @@
 import { InventoryLevelDTO, InventoryTypes } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import {
   createInventoryLevelsStep,
   deleteInventoryLevelsFromItemAndLocationsStep,
@@ -14,9 +18,11 @@ export const bulkCreateDeleteLevelsWorkflowId =
   "bulk-create-delete-levels-workflow"
 export const bulkCreateDeleteLevelsWorkflow = createWorkflow(
   bulkCreateDeleteLevelsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<InventoryLevelDTO[]> => {
+  (
+    input: WorkflowData<WorkflowInput>
+  ): WorkflowResponse<InventoryLevelDTO[]> => {
     deleteInventoryLevelsFromItemAndLocationsStep(input.deletes)
 
-    return createInventoryLevelsStep(input.creates)
+    return new WorkflowResponse(createInventoryLevelsStep(input.creates))
   }
 )

packages/core/core-flows/src/inventory/workflows/create-inventory-items.ts

@@ -1,5 +1,6 @@
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   transform,
 } from "@medusajs/workflows-sdk"
@@ -81,6 +82,6 @@ export const createInventoryItemsWorkflow = createWorkflow(
 
     createInventoryLevelsWorkflow.runAsStep(inventoryLevelsInput)
 
-    return items
+    return new WorkflowResponse(items)
   }
 )

packages/core/core-flows/src/inventory/workflows/create-inventory-levels.ts

@@ -1,5 +1,9 @@
 import { InventoryLevelDTO, InventoryTypes } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import {
   createInventoryLevelsStep,
   validateInventoryLocationsStep,
@@ -12,9 +16,13 @@ export const createInventoryLevelsWorkflowId =
   "create-inventory-levels-workflow"
 export const createInventoryLevelsWorkflow = createWorkflow(
   createInventoryLevelsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<InventoryLevelDTO[]> => {
+  (
+    input: WorkflowData<WorkflowInput>
+  ): WorkflowResponse<InventoryLevelDTO[]> => {
     validateInventoryLocationsStep(input.inventory_levels)
 
-    return createInventoryLevelsStep(input.inventory_levels)
+    return new WorkflowResponse(
+      createInventoryLevelsStep(input.inventory_levels)
+    )
   }
 )

packages/core/core-flows/src/inventory/workflows/delete-inventory-items.ts

@@ -1,4 +1,8 @@
-import { createWorkflow, WorkflowData } from "@medusajs/workflows-sdk"
+import {
+  createWorkflow,
+  WorkflowData,
+  WorkflowResponse,
+} from "@medusajs/workflows-sdk"
 
 import { deleteInventoryItemStep } from "../steps"
 import { removeRemoteLinkStep } from "../../common/steps/remove-remote-links"
@@ -7,12 +11,11 @@ import { Modules } from "@medusajs/utils"
 export const deleteInventoryItemWorkflowId = "delete-inventory-item-workflow"
 export const deleteInventoryItemWorkflow = createWorkflow(
   deleteInventoryItemWorkflowId,
-  (input: WorkflowData<string[]>): WorkflowData<string[]> => {
+  (input: WorkflowData<string[]>): WorkflowResponse<string[]> => {
     deleteInventoryItemStep(input)
-
     removeRemoteLinkStep({
       [Modules.INVENTORY]: { inventory_item_id: input },
     })
-    return input
+    return new WorkflowResponse(input)
   }
 )

packages/core/core-flows/src/inventory/workflows/delete-inventory-levels.ts

@@ -1,4 +1,8 @@
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 
 import { deleteInventoryLevelsStep } from "../steps"
 
@@ -9,7 +13,7 @@ export const deleteInventoryLevelsWorkflowId =
   "delete-inventory-levels-workflow"
 export const deleteInventoryLevelsWorkflow = createWorkflow(
   deleteInventoryLevelsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<string[]> => {
-    return deleteInventoryLevelsStep(input.ids)
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<string[]> => {
+    return new WorkflowResponse(deleteInventoryLevelsStep(input.ids))
   }
 )

packages/core/core-flows/src/inventory/workflows/update-inventory-items.ts

@@ -1,4 +1,8 @@
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 
 import { InventoryTypes } from "@medusajs/types"
 import { updateInventoryItemsStep } from "../steps"
@@ -11,7 +15,7 @@ export const updateInventoryItemsWorkflow = createWorkflow(
   updateInventoryItemsWorkflowId,
   (
     input: WorkflowData<WorkflowInput>
-  ): WorkflowData<InventoryTypes.InventoryItemDTO[]> => {
-    return updateInventoryItemsStep(input.updates)
+  ): WorkflowResponse<InventoryTypes.InventoryItemDTO[]> => {
+    return new WorkflowResponse(updateInventoryItemsStep(input.updates))
   }
 )

packages/core/core-flows/src/inventory/workflows/update-inventory-levels.ts

@@ -1,5 +1,9 @@
 import { InventoryLevelDTO, InventoryTypes } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 
 import { updateInventoryLevelsStep } from "../steps/update-inventory-levels"
 
@@ -10,7 +14,9 @@ export const updateInventoryLevelsWorkflowId =
   "update-inventory-levels-workflow"
 export const updateInventoryLevelsWorkflow = createWorkflow(
   updateInventoryLevelsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<InventoryLevelDTO[]> => {
-    return updateInventoryLevelsStep(input.updates)
+  (
+    input: WorkflowData<WorkflowInput>
+  ): WorkflowResponse<InventoryLevelDTO[]> => {
+    return new WorkflowResponse(updateInventoryLevelsStep(input.updates))
   }
 )

packages/core/core-flows/src/invite/workflows/accept-invite.ts

@@ -1,6 +1,7 @@
 import { InviteWorkflow, UserDTO } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   transform,
 } from "@medusajs/workflows-sdk"
@@ -14,7 +15,7 @@ export const acceptInviteWorkflow = createWorkflow(
   acceptInviteWorkflowId,
   (
     input: WorkflowData<InviteWorkflow.AcceptInviteWorkflowInputDTO>
-  ): WorkflowData<UserDTO[]> => {
+  ): WorkflowResponse<UserDTO[]> => {
     const invite = validateTokenStep(input.invite_token)
 
     const createUserInput = transform(
@@ -44,6 +45,6 @@ export const acceptInviteWorkflow = createWorkflow(
     setAuthAppMetadataStep(authUserInput)
     deleteInvitesStep([invite.id])
 
-    return users
+    return new WorkflowResponse(users)
   }
 )

packages/core/core-flows/src/invite/workflows/create-invites.ts

@@ -1,4 +1,8 @@
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createInviteStep } from "../steps"
 import { InviteDTO, InviteWorkflow } from "@medusajs/types"
 
@@ -7,7 +11,7 @@ export const createInvitesWorkflow = createWorkflow(
   createInvitesWorkflowId,
   (
     input: WorkflowData<InviteWorkflow.CreateInvitesWorkflowInputDTO>
-  ): WorkflowData<InviteDTO[]> => {
-    return createInviteStep(input.invites)
+  ): WorkflowResponse<InviteDTO[]> => {
+    return new WorkflowResponse(createInviteStep(input.invites))
   }
 )

packages/core/core-flows/src/invite/workflows/refresh-invite-tokens.ts

@@ -1,5 +1,9 @@
 import { InviteDTO, InviteWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 
 import { refreshInviteTokensStep } from "../steps/refresh-invite-tokens"
 
@@ -8,7 +12,7 @@ export const refreshInviteTokensWorkflow = createWorkflow(
   refreshInviteTokensWorkflowId,
   (
     input: WorkflowData<InviteWorkflow.ResendInvitesWorkflowInputDTO>
-  ): WorkflowData<InviteDTO[]> => {
-    return refreshInviteTokensStep(input.invite_ids)
+  ): WorkflowResponse<InviteDTO[]> => {
+    return new WorkflowResponse(refreshInviteTokensStep(input.invite_ids))
   }
 )

packages/core/core-flows/src/order/workflows/add-line-items.ts

@@ -2,6 +2,7 @@ import { OrderLineItemDTO, OrderWorkflow } from "@medusajs/types"
 import { MathBN, MedusaError } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -59,7 +60,7 @@ export const addOrderLineItemsWorkflow = createWorkflow(
   addOrderLineItemsWorkflowId,
   (
     input: WorkflowData<OrderWorkflow.OrderAddLineItemWorkflowInput>
-  ): WorkflowData<OrderLineItemDTO[]> => {
+  ): WorkflowResponse<OrderLineItemDTO[]> => {
     const order = useRemoteQueryStep({
       entry_point: "orders",
       fields: [
@@ -141,8 +142,10 @@ export const addOrderLineItemsWorkflow = createWorkflow(
       prepareLineItems
     )
 
-    return createOrderLineItemsStep({
-      items: lineItems,
-    })
+    return new WorkflowResponse(
+      createOrderLineItemsStep({
+        items: lineItems,
+      })
+    )
   }
 )

packages/core/core-flows/src/order/workflows/archive-orders.ts

@@ -1,5 +1,9 @@
 import { OrderDTO } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { archiveOrdersStep } from "../steps"
 
 type ArchiveOrdersStepInput = {
@@ -9,9 +13,9 @@ type ArchiveOrdersStepInput = {
 export const archiveOrderWorkflowId = "archive-order-workflow"
 export const archiveOrderWorkflow = createWorkflow(
   archiveOrderWorkflowId,
-  (input: WorkflowData<ArchiveOrdersStepInput>): WorkflowData<OrderDTO[]> => {
-    const archivedOrders = archiveOrdersStep(input)
-
-    return archivedOrders
+  (
+    input: WorkflowData<ArchiveOrdersStepInput>
+  ): WorkflowResponse<OrderDTO[]> => {
+    return new WorkflowResponse(archiveOrdersStep(input))
   }
 )

packages/core/core-flows/src/order/workflows/claim/begin-order-claim.ts

@@ -1,6 +1,7 @@
 import { OrderChangeDTO, OrderDTO, OrderWorkflow } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -22,7 +23,7 @@ export const beginClaimOrderWorkflow = createWorkflow(
   beginClaimOrderWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.BeginOrderClaimWorkflowInput>
-  ): WorkflowData<OrderChangeDTO> {
+  ): WorkflowResponse<OrderChangeDTO> {
     const order: OrderDTO = useRemoteQueryStep({
       entry_point: "orders",
       fields: ["id", "status"],
@@ -54,6 +55,6 @@ export const beginClaimOrderWorkflow = createWorkflow(
         }
       }
     )
-    return createOrderChangeStep(orderChangeInput)
+    return new WorkflowResponse(createOrderChangeStep(orderChangeInput))
   }
 )

packages/core/core-flows/src/order/workflows/claim/claim-add-new-item.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -42,7 +43,7 @@ export const orderClaimAddNewItemWorkflow = createWorkflow(
   orderClaimAddNewItemWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.OrderClaimAddNewItemWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "order_id", "canceled_at"],
@@ -109,6 +110,6 @@ export const orderClaimAddNewItemWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(orderClaim.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderClaim.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/claim-item.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -41,7 +42,7 @@ export const orderClaimItemWorkflow = createWorkflow(
   orderClaimItemWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.OrderClaimItemWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "order_id", "canceled_at"],
@@ -100,6 +101,6 @@ export const orderClaimItemWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(orderClaim.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderClaim.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/claim-request-item-return.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -52,7 +53,7 @@ export const orderClaimRequestItemReturnWorkflow = createWorkflow(
   orderClaimRequestItemReturnWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.OrderClaimRequestItemReturnWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "order_id", "return_id", "canceled_at"],
@@ -165,6 +166,6 @@ export const orderClaimRequestItemReturnWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(orderClaim.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderClaim.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/confirm-claim-request.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, Modules, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   parallelize,
@@ -167,7 +168,7 @@ function extractShippingOption({ orderPreview, orderClaim, returnId }) {
 export const confirmClaimRequestWorkflowId = "confirm-claim-request"
 export const confirmClaimRequestWorkflow = createWorkflow(
   confirmClaimRequestWorkflowId,
-  function (input: WorkflowInput): WorkflowData<OrderDTO> {
+  function (input: WorkflowInput): WorkflowResponse<OrderDTO> {
     const orderClaim: OrderClaimDTO = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -356,6 +357,6 @@ export const confirmClaimRequestWorkflow = createWorkflow(
       })
     })
 
-    return orderPreview
+    return new WorkflowResponse(orderPreview)
   }
 )

packages/core/core-flows/src/order/workflows/claim/create-claim-shipping-method.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -46,7 +47,7 @@ export const createClaimShippingMethodWorkflow = createWorkflow(
     claim_id?: string
     shipping_option_id: string
     custom_price?: BigNumberInput
-  }): WorkflowData<OrderDTO> {
+  }): WorkflowResponse<OrderDTO> {
     const orderClaim: OrderClaimDTO = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -164,6 +165,6 @@ export const createClaimShippingMethodWorkflow = createWorkflow(
 
     createOrderChangeActionsStep([orderChangeActionInput])
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/remove-claim-item-action.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
 } from "@medusajs/workflows-sdk"
@@ -57,7 +58,7 @@ export const removeItemClaimActionWorkflow = createWorkflow(
   removeItemClaimActionWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.DeleteOrderClaimItemActionWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim: OrderClaimDTO = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -91,6 +92,6 @@ export const removeItemClaimActionWorkflow = createWorkflow(
 
     deleteOrderChangeActionsStep({ ids: [input.action_id] })
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/remove-claim-shipping-method.ts

@@ -2,11 +2,13 @@ import {
   OrderChangeActionDTO,
   OrderChangeDTO,
   OrderClaimDTO,
+  OrderDTO,
   OrderWorkflow,
 } from "@medusajs/types"
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   parallelize,
@@ -57,7 +59,7 @@ export const removeClaimShippingMethodWorkflow = createWorkflow(
   removeClaimShippingMethodWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.DeleteClaimShippingMethodWorkflowInput>
-  ): WorkflowData {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim: OrderClaimDTO = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -100,6 +102,6 @@ export const removeClaimShippingMethodWorkflow = createWorkflow(
       deleteOrderShippingMethods({ ids: [dataToRemove.shippingMethodId] })
     )
 
-    return previewOrderChangeStep(orderClaim.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderClaim.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/update-claim-add-item.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -61,7 +62,7 @@ export const updateClaimAddItemWorkflow = createWorkflow(
   updateClaimAddItemWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.UpdateClaimAddNewItemWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim: OrderClaimDTO = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -113,6 +114,6 @@ export const updateClaimAddItemWorkflow = createWorkflow(
 
     updateOrderChangeActionsStep([updateData])
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/update-claim-item.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -61,7 +62,7 @@ export const updateClaimItemWorkflow = createWorkflow(
   updateClaimItemWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.UpdateClaimItemWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim: OrderClaimDTO = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -113,6 +114,6 @@ export const updateClaimItemWorkflow = createWorkflow(
 
     updateOrderChangeActionsStep([updateData])
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/claim/update-claim-shipping-method.ts

@@ -2,11 +2,13 @@ import {
   OrderChangeActionDTO,
   OrderChangeDTO,
   OrderClaimDTO,
+  OrderDTO,
   OrderWorkflow,
 } from "@medusajs/types"
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   parallelize,
@@ -59,7 +61,7 @@ export const updateClaimShippingMethodWorkflow = createWorkflow(
   updateClaimShippingMethodWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.UpdateClaimShippingMethodWorkflowInput>
-  ): WorkflowData {
+  ): WorkflowResponse<OrderDTO> {
     const orderClaim: OrderClaimDTO = useRemoteQueryStep({
       entry_point: "order_claim",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -115,6 +117,6 @@ export const updateClaimShippingMethodWorkflow = createWorkflow(
       updateOrderShippingMethodsStep([updateData.shippingMethod!])
     )
 
-    return previewOrderChangeStep(orderClaim.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderClaim.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/complete-orders.ts

@@ -1,5 +1,9 @@
 import { OrderDTO } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { completeOrdersStep } from "../steps"
 
 type CompleteOrdersStepInput = {
@@ -9,12 +13,9 @@ type CompleteOrdersStepInput = {
 export const completeOrderWorkflowId = "complete-order-workflow"
 export const completeOrderWorkflow = createWorkflow(
   completeOrderWorkflowId,
-  (input: WorkflowData<CompleteOrdersStepInput>): WorkflowData<OrderDTO[]> => {
-    const completedOrders = completeOrdersStep(input)
-
-    // TODO: Emit event "OrderModuleService.Order.COMPLETED"
-    // id + no_notification
-
-    return completedOrders
+  (
+    input: WorkflowData<CompleteOrdersStepInput>
+  ): WorkflowResponse<OrderDTO[]> => {
+    return new WorkflowResponse(completeOrdersStep(input))
   }
 )

packages/core/core-flows/src/order/workflows/create-fulfillment.ts

@@ -9,6 +9,7 @@ import {
 import { MathBN, MedusaError, Modules } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   parallelize,
@@ -193,7 +194,7 @@ export const createOrderFulfillmentWorkflow = createWorkflow(
   createOrderFulfillmentWorkflowId,
   (
     input: WorkflowData<OrderWorkflow.CreateOrderFulfillmentWorkflowInput>
-  ): WorkflowData<FulfillmentDTO> => {
+  ): WorkflowResponse<FulfillmentDTO> => {
     const order: OrderDTO = useRemoteQueryStep({
       entry_point: "orders",
       fields: [
@@ -296,6 +297,6 @@ export const createOrderFulfillmentWorkflow = createWorkflow(
     )
 
     // trigger event OrderModuleService.Events.FULFILLMENT_CREATED
-    return fulfillment
+    return new WorkflowResponse(fulfillment)
   }
 )

packages/core/core-flows/src/order/workflows/create-order-change-actions.ts

@@ -2,7 +2,11 @@ import {
   CreateOrderChangeActionDTO,
   OrderChangeActionDTO,
 } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createOrderChangeActionsStep } from "../steps"
 
 export const createOrderChangeActionsWorkflowId = "create-order-change-actions"
@@ -10,7 +14,7 @@ export const createOrderChangeActionsWorkflow = createWorkflow(
   createOrderChangeActionsWorkflowId,
   (
     input: WorkflowData<CreateOrderChangeActionDTO[]>
-  ): WorkflowData<OrderChangeActionDTO[]> => {
-    return createOrderChangeActionsStep(input)
+  ): WorkflowResponse<OrderChangeActionDTO[]> => {
+    return new WorkflowResponse(createOrderChangeActionsStep(input))
   }
 )

packages/core/core-flows/src/order/workflows/create-order-change.ts

@@ -1,11 +1,17 @@
 import { CreateOrderChangeDTO, OrderChangeDTO } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createOrderChangeStep } from "../steps"
 
 export const createOrderChangeWorkflowId = "create-order-change"
 export const createOrderChangeWorkflow = createWorkflow(
   createOrderChangeWorkflowId,
-  (input: WorkflowData<CreateOrderChangeDTO>): WorkflowData<OrderChangeDTO> => {
-    return createOrderChangeStep(input)
+  (
+    input: WorkflowData<CreateOrderChangeDTO>
+  ): WorkflowResponse<OrderChangeDTO> => {
+    return new WorkflowResponse(createOrderChangeStep(input))
   }
 )

packages/core/core-flows/src/order/workflows/create-orders.ts

@@ -2,6 +2,7 @@ import { CreateOrderDTO, OrderDTO } from "@medusajs/types"
 import { MathBN, MedusaError, isPresent } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -84,7 +85,7 @@ function getOrderInput(data) {
 export const createOrdersWorkflowId = "create-orders"
 export const createOrdersWorkflow = createWorkflow(
   createOrdersWorkflowId,
-  (input: WorkflowData<CreateOrderDTO>): WorkflowData<OrderDTO> => {
+  (input: WorkflowData<CreateOrderDTO>): WorkflowResponse<OrderDTO> => {
     const variantIds = transform({ input }, (data) => {
       return (data.input.items ?? [])
         .map((item) => item.variant_id)
@@ -176,6 +177,6 @@ export const createOrdersWorkflow = createWorkflow(
 
     updateOrderTaxLinesStep({ order_id: order.id })
 
-    return order
+    return new WorkflowResponse(order)
   }
 )

packages/core/core-flows/src/order/workflows/exchange/begin-order-exchange.ts

@@ -1,6 +1,7 @@
 import { OrderChangeDTO, OrderDTO, OrderWorkflow } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -22,7 +23,7 @@ export const beginExchangeOrderWorkflow = createWorkflow(
   beginExchangeOrderWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.BeginOrderExchangeWorkflowInput>
-  ): WorkflowData<OrderChangeDTO> {
+  ): WorkflowResponse<OrderChangeDTO> {
     const order: OrderDTO = useRemoteQueryStep({
       entry_point: "orders",
       fields: ["id", "status"],
@@ -53,6 +54,6 @@ export const beginExchangeOrderWorkflow = createWorkflow(
         }
       }
     )
-    return createOrderChangeStep(orderChangeInput)
+    return new WorkflowResponse(createOrderChangeStep(orderChangeInput))
   }
 )

packages/core/core-flows/src/order/workflows/exchange/create-exchange-return-shipping-method.ts

@@ -1,5 +1,6 @@
 import {
   BigNumberInput,
+  OrderChangeActionDTO,
   OrderChangeDTO,
   OrderDTO,
   OrderExchangeDTO,
@@ -7,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -41,7 +43,7 @@ export const createExchangeReturnShippingMethodWorkflow = createWorkflow(
     exchangeId: string
     shippingOptionId: string
     customShippingPrice?: BigNumberInput
-  }): WorkflowData {
+  }): WorkflowResponse<OrderChangeActionDTO[]> {
     const orderExchange: OrderExchangeDTO = useRemoteQueryStep({
       entry_point: "order_exchange",
       fields: ["id", "status", "order_id", "return_id"],
@@ -148,6 +150,7 @@ export const createExchangeReturnShippingMethodWorkflow = createWorkflow(
       }
     )
 
-    return createOrderChangeActionsStep([orderChangeActionInput])
+    const response = createOrderChangeActionsStep([orderChangeActionInput])
+    return new WorkflowResponse(response)
   }
 )

packages/core/core-flows/src/order/workflows/exchange/exchange-add-new-item.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -43,7 +44,7 @@ export const orderExchangeAddNewItemWorkflow = createWorkflow(
   orderExchangeAddNewItemWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.OrderExchangeAddNewItemWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderExchange = useRemoteQueryStep({
       entry_point: "order_exchange",
       fields: ["id", "order_id", "canceled_at"],
@@ -110,6 +111,6 @@ export const orderExchangeAddNewItemWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(orderExchange.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderExchange.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/exchange/exchange-request-item-return.ts

@@ -12,6 +12,7 @@ import {
   transform,
   when,
   WorkflowData,
+  WorkflowResponse,
 } from "@medusajs/workflows-sdk"
 import { useRemoteQueryStep } from "../../../common"
 import { createOrderChangeActionsStep } from "../../steps/create-order-change-actions"
@@ -54,7 +55,7 @@ export const orderExchangeRequestItemReturnWorkflow = createWorkflow(
   orderExchangeRequestItemReturnWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.OrderExchangeRequestItemReturnWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderExchange = useRemoteQueryStep({
       entry_point: "order_exchange",
       fields: ["id", "order_id", "return_id", "canceled_at"],
@@ -163,6 +164,6 @@ export const orderExchangeRequestItemReturnWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(orderExchange.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderExchange.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/get-order-detail.ts

@@ -2,6 +2,7 @@ import { OrderDTO, OrderDetailDTO } from "@medusajs/types"
 import { deduplicate } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   transform,
 } from "@medusajs/workflows-sdk"
@@ -19,7 +20,7 @@ export const getOrderDetailWorkflow = createWorkflow(
       fields: string[]
       order_id: string
     }>
-  ): WorkflowData<OrderDetailDTO> => {
+  ): WorkflowResponse<OrderDetailDTO> => {
     const fields = transform(input, ({ fields }) => {
       return deduplicate([
         ...fields,
@@ -51,6 +52,6 @@ export const getOrderDetailWorkflow = createWorkflow(
       return order_
     })
 
-    return aggregatedOrder
+    return new WorkflowResponse(aggregatedOrder)
   }
 )

packages/core/core-flows/src/order/workflows/get-orders-list.ts

@@ -2,6 +2,7 @@ import { OrderDTO, OrderDetailDTO } from "@medusajs/types"
 import { deduplicate } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   transform,
 } from "@medusajs/workflows-sdk"
@@ -26,7 +27,7 @@ export const getOrdersListWorkflow = createWorkflow(
       fields: string[]
       variables?: Record<string, any>
     }>
-  ): WorkflowData<OrderOutput> => {
+  ): WorkflowResponse<OrderOutput> => {
     const fields = transform(input, ({ fields }) => {
       return deduplicate([
         ...fields,
@@ -89,6 +90,6 @@ export const getOrdersListWorkflow = createWorkflow(
       }
     )
 
-    return aggregatedOrders
+    return new WorkflowResponse(aggregatedOrders)
   }
 )

packages/core/core-flows/src/order/workflows/return/begin-receive-return.ts

@@ -6,6 +6,7 @@ import {
 } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -36,7 +37,7 @@ export const beginReceiveReturnWorkflow = createWorkflow(
   beginReceiveReturnWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.BeginReceiveOrderReturnWorkflowInput>
-  ): WorkflowData<OrderChangeDTO> {
+  ): WorkflowResponse<OrderChangeDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -68,6 +69,6 @@ export const beginReceiveReturnWorkflow = createWorkflow(
         }
       }
     )
-    return createOrderChangeStep(orderChangeInput)
+    return new WorkflowResponse(createOrderChangeStep(orderChangeInput))
   }
 )

packages/core/core-flows/src/order/workflows/return/begin-return.ts

@@ -1,6 +1,7 @@
 import { OrderChangeDTO, OrderDTO, OrderWorkflow } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -21,7 +22,7 @@ export const beginReturnOrderWorkflow = createWorkflow(
   beginReturnOrderWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.BeginOrderReturnWorkflowInput>
-  ): WorkflowData<OrderChangeDTO> {
+  ): WorkflowResponse<OrderChangeDTO> {
     const order: OrderDTO = useRemoteQueryStep({
       entry_point: "orders",
       fields: ["id", "status"],
@@ -53,6 +54,6 @@ export const beginReturnOrderWorkflow = createWorkflow(
         }
       }
     )
-    return createOrderChangeStep(orderChangeInput)
+    return new WorkflowResponse(createOrderChangeStep(orderChangeInput))
   }
 )

packages/core/core-flows/src/order/workflows/return/confirm-receive-return-request.ts

@@ -15,6 +15,7 @@ import {
 } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   parallelize,
@@ -128,7 +129,7 @@ function prepareInventoryUpdate({ orderReturn, returnedQuantityMap }) {
 export const confirmReturnReceiveWorkflowId = "confirm-return-receive"
 export const confirmReturnReceiveWorkflow = createWorkflow(
   confirmReturnReceiveWorkflowId,
-  function (input: WorkflowInput): WorkflowData<OrderDTO> {
+  function (input: WorkflowInput): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: [
@@ -276,6 +277,6 @@ export const confirmReturnReceiveWorkflow = createWorkflow(
       adjustInventoryLevelsStep(inventoryAdjustment)
     )
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/confirm-return-request.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, Modules, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -124,7 +125,7 @@ function extractReturnShippingOptionId({ orderPreview, orderReturn }) {
 export const confirmReturnRequestWorkflowId = "confirm-return-request"
 export const confirmReturnRequestWorkflow = createWorkflow(
   confirmReturnRequestWorkflowId,
-  function (input: WorkflowInput): WorkflowData<OrderDTO> {
+  function (input: WorkflowInput): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "location_id", "canceled_at"],
@@ -234,6 +235,6 @@ export const confirmReturnRequestWorkflow = createWorkflow(
 
     confirmOrderChanges({ changes: [orderChange], orderId: order.id })
 
-    return orderPreview
+    return new WorkflowResponse(orderPreview)
   }
 )

packages/core/core-flows/src/order/workflows/return/create-return-shipping-method.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -47,7 +48,7 @@ export const createReturnShippingMethodWorkflow = createWorkflow(
     exchange_id?: string
     shipping_option_id: string
     custom_price?: BigNumberInput
-  }): WorkflowData<OrderDTO> {
+  }): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -167,6 +168,6 @@ export const createReturnShippingMethodWorkflow = createWorkflow(
 
     createOrderChangeActionsStep([orderChangeActionInput])
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/dismiss-item-return-request.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -48,7 +49,7 @@ export const dismissItemReturnRequestWorkflow = createWorkflow(
   dismissItemReturnRequestWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.ReceiveOrderReturnItemsWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at", "items.*"],
@@ -102,6 +103,6 @@ export const dismissItemReturnRequestWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/receive-complete-return.ts

@@ -1,6 +1,7 @@
 import { OrderWorkflow } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
 } from "@medusajs/workflows-sdk"
@@ -35,7 +36,7 @@ export const receiveAndCompleteReturnOrderWorkflow = createWorkflow(
   receiveAndCompleteReturnOrderWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.ReceiveCompleteOrderReturnWorkflowInput>
-  ): WorkflowData<ReturnDTO> {
+  ): WorkflowResponse<ReturnDTO | undefined> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "returns",
       fields: ["id", "canceled_at", "items.*"],
@@ -46,8 +47,6 @@ export const receiveAndCompleteReturnOrderWorkflow = createWorkflow(
 
     validationStep({ orderReturn, input })
 
-    const received = receiveReturnStep(input)
-
-    return received
+    return new WorkflowResponse(receiveReturnStep(input))
   }
 )

packages/core/core-flows/src/order/workflows/return/receive-item-return-request.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -48,7 +49,7 @@ export const receiveItemReturnRequestWorkflow = createWorkflow(
   receiveItemReturnRequestWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.ReceiveOrderReturnItemsWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at", "items.*"],
@@ -102,6 +103,6 @@ export const receiveItemReturnRequestWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/remove-item-receive-return-action.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
 } from "@medusajs/workflows-sdk"
@@ -65,7 +66,7 @@ export const removeItemReceiveReturnActionWorkflow = createWorkflow(
   removeItemReceiveReturnActionWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.DeleteRequestItemReceiveReturnWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -99,6 +100,6 @@ export const removeItemReceiveReturnActionWorkflow = createWorkflow(
 
     deleteOrderChangeActionsStep({ ids: [input.action_id] })
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/remove-item-return-action.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
 } from "@medusajs/workflows-sdk"
@@ -59,7 +60,7 @@ export const removeItemReturnActionWorkflow = createWorkflow(
   removeItemReturnActionWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.DeleteRequestItemReturnWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -93,6 +94,6 @@ export const removeItemReturnActionWorkflow = createWorkflow(
 
     deleteOrderChangeActionsStep({ ids: [input.action_id] })
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/remove-return-shipping-method.ts

@@ -1,12 +1,14 @@
 import {
   OrderChangeActionDTO,
   OrderChangeDTO,
+  OrderDTO,
   OrderWorkflow,
   ReturnDTO,
 } from "@medusajs/types"
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   parallelize,
@@ -57,7 +59,7 @@ export const removeReturnShippingMethodWorkflow = createWorkflow(
   removeReturnShippingMethodWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.DeleteReturnShippingMethodWorkflowInput>
-  ): WorkflowData {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -100,6 +102,6 @@ export const removeReturnShippingMethodWorkflow = createWorkflow(
       deleteOrderShippingMethods({ ids: [dataToRemove.shippingMethodId] })
     )
 
-    return previewOrderChangeStep(orderReturn.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderReturn.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/return/request-item-return.ts

@@ -7,6 +7,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -54,7 +55,7 @@ export const requestItemReturnWorkflow = createWorkflow(
   requestItemReturnWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.RequestItemReturnWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -110,6 +111,6 @@ export const requestItemReturnWorkflow = createWorkflow(
 
     createOrderChangeActionsStep(orderChangeActionInput)
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/update-receive-item-return-request.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -69,7 +70,7 @@ export const updateReceiveItemReturnRequestWorkflow = createWorkflow(
   updateReceiveItemReturnRequestWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.UpdateReceiveItemReturnRequestWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -121,6 +122,6 @@ export const updateReceiveItemReturnRequestWorkflow = createWorkflow(
 
     updateOrderChangeActionsStep([updateData])
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/update-request-item-return.ts

@@ -8,6 +8,7 @@ import {
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -74,7 +75,7 @@ export const updateRequestItemReturnWorkflow = createWorkflow(
   updateRequestItemReturnWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.UpdateRequestItemReturnWorkflowInput>
-  ): WorkflowData<OrderDTO> {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -127,6 +128,6 @@ export const updateRequestItemReturnWorkflow = createWorkflow(
 
     updateOrderChangeActionsStep([updateData])
 
-    return previewOrderChangeStep(order.id)
+    return new WorkflowResponse(previewOrderChangeStep(order.id))
   }
 )

packages/core/core-flows/src/order/workflows/return/update-return-shipping-method.ts

@@ -1,12 +1,14 @@
 import {
   OrderChangeActionDTO,
   OrderChangeDTO,
+  OrderDTO,
   OrderWorkflow,
   ReturnDTO,
 } from "@medusajs/types"
 import { ChangeActionType, OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   parallelize,
@@ -59,7 +61,7 @@ export const updateReturnShippingMethodWorkflow = createWorkflow(
   updateReturnShippingMethodWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.UpdateReturnShippingMethodWorkflowInput>
-  ): WorkflowData {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -115,6 +117,6 @@ export const updateReturnShippingMethodWorkflow = createWorkflow(
       updateOrderShippingMethodsStep([updateData.shippingMethod!])
     )
 
-    return previewOrderChangeStep(orderReturn.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderReturn.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/return/update-return.ts

@@ -1,7 +1,13 @@
-import { OrderChangeDTO, OrderWorkflow, ReturnDTO } from "@medusajs/types"
+import {
+  OrderChangeDTO,
+  OrderDTO,
+  OrderWorkflow,
+  ReturnDTO,
+} from "@medusajs/types"
 import { OrderChangeStatus } from "@medusajs/utils"
 import {
   WorkflowData,
+  WorkflowResponse,
   createStep,
   createWorkflow,
   transform,
@@ -33,7 +39,7 @@ export const updateReturnWorkflow = createWorkflow(
   updateReturnWorkflowId,
   function (
     input: WorkflowData<OrderWorkflow.UpdateReturnWorkflowInput>
-  ): WorkflowData {
+  ): WorkflowResponse<OrderDTO> {
     const orderReturn: ReturnDTO = useRemoteQueryStep({
       entry_point: "return",
       fields: ["id", "status", "order_id", "canceled_at"],
@@ -68,6 +74,6 @@ export const updateReturnWorkflow = createWorkflow(
 
     updateReturnsStep([updateData])
 
-    return previewOrderChangeStep(orderReturn.order_id)
+    return new WorkflowResponse(previewOrderChangeStep(orderReturn.order_id))
   }
 )

packages/core/core-flows/src/order/workflows/update-order-change-actions.ts

@@ -2,7 +2,11 @@ import {
   OrderChangeActionDTO,
   UpdateOrderChangeActionDTO,
 } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateOrderChangeActionsStep } from "../steps"
 
 export const updateOrderChangeActionsWorkflowId = "update-order-change-actions"
@@ -10,7 +14,7 @@ export const updateOrderChangeActionsWorkflow = createWorkflow(
   updateOrderChangeActionsWorkflowId,
   (
     input: WorkflowData<UpdateOrderChangeActionDTO[]>
-  ): WorkflowData<OrderChangeActionDTO[]> => {
-    return updateOrderChangeActionsStep(input)
+  ): WorkflowResponse<OrderChangeActionDTO[]> => {
+    return new WorkflowResponse(updateOrderChangeActionsStep(input))
   }
 )

packages/core/core-flows/src/payment-collection/workflows/create-payment-session.ts

@@ -1,6 +1,7 @@
 import { PaymentProviderContext, PaymentSessionDTO } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -19,7 +20,7 @@ interface WorkflowInput {
 export const createPaymentSessionsWorkflowId = "create-payment-sessions"
 export const createPaymentSessionsWorkflow = createWorkflow(
   createPaymentSessionsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<PaymentSessionDTO> => {
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<PaymentSessionDTO> => {
     const paymentCollection = useRemoteQueryStep({
       entry_point: "payment_collection",
       fields: ["id", "amount", "currency_code", "payment_sessions.*"],
@@ -62,6 +63,6 @@ export const createPaymentSessionsWorkflow = createWorkflow(
       })
     )
 
-    return created
+    return new WorkflowResponse(created)
   }
 )

packages/core/core-flows/src/payment-collection/workflows/delete-payment-sessions.ts

@@ -1,4 +1,8 @@
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import {
   deletePaymentSessionsStep,
   validateDeletedPaymentSessionsStep,
@@ -19,6 +23,6 @@ export const deletePaymentSessionsWorkflow = createWorkflow(
       idsDeleted,
     })
 
-    return idsDeleted
+    return new WorkflowResponse(idsDeleted)
   }
 )

packages/core/core-flows/src/payment/workflows/capture-payment.ts

@@ -1,6 +1,10 @@
 import { BigNumberInput, PaymentDTO } from "@medusajs/types"
 import { PaymentEvents } from "@medusajs/utils"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { emitEventStep } from "../../common"
 import { capturePaymentStep } from "../steps/capture-payment"
 
@@ -13,7 +17,7 @@ export const capturePaymentWorkflow = createWorkflow(
       captured_by?: string
       amount?: BigNumberInput
     }>
-  ): WorkflowData<PaymentDTO> => {
+  ): WorkflowResponse<PaymentDTO> => {
     const payment = capturePaymentStep(input)
 
     emitEventStep({
@@ -21,6 +25,6 @@ export const capturePaymentWorkflow = createWorkflow(
       data: { id: payment.id },
     })
 
-    return payment
+    return new WorkflowResponse(payment)
   }
 )

packages/core/core-flows/src/payment/workflows/refund-payment.ts

@@ -1,6 +1,10 @@
 import { BigNumberInput } from "@medusajs/types"
 import { PaymentEvents } from "@medusajs/utils"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { emitEventStep } from "../../common"
 import { refundPaymentStep } from "../steps/refund-payment"
 
@@ -21,6 +25,6 @@ export const refundPaymentWorkflow = createWorkflow(
       data: { id: payment.id },
     })
 
-    return payment
+    return new WorkflowResponse(payment)
   }
 )

packages/core/core-flows/src/price-list/workflows/batch-price-list-prices.ts

@@ -4,6 +4,7 @@ import {
 } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -19,7 +20,7 @@ export const batchPriceListPricesWorkflow = createWorkflow(
     input: WorkflowData<{
       data: BatchPriceListPricesWorkflowDTO
     }>
-  ): WorkflowData<BatchPriceListPricesWorkflowResult> => {
+  ): WorkflowResponse<BatchPriceListPricesWorkflowResult> => {
     const createInput = transform({ input: input.data }, (data) => [
       { id: data.input.id, prices: data.input.create },
     ])
@@ -34,6 +35,8 @@ export const batchPriceListPricesWorkflow = createWorkflow(
       removePriceListPricesWorkflowStep(input.data.delete)
     )
 
-    return transform({ created, updated, deleted }, (data) => data)
+    return new WorkflowResponse(
+      transform({ created, updated, deleted }, (data) => data)
+    )
   }
 )

packages/core/core-flows/src/price-list/workflows/create-price-list-prices.ts

@@ -1,6 +1,7 @@
 import { CreatePriceListPricesWorkflowDTO, PricingTypes } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
 } from "@medusajs/workflows-sdk"
@@ -15,15 +16,17 @@ export const createPriceListPricesWorkflow = createWorkflow(
     input: WorkflowData<{
       data: CreatePriceListPricesWorkflowDTO[]
     }>
-  ): WorkflowData<PricingTypes.PriceDTO[]> => {
+  ): WorkflowResponse<PricingTypes.PriceDTO[]> => {
     const [_, variantPriceMap] = parallelize(
       validatePriceListsStep(input.data),
       validateVariantPriceLinksStep(input.data)
     )
 
-    return createPriceListPricesStep({
-      data: input.data,
-      variant_price_map: variantPriceMap,
-    })
+    return new WorkflowResponse(
+      createPriceListPricesStep({
+        data: input.data,
+        variant_price_map: variantPriceMap,
+      })
+    )
   }
 )

packages/core/core-flows/src/price-list/workflows/create-price-lists.ts

@@ -1,5 +1,9 @@
 import { CreatePriceListWorkflowInputDTO, PriceListDTO } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createPriceListsStep, validateVariantPriceLinksStep } from "../steps"
 
 type WorkflowInput = { price_lists_data: CreatePriceListWorkflowInputDTO[] }
@@ -7,14 +11,16 @@ type WorkflowInput = { price_lists_data: CreatePriceListWorkflowInputDTO[] }
 export const createPriceListsWorkflowId = "create-price-lists"
 export const createPriceListsWorkflow = createWorkflow(
   createPriceListsWorkflowId,
-  (input: WorkflowData<WorkflowInput>): WorkflowData<PriceListDTO[]> => {
+  (input: WorkflowData<WorkflowInput>): WorkflowResponse<PriceListDTO[]> => {
     const variantPriceMap = validateVariantPriceLinksStep(
       input.price_lists_data
     )
 
-    return createPriceListsStep({
-      data: input.price_lists_data,
-      variant_price_map: variantPriceMap,
-    })
+    return new WorkflowResponse(
+      createPriceListsStep({
+        data: input.price_lists_data,
+        variant_price_map: variantPriceMap,
+      })
+    )
   }
 )

packages/core/core-flows/src/price-list/workflows/remove-price-list-prices.ts

@@ -1,10 +1,14 @@
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { removePriceListPricesStep } from "../steps/remove-price-list-prices"
 
 export const removePriceListPricesWorkflowId = "remove-price-list-prices"
 export const removePriceListPricesWorkflow = createWorkflow(
   removePriceListPricesWorkflowId,
-  (input: WorkflowData<{ ids: string[] }>): WorkflowData<string[]> => {
-    return removePriceListPricesStep(input.ids)
+  (input: WorkflowData<{ ids: string[] }>): WorkflowResponse<string[]> => {
+    return new WorkflowResponse(removePriceListPricesStep(input.ids))
   }
 )

packages/core/core-flows/src/price-list/workflows/update-price-list-prices.ts

@@ -1,6 +1,7 @@
 import { PricingTypes, UpdatePriceListPricesWorkflowDTO } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
 } from "@medusajs/workflows-sdk"
@@ -15,15 +16,17 @@ export const updatePriceListPricesWorkflow = createWorkflow(
     input: WorkflowData<{
       data: UpdatePriceListPricesWorkflowDTO[]
     }>
-  ): WorkflowData<PricingTypes.PriceDTO[]> => {
+  ): WorkflowResponse<PricingTypes.PriceDTO[]> => {
     const [_, variantPriceMap] = parallelize(
       validatePriceListsStep(input.data),
       validateVariantPriceLinksStep(input.data)
     )
 
-    return updatePriceListPricesStep({
-      data: input.data,
-      variant_price_map: variantPriceMap,
-    })
+    return new WorkflowResponse(
+      updatePriceListPricesStep({
+        data: input.data,
+        variant_price_map: variantPriceMap,
+      })
+    )
   }
 )

packages/core/core-flows/src/pricing/workflows/create-price-preferences.ts

@@ -1,5 +1,9 @@
 import { PricingWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createPricePreferencesStep } from "../steps"
 
 type WorkflowInputData = PricingWorkflow.CreatePricePreferencesWorkflowInput[]
@@ -8,6 +12,6 @@ export const createPricePreferencesWorkflowId = "create-price-preferences"
 export const createPricePreferencesWorkflow = createWorkflow(
   createPricePreferencesWorkflowId,
   (input: WorkflowData<WorkflowInputData>) => {
-    return createPricePreferencesStep(input)
+    return new WorkflowResponse(createPricePreferencesStep(input))
   }
 )

packages/core/core-flows/src/pricing/workflows/update-price-preferences.ts

@@ -1,5 +1,9 @@
 import { PricingWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updatePricePreferencesStep } from "../steps"
 
 type WorkflowInputData = PricingWorkflow.UpdatePricePreferencesWorkflowInput
@@ -8,6 +12,6 @@ export const updatePricePreferencesWorkflowId = "update-price-preferences"
 export const updatePricePreferencesWorkflow = createWorkflow(
   updatePricePreferencesWorkflowId,
   (input: WorkflowData<WorkflowInputData>) => {
-    return updatePricePreferencesStep(input)
+    return new WorkflowResponse(updatePricePreferencesStep(input))
   }
 )

packages/core/core-flows/src/product-category/workflows/create-product-categories.ts

@@ -1,5 +1,9 @@
 import { ProductCategoryWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createProductCategoriesStep } from "../steps"
 
 type WorkflowInputData =
@@ -9,6 +13,6 @@ export const createProductCategoriesWorkflowId = "create-product-categories"
 export const createProductCategoriesWorkflow = createWorkflow(
   createProductCategoriesWorkflowId,
   (input: WorkflowData<WorkflowInputData>) => {
-    return createProductCategoriesStep(input)
+    return new WorkflowResponse(createProductCategoriesStep(input))
   }
 )

packages/core/core-flows/src/product-category/workflows/update-product-categories.ts

@@ -1,5 +1,9 @@
 import { ProductCategoryWorkflow } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { updateProductCategoriesStep } from "../steps"
 
 type WorkflowInputData =
@@ -9,6 +13,6 @@ export const updateProductCategoriesWorkflowId = "update-product-categories"
 export const updateProductCategoriesWorkflow = createWorkflow(
   updateProductCategoriesWorkflowId,
   (input: WorkflowData<WorkflowInputData>) => {
-    return updateProductCategoriesStep(input)
+    return new WorkflowResponse(updateProductCategoriesStep(input))
   }
 )

packages/core/core-flows/src/product/workflows/batch-product-variants.ts

@@ -1,5 +1,6 @@
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -25,7 +26,7 @@ export const batchProductVariantsWorkflow = createWorkflow(
         UpdateProductVariantWorkflowInputDTO
       >
     >
-  ): WorkflowData<BatchWorkflowOutput<ProductTypes.ProductVariantDTO>> => {
+  ): WorkflowResponse<BatchWorkflowOutput<ProductTypes.ProductVariantDTO>> => {
     const normalizedInput = transform({ input }, (data) => {
       return {
         create: data.input.create ?? [],
@@ -46,7 +47,7 @@ export const batchProductVariantsWorkflow = createWorkflow(
       })
     )
 
-    return transform({ res, input }, (data) => {
+    const response = transform({ res, input }, (data) => {
       return {
         created: data.res[0],
         updated: data.res[1],
@@ -57,5 +58,7 @@ export const batchProductVariantsWorkflow = createWorkflow(
         },
       }
     })
+
+    return new WorkflowResponse(response)
   }
 )

packages/core/core-flows/src/product/workflows/batch-products.ts

@@ -7,6 +7,7 @@ import {
 } from "@medusajs/types"
 import {
   WorkflowData,
+  WorkflowResponse,
   createWorkflow,
   parallelize,
   transform,
@@ -25,7 +26,7 @@ export const batchProductsWorkflow = createWorkflow(
         UpdateProductWorkflowInputDTO
       >
     >
-  ): WorkflowData<BatchWorkflowOutput<ProductTypes.ProductDTO>> => {
+  ): WorkflowResponse<BatchWorkflowOutput<ProductTypes.ProductDTO>> => {
     const res = parallelize(
       createProductsWorkflow.runAsStep({
         input: { products: input.create ?? [] },
@@ -38,16 +39,18 @@ export const batchProductsWorkflow = createWorkflow(
       })
     )
 
-    return transform({ res, input }, (data) => {
-      return {
-        created: data.res[0],
-        updated: data.res[1],
-        deleted: {
-          ids: data.input.delete ?? [],
-          object: "product",
-          deleted: true,
-        },
-      }
-    })
+    return new WorkflowResponse(
+      transform({ res, input }, (data) => {
+        return {
+          created: data.res[0],
+          updated: data.res[1],
+          deleted: {
+            ids: data.input.delete ?? [],
+            object: "product",
+            deleted: true,
+          },
+        }
+      })
+    )
   }
 )

packages/core/core-flows/src/product/workflows/create-collections.ts

@@ -1,5 +1,9 @@
 import { ProductTypes } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
 import { createCollectionsStep } from "../steps"
 
 type WorkflowInput = { collections: ProductTypes.CreateProductCollectionDTO[] }
@@ -9,7 +13,7 @@ export const createCollectionsWorkflow = createWorkflow(
   createCollectionsWorkflowId,
   (
     input: WorkflowData<WorkflowInput>
-  ): WorkflowData<ProductTypes.ProductCollectionDTO[]> => {
-    return createCollectionsStep(input.collections)
+  ): WorkflowResponse<ProductTypes.ProductCollectionDTO[]> => {
+    return new WorkflowResponse(createCollectionsStep(input.collections))
   }
 )