weichen
c7736a1917
Details (original message): Integrate the Agent-Squad engine: squadadapter maps RoleRegistry/RulePlanner/BackendResolver/ProgressSink/HITL onto the unified Run and event model (step subagents, plan updates, deliverable artifacts, confirmation cards); step-wise engine driving loop; model routing respects profile routes with layered resolver + integration tests; context assembly observation flows to run.completed; mock outputs and e2e coverage.
48 lines
1.8 KiB
Go
48 lines
1.8 KiB
Go
package runtime
|
|
|
|
import (
|
|
"testing"
|
|
|
|
core "agent-system-core"
|
|
)
|
|
|
|
func effOf(kv map[string]any) core.EffectiveSettings { return core.EffectiveSettings(kv) }
|
|
|
|
// 三层覆盖链解析：defaults → user → project 由 settings 服务合并（server 集成测试覆盖）；
|
|
// 此处验证解析器对「设置层结果 + 画像层」的最后一级覆盖与回落。
|
|
func TestResolveChatModelLayers(t *testing.T) {
|
|
eff := effOf(map[string]any{"models": map[string]any{"chat": map[string]any{
|
|
"model": "model-settings", "baseUrl": "https://s.example", "apiKey": "k0",
|
|
}}})
|
|
prof := core.AgentProfile{}
|
|
|
|
// 仅设置层。
|
|
got := ResolveChatModel(prof, eff)
|
|
if got.Model != "model-settings" || got.Base != "https://s.example" || got.Key != "k0" {
|
|
t.Fatalf("settings layer: %+v", got)
|
|
}
|
|
// 画像覆盖模型（端点沿用设置层）。
|
|
prof.Model.Chat = "model-profile"
|
|
got = ResolveChatModel(prof, eff)
|
|
if got.Model != "model-profile" || got.Base != "https://s.example" {
|
|
t.Fatalf("profile layer should override model only: %+v", got)
|
|
}
|
|
}
|
|
|
|
func TestResolveLightweightFallback(t *testing.T) {
|
|
eff := effOf(map[string]any{"models": map[string]any{
|
|
"chat": map[string]any{"model": "main-m", "baseUrl": "https://main", "apiKey": "k"},
|
|
"lightweight": map[string]any{"model": ""},
|
|
}})
|
|
got := ResolveLightweightModel(core.AgentProfile{}, eff)
|
|
if got.Model != "main-m" || got.Base != "https://main" {
|
|
t.Fatalf("lightweight should fall back to chat: %+v", got)
|
|
}
|
|
// 轻量档只配模型名：模型用轻量、端点回落主档。
|
|
eff["models"].(map[string]any)["lightweight"] = map[string]any{"model": "lite-m"}
|
|
got = ResolveLightweightModel(core.AgentProfile{Model: core.ModelRoute{Lightweight: "prof-lite"}}, eff)
|
|
if got.Model != "prof-lite" || got.Base != "https://main" {
|
|
t.Fatalf("profile lightweight + chat endpoint: %+v", got)
|
|
}
|
|
}
|