mrcharles
591d345371
harness.toml loading via a stdlib-only TOML subset parser (tables, bare
keys, strings/ints/bools, multi-line arrays; anything richer fails loudly).
Secrets are refs only (env:/file:/literal:, bw: reserved) and are redacted
from every error path. Model routing v0: static class -> tier alias ->
concrete model map per the DESIGN model-selection layer; requests carry the
resolved concrete model and unknown classes fail hard so routing stays
auditable. Table-driven tests cover the parser, validation, key refs, and
routing decisions.
💘 Generated with Crush
Assisted-by: Crush:glm-5.2
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package models
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func testRouter(t *testing.T) *Router {
|
|
t.Helper()
|
|
tiers := map[string]string{
|
|
"mopac-study": "glm-4.7-flash",
|
|
"mopac-code": "glm-5.2",
|
|
"mopac-review": "glm-5-turbo",
|
|
"mopac-primary": "glm-5.3",
|
|
"mopac-vision": "glm-4.6v",
|
|
}
|
|
classes := map[string]string{
|
|
"study": "mopac-study",
|
|
"read": "mopac-study",
|
|
"code": "mopac-code",
|
|
"review": "mopac-review",
|
|
"primary": "mopac-primary",
|
|
}
|
|
r, err := NewRouter(tiers, classes, "mopac-primary")
|
|
if err != nil {
|
|
t.Fatalf("NewRouter: %v", err)
|
|
}
|
|
return r
|
|
}
|
|
|
|
func TestResolve(t *testing.T) {
|
|
cases := []struct {
|
|
class string
|
|
tier string
|
|
model string
|
|
}{
|
|
{"study", "mopac-study", "glm-4.7-flash"},
|
|
{"read", "mopac-study", "glm-4.7-flash"},
|
|
{"code", "mopac-code", "glm-5.2"},
|
|
{"review", "mopac-review", "glm-5-turbo"},
|
|
{"primary", "mopac-primary", "glm-5.3"},
|
|
{"", "mopac-primary", "glm-5.3"}, // no class -> default tier
|
|
}
|
|
r := testRouter(t)
|
|
for _, tc := range cases {
|
|
d, err := r.Resolve(tc.class)
|
|
if err != nil {
|
|
t.Fatalf("Resolve(%q): %v", tc.class, err)
|
|
}
|
|
if d.Tier != tc.tier || d.Model != tc.model {
|
|
t.Errorf("Resolve(%q) = %s/%s, want %s/%s", tc.class, d.Tier, d.Model, tc.tier, tc.model)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestResolveUnknownClassIsError(t *testing.T) {
|
|
r := testRouter(t)
|
|
_, err := r.Resolve("urgent")
|
|
if err == nil {
|
|
t.Fatal("unknown class must error (routing stays auditable)")
|
|
}
|
|
if !strings.Contains(err.Error(), `[models.classes]`) {
|
|
t.Errorf("error should point at the config section: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestNewRouterValidation(t *testing.T) {
|
|
tiers := map[string]string{"mopac-primary": "glm-5.3"}
|
|
if _, err := NewRouter(tiers, nil, "mopac-missing"); err == nil || !strings.Contains(err.Error(), "default tier") {
|
|
t.Errorf("missing default tier: err=%v", err)
|
|
}
|
|
if _, err := NewRouter(tiers, map[string]string{"x": "mopac-nope"}, "mopac-primary"); err == nil || !strings.Contains(err.Error(), "unknown tier") {
|
|
t.Errorf("class to unknown tier: err=%v", err)
|
|
}
|
|
if _, err := NewRouter(nil, nil, "mopac-primary"); err == nil {
|
|
t.Errorf("empty tiers must error")
|
|
}
|
|
}
|