【capacity 唤醒式核心(77)】internal/runtime/capacity:(Decider L1启发式+L2上下文+人格覆盖→三态静息/唤醒/全能→允许工具集+capacity_awaken事件)。loop 外围允许集过滤(不硬阻断=能力≠边界红线)。config 加 Capacity(Enabled/MinConfidence/ToolSets)。wire buildCapacity 接入 loop。 【L4 记忆驱动意图(80)】recall.go:从记忆召回用户习惯(搞一下xx=办事)→唤醒L1/L2漏掉/升置信;群聊陌生人不读owner习惯(隔离红线);RecordIntent闭环(判定写回记忆越用越准);默认保守静息。 【会话消息编排(78)】gateway/session_orchestrator.go:每会话(模式/pending有界5/tail上下文尾巴20);Submit非阻塞(idle立即/busy入队ack/满拒绝);完成回调按序续接(tail+新消息合并增量注入非全量重放);Interrupt中断;REST 202/409+WS session_busy/pending_ack事件;不重造任务引擎复用loop。 【前端集成(79)】ws.ts 加 capacity_awaken/session_busy/pending_ack 解析;store 加 awakenState/sessionBusy;TopBar/ChatShell 加唤醒态徽标+忙态提示(不阻塞输入框);状态组件 status.tsx;Miuix风格。 【管理面板写CRUD(81)】20个写端点(POST/PUT/DELETE /admin/*,owner+审计落库):personas(真持久化YAML)/users apikey(签发掩码+撤销)/iot控制(真下发)/automations(建/触发/删)/llm(test真调用+临时)/channels(connect/close/身份测试actor白名单)/plugins(卸载)/memories(search真检索)/sessions(export)。无存储如实501不伪造。拆4文件满足≤500红线。前端9模块接真写。 【wire收口】buildCapacity接入loop+OnCapacity事件;server.go newOrchestrator接hub。 实机验证:iot ac.living control 200 真实改变 on:true;observability/metrics 200、health ok;全仓 build/vet/test 47包绿+前端tsc/vite。
98 lines
3.7 KiB
Go
98 lines
3.7 KiB
Go
package loop
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/capability/restrict"
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/capacity"
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/memory"
|
|
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/model"
|
|
)
|
|
|
|
// seedL4Habit writes a canonical intent record memory (the closed-loop write-back
|
|
// format produced by capacity.RecordIntent) recording that "搞一下这个部署" is a task
|
|
// for persona-a.
|
|
func seedL4Habit(t *testing.T, store *memory.Store) {
|
|
t.Helper()
|
|
ctx := context.Background()
|
|
if _, err := store.Add(ctx, memory.MemoryItem{
|
|
PersonaID: "persona-a",
|
|
Scope: memory.ScopeUser,
|
|
Kind: memory.KindRelationship,
|
|
Content: "intent:isTask=true;taskKind=ops;confidence=0.85;text=搞一下这个部署",
|
|
}); err != nil {
|
|
t.Fatalf("seed L4 habit memory: %v", err)
|
|
}
|
|
}
|
|
|
|
// TestLoop_Capacity_L4_WakesOnLearnedHabit_OwnerEnv proves the L4 feed: with a
|
|
// retrieved user habit recording that the phrase is a task, an otherwise-undecided
|
|
// turn wakes in the owner-full env, and the model is advertised the awake tool set.
|
|
func TestLoop_Capacity_L4_WakesOnLearnedHabit_OwnerEnv(t *testing.T) {
|
|
t.Parallel()
|
|
store := newTestMemoryStore(t)
|
|
seedL4Habit(t, store)
|
|
td := capacityToolSet(t)
|
|
ad := &mockAdapter{stream: []mgResp{{resp: model.Response{Content: "好的"}}}}
|
|
a := newTestAgent(t, ad, td)
|
|
dec := capacity.New(capacity.Config{Enabled: true, MinConfidence: 0.6})
|
|
l := New(Options{
|
|
Agent: a,
|
|
MaxRounds: 8,
|
|
OnStream: func(model.StreamEvent) {},
|
|
Capacity: dec,
|
|
Retriever: memory.NewRetriever(store),
|
|
PersonaID: "persona-a",
|
|
})
|
|
ctx := WithSessionEnv(context.Background(), OwnerEnv())
|
|
if _, err := l.Run(ctx, "sess", []model.Message{model.TextMessage(model.RoleUser, "搞一下那个部署")}); err != nil {
|
|
t.Fatalf("Run: %v", err)
|
|
}
|
|
if len(ad.streamReqs) == 0 {
|
|
t.Fatal("no streaming request captured")
|
|
}
|
|
set := toolSet(ad.streamReqs[0].Tools)
|
|
// L4 read the learned habit → awake → web-search is advertised (awake set).
|
|
if !hasAll(set, "terminal", "file", "web-search") {
|
|
t.Fatalf("owner L4 tools = %v, want awake set incl web-search", toolNames(ad.streamReqs[0].Tools))
|
|
}
|
|
}
|
|
|
|
// TestLoop_Capacity_L4_RestrictedEnv_NoOwnerHabitRead proves the L4 isolation red
|
|
// line: in a group/stranger context the loop passes no recall, so the decider falls
|
|
// back to L1/L2 and the owner's habit memory is never used to wake the intent.
|
|
func TestLoop_Capacity_L4_RestrictedEnv_NoOwnerHabitRead(t *testing.T) {
|
|
t.Parallel()
|
|
store := newTestMemoryStore(t)
|
|
seedL4Habit(t, store)
|
|
td := capacityToolSet(t)
|
|
ad := &mockAdapter{stream: []mgResp{{resp: model.Response{Content: "好的"}}}}
|
|
a := newTestAgent(t, ad, td)
|
|
dec := capacity.New(capacity.Config{Enabled: true, MinConfidence: 0.6})
|
|
l := New(Options{
|
|
Agent: a,
|
|
MaxRounds: 8,
|
|
OnStream: func(model.StreamEvent) {},
|
|
Capacity: dec,
|
|
Retriever: memory.NewRetriever(store),
|
|
PersonaID: "persona-a",
|
|
})
|
|
ctx := WithSessionEnv(context.Background(), SessionEnv{ActorEnv: restrict.ActorGroupMember, MemWritable: false})
|
|
if _, err := l.Run(ctx, "sess", []model.Message{model.TextMessage(model.RoleUser, "搞一下那个部署")}); err != nil {
|
|
t.Fatalf("Run: %v", err)
|
|
}
|
|
if len(ad.streamReqs) == 0 {
|
|
t.Fatal("no streaming request captured")
|
|
}
|
|
set := toolSet(ad.streamReqs[0].Tools)
|
|
// Restricted env → L4 silent → undecided turn stays quiet → only quiet tools, no
|
|
// web-search, no owner habit read.
|
|
if !hasAll(set, "terminal", "file") {
|
|
t.Fatalf("restricted tools = %v, want quiet set terminal+file", toolNames(ad.streamReqs[0].Tools))
|
|
}
|
|
if set["web-search"] {
|
|
t.Fatalf("restricted tools = %v, must NOT wake (owner habit not read)", toolNames(ad.streamReqs[0].Tools))
|
|
}
|
|
}
|