diff --git a/apps/webapp/app/v3/services/rescheduleTaskRun.server.ts b/apps/webapp/app/v3/services/rescheduleTaskRun.server.ts index 122fcc2c59..43163fb4fb 100644 --- a/apps/webapp/app/v3/services/rescheduleTaskRun.server.ts +++ b/apps/webapp/app/v3/services/rescheduleTaskRun.server.ts @@ -3,6 +3,7 @@ import { TaskRun } from "@trigger.dev/database"; import { parseDelay } from "~/utils/delays"; import { BaseService, ServiceValidationError } from "./baseService.server"; import { EnqueueDelayedRunService } from "./enqueueDelayedRun.server"; +import { engine } from "../runEngine.server"; export class RescheduleTaskRunService extends BaseService { public async call(taskRun: TaskRun, body: RescheduleRunRequestBody) { @@ -22,11 +23,15 @@ export class RescheduleTaskRunService extends BaseService { }, data: { delayUntil: delay, + queueTimestamp: delay, }, }); - await EnqueueDelayedRunService.reschedule(taskRun.id, delay); - - return updatedRun; + if (updatedRun.engine === "V1") { + await EnqueueDelayedRunService.reschedule(taskRun.id, delay); + return updatedRun; + } else { + return engine.rescheduleDelayedRun({ runId: taskRun.id, delayUntil: delay }); + } } }