Function setupProviderConfig

core/config/onboarding.ts:132–171

(
  config: ConfigYaml,
  provider: string,
  apiKey: string,
)

}

export function setupProviderConfig(
  config: ConfigYaml,
  provider: string,
  apiKey: string,
): ConfigYaml {
  let newModels: OnboardingModel[];

  switch (provider) {
    case "openai":
      newModels = OPENAI_ONBOARDING_MODELS(apiKey);
      break;
    case "anthropic":
      newModels = ANTHROPIC_ONBOARDING_MODELS(apiKey);
      break;
    case "gemini":
      newModels = GEMINI_ONBOARDING_MODELS(apiKey);
      break;
    default:
      throw new Error(`Unknown provider: ${provider}`);
  }

  const existingModels = config.models ?? [];

  const isSameModel = (m: OnboardingModel, n: OnboardingModel) =>
    "provider" in m &&
    "provider" in n &&
    m.provider === n.provider &&
    m.model === n.model;

  // Update API key on existing models; add new entries for any missing models
  const updatedModels = existingModels.map((m) => {
    const match = newModels.find((n) => isSameModel(m, n));
    return match ? { ...m, apiKey } : m;
  });
  const modelsToAdd = newModels.filter(
    (n) => !existingModels.some((m) => isSameModel(m, n)),
  );

  return { ...config, models: [...updatedModels, ...modelsToAdd] };
}