feat: Refactor taskType to use RetryTaskType in RetryHandlerTask and RetryableTask

Author: YunchuWang

packages/durabletask-js/src/task/retry-handler-task.ts

@@ -2,7 +2,7 @@
 // Licensed under the MIT License.
 
 import * as pb from "../proto/orchestrator_service_pb";
-import { RetryTaskBase } from "./retry-task-base";
+import { RetryTaskBase, RetryTaskType } from "./retry-task-base";
 import { AsyncRetryHandler } from "./retry/retry-handler";
 import { createRetryContext } from "./retry/retry-context";
 import { TaskFailureDetails } from "./failure-details";
@@ -34,7 +34,7 @@ export class RetryHandlerTask<T> extends RetryTaskBase<T> {
     handler: AsyncRetryHandler,
     action: pb.OrchestratorAction,
     startTime: Date,
-    taskType: "activity" | "subOrchestration",
+    taskType: RetryTaskType,
   ) {
     super(action, startTime, taskType);
     this._handler = handler;

packages/durabletask-js/src/task/retryable-task.ts

@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
-import { RetryTaskBase } from "./retry-task-base";
+import { RetryTaskBase, RetryTaskType } from "./retry-task-base";
 import { RetryPolicy } from "./retry/retry-policy";
 import * as pb from "../proto/orchestrator_service_pb";
 
@@ -28,7 +28,7 @@ export class RetryableTask<T> extends RetryTaskBase<T> {
     retryPolicy: RetryPolicy,
     action: pb.OrchestratorAction,
     startTime: Date,
-    taskType: "activity" | "subOrchestration",
+    taskType: RetryTaskType,
   ) {
     super(action, startTime, taskType);
     this._retryPolicy = retryPolicy;

packages/durabletask-js/src/worker/runtime-orchestration-context.ts

@@ -8,7 +8,7 @@ import { ParentOrchestrationInstance } from "../types/parent-orchestration-insta
 import * as pb from "../proto/orchestrator_service_pb";
 import * as ph from "../utils/pb-helper.util";
 import { CompletableTask } from "../task/completable-task";
-import { RetryTaskBase } from "../task/retry-task-base";
+import { RetryTaskBase, RetryTaskType } from "../task/retry-task-base";
 import { RetryableTask } from "../task/retryable-task";
 import { RetryHandlerTask } from "../task/retry-handler-task";
 import { RetryTimerTask } from "../task/retry-timer-task";
@@ -518,7 +518,7 @@ export class RuntimeOrchestrationContext extends OrchestrationContext {
     action: pb.OrchestratorAction,
     id: number,
     options: TaskOptions | SubOrchestrationOptions | undefined,
-    taskType: "activity" | "subOrchestration",
+    taskType: RetryTaskType
   ): CompletableTask<TOutput> {
     if (options?.retry && isRetryPolicy(options.retry)) {
       const retryableTask = new RetryableTask<TOutput>(