diff --git a/Taskfile.yaml b/Taskfile.yaml index 4a62fa4e52939be69dbf5cfab1fcfb8cc061c26c..9bb3f6ecbc7100bcd547887a105247d282df355b 100644 --- a/Taskfile.yaml +++ b/Taskfile.yaml @@ -16,6 +16,11 @@ tasks: cmds: - go fmt ./... + test: + desc: Run tests + cmds: + - go test {{.CLI_ARGS}} ./... + modernize: desc: Modernize code cmds: diff --git a/internal/providers/providers_test.go b/internal/providers/providers_test.go new file mode 100644 index 0000000000000000000000000000000000000000..ffaa47744ed1a6453734042d5c954fa5e15912e3 --- /dev/null +++ b/internal/providers/providers_test.go @@ -0,0 +1,23 @@ +package providers + +import ( + "slices" + "testing" +) + +func TestValidDefaultModels(t *testing.T) { + for _, p := range GetAll() { + t.Run(p.Name, func(t *testing.T) { + var modelIds []string + for _, m := range p.Models { + modelIds = append(modelIds, m.ID) + } + if !slices.Contains(modelIds, p.DefaultLargeModelID) { + t.Errorf("Default large model %q not found in provider %q", p.DefaultLargeModelID, p.Name) + } + if !slices.Contains(modelIds, p.DefaultSmallModelID) { + t.Errorf("Default small model %q not found in provider %q", p.DefaultSmallModelID, p.Name) + } + }) + } +}