Function buildAIProviderFromRow

cli/aibridged.go:209–305

buildAIProviderFromRow decodes the settings blob and constructs the appropriate [aibridge.Provider] for a single ai_providers row. Disabled rows return a Provider stub carrying only Name and Disabled: true; settings decode, key loading, and credential checks are skipped because the provider will nev

(
	row database.AIProvider,
	keys []database.AIProviderKey,
	cfg codersdk.AIBridgeConfig,
)

// Disabled: true; settings decode, key loading, and credential checks
// are skipped because the provider will never call upstream.
func buildAIProviderFromRow(
	row database.AIProvider,
	keys []database.AIProviderKey,
	cfg codersdk.AIBridgeConfig,
) (aibridge.Provider, error) {
	if !row.Enabled {
		return disabledProviderFromRow(row)
	}

	settings, err := db2sdk.AIProviderSettings(row.Settings)
	if err != nil {
		return nil, xerrors.Errorf("decode settings: %w", err)
	}

	cbCfg := circuitBreakerConfig(cfg)
	sendActorHeaders := cfg.SendActorHeaders.Value()
	dumpDir := cfg.APIDumpDir.Value()

	// aibridge currently has native support for OpenAI and Anthropic
	// only. The other ai_provider_type values (azure, google,
	// openai-compat, openrouter, vercel) route through the OpenAI
	// provider because chatd configures them against their
	// OpenAI-compatible endpoints. Bedrock routes through the Anthropic
	// provider with a Bedrock discriminator in Settings.
	switch row.Type {
	case database.AiProviderTypeOpenai,
		database.AiProviderTypeAzure,
		database.AiProviderTypeGoogle,
		database.AiProviderTypeOpenaiCompat,
		database.AiProviderTypeOpenrouter,
		database.AiProviderTypeVercel:
		if len(keys) == 0 && !cfg.AllowBYOK.Value() {
			return nil, xerrors.Errorf("%s provider has no api keys configured and BYOK is not enabled", row.Type)
		}
		var pool *keypool.Pool
		if len(keys) > 0 {
			var err error
			pool, err = buildAIProviderKeyPool(keys)
			if err != nil {
				return nil, xerrors.Errorf("%s key pool: %w", row.Type, err)
			}
		}
		return aibridge.NewOpenAIProvider(aibridge.OpenAIConfig{
			Name:             row.Name,
			BaseURL:          row.BaseUrl,
			KeyPool:          pool,
			APIDumpDir:       dumpDir,
			CircuitBreaker:   cbCfg,
			SendActorHeaders: sendActorHeaders,
		}), nil

	case database.AiProviderTypeAnthropic, database.AiProviderTypeBedrock:
		bedrock := bedrockConfigFromRow(row, settings)
		// A row typed 'bedrock' authenticates exclusively via settings;
		// without populated Bedrock credentials it cannot make upstream
		// calls, so refuse rather than falling back to an unsigned
		// Anthropic client.
		if row.Type == database.AiProviderTypeBedrock && bedrock == nil {