mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 06:47:29 +00:00
4e70b8d5d8
批 1，不含 SDK 仓与目录改名。 - go.mod 第 1 行 module 路径 - 366 个 .go 文件的完整前缀 github.com/JianFeeeee/HomeAgent 机械替换 （857 处，仅完整前缀；展示名/注释里的 HomeAgent 字样留给批 5） - 未改：third_party 目录名、SDK 自身 go.mod module、go.mod replace 目标、 homed/waiter 可执行名、.hmap 后缀
52 lines
1.8 KiB
Go
52 lines
1.8 KiB
Go
package core
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
agentAPI "github.com/JianFeeeee/HologramAgent/internal/agent/api"
|
||
)
|
||
|
||
// windowProvider 是只声明窗口的 Provider 桩（预算是纯函数，不需要真推理）。
|
||
type windowProvider struct{ n int }
|
||
|
||
func (windowProvider) Name() string { return "window-stub" }
|
||
func (windowProvider) Chat(context.Context, *agentAPI.CompletionRequest) (*agentAPI.CompletionResponse, error) {
|
||
return nil, nil
|
||
}
|
||
func (windowProvider) ChatStream(context.Context, *agentAPI.CompletionRequest) (<-chan agentAPI.StreamChunk, error) {
|
||
return nil, nil
|
||
}
|
||
func (p windowProvider) MaxContextTokens() int { return p.n }
|
||
|
||
// 回归（2026-09-19）：窗口声明为 1M 后，工作面不能被 80% 带到 838K ——
|
||
// 该源最优区间是 600K，超过就只剩成本与延迟。
|
||
func TestComputeTokenBudgetCapsTargetAtWorkingBand(t *testing.T) {
|
||
const sys = "系统提示词"
|
||
b := ComputeTokenBudget(windowProvider{1048576}, sys)
|
||
|
||
if b.MaxContext != 1048576 {
|
||
t.Errorf("MaxContext 应保留真实窗口：%d", b.MaxContext)
|
||
}
|
||
if b.TargetUsage != 600000 {
|
||
t.Errorf("TargetUsage 应封顶在 600K（最优区间），实际 %d", b.TargetUsage)
|
||
}
|
||
|
||
avail := 600000 - EstimateTokens(sys)
|
||
if b.MemoryTokens != avail/3 {
|
||
t.Errorf("MemoryTokens=%d want %d", b.MemoryTokens, avail/3)
|
||
}
|
||
if b.MemoryTokens+b.ContextTokens != avail {
|
||
t.Errorf("两部分应恰好用完可用预算：%d + %d != %d", b.MemoryTokens, b.ContextTokens, avail)
|
||
}
|
||
}
|
||
|
||
// 封顶只作用于“窗口远大于工作区间”的情形；普通窗口仍按 80% 走，
|
||
// 不能让这条约束把小窗口的预算也一起改掉。
|
||
func TestComputeTokenBudgetSmallWindowUnchanged(t *testing.T) {
|
||
b := ComputeTokenBudget(windowProvider{32768}, "x")
|
||
if b.TargetUsage != 26214 {
|
||
t.Errorf("小窗口应仍取 80%%：got %d want %d", b.TargetUsage, 26214)
|
||
}
|
||
}
|