Add multi-agent config and registry runtime flow

2026-05-06 01:47:30 +08:00 · 2026-03-06 12:47:05 +08:00
parent 959870e6f7
commit 6902f65c54
29 changed files with 4654 additions and 76 deletions
--- a/pkg/agent/router_dispatch_test.go
+++ b/pkg/agent/router_dispatch_test.go
@@ -0,0 +1,127 @@
+package agent
+
+import (
+	"context"
+	"testing"
+
+	"clawgo/pkg/bus"
+	"clawgo/pkg/config"
+	"clawgo/pkg/runtimecfg"
+	"clawgo/pkg/tools"
+)
+
+func TestResolveAutoRouteTarget(t *testing.T) {
+	cfg := config.DefaultConfig()
+	cfg.Agents.Router.Enabled = true
+	cfg.Agents.Subagents["coder"] = config.SubagentConfig{Enabled: true}
+
+	agentID, task := resolveAutoRouteTarget(cfg, "@coder fix login")
+	if agentID != "coder" || task != "fix login" {
+		t.Fatalf("unexpected route target: %s / %s", agentID, task)
+	}
+}
+
+func TestResolveAutoRouteTargetRulesFirst(t *testing.T) {
+	cfg := config.DefaultConfig()
+	cfg.Agents.Router.Enabled = true
+	cfg.Agents.Router.Strategy = "rules_first"
+	cfg.Agents.Subagents["coder"] = config.SubagentConfig{Enabled: true, Role: "coding"}
+	cfg.Agents.Subagents["tester"] = config.SubagentConfig{Enabled: true, Role: "testing"}
+	cfg.Agents.Router.Rules = []config.AgentRouteRule{{AgentID: "coder", Keywords: []string{"登录", "bug"}}}
+
+	agentID, task := resolveAutoRouteTarget(cfg, "请帮我修复登录接口的 bug 并改代码")
+	if agentID != "coder" || task == "" {
+		t.Fatalf("expected coder route, got %s / %s", agentID, task)
+	}
+}
+
+func TestMaybeAutoRouteDispatchesExplicitAgentMention(t *testing.T) {
+	cfg := config.DefaultConfig()
+	cfg.Agents.Router.Enabled = true
+	cfg.Agents.Router.DefaultTimeoutSec = 5
+	cfg.Agents.Subagents["coder"] = config.SubagentConfig{Enabled: true}
+	runtimecfg.Set(cfg)
+	t.Cleanup(func() { runtimecfg.Set(config.DefaultConfig()) })
+
+	workspace := t.TempDir()
+	manager := tools.NewSubagentManager(nil, workspace, nil, nil)
+	manager.SetRunFunc(func(ctx context.Context, task *tools.SubagentTask) (string, error) {
+		return "auto-routed", nil
+	})
+	loop := &AgentLoop{
+		subagentManager: manager,
+		subagentRouter:  tools.NewSubagentRouter(manager),
+	}
+	out, ok, err := loop.maybeAutoRoute(context.Background(), bus.InboundMessage{
+		Channel:    "cli",
+		ChatID:     "direct",
+		SessionKey: "main",
+		Content:    "@coder implement auth",
+	})
+	if err != nil {
+		t.Fatalf("auto route failed: %v", err)
+	}
+	if !ok {
+		t.Fatalf("expected auto route to trigger")
+	}
+	if out == "" {
+		t.Fatalf("expected merged output")
+	}
+}
+
+func TestMaybeAutoRouteSkipsNormalMessages(t *testing.T) {
+	cfg := config.DefaultConfig()
+	cfg.Agents.Router.Enabled = true
+	cfg.Agents.Subagents["coder"] = config.SubagentConfig{Enabled: true}
+	runtimecfg.Set(cfg)
+	t.Cleanup(func() { runtimecfg.Set(config.DefaultConfig()) })
+
+	loop := &AgentLoop{}
+	out, ok, err := loop.maybeAutoRoute(context.Background(), bus.InboundMessage{
+		Channel:    "cli",
+		ChatID:     "direct",
+		SessionKey: "main",
+		Content:    "please help with auth",
+	})
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if ok || out != "" {
+		t.Fatalf("expected normal message to skip auto route, got ok=%v out=%q", ok, out)
+	}
+}
+
+func TestMaybeAutoRouteDispatchesRulesFirstMatch(t *testing.T) {
+	cfg := config.DefaultConfig()
+	cfg.Agents.Router.Enabled = true
+	cfg.Agents.Router.Strategy = "rules_first"
+	cfg.Agents.Router.DefaultTimeoutSec = 5
+	cfg.Agents.Subagents["tester"] = config.SubagentConfig{Enabled: true, Role: "testing"}
+	runtimecfg.Set(cfg)
+	t.Cleanup(func() { runtimecfg.Set(config.DefaultConfig()) })
+
+	workspace := t.TempDir()
+	manager := tools.NewSubagentManager(nil, workspace, nil, nil)
+	manager.SetRunFunc(func(ctx context.Context, task *tools.SubagentTask) (string, error) {
+		return "tested", nil
+	})
+	loop := &AgentLoop{
+		subagentManager: manager,
+		subagentRouter:  tools.NewSubagentRouter(manager),
+	}
+	out, ok, err := loop.maybeAutoRoute(context.Background(), bus.InboundMessage{
+		Channel:    "cli",
+		ChatID:     "direct",
+		SessionKey: "main",
+		Content:    "请做一次回归测试并验证这个修复",
+	})
+	if err != nil {
+		t.Fatalf("rules-first auto route failed: %v", err)
+	}
+	if !ok {
+		t.Fatalf("expected rules-first auto route to trigger")
+	}
+	if out == "" {
+		t.Fatalf("expected merged output")
+	}
+}