mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-08-28 05:27:24 +00:00
37 lines
1.3 KiB
TypeScript
37 lines
1.3 KiB
TypeScript
import type { AssertEqual, Equals, Keys, Values, ExperimentId, Experiments } from "@roo-code/types"
|
|
|
|
export const EXPERIMENT_IDS = {
|
|
MULTI_FILE_APPLY_DIFF: "multiFileApplyDiff",
|
|
POWER_STEERING: "powerSteering",
|
|
PREVENT_FOCUS_DISRUPTION: "preventFocusDisruption",
|
|
IMAGE_GENERATION: "imageGeneration",
|
|
RUN_SLASH_COMMAND: "runSlashCommand",
|
|
} as const satisfies Record<string, ExperimentId>
|
|
|
|
type _AssertExperimentIds = AssertEqual<Equals<ExperimentId, Values<typeof EXPERIMENT_IDS>>>
|
|
|
|
type ExperimentKey = Keys<typeof EXPERIMENT_IDS>
|
|
|
|
interface ExperimentConfig {
|
|
enabled: boolean
|
|
}
|
|
|
|
export const experimentConfigsMap: Record<ExperimentKey, ExperimentConfig> = {
|
|
MULTI_FILE_APPLY_DIFF: { enabled: false },
|
|
POWER_STEERING: { enabled: false },
|
|
PREVENT_FOCUS_DISRUPTION: { enabled: false },
|
|
IMAGE_GENERATION: { enabled: false },
|
|
RUN_SLASH_COMMAND: { enabled: false },
|
|
}
|
|
|
|
export const experimentDefault = Object.fromEntries(
|
|
Object.entries(experimentConfigsMap).map(([_, config]) => [
|
|
EXPERIMENT_IDS[_ as keyof typeof EXPERIMENT_IDS] as ExperimentId,
|
|
config.enabled,
|
|
]),
|
|
) as Record<ExperimentId, boolean>
|
|
|
|
export const experiments = {
|
|
get: (id: ExperimentKey): ExperimentConfig | undefined => experimentConfigsMap[id],
|
|
isEnabled: (experimentsConfig: Experiments, id: ExperimentId) => experimentsConfig[id] ?? experimentDefault[id],
|
|
} as const
|