Kode-cli/src/utils/betas.ts
2025-08-10 19:57:17 +08:00

21 lines
623 B
TypeScript

import { memoize } from 'lodash-es'
import { checkGate } from '../services/statsig'
import {
GATE_TOKEN_EFFICIENT_TOOLS,
BETA_HEADER_TOKEN_EFFICIENT_TOOLS,
CLAUDE_CODE_20250219_BETA_HEADER,
} from '../constants/betas.js'
export const getBetas = memoize(async (): Promise<string[]> => {
const betaHeaders = [CLAUDE_CODE_20250219_BETA_HEADER]
if (process.env.USER_TYPE === 'ant' || process.env.SWE_BENCH) {
const useTokenEfficientTools = await checkGate(GATE_TOKEN_EFFICIENT_TOOLS)
if (useTokenEfficientTools) {
betaHeaders.push(BETA_HEADER_TOKEN_EFFICIENT_TOOLS)
}
}
return betaHeaders
})