yetone
2cdae80a10
169c23c1cut only `@wsl:` from agent ids, and78d90571(four minutes earlier) made each omp profile an agent `omp#<name>`, so the profile name went to PostHog. modelLabel sent `<preset>/<model>` for any model on a preset's provider, and on Ollama, LM Studio or a fine-tune that id is the user's own text. docs/reference.md promised "No names". Model ids on subscriptions and plugins are the vendor's and are unchanged; known models on a key's provider still go by id (deepseek/deepseek-v4). Red before: with the old usage.go, TestAgentLabel fails ("omp#acme-secret", want "omp") and TestModelLabelKeepsTheUsersOwnModelIDs fails ("openai/ft:gpt-4o:acme-corp::abc123", want "openai/other"; "ollama/acme-internal-llm", want "ollama/other"); both pass with it. Run in docker golang:1.26.8 (this Mac's syspolicyd is pegged and new test binaries hang before start): gofmt -l, go vet -tags nogui on stats/agent/catalog, GOOS=windows go build, go build -tags nogui, and go test -tags nogui ./... — all pass except internal/agent's aside chmod-0500 tests, which root ignores; internal/agent passes as uid 1000.
110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
package stats
|
|
|
|
import (
|
|
"slices"
|
|
"strings"
|
|
|
|
"github.com/yetone/magpie/internal/agent"
|
|
"github.com/yetone/magpie/internal/catalog"
|
|
"github.com/yetone/magpie/internal/provider"
|
|
)
|
|
|
|
// Usage is what the day's event says magpie is used with, when the user
|
|
// shares it: built-in ids only. A provider of the user's own is "custom",
|
|
// and so is a model on one; a plugin no one knows is "plugin".
|
|
type Usage struct {
|
|
Agents []string // the agents connected to magpie: claude, codex
|
|
Providers []string // the providers on: deepseek, codex (a subscription), plugin:kiro
|
|
Models []string // the models the connected agents are set to: deepseek/deepseek-v4, custom, group
|
|
Groups int // routing groups shown
|
|
}
|
|
|
|
// most of each list sent: far more than anyone has
|
|
const most = 50
|
|
|
|
// usage reads Usage; tests put their own.
|
|
var usage = readUsage
|
|
|
|
func readUsage() Usage {
|
|
defer provider.Hold()()
|
|
var u Usage
|
|
for _, a := range agent.All() {
|
|
if !a.Wired() {
|
|
continue
|
|
}
|
|
u.Agents = with(u.Agents, agentLabel(a.ID))
|
|
for _, ref := range a.Models() {
|
|
u.Models = with(u.Models, modelLabel(ref))
|
|
}
|
|
}
|
|
for _, p := range provider.All() {
|
|
if p.On() && !p.Hidden {
|
|
u.Providers = with(u.Providers, providerLabel(p))
|
|
}
|
|
}
|
|
for _, g := range provider.Groups() {
|
|
if !g.Hidden && !g.Disabled {
|
|
u.Groups++
|
|
}
|
|
}
|
|
return u
|
|
}
|
|
|
|
func with(xs []string, x string) []string {
|
|
if x == "" || slices.Contains(xs, x) || len(xs) >= most {
|
|
return xs
|
|
}
|
|
return append(xs, x)
|
|
}
|
|
|
|
// providerLabel is p as stats name it: nothing the user typed.
|
|
func providerLabel(p provider.Provider) string {
|
|
switch {
|
|
case p.IsPlugin():
|
|
if id := p.PluginProvider(); provider.Movable(id) || provider.Preset(id) != nil {
|
|
return "plugin:" + id
|
|
}
|
|
return "plugin"
|
|
case p.Account != nil:
|
|
return p.Account.Agent
|
|
case p.Preset != "":
|
|
if d := provider.Preset(p.Preset); d != nil {
|
|
return d.ID
|
|
}
|
|
}
|
|
return "custom"
|
|
}
|
|
|
|
// agentLabel is an agent as stats name it: its built-in id, without the
|
|
// WSL distro or the omp profile (omp#<name>) the user named.
|
|
func agentLabel(id string) string {
|
|
id, _, _ = strings.Cut(id, "@wsl:")
|
|
id, _, _ = strings.Cut(id, "#")
|
|
return id
|
|
}
|
|
|
|
// modelLabel is an agent's model as stats name it: the provider's label
|
|
// and the vendor's model id, "group" for a routing group, and the label
|
|
// alone for a provider of the user's own; "" for one magpie can't find.
|
|
// On a key's provider the model id is the user's to type (an Ollama
|
|
// model they named, a fine-tune with their organisation in it), so it
|
|
// goes only when models.dev knows it, else as "other"; a subscription's
|
|
// or plugin's ids are its vendor's.
|
|
func modelLabel(ref string) string {
|
|
if strings.HasPrefix(ref, provider.GroupPrefix) {
|
|
return "group"
|
|
}
|
|
p, m, ok := provider.Resolve(ref)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
l := providerLabel(p)
|
|
if l == "custom" || l == "plugin" {
|
|
return l
|
|
}
|
|
if p.Account == nil && !p.IsPlugin() && !catalog.Knows(m) {
|
|
return l + "/other"
|
|
}
|
|
return l + "/" + m
|
|
}
|