@@ -2,7 +2,7 @@ import { WorkerInstanceGroup, WorkerInstanceGroupType } from "@trigger.dev/datab
2
2
import { WithRunEngine } from "../baseService.server" ;
3
3
import { WorkerGroupTokenService } from "./workerGroupTokenService.server" ;
4
4
import { logger } from "~/services/logger.server" ;
5
- import { makeFlags , makeSetFlags } from "~/v3/featureFlags.server" ;
5
+ import { FEATURE_FLAG , makeFlags , makeSetFlags } from "~/v3/featureFlags.server" ;
6
6
7
7
export class WorkerGroupService extends WithRunEngine {
8
8
private readonly defaultNamePrefix = "worker_group" ;
@@ -49,14 +49,14 @@ export class WorkerGroupService extends WithRunEngine {
49
49
50
50
const getFlag = makeFlags ( this . _prisma ) ;
51
51
const defaultWorkerInstanceGroupId = await getFlag ( {
52
- key : " defaultWorkerInstanceGroupId" ,
52
+ key : FEATURE_FLAG . defaultWorkerInstanceGroupId ,
53
53
} ) ;
54
54
55
55
// If there's no global default yet we should set it to the new worker group
56
56
if ( ! defaultWorkerInstanceGroupId ) {
57
57
const setFlag = makeSetFlags ( this . _prisma ) ;
58
58
await setFlag ( {
59
- key : " defaultWorkerInstanceGroupId" ,
59
+ key : FEATURE_FLAG . defaultWorkerInstanceGroupId ,
60
60
value : workerGroup . id ,
61
61
} ) ;
62
62
}
@@ -169,7 +169,7 @@ export class WorkerGroupService extends WithRunEngine {
169
169
const flags = makeFlags ( this . _prisma ) ;
170
170
171
171
const defaultWorkerInstanceGroupId = await flags ( {
172
- key : " defaultWorkerInstanceGroupId" ,
172
+ key : FEATURE_FLAG . defaultWorkerInstanceGroupId ,
173
173
} ) ;
174
174
175
175
if ( ! defaultWorkerInstanceGroupId ) {
0 commit comments