Function AIProvider

coderd/database/dbgen/dbgen.go:195–226

(t testing.TB, db database.Store, seed database.AIProvider, munge ...func(*database.InsertAIProviderParams))

}

func AIProvider(t testing.TB, db database.Store, seed database.AIProvider, munge ...func(*database.InsertAIProviderParams)) database.AIProvider {
	t.Helper()
	id := seed.ID
	if id == uuid.Nil {
		id = uuid.New()
	}
	provType := seed.Type
	if provType == "" {
		provType = database.AiProviderTypeOpenai
	}
	name := takeFirst(seed.Name, testutil.GetRandomNameHyphenated(t))
	displayName := seed.DisplayName
	if !displayName.Valid {
		displayName = sql.NullString{String: name, Valid: true}
	}
	params := database.InsertAIProviderParams{
		ID:            id,
		Type:          provType,
		Name:          name,
		DisplayName:   displayName,
		Enabled:       takeFirst(seed.Enabled, true),
		BaseUrl:       takeFirst(seed.BaseUrl, "https://api.example.com/"),
		Settings:      seed.Settings,
		SettingsKeyID: seed.SettingsKeyID,
	}
	for _, fn := range munge {
		fn(&params)
	}
	provider, err := db.InsertAIProvider(genCtx, params)
	require.NoError(t, err, "insert ai provider")
	return provider
}

func AIProviderKey(t testing.TB, db database.Store, seed database.AIProviderKey, munge ...func(*database.InsertAIProviderKeyParams)) database.AIProviderKey {
	t.Helper()