Function ChatProvider

coderd/database/dbgen/dbgen.go:271–318

(t testing.TB, db database.Store, seed database.ChatProvider, munge ...func(*database.InsertChatProviderParams))

}

func ChatProvider(t testing.TB, db database.Store, seed database.ChatProvider, munge ...func(*database.InsertChatProviderParams)) database.ChatProvider {
	t.Helper()
	params := database.InsertChatProviderParams{
		Provider:                   takeFirst(seed.Provider, "openai"),
		DisplayName:                takeFirst(seed.DisplayName, seed.Provider, "openai"),
		APIKey:                     takeFirst(seed.APIKey, "test-key"),
		BaseUrl:                    seed.BaseUrl,
		ApiKeyKeyID:                seed.ApiKeyKeyID,
		CreatedBy:                  seed.CreatedBy,
		Enabled:                    takeFirst(seed.Enabled, true),
		CentralApiKeyEnabled:       takeFirst(seed.CentralApiKeyEnabled, true),
		AllowUserApiKey:            seed.AllowUserApiKey,
		AllowCentralApiKeyFallback: seed.AllowCentralApiKeyFallback,
	}
	for _, fn := range munge {
		fn(&params)
	}
	provider := AIProvider(t, db, database.AIProvider{
		Type:        database.AIProviderType(params.Provider),
		Name:        "test-" + uuid.NewString(),
		DisplayName: sql.NullString{String: params.DisplayName, Valid: params.DisplayName != ""},
		BaseUrl:     params.BaseUrl,
	}, func(p *database.InsertAIProviderParams) {
		p.Enabled = params.Enabled
	})
	if params.APIKey != "" {
		AIProviderKey(t, db, database.AIProviderKey{
			ProviderID:  provider.ID,
			APIKey:      params.APIKey,
			ApiKeyKeyID: params.ApiKeyKeyID,
		})
	}
	return database.ChatProvider{
		ID:                         provider.ID,
		Provider:                   params.Provider,
		DisplayName:                params.DisplayName,
		APIKey:                     params.APIKey,
		BaseUrl:                    params.BaseUrl,
		ApiKeyKeyID:                params.ApiKeyKeyID,
		CreatedBy:                  params.CreatedBy,
		Enabled:                    params.Enabled,
		CentralApiKeyEnabled:       params.CentralApiKeyEnabled,
		AllowUserApiKey:            params.AllowUserApiKey,
		AllowCentralApiKeyFallback: params.AllowCentralApiKeyFallback,
		CreatedAt:                  provider.CreatedAt,
		UpdatedAt:                  provider.UpdatedAt,
	}
}

func MCPServerConfig(t testing.TB, db database.Store, seed database.MCPServerConfig) database.MCPServerConfig {
	t.Helper()