mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 09:28:14 +00:00
98d67559d1
两处都不是风格差异，而是真的偏离设计语义（其一为回归），
已按“先写判据确认失败、再修”的方式处理，判据保留为回归测试。
1. 空闲时到达的中断永远不会被处理（设计 §5.1 ③ 未落地）
调度器空闲时只阻塞在 select{InputChan, selfInputCh, ctx.Done}，
而 pendingInterrupts 不是 channel——interceptLoop 把中断入队后
没有任何东西唤醒调度器，中断要等“下一条输入”才被看到。
修复：调度器加 wake channel，enqueueInterrupt 非阻塞 signalWake，
空闲分支增加 wake 分支。
2. 临界区内只“不让位”却仍被“取消”（设计 §4.3/§5.2）
requestPreempt 不判临界区，interceptLoop 照常 cancelLLM，
于是正在流式的记忆整理被中断，stepLLM 以 error 提前结束——
整理任务被砍掉一半，而设计要求的是“请求排队等它结束”。
修复：scheduler 增加原子 critical 标志（帧仍只由调度器读写），
runInputTask 在 prepare 后设置、结束（含挂起）时清除；
requestPreempt 在临界区内不 arm、不取消，中断只入队。
- 新增 scheduler_regression_test.go 2 项（先失败后通过）
- 验收：agent 全量 + -race；全仓 vet 通过
102 lines
3.3 KiB
Go
102 lines
3.3 KiB
Go
package core
|
||
|
||
// 回归判据：M3 实现与设计稿 §5.1/§4.3 的两处偏离。
|
||
//
|
||
// 这两条是**先写判据、确认失败、再修**的（修完保留为回归测试）。
|
||
|
||
import (
|
||
"context"
|
||
"sync"
|
||
"sync/atomic"
|
||
"testing"
|
||
"time"
|
||
|
||
agentAPI "gitcode.com/JianFeeeee/HomeAgent/internal/agent/api"
|
||
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
|
||
)
|
||
|
||
// cancelAwareProvider 第一次调用进入后阻塞，直到 ctx 取消；记录是否被取消。
|
||
type cancelAwareProvider struct {
|
||
once sync.Once
|
||
entered chan struct{}
|
||
canceled atomic.Bool
|
||
}
|
||
|
||
func newCancelAwareProvider() *cancelAwareProvider {
|
||
return &cancelAwareProvider{entered: make(chan struct{})}
|
||
}
|
||
|
||
func (p *cancelAwareProvider) Name() string { return "cancel-aware" }
|
||
func (p *cancelAwareProvider) Chat(ctx context.Context, req *agentAPI.CompletionRequest) (*agentAPI.CompletionResponse, error) {
|
||
p.once.Do(func() { close(p.entered) })
|
||
<-ctx.Done()
|
||
p.canceled.Store(true)
|
||
return nil, ctx.Err()
|
||
}
|
||
func (p *cancelAwareProvider) ChatStream(ctx context.Context, req *agentAPI.CompletionRequest) (<-chan agentAPI.StreamChunk, error) {
|
||
return nil, context.Canceled
|
||
}
|
||
func (p *cancelAwareProvider) MaxContextTokens() int { return 8192 }
|
||
|
||
// 设计 §5.1 ③：interruptLoop 定级/决策后必须**唤醒调度器**。
|
||
// 现状：中断只被放进 pendingInterrupts，而调度器空闲时阻塞在 select（只看
|
||
// InputChan/selfInputCh/ctx.Done）——没有任何东西会把它叫醒。
|
||
func TestGap_IdleInterruptIsProcessed(t *testing.T) {
|
||
sp := &countingProvider{}
|
||
a := New(AgentConfig{
|
||
ID: "idle-intr",
|
||
Provider: sp,
|
||
ProviderManager: agentAPI.NewProviderManager(),
|
||
IO: agentIO.NewIOManager(),
|
||
StageHost: NewStageHost(),
|
||
})
|
||
a.Start()
|
||
defer a.Stop()
|
||
|
||
// 完全空闲时投递一条中断（模拟定时器通知/插件提醒）。
|
||
a.io.InjectInterruptText("qq", "cli", "空闲时的通知")
|
||
|
||
deadline := time.Now().Add(3 * time.Second)
|
||
for sp.n.Load() == 0 && time.Now().Before(deadline) {
|
||
time.Sleep(20 * time.Millisecond)
|
||
}
|
||
if sp.n.Load() == 0 {
|
||
snap := a.DumpScheduler()
|
||
t.Fatalf("空闲时到达的中断未被处理：pendingInterrupts=%d（调度器未被唤醒）",
|
||
len(snap.PendingInterrupts))
|
||
}
|
||
}
|
||
|
||
// 设计 §4.3/§5.2：`_consolidation_` 是整任务临界区——抢占请求必须排队等它结束，
|
||
// 而不是取消它。现状：requestPreempt 不判临界区，interceptLoop 照常 cancelLLM，
|
||
// 于是正在流式的记忆整理被中断 → stepLLM 直接以 error 结束（整理丢一半）。
|
||
func TestGap_ConsolidationMustNotBeCancelled(t *testing.T) {
|
||
sp := newCancelAwareProvider()
|
||
a := New(AgentConfig{
|
||
ID: "consol",
|
||
Provider: sp,
|
||
ProviderManager: agentAPI.NewProviderManager(),
|
||
IO: agentIO.NewIOManager(),
|
||
StageHost: NewStageHost(),
|
||
})
|
||
a.Start()
|
||
defer a.Stop()
|
||
|
||
// 走自循环通道发起一次记忆整理。
|
||
a.selfInputCh <- selfInputMsg{text: "合并实体", channel: channelConsolidation}
|
||
|
||
select {
|
||
case <-sp.entered:
|
||
case <-time.After(3 * time.Second):
|
||
t.Fatal("记忆整理未进入 LLM 调用")
|
||
}
|
||
|
||
// L4 中断到达。
|
||
a.io.InjectInterruptText("cli", "cli", "L4 打断")
|
||
|
||
time.Sleep(500 * time.Millisecond)
|
||
if sp.canceled.Load() {
|
||
t.Fatal("记忆整理是临界区，其 LLM 不该被取消（设计 §4.3/§5.2）")
|
||
}
|
||
}
|