Function collectModels

packages/setupWizard/src/models.ts:298–378

()

}

export async function collectModels(): Promise<{ models: LanguageModel[]; env: EnvVars }> {
    const models: LanguageModel[] = [];
    const env: EnvVars = {};

    note(
        [
            'AI features include Ask, which lets you ask questions about your codebase',
            'in natural language and get answers grounded in your indexed code.',
            '  https://docs.sourcebot.dev/docs/features/ask/ask-sourcebot',
            '',
            'You\'ll need an API key from at least one supported provider',
            '(Anthropic, OpenAI, Google, etc.) to enable these features.',
        ].join('\n'),
        'AI features',
    );

    const wantsAI = await confirm({
        message: 'Would you like to configure AI features?',
        default: true,
    });

    if (!wantsAI) {
        return { models, env };
    }

    // eslint-disable-next-line no-constant-condition
    while (true) {
        const provider = await select<Provider>({
            message: 'Which AI provider?',
            loop: false,
            choices: [
                { value: 'anthropic', name: 'Anthropic' },
                { value: 'openai', name: 'OpenAI' },
                { value: 'openai-compatible', name: 'OpenAI-compatible', description: 'self-hosted / custom endpoint' },
                { value: 'amazon-bedrock', name: 'Amazon Bedrock' },
                { value: 'google-generative-ai', name: 'Google Gemini' },
                { value: 'google-vertex', name: 'Google Vertex AI', description: 'Gemini via Vertex' },
                { value: 'google-vertex-anthropic', name: 'Google Vertex AI (Anthropic)', description: 'Claude via Vertex' },
                { value: 'azure', name: 'Azure OpenAI' },
                { value: 'deepseek', name: 'DeepSeek' },
                { value: 'mistral', name: 'Mistral' },
                { value: 'openrouter', name: 'OpenRouter' },
                { value: 'xai', name: 'xAI', description: 'Grok' },
            ],
        });

        const modelOptions = provider === 'openai-compatible'
            ? null
            : await getModelOptionsForProvider(provider);
        const model = modelOptions && modelOptions.length > 0
            ? await searchModel({
                message: 'Model name',
                models: modelOptions,
            })
            : await input({
                message: 'Model name',
                validate: (v) => !v?.trim() ? 'Model name is required' : true,
            });