Function AIProviderKey

coderd/database/dbgen/dbgen.go:228–249

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

}

func AIProviderKey(t testing.TB, db database.Store, seed database.AIProviderKey, munge ...func(*database.InsertAIProviderKeyParams)) database.AIProviderKey {
	t.Helper()
	id := seed.ID
	if id == uuid.Nil {
		id = uuid.New()
	}
	now := dbtime.Now()
	params := database.InsertAIProviderKeyParams{
		ID:          id,
		ProviderID:  seed.ProviderID,
		APIKey:      takeFirst(seed.APIKey, "test-key"),
		ApiKeyKeyID: seed.ApiKeyKeyID,
		CreatedAt:   takeFirst(seed.CreatedAt, now),
		UpdatedAt:   takeFirst(seed.UpdatedAt, now),
	}
	for _, fn := range munge {
		fn(&params)
	}
	key, err := db.InsertAIProviderKey(genCtx, params)
	require.NoError(t, err, "insert ai provider key")
	return key
}

// AIProviderWithOptionalKey inserts an AI provider and, when apiKey is not
// empty, inserts a provider-scoped key for it.