Files
magpie/groups_cli.go
T
yetone 10b940233c routing: a group's member can be sent in its vendor's fast mode, beside its own reasoning (vs on Discord: a member "gpt-6.1 sol · high · fast") — a member's effort was already its own (#189, "provider/model:effort") but there was no way to ask for speed, so a group could not hold "gpt-6.1-sol at high, priority" next to slower members. A group now lists in fast the members it sends fast, as Members spells them, kept beside Members rather than in their ids so an older version still routes to them, only not fast (checked: HEAD's build on the same providers.json sends the same members at their effort, no tier, and leaves the file as it was). When the gateway sends to such a member it asks in the upstream's own words, only where the model has a fast mode magpie can ask for (provider.CanFast): service_tier "priority" for a ChatGPT account's GPT models and an OpenAI key's GPT/o-series models on its Responses API, Cursor's -fast variant, and speed "fast" with the fast-mode-2026-02-01 beta for Claude Opus 4.8/5/5.5 on an Anthropic key (api.anthropic.com); a relay, Bedrock, a Claude subscription, a model without one, or a provider since moved off those hosts is sent untouched. It is applied in the agent's own API and carried through translation (parseAnthropic now reads speed), so Chat, Responses and Anthropic agents all get it. A translated request for OpenAI's Chat API is still told no tier (TestCursorPluginFast), hence the Responses requirement. The trace marks the try, the order and the group (fast), shown as "fast" beside the reasoning in Requests and the order. Typed, a member is provider/model[:effort]:fast: magpie group add/set models=… takes it, fast=<m1>,… sets them, models-= takes either spelling, and the group lists and shows members as typed with "· fast"; a model with no fast mode, or a group in the group, is refused. The GUI group editor shows a "Standard speed"/"Fast" toggle (标准速度/快速) beside a member's reasoning only for a model that has one, the card reads "· fast" after the reasoning, and fast follows a member's reasoning change and leaves with a removed member; the TUI's group view takes E to step a member's effort and F to toggle fast. Provider renames carry it. TestFastMemberOpenAI, TestFastMemberAnthropic, TestWithFastWire (gateway, per wire format, a fake vendor recording body and anthropic-beta), TestCanFast, TestSaveGroupFast, TestGroupMembersFast, TestGroupMemberEffortAndFast, TestGroupSaveFast and group-member-fast.test.cjs fail without the change. A sandbox magpie (own HOME, gateway on 3966) with providers at http://api.openai.com and http://api.anthropic.com through a local recording proxy sent priority+high to OpenAI and speed fast + the beta to Anthropic from Chat, Responses and Anthropic clients, and nothing extra for Sonnet or a non-fast member. go vet, go build -tags nogui (darwin, GOOS=linux), GOOS=windows go build, go test -tags nogui ./... pass; group-member-fast, click-scroll, old-webkit, routing-levels, routing-manual, routing-effort, settings-groups pass (Chromium, WebKit, en/zh).
2026-10-01 17:13:34 +08:00

881 lines
30 KiB
Go

package main
import (
"fmt"
"slices"
"strings"
"github.com/charmbracelet/lipgloss"
"github.com/yetone/magpie/internal/agent"
"github.com/yetone/magpie/internal/provider"
)
// Routing groups from the terminal: what the Routing view does, through
// the same provider.SaveGroup, DeleteGroup and ShowGroup.
const groupUsage = `usage:
magpie groups list routing groups: yours, then those magpie found
magpie group <id> show one group and its models (also: magpie group show <id>)
magpie group add <name> models=<m1>[,m2…] [routing=…] [stays=…]
make a group; agents pick it as group/<id>, the id made from the name;
a name in use replaces that group
magpie group set <id> k=v… change one: name, models (the whole list, in order),
models+=<m> (append), models-=<m> (drop), routing, stays,
context (how long a request agents are told it takes: 272k; empty is
its shortest model's), levels (the reasoning levels agents are offered:
levels=none,low,medium,high,xhigh,max; empty is those every model has —
a model without the one asked is sent its nearest),
family (a tag: magpie visible shows agents families, not each group),
id (what agents pick it as: id=gpt-6-astra drops auto-; the groups
it is in follow; an agent set to the old id needs setting again),
effort=auto (the classifier picks each turn's reasoning; needs classifier=),
effort=agent (the agent's again), classifier=<provider/model>|group/<id>,
pick=<model> (routing=manual, every request to that one of its models),
fast=<m1>[,m2…] (the models sent in their vendor's fast mode; empty for none)
magpie group pick <id> <model> route the group manually, every request to that one of its models
(as clicking it on the group's card in the Routing view does)
magpie group rm <id> remove a group (one magpie found is hidden instead)
magpie group restore <id> bring back a group magpie found that you removed
magpie group rule add|rm|mv <id> … rules: which model a turn goes to first, by its length, an image,
the reasoning asked for or the agent (magpie group rule help)
magpie finds a group for each model two or more providers serve (auto-<model>, never stored);
removing one stores {"id":…,"hidden":true} in providers.json, which is what keeps it removed:
take that record out of the file and the group is back
models provider/model ids as magpie models lists them; a bare model id works when one provider serves it;
group/<id> puts another group in it, routed by its own routing and rules in its place —
never one the group is in already (that would put it in itself), at most 8 groups deep
provider/model:<effort> fixes the member's reasoning — none, minimal, low, medium, high, xhigh
or max — sent whatever the agent asks or effort=auto picks, at the level the model has nearest;
without one it reasons as the group's effort says. The same model at two efforts is two members
(a rule can send to either). A model whose own id ends so (a :free, :7b or :0) stays as it is
provider/model[:effort]:fast sends the member in its vendor's fast mode: priority processing on a
ChatGPT account (Codex's Fast) or an OpenAI key, Cursor's -fast model, Claude's fast mode on an
Anthropic key for Opus 4.8 and 5; a model without one is refused (codex/gpt-6.1-sol:high:fast)
routing smart (default) of the subscriptions with quota to spare, the one renewing soonest first
order the first model until it can't answer, then the next
rotate each conversation's next turn goes to the next member's account or key
usage the account or key with the most of its allowance left first
manual only the model you pick (pick=, or click it on the group's card): the others and
the rules wait until you pick another or route it otherwise; no failover to them
stays auto (default) with the account or key that answered, while its cache is worth keeping
session for the whole session
turn within a turn only; routing decides afresh when you speak again
off every request routed afresh
effort agent (default) each request reasons as much as the agent asked, but at a member's own :<effort>
auto as a turn begins, the group's classifier — Jev, from a TypeSafe provider
(magpie provider add typesafe key=…), or any model, as Jev Router on OpenRouter,
asked in words — rates how hard it is, and the turn's requests
reason at low, medium, high or xhigh; only those the agent asked to reason
e.g. magpie group add "Opus anywhere" models=claude/claude-opus-5-5,copilot/claude-opus-5.5 routing=order
magpie group set opus-anywhere stays=session models+=openrouter/anthropic/claude-opus-5.5
magpie group set auto-gpt-6-astra id=gpt-6-astra
magpie group add Everything models=group/opus-anywhere,deepseek/deepseek-v4-flash routing=order
magpie group set opus-anywhere effort=auto classifier=typesafe/jev-latest
magpie group add Fast models=codex/gpt-5.6-luna:low,deepseek/deepseek-v4-flash,glm/glm-5.3-flash:high
magpie group set fast models+=gcloud/gemini-3.8-flash:medium
magpie group pick opus-anywhere copilot/claude-opus-5.5
magpie claude group/opus-anywhere`
// routingNames: each routing's value in the file, what the CLI calls it,
// and the other spellings it takes.
var routingNames = []struct {
value, name string
also []string
}{
{"", "smart", []string{"default", "auto"}},
{provider.Ordered, "order", []string{"ordered", "in-order"}},
{provider.Rotate, "rotate", []string{"in-turn", "round-robin"}},
{provider.LeastUsed, "usage", []string{"least-used"}},
{provider.Manual, "manual", []string{"pick", "picked", "pinned"}},
}
var staysNames = []struct {
value, name string
also []string
}{
{"", "auto", []string{"default"}},
{provider.AffinitySession, "session", nil},
{provider.AffinityTurn, "turn", []string{"within-a-turn"}},
{provider.AffinityOff, "off", []string{"none", "never"}},
}
func parseRouting(v string) (string, error) {
v = strings.ToLower(strings.TrimSpace(v))
var names []string
for _, r := range routingNames {
if v == r.value || v == r.name || slices.Contains(r.also, v) {
return r.value, nil
}
names = append(names, r.name)
}
return "", fmt.Errorf("routing %q is not one magpie has: %s", v, strings.Join(names, ", "))
}
func parseStays(v string) (string, error) {
v = strings.ToLower(strings.TrimSpace(v))
var names []string
for _, s := range staysNames {
if v == s.value || v == s.name || slices.Contains(s.also, v) {
return s.value, nil
}
names = append(names, s.name)
}
return "", fmt.Errorf("stays %q is not one magpie has: %s", v, strings.Join(names, ", "))
}
func routingName(v string) string {
for _, r := range routingNames {
if r.value == v {
return r.name
}
}
return "smart"
}
func staysName(v string) string {
for _, s := range staysNames {
if s.value == v {
return s.name
}
}
return "auto"
}
func splitList(v string) []string {
return strings.FieldsFunc(v, func(r rune) bool { return r == ',' || r == ' ' })
}
// memberResolver spells a model the user typed as the catalog's
// provider/model id: the id itself, or a bare model id one provider
// serves — with the effort it is fixed at after it, when one was typed
// ("glm/glm-5.3-flash:high"; see provider.MemberEffort). keep are ids the
// group has already, kept though no provider serves them now (the Routing
// view keeps them too, skipped).
func memberResolver(keep []string) func(string) (string, error) {
var ids []string
byModel := map[string][]string{}
for _, e := range provider.Served() {
if e.Group != "" {
continue
}
ids = append(ids, e.ID)
byModel[strings.ToLower(e.Model)] = append(byModel[strings.ToLower(e.Model)], e.ID)
}
var resolve func(string) (string, error)
resolve = func(in string) (string, error) {
id := strings.TrimPrefix(strings.TrimSpace(in), "magpie/")
if slices.Contains(keep, id) {
return id, nil
}
if member, fast := provider.MemberFast(id); fast {
// sent fast (Group.Fast): the member as any other, SaveGroup
// takes the :fast off
m, err := resolve(member)
if err != nil {
return "", err
}
return m + ":" + provider.FastWord, nil
}
if model, effort := provider.MemberEffort(id); effort != "" {
// a model at an effort of its own: the model as any other is
if strings.HasPrefix(model, provider.GroupPrefix) {
return "", fmt.Errorf("%s is a group: its models reason as it says, so it takes no :%s", model, effort)
}
m, err := resolve(model)
if err != nil {
return "", err
}
return provider.WithMemberEffort(m, effort), nil
}
if gid, ok := strings.CutPrefix(id, provider.GroupPrefix); ok {
// a group in the group: SaveGroup refuses one it would be in itself through
for _, g := range provider.Groups() {
if strings.EqualFold(g.ID, gid) && !g.Hidden {
return provider.GroupPrefix + g.ID, nil
}
}
return "", fmt.Errorf("magpie has no group %q (magpie groups lists them)", gid)
}
if slices.Contains(ids, id) || slices.Contains(keep, id) {
return id, nil
}
for _, x := range ids { // a provider/model id in another case
if strings.EqualFold(x, id) {
return x, nil
}
}
switch hits := byModel[strings.ToLower(id)]; len(hits) {
case 0:
case 1:
return hits[0], nil
default:
return "", fmt.Errorf("%s is served by %d providers; name one: %s", id, len(hits), strings.Join(hits, ", "))
}
msg := fmt.Sprintf("magpie knows no model %q", id)
if near := closeMatches(id, ids, 6); len(near) > 0 {
msg += "; did you mean " + strings.Join(near, ", ") + "?"
}
return "", fmt.Errorf("%s (magpie models lists them)", msg)
}
return resolve
}
// closeMatches are the ids most like what was typed: those containing it,
// then those a few edits away.
func closeMatches(in string, ids []string, n int) []string {
q := strings.ToLower(in)
if _, m, ok := strings.Cut(q, "/"); ok && m != "" {
q = m // the provider may be the part that's wrong
}
type hit struct {
id string
d int
}
var hits []hit
for _, id := range ids {
l := strings.ToLower(id)
d := editDistance(q, l[strings.Index(l, "/")+1:])
switch {
case strings.Contains(l, q):
d = -1
case d > max(2, len(q)/3):
continue
}
hits = append(hits, hit{id, d})
}
slices.SortStableFunc(hits, func(a, b hit) int { return a.d - b.d })
var out []string
for _, h := range hits {
if len(out) == n {
break
}
out = append(out, h.id)
}
return out
}
func editDistance(a, b string) int {
prev := make([]int, len(b)+1)
for j := range prev {
prev[j] = j
}
for i := 1; i <= len(a); i++ {
cur := make([]int, len(b)+1)
cur[0] = i
for j := 1; j <= len(b); j++ {
c := 1
if a[i-1] == b[j-1] {
c = 0
}
cur[j] = min(prev[j]+1, cur[j-1]+1, prev[j-1]+c)
}
prev = cur
}
return prev[len(b)]
}
// applyGroupPairs sets a group's fields from k=v pairs. id is taken only
// where allowID (a new group's).
func applyGroupPairs(g *provider.Group, pairs []string, resolve func(string) (string, error), allowID bool) error {
members := func(v string) ([]string, error) {
var out []string
for _, m := range splitList(v) {
id, err := resolve(m)
if err != nil {
return nil, err
}
if !slices.Contains(out, id) {
out = append(out, id)
}
}
return out, nil
}
for _, kv := range pairs {
k, v, ok := strings.Cut(kv, "=")
if !ok {
return fmt.Errorf("expected key=value, got %q (magpie group help)", kv)
}
var err error
switch strings.ToLower(strings.TrimSpace(k)) {
case "id":
if !allowID {
return fmt.Errorf("a group's id stays as it is; add a new group for another")
}
g.ID = v
case "name":
g.Name = v
case "models", "members", "model":
g.Members, err = members(v)
g.Fast = nil // the list as typed: :fast on those sent fast
case "models+", "members+", "model+":
var add []string
if add, err = members(v); err == nil {
for _, id := range add {
if !slices.Contains(g.Members, id) {
g.Members = append(g.Members, id)
}
}
}
case "models-", "members-", "model-":
for _, m := range splitList(v) {
m, _ = provider.MemberFast(strings.TrimPrefix(m, "magpie/"))
i := slices.IndexFunc(g.Members, func(x string) bool { return strings.EqualFold(x, m) })
if i < 0 { // a bare model id, as models= takes it
i = slices.IndexFunc(g.Members, func(x string) bool {
_, bare, _ := strings.Cut(x, "/")
return strings.EqualFold(bare, m)
})
}
if i < 0 {
return fmt.Errorf("%s is not in the group (its models: %s)", m, strings.Join(g.Members, ", "))
}
g.Members = slices.Delete(g.Members, i, i+1)
}
case "routing", "strategy":
g.Routing, err = parseRouting(v)
case "stays", "stay", "affinity":
g.Affinity, err = parseStays(v)
case "context":
// what agents are told the group takes; empty or 0 is its
// shortest model's again
g.Context = 0
if strings.TrimSpace(v) != "" {
g.Context, err = parseTokens(v)
}
case "levels", "level":
// the reasoning levels agents are offered; empty is those every
// member has again
g.Levels = nil
if strings.TrimSpace(v) != "" {
g.Levels, err = provider.CleanLevels(splitList(v))
}
case "family", "tag":
g.Family = strings.TrimSpace(v)
case "effort", "reasoning":
switch strings.ToLower(strings.TrimSpace(v)) {
case "auto", "jev":
g.Effort = provider.EffortAuto
case "", "agent", "off":
g.Effort = ""
default:
err = fmt.Errorf("effort=auto (the classifier picks each turn's) or effort=agent (the agent's), not %q", v)
}
case "classifier", "classify":
g.Classifier = strings.TrimPrefix(strings.TrimSpace(v), "magpie/")
case "fast":
// the members sent in their vendor's fast mode, as they are in
// the group
g.Fast = nil
for _, m := range splitList(v) {
m, _ = provider.MemberFast(m)
id, perr := pickMember(g.Members, m)
if perr != nil {
return perr
}
g.Fast = append(g.Fast, id)
}
case "pick", "use":
// the one member a manual group sends to (#317): routing is
// manual then
g.Pick, err = pickMember(g.Members, v)
g.Routing = provider.Manual
default:
return fmt.Errorf("unknown field %q (fields: name, models, models+, models-, routing, pick, stays, context, levels, family, effort, classifier, fast; magpie group help)", k)
}
if err != nil {
return err
}
}
return nil
}
// pickMember is the group's member the user named: its id, or a bare
// model id one member has.
func pickMember(members []string, v string) (string, error) {
v = strings.TrimPrefix(strings.TrimSpace(v), "magpie/")
if i := slices.IndexFunc(members, func(x string) bool { return strings.EqualFold(x, v) }); i >= 0 {
return members[i], nil
}
var hits []string
for _, x := range members {
if _, bare, _ := strings.Cut(x, "/"); strings.EqualFold(bare, v) {
hits = append(hits, x)
}
}
switch len(hits) {
case 1:
return hits[0], nil
case 0:
return "", fmt.Errorf("%s is not in the group (its models: %s)", v, strings.Join(members, ", "))
}
return "", fmt.Errorf("%s is %d of the group's models; name one: %s", v, len(hits), strings.Join(hits, ", "))
}
// findGroup looks a group up by its id, its catalog id or its name.
func findGroup(ref string) (provider.Group, error) {
all := provider.Groups()
ref = strings.TrimPrefix(strings.TrimPrefix(strings.TrimSpace(ref), "magpie/"), provider.GroupPrefix)
for _, match := range []func(provider.Group) bool{
func(g provider.Group) bool { return g.ID == ref },
func(g provider.Group) bool { return strings.EqualFold(g.ID, ref) },
func(g provider.Group) bool { return strings.EqualFold(g.Name, ref) && !g.Hidden },
} {
if i := slices.IndexFunc(all, match); i >= 0 {
return all[i], nil
}
}
var ids []string
for _, g := range all {
if !g.Hidden {
ids = append(ids, g.ID)
}
}
if len(ids) == 0 {
return provider.Group{}, fmt.Errorf("no group %q; there are none yet (magpie group add <name> models=…)", ref)
}
return provider.Group{}, fmt.Errorf("no group %q (groups: %s)", ref, strings.Join(ids, ", "))
}
// newGroupID is the id a new group gets, as the Routing view makes it:
// its name's, else "group", numbered past one taken.
func newGroupID(name string) string {
base := provider.Slug(name)
if base == "" {
base = "group"
}
taken := map[string]bool{}
for _, g := range provider.Groups() {
taken[g.ID] = true
}
id := base
for n := 2; taken[id]; n++ {
id = fmt.Sprintf("%s-%d", base, n)
}
return id
}
// groupCmd: `magpie group <verb> …`
func groupCmd(args []string) error {
if len(args) < 2 {
return fmt.Errorf("%s", groupUsage)
}
verb, rest := args[1], args[2:]
switch verb {
case "-h", "--help", "help":
fmt.Println(groupUsage)
return nil
case "add", "new":
if len(rest) == 0 || strings.Contains(rest[0], "=") {
return fmt.Errorf("magpie group add <name> models=<m1>[,m2…] [routing=…] [stays=…]\n\n%s", groupUsage)
}
verb := "added"
if slices.ContainsFunc(provider.Groups(), func(o provider.Group) bool {
return !o.Hidden && strings.EqualFold(o.Name, strings.TrimSpace(rest[0])) && !slices.ContainsFunc(rest[1:], func(kv string) bool { return strings.HasPrefix(kv, "id=") })
}) {
verb = "replaced"
}
g, err := addGroup(rest[0], rest[1:])
if err != nil {
return err
}
fmt.Println(green.Render("✓"), verb, bold.Render(g.Name), muted.Render("· agents pick it as "+provider.GroupPrefix+g.ID))
return showGroup(g)
case "set", "edit":
if len(rest) < 2 {
return fmt.Errorf("magpie group set <id> k=v…\n\n%s", groupUsage)
}
g, err := setGroup(rest[0], rest[1:])
if err != nil {
return err
}
fmt.Println(green.Render("✓"), "saved", bold.Render(g.Name))
return showGroup(g)
case "rm", "remove", "delete":
if len(rest) != 1 {
return fmt.Errorf("magpie group rm <id>")
}
g, err := removeGroup(rest[0])
if err != nil {
return err
}
if g.Auto {
fmt.Println(green.Render("✓"), "removed", bold.Render(g.Name), muted.Render("· magpie found it, so it's hidden: magpie group restore "+g.ID+" brings it back"))
} else {
fmt.Println(green.Render("✓"), "removed", bold.Render(g.Name))
}
return nil
case "pick", "use":
if len(rest) != 2 {
return fmt.Errorf("magpie group pick <id> <model>")
}
g, err := setGroup(rest[0], []string{"pick=" + rest[1]})
if err != nil {
return err
}
fmt.Println(green.Render("✓"), bold.Render(g.Name), "sends every request to", bold.Render(g.Picked()), muted.Render("· routing manual"))
return showGroup(g)
case "rule", "rules":
return ruleCmd(rest)
case "show":
if len(rest) != 1 {
return fmt.Errorf("magpie group show <id>")
}
g, err := findGroup(rest[0])
if err != nil {
return err
}
return showGroup(g)
case "restore", "unhide":
if len(rest) != 1 {
return fmt.Errorf("magpie group restore <id>")
}
g, err := restoreGroup(rest[0])
if err != nil {
return err
}
if len(g.Members) == 0 {
fmt.Println(green.Render("✓"), bold.Render(g.ID), "is no longer removed", muted.Render("· it shows once two providers serve its model"))
return nil
}
fmt.Println(green.Render("✓"), bold.Render(g.Name), "is back")
return showGroup(g)
}
if len(rest) > 0 {
return fmt.Errorf("magpie group has no %q\n\n%s", verb, groupUsage)
}
g, err := findGroup(verb)
if err != nil {
return err
}
return showGroup(g)
}
func addGroup(name string, pairs []string) (provider.Group, error) {
g := provider.Group{Name: strings.TrimSpace(name)}
if err := applyGroupPairs(&g, pairs, memberResolver(nil), true); err != nil {
return g, err
}
if g.ID = strings.ToLower(strings.TrimSpace(g.ID)); g.ID != "" {
if old, err := findGroup(g.ID); err == nil && old.ID == g.ID {
return g, fmt.Errorf("there is a group %s already: magpie group set %s k=v… changes it", g.ID, g.ID)
}
} else {
// a group is known by its name: adding one under a name taken
// replaces that group (its id stays), rather than making name-2
g.ID = newGroupID(g.Name)
for _, o := range provider.Groups() {
if !o.Hidden && strings.EqualFold(o.Name, g.Name) {
g.ID = o.ID
break
}
}
}
if strings.TrimSpace(g.Name) == "" {
g.Name = g.ID
}
if len(g.Members) == 0 {
return g, fmt.Errorf("a group needs a model in it: models=<provider/model>[,…] (magpie models lists them)")
}
if err := provider.SaveGroup(g); err != nil {
return g, err
}
return findGroup(g.ID)
}
func setGroup(ref string, pairs []string) (provider.Group, error) {
g, err := findGroup(ref)
if err != nil {
return g, err
}
if g.Hidden {
return g, fmt.Errorf("%s was removed: magpie group restore %s brings it back first", g.ID, g.ID)
}
from := g.ID
if err := applyGroupPairs(&g, pairs, memberResolver(g.Members), true); err != nil {
return g, err
}
if len(g.Members) == 0 {
return g, fmt.Errorf("a group needs a model in it; magpie group rm %s removes it", from)
}
pruneRules(&g)
to := strings.ToLower(strings.TrimSpace(g.ID))
if to != from && (to == "" || to != provider.Slug(to)) {
return g, fmt.Errorf("a group's id must be lowercase letters, digits and dashes, not %q", g.ID)
}
g.ID = from
if err := provider.SaveGroup(g); err != nil { // one magpie found is the user's now
return g, err
}
if to != from {
if err := provider.RenameGroup(from, to); err != nil {
return g, err
}
fmt.Println(amber.Render("!"), "agents set to "+provider.GroupPrefix+from+" need "+provider.GroupPrefix+to+" now")
}
return findGroup(to)
}
// removedOnly is a removed found group magpie doesn't find now (its model
// is down to one provider): only its record is left, to keep it removed.
func removedOnly(ref string) (string, bool) {
ref = strings.TrimPrefix(strings.TrimPrefix(strings.TrimSpace(ref), "magpie/"), provider.GroupPrefix)
for _, id := range provider.RemovedGroups() {
if strings.EqualFold(id, ref) {
return id, true
}
}
return "", false
}
func removeGroup(ref string) (provider.Group, error) {
g, err := findGroup(ref)
if err != nil {
if id, ok := removedOnly(ref); ok {
return g, fmt.Errorf("%s is removed already (and no two providers serve its model now); magpie group restore %s brings it back", id, id)
}
return g, err
}
if g.Hidden {
return g, fmt.Errorf("%s is removed already; magpie group restore %s brings it back", g.ID, g.ID)
}
return g, provider.DeleteGroup(g.ID)
}
func restoreGroup(ref string) (provider.Group, error) {
g, err := findGroup(ref)
if err != nil {
if id, ok := removedOnly(ref); ok {
// its record goes; the group comes back once two providers serve its model
return provider.Group{ID: id, Name: id}, provider.ShowGroup(id)
}
return g, err
}
if !g.Hidden {
return g, fmt.Errorf("%s isn't removed", g.ID)
}
if err := provider.ShowGroup(g.ID); err != nil {
return g, err
}
return findGroup(g.ID)
}
// groupUses: the agents set to each group, by its id.
func groupUses() map[string][]string {
out := map[string][]string{}
for _, a := range agent.Detected() {
if len(a.Fields) == 0 {
continue
}
v := strings.TrimPrefix(a.Fields[0].Get(), "magpie/")
if id, ok := strings.CutPrefix(v, provider.GroupPrefix); ok {
out[id] = append(out[id], a.Name)
}
}
return out
}
// memberLabel: a member as the Routing view shows it, its provider and
// the model's name; ok is false for one no provider serves now.
func memberLabel(id string, names map[string]provider.Entry) (string, bool) {
if gid, ok := strings.CutPrefix(id, provider.GroupPrefix); ok {
for _, g := range provider.Groups() {
if g.ID == gid && !g.Hidden {
return "routing group " + g.Name + " · " + routingName(g.Routing) + " · " + strings.Join(g.Members, ", "), true
}
}
return "", false
}
model, effort := provider.MemberEffort(id)
if e, ok := names[model]; ok {
n := e.Name
if n == "" {
n = e.Model
}
if effort != "" {
n += " · " + effort + " reasoning, whatever the agent asks"
}
return e.Provider.Name + " · " + n, true
}
return "", false
}
// typedMember is the group's member as it is typed: with :fast after it
// when the group sends it fast.
func typedMember(g provider.Group, id string) string {
if g.IsFast(id) {
return id + ":" + provider.FastWord
}
return id
}
func catalogByID() map[string]provider.Entry {
out := map[string]provider.Entry{}
for _, e := range provider.Served() {
if e.Group == "" {
out[e.ID] = e
}
}
return out
}
// groups: `magpie groups`
func groups() error {
all := provider.Groups()
var shown, hidden []provider.Group
for _, g := range all {
if g.Hidden {
hidden = append(hidden, g)
} else {
shown = append(shown, g)
}
}
if len(shown) == 0 {
fmt.Println(muted.Render("no routing groups yet ·"), "magpie group add <name> models=<m1>,<m2>", muted.Render("· magpie group help"))
}
names, uses := catalogByID(), groupUses()
for _, e := range provider.Served() {
if e.Group != "" {
names[e.ID] = e // a group in a group is served when it is
}
}
type row struct{ name, id, how, members, uses string }
var rows []row
w := [3]int{}
for _, g := range shown {
r := row{name: bold.Render(g.Name), id: muted.Render(provider.GroupPrefix + g.ID)}
if g.Auto {
r.name += " " + faint.Render("found")
}
r.how = routingName(g.Routing)
if g.Affinity != "" {
r.how += muted.Render(" · stays " + staysName(g.Affinity))
}
if n := len(g.Rules); n > 0 {
r.how += muted.Render(fmt.Sprintf(" · %d rule%s", n, map[bool]string{true: "", false: "s"}[n == 1]))
}
sep, ready := muted.Render(" · "), false
if g.Routing == provider.Ordered {
sep = muted.Render(" → ")
}
var ms []string
for _, id := range g.Members {
model, _ := provider.MemberEffort(id)
served := names[model].ID != ""
picked := g.Routing == provider.Manual && id == g.Picked()
if served && (picked || g.Routing != provider.Manual) {
ready = true
}
shown := typedMember(g, id)
switch {
case !served:
ms = append(ms, faint.Render(shown+" (not served)"))
case picked:
ms = append(ms, green.Render("● "+shown))
case g.Routing == provider.Manual:
ms = append(ms, faint.Render(shown))
default:
ms = append(ms, shown)
}
}
r.members = strings.Join(ms, sep)
if !ready {
r.how += " " + amber.Render("no member ready")
}
if u := uses[g.ID]; len(u) > 0 {
r.uses = green.Render(" ← " + strings.Join(u, ", "))
}
for i, s := range []string{r.name, r.id, r.how} {
w[i] = max(w[i], lipgloss.Width(s))
}
rows = append(rows, r)
}
for _, r := range rows {
fmt.Printf(" %s %s %s %s%s\n", pad(r.name, w[0]), pad(r.id, w[1]), pad(r.how, w[2]), r.members, r.uses)
}
var ids []string
for _, g := range hidden {
ids = append(ids, g.ID)
}
for _, id := range provider.RemovedGroups() {
if !slices.Contains(ids, id) {
ids = append(ids, id) // its model is down to one provider now
}
}
if len(ids) > 0 {
fmt.Println()
fmt.Println(" ", muted.Render("removed: "+strings.Join(ids, ", ")+" · magpie group restore <id> brings one back"))
fmt.Println(" ", muted.Render(" (each is a {\"hidden\": true} record in providers.json that keeps it removed; deleting the record brings it back)"))
}
return nil
}
func showGroup(g provider.Group) error {
kv := func(k, v string) { fmt.Printf(" %s %s\n", muted.Render(pad(k, 9)), v) }
head := bold.Render(g.Name) + muted.Render(" "+provider.GroupPrefix+g.ID)
if g.Auto {
head += faint.Render(" found by magpie — changing it makes it yours")
}
if g.Hidden {
head += amber.Render(" removed") + muted.Render(" · magpie group restore "+g.ID)
}
fmt.Println(" ", head)
if g.Routing == provider.Manual {
kv("routing", "manual"+muted.Render(" every request to "+g.Picked()+"; magpie group pick "+g.ID+" <model> picks another"))
} else {
kv("routing", routingName(g.Routing))
}
kv("stays", staysName(g.Affinity))
names := catalogByID()
for i, id := range g.Members {
k := ""
if i == 0 {
k = "models"
}
line := fmt.Sprintf("%d %s", i+1, typedMember(g, id))
if g.Routing == provider.Manual && id == g.Picked() {
line = green.Render(fmt.Sprintf("%d %s ●", i+1, typedMember(g, id)))
}
if l, ok := memberLabel(id, names); ok {
if g.IsFast(id) {
l += " · fast"
}
line += muted.Render(" " + l)
} else {
line = faint.Render(line) + amber.Render(" not served now, skipped")
}
kv(k, line)
}
for i, r := range g.Rules {
k := ""
if i == 0 {
k = "rules"
}
line := fmt.Sprintf("%d %s", i+1, ruleLine(r))
if g.Routing == provider.Manual {
line = faint.Render(line + " (waits: routing is manual)")
}
kv(k, line)
}
if len(g.Levels) > 0 {
kv("levels", strings.Join(g.Levels, ", ")+muted.Render(" offered to agents; a model without the one asked is sent its nearest"))
}
if g.Effort == provider.EffortAuto {
kv("effort", "auto"+muted.Render(" the classifier picks each turn's reasoning"))
}
if g.Classifier != "" {
what := " tells which intent a message is"
if !slices.ContainsFunc(g.Rules, func(r provider.Rule) bool { return r.Intent != "" }) {
what = " rates how hard each turn is"
}
kv("classifier", g.Classifier+muted.Render(what))
}
if u := groupUses()[g.ID]; len(u) > 0 {
kv("used by", green.Render(strings.Join(u, ", ")))
}
return nil
}