Function getDefaultEffortForModel

src/utils/effort.ts:302–358

(
  model: string,
)

// @[MODEL LAUNCH]: Update the default effort levels for new models
export function getDefaultEffortForModel(
  model: string,
): EffortValue | undefined {
  const ncodeModel = resolveNCodeManagedModel(model)
  if (ncodeModel) {
    return ncodeModel.defaultEffortLevel
  }

  if (isInternalBuild()) {
    const config = getAntModelOverrideConfig()
    const isDefaultModel =
      config?.defaultModel !== undefined &&
      model.toLowerCase() === config.defaultModel.toLowerCase()
    if (isDefaultModel && config?.defaultModelEffortLevel) {
      return config.defaultModelEffortLevel
    }
    const antModel = resolveAntModel(model)
    if (antModel) {
      if (antModel.defaultEffortLevel) {
        return antModel.defaultEffortLevel
      }
      if (antModel.defaultEffortValue !== undefined) {
        return antModel.defaultEffortValue
      }
    }
    // Always default ants to undefined/high
    return undefined
  }

  // IMPORTANT: Do not change the default effort level without notifying
  // the model launch DRI and research. Default effort is a sensitive setting
  // that can greatly affect model quality and bashing.

  // Default effort on Opus 4.6 to medium for Pro.
  // Max/Team also get medium when the ncode_grey_step2 config is enabled.
  if (model.toLowerCase().includes('opus-4-6')) {
    const session = getCurrentSubscriptionSessionState()
    if (session.isProSubscriber) {
      return 'medium'
    }
    if (
      getOpusDefaultEffortConfig().enabled &&
      (session.isMaxSubscriber || session.isTeamSubscriber)
    ) {
      return 'medium'
    }
  }

  // When ultrathink feature is on, default effort to medium (ultrathink bumps to high)
  if (isUltrathinkEnabled() && modelSupportsEffort(model)) {
    return 'medium'
  }

  // Fallback to undefined, which means we don't set an effort level. This
  // should resolve to high effort level in the API.
  return undefined
}