98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/mathiasbq/supervisor/internal/config"
|
|
iexec "github.com/mathiasbq/supervisor/internal/exec"
|
|
"github.com/mathiasbq/supervisor/internal/mcp"
|
|
"github.com/mathiasbq/supervisor/internal/registry"
|
|
"github.com/mathiasbq/supervisor/internal/skills/brain"
|
|
"github.com/mathiasbq/supervisor/internal/skills/org"
|
|
"github.com/mathiasbq/supervisor/internal/skills/retrospective"
|
|
"github.com/mathiasbq/supervisor/internal/skills/sessionlog"
|
|
"github.com/mathiasbq/supervisor/internal/skills/tdd"
|
|
"github.com/mathiasbq/supervisor/internal/tier"
|
|
)
|
|
|
|
func main() {
|
|
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
logger.Error("load config", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
models, err := config.LoadModels(cfg.ModelsFile)
|
|
if err != nil {
|
|
logger.Error("load models", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
systemPrompt, err := os.ReadFile(cfg.ConfigDir + "/CLAUDE.md")
|
|
if err != nil {
|
|
logger.Error("read supervisor CLAUDE.md", "path", cfg.ConfigDir+"/CLAUDE.md", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
tddPrompt, err := os.ReadFile(cfg.ConfigDir + "/tdd.md")
|
|
if err != nil {
|
|
logger.Error("read tdd.md", "path", cfg.ConfigDir+"/tdd.md", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
retroPrompt, err := os.ReadFile(cfg.ConfigDir + "/retrospective.md")
|
|
if err != nil {
|
|
logger.Error("read retrospective.md", "path", cfg.ConfigDir+"/retrospective.md", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
executor := iexec.New(iexec.Config{
|
|
SystemPrompt: string(systemPrompt),
|
|
LiteLLMBaseURL: cfg.LiteLLMBaseURL,
|
|
LiteLLMAPIKey: cfg.LiteLLMAPIKey,
|
|
})
|
|
|
|
tierFn := func(ctx context.Context) tier.Info {
|
|
return tier.Detect(ctx, "https://api.anthropic.com", cfg.LiteLLMBaseURL)
|
|
}
|
|
|
|
reg := registry.New()
|
|
reg.Register(tdd.New(tdd.Config{
|
|
SystemPrompt: string(systemPrompt),
|
|
SkillPrompt: string(tddPrompt),
|
|
DefaultModel: models.Resolve("tdd", ""),
|
|
ExecutorFn: executor.Run,
|
|
}))
|
|
reg.Register(brain.New(brain.Config{
|
|
IngestBaseURL: cfg.IngestBaseURL,
|
|
}))
|
|
reg.Register(org.New(org.Config{
|
|
TierFn: tierFn,
|
|
}))
|
|
reg.Register(sessionlog.New(sessionlog.Config{
|
|
SessionsDir: cfg.SessionsDir,
|
|
}))
|
|
reg.Register(retrospective.New(retrospective.Config{
|
|
SkillPrompt: string(retroPrompt),
|
|
DefaultModel: models.Resolve("retrospective", ""),
|
|
SessionsDir: cfg.SessionsDir,
|
|
ExecutorFn: executor.Run,
|
|
}))
|
|
|
|
srv := mcp.NewServer(reg)
|
|
mux := http.NewServeMux()
|
|
mux.Handle("/mcp", srv)
|
|
|
|
addr := ":" + cfg.Port
|
|
logger.Info("supervisor starting", "addr", addr)
|
|
if err := http.ListenAndServe(addr, mux); err != nil {
|
|
logger.Error("server stopped", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|