mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 09:28:14 +00:00
feat(scheduler): M3a+M3b 任务生命周期重构 + 四级优先级抢占
设计依据 docs/zh/input-scheduler-design.md §3–§8、§14。 M3a（行为等价的所有权重构）： - processInput 拆为 prepareInputTask / runTaskSteps / finishInputTask， 帧覆盖 prepare→step…→finish；提交与回执只在 finish 段发生一次， 为安全点挂起做准备（挂起不重复提交） - process() 不再持 a.mu（挂起不能持锁），a.mu 字段随之移除 - TaskFrame 增加任务层现场（Evt/CleanInput/IsInterrupt/StartedAt/Terminal/ Level/SeedMsgs）与 taskTerminal / outcomeSuspended - 新增 task_lifecycle_test.go 5 项：正常恰好一次终态、去重 skipped、 on_input 短路、错误终态、consolidation 路由 M3b（优先级与抢占）： - interceptLoop 重写：只做「收中断 → 定级 → requestPreempt → 必要时取消 LLM」，绝不触碰帧（不变量 I2）；三条降级路径与 interceptCh 兜底退场， 改为统一的 pendingInterrupts - scheduler：pendingInterrupts / suspendPool / 让位信号，nextRef 在三集合上 按统一排序键取值；深度上限 4（canSuspend 在安全点拦下） - 抢占判据 incoming.level > running.level；相等与更低只入队 - 安全点只在 step 之间；执行中的 step（工具 RPC/ONNX/CAS）天然不可抢占； _consolidation_ 整任务视为临界区 - 恢复走 resumeTask：从 frame.Step 继续，不重跑 prepare - D1=A：suspend 把被打断任务的只读前缀交给抢占比它的中断任务（SeedMsgs） - 新增 scheduler_preempt_test.go 5 项：抢占-挂起-恢复（含 R1/R5）、同级更低 不抢占、深度上限、空闲中断不丢、seed 路径不污染标志位 验收：agent 全量 + -race 通过；全仓 build/vet 通过
This commit is contained in:
JianFeeeee
204
internal/agent/core/task_lifecycle_test.go
Normal file
204
internal/agent/core/task_lifecycle_test.go
Normal file
@ -0,0 +1,204 @@
|
||||
package core
|
||||
|
||||
// M3a 验收测试：任务生命周期（prepare → run → finish）与「每任务恰一次终态」。
|
||||
//
|
||||
// 设计依据 docs/zh/input-scheduler-design.md §11.3（X2/X3/X4 的 M3a 形态）：
|
||||
// 帧覆盖全生命周期后，提交（context.Append）与回执（emitResponse）只能在
|
||||
// finish 段发生一次——挂起不会重复提交。
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
agentAPI "gitcode.com/JianFeeeee/HomeAgent/internal/agent/api"
|
||||
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
|
||||
"gitcode.com/JianFeeeee/HomeAgent/internal/events"
|
||||
sdk "gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
|
||||
)
|
||||
|
||||
func newLifecycleAgent(t *testing.T, sp agentAPI.Provider, bus *events.Bus, sh *StageHost) *Agent {
|
||||
t.Helper()
|
||||
return New(AgentConfig{
|
||||
ID: "lifecycle",
|
||||
Provider: sp,
|
||||
ProviderManager: agentAPI.NewProviderManager(),
|
||||
IO: agentIO.NewIOManager(),
|
||||
StageHost: sh,
|
||||
EventBus: bus,
|
||||
})
|
||||
}
|
||||
|
||||
func textEvent(source, content string) (*agentIO.InputEvent, chan *agentIO.OutputEvent) {
|
||||
ch := make(chan *agentIO.OutputEvent, 1)
|
||||
return &agentIO.InputEvent{
|
||||
RequestID: "req-1",
|
||||
Source: source,
|
||||
Type: "text",
|
||||
Payload: map[string]interface{}{"content": content},
|
||||
OutputChannel: source,
|
||||
ResponseCh: ch,
|
||||
}, ch
|
||||
}
|
||||
|
||||
// X3（M3a 形态）：正常任务在 finish 段**恰好**提交一次并回执一次。
|
||||
func TestLifecycle_NormalCommitsOnceAndReplies(t *testing.T) {
|
||||
bus := events.NewBus()
|
||||
var outputs, rawInputs int
|
||||
bus.Subscribe(events.EventAgentOutput, func(*events.Event) { outputs++ })
|
||||
bus.Subscribe(events.EventRawInput, func(*events.Event) { rawInputs++ })
|
||||
|
||||
sp := &scriptProvider{script: []*agentAPI.CompletionResponse{{Content: "答复"}}}
|
||||
a := newLifecycleAgent(t, sp, bus, NewStageHost())
|
||||
|
||||
evt, respCh := textEvent("cli", "你好")
|
||||
if _, out := a.runInputTask(evt, nil); out != outcomeDone {
|
||||
t.Fatalf("runInputTask=%v，期望 outcomeDone", out)
|
||||
}
|
||||
|
||||
select {
|
||||
case r := <-respCh:
|
||||
if got, _ := r.Payload["content"].(string); got != "答复" {
|
||||
t.Fatalf("回执内容=%q，期望 答复", got)
|
||||
}
|
||||
if !r.Done {
|
||||
t.Fatal("回执必须带 Done=true")
|
||||
}
|
||||
default:
|
||||
t.Fatal("同步回执缺失：finish 段必须写 ResponseCh")
|
||||
}
|
||||
|
||||
if outputs != 1 {
|
||||
t.Fatalf("agent_output 事件=%d，期望恰好 1（每任务一次终态）", outputs)
|
||||
}
|
||||
if rawInputs != 1 {
|
||||
t.Fatalf("raw_input 事件=%d，期望 1", rawInputs)
|
||||
}
|
||||
if a.context.Len() != 2 {
|
||||
t.Fatalf("上下文事件=%d，期望 2（输入事件 + 本轮事件）", a.context.Len())
|
||||
}
|
||||
}
|
||||
|
||||
// X2：被去重的输入以 skipped 终态结束——不提交、不回执、不发输出事件。
|
||||
func TestLifecycle_DuplicateSkippedHasTerminal(t *testing.T) {
|
||||
bus := events.NewBus()
|
||||
outputs := 0
|
||||
bus.Subscribe(events.EventAgentOutput, func(*events.Event) { outputs++ })
|
||||
|
||||
sp := &scriptProvider{script: []*agentAPI.CompletionResponse{
|
||||
{Content: "第一次"}, {Content: "第二次"},
|
||||
}}
|
||||
a := newLifecycleAgent(t, sp, bus, NewStageHost())
|
||||
|
||||
e1, _ := textEvent("webui", "同样的消息")
|
||||
if _, out := a.runInputTask(e1, nil); out != outcomeDone {
|
||||
t.Fatalf("首次输入=%v，期望 outcomeDone", out)
|
||||
}
|
||||
after1, outputs1 := a.context.Len(), outputs
|
||||
|
||||
e2, ch2 := textEvent("webui", "同样的消息")
|
||||
if _, out := a.runInputTask(e2, nil); out != outcomeDone {
|
||||
t.Fatalf("去重输入应正常返回（不挂起），实际 %v", out)
|
||||
}
|
||||
if a.context.Len() != after1 {
|
||||
t.Fatalf("去重命中不得提交上下文：%d → %d", after1, a.context.Len())
|
||||
}
|
||||
if len(ch2) != 0 {
|
||||
t.Fatal("去重命中不得回执（原实现静默 return）")
|
||||
}
|
||||
if outputs != outputs1 {
|
||||
t.Fatalf("去重命中不得发输出事件：%d → %d", outputs1, outputs)
|
||||
}
|
||||
}
|
||||
|
||||
// 被 on_input 阶段短路：回执阶段给的响应，且不提交上下文（与原实现一致）。
|
||||
func TestLifecycle_OnInputShortCircuit(t *testing.T) {
|
||||
sh := NewStageHost()
|
||||
reply := "被插件短路"
|
||||
sh.RegisterStage(sdk.StageOnInput, func(ctx *sdk.StageContext) error {
|
||||
ctx.Response = &reply
|
||||
return nil
|
||||
})
|
||||
|
||||
sp := &scriptProvider{} // 不应被调用到
|
||||
a := newLifecycleAgent(t, sp, events.NewBus(), sh)
|
||||
|
||||
evt, respCh := textEvent("cli", "任意")
|
||||
if _, out := a.runInputTask(evt, nil); out != outcomeDone {
|
||||
t.Fatalf("短路任务=%v，期望 outcomeDone", out)
|
||||
}
|
||||
select {
|
||||
case r := <-respCh:
|
||||
if got, _ := r.Payload["content"].(string); got != reply {
|
||||
t.Fatalf("短路响应=%q，期望 %q", got, reply)
|
||||
}
|
||||
default:
|
||||
t.Fatal("短路路径必须回执")
|
||||
}
|
||||
if a.context.Len() != 0 {
|
||||
t.Fatalf("短路路径不得提交上下文，实际 %d 条", a.context.Len())
|
||||
}
|
||||
if len(sp.reqs) != 0 {
|
||||
t.Fatal("短路路径不得调用 LLM")
|
||||
}
|
||||
}
|
||||
|
||||
// 错误路径：以 error 终态结束，回执错误文本，且提交的是**错误事件**（无 turn 事件）。
|
||||
func TestLifecycle_ErrorPathTerminal(t *testing.T) {
|
||||
bus := events.NewBus()
|
||||
outputs := 0
|
||||
bus.Subscribe(events.EventAgentOutput, func(*events.Event) { outputs++ })
|
||||
|
||||
sp := &scriptProvider{err: &agentAPI.ProviderError{StatusCode: 401, Message: "bad key"}}
|
||||
a := newLifecycleAgent(t, sp, bus, NewStageHost())
|
||||
|
||||
evt, respCh := textEvent("cli", "会失败")
|
||||
if _, out := a.runInputTask(evt, nil); out != outcomeFailed {
|
||||
t.Fatalf("runInputTask=%v，期望 outcomeFailed", out)
|
||||
}
|
||||
select {
|
||||
case r := <-respCh:
|
||||
got, _ := r.Payload["content"].(string)
|
||||
if !strings.HasPrefix(got, "处理错误:") {
|
||||
t.Fatalf("错误回执=%q，期望以 处理错误: 开头", got)
|
||||
}
|
||||
default:
|
||||
t.Fatal("错误路径必须回执（否则同步调用方永久挂起）")
|
||||
}
|
||||
if outputs != 1 {
|
||||
t.Fatalf("错误路径的 agent_output 事件=%d，期望 1", outputs)
|
||||
}
|
||||
// 输入事件 + 错误事件 = 2；不得出现带 ToolsUsed 的 turn 事件。
|
||||
if a.context.Len() != 2 {
|
||||
t.Fatalf("错误路径上下文事件=%d，期望 2", a.context.Len())
|
||||
}
|
||||
recent := a.context.Recent(10)
|
||||
last := recent[len(recent)-1]
|
||||
if last.Response == "" {
|
||||
t.Fatal("错误事件必须带 Response")
|
||||
}
|
||||
}
|
||||
|
||||
// _consolidation_ 走记忆整理专用路径：不回执、不提交上下文。
|
||||
func TestLifecycle_ConsolidationRouted(t *testing.T) {
|
||||
bus := events.NewBus()
|
||||
outputs := 0
|
||||
bus.Subscribe(events.EventAgentOutput, func(*events.Event) { outputs++ })
|
||||
|
||||
sp := &scriptProvider{script: []*agentAPI.CompletionResponse{{Content: "整理完毕"}}}
|
||||
a := newLifecycleAgent(t, sp, bus, NewStageHost())
|
||||
|
||||
evt, respCh := textEvent("system", "整理任务")
|
||||
evt.OutputChannel = channelConsolidation
|
||||
if _, out := a.runInputTask(evt, nil); out != outcomeDone {
|
||||
t.Fatalf("consolidation=%v，期望 outcomeDone", out)
|
||||
}
|
||||
if len(respCh) != 0 {
|
||||
t.Fatal("consolidation 路径不得回执")
|
||||
}
|
||||
if outputs != 0 {
|
||||
t.Fatalf("consolidation 路径不得发输出事件，实际 %d", outputs)
|
||||
}
|
||||
if a.context.Len() != 0 {
|
||||
t.Fatalf("consolidation 路径不得写用户上下文，实际 %d", a.context.Len())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user