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}