providers_test.go

 1package providers
 2
 3import (
 4	"slices"
 5	"testing"
 6)
 7
 8func TestValidDefaultModels(t *testing.T) {
 9	for _, p := range GetAll() {
10		t.Run(p.Name, func(t *testing.T) {
11			var modelIds []string
12			for _, m := range p.Models {
13				modelIds = append(modelIds, m.ID)
14			}
15			if !slices.Contains(modelIds, p.DefaultLargeModelID) {
16				t.Errorf("Default large model %q not found in provider %q", p.DefaultLargeModelID, p.Name)
17			}
18			if !slices.Contains(modelIds, p.DefaultSmallModelID) {
19				t.Errorf("Default small model %q not found in provider %q", p.DefaultSmallModelID, p.Name)
20			}
21		})
22	}
23}