mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-23 10:28:06 +00:00
b792a94b84
问题：召回（把 L2/L3 相关记忆注入本轮）此前不可声明、也不受任何 SDK 字段 控制——它只在任务开始时对 f.Input 无条件跑一次。于是 qq_get_message 取回 真实正文后只触发 Prune（裁剪），从不触发召回；而中断通知的 meta 文本反而 会去召回（词不对题，命中一堆泛实体）。 改动： - SDK 新增 RecallPolicy(none|auto) 轴，落在 InjectOptions / ChannelDef / ToolDef 三个声明面，与 ContextPolicy 正交（裁剪 vs 召回）。默认值与 prune 刻意相反：输入/注入默认 auto（保持既有「每条输入都召回」）， 工具默认 none（工具输出多为噪声，按需声明）。 - 内核：recallDeclared 按 注入点 > 通道 > 默认auto 解析；输入侧用它决定 是否注入记忆索引；工具侧 ContextPolicy/RecallPolicy 共用同一份清洗后 query，一次相关性过程分别 prune / recall；召回以 system 消息挂到消息 末尾（同任务内替换而非累加）。 - 管线：proc RPC（inject/register + 校验）、lua 键、io payload 全量透传。 - QQ 插件：中断与 qq 通道声明 RecallPolicy=none（meta 不是内容）； qq_get_message 声明 RecallPolicy=auto（取回正文后据正文召回）。 测试：新增 recallpolicy_test.go（core）与 proc 校验用例； go build ./... 通过，go test ./internal/... 全通过，SDK 模块与 qq 插件测试通过。
65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
package proc
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
|
||
pubsdk "gitcode.com/JianFeeeee/homeagent-sdk/sdk"
|
||
)
|
||
|
||
// handleRegister 处理注册面：工具 / stage / 输出通道 / 插件 API / 入站通道。
|
||
//
|
||
// 本函数体是 corehandler.go 里 Handle 那一个大 switch 的**整块平移**：
|
||
// case 标签与 case 体逐字保留，只换了宿主函数（§3.2 的平移原则）。
|
||
func (h *coreHandler) handleRegister(method string, params json.RawMessage) (interface{}, error) {
|
||
switch method {
|
||
// ---- 注册面（原 case 1/2/3/4/46）----
|
||
case MethodToolRegister:
|
||
return h.toolRegister(params)
|
||
case MethodStageRegister:
|
||
return h.stageRegister(params)
|
||
case MethodOutputRegister:
|
||
return h.outputRegister(params)
|
||
case MethodAPIRegister:
|
||
var p struct {
|
||
Name string `json:"name"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
return nil, h.sdk.RegisterPluginAPI(p.Name)
|
||
case MethodInputRegister:
|
||
var p struct {
|
||
Name string `json:"name"`
|
||
Def pubsdk.ChannelDef `json:"def"`
|
||
HasCleaner bool `json:"has_cleaner"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
if p.Name == "" {
|
||
return nil, fmt.Errorf("input.register: 缺少 name")
|
||
}
|
||
cleaner, err := h.cleanerProxy(CleanerScopeInput, p.Name, p.HasCleaner)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("input.register: %w", err)
|
||
}
|
||
if err := validateContextPolicy("input.register", p.Def.ContextPolicy); err != nil {
|
||
return nil, err
|
||
}
|
||
if err := validateRecallPolicy("input.register", p.Def.RecallPolicy); err != nil {
|
||
return nil, err
|
||
}
|
||
// 整体传 p.Def（只是把函数型的 Cleaner 换成代理），不要手写字段白名单：
|
||
// 白名单会让新增字段静默丢失。
|
||
def := p.Def
|
||
def.Cleaner = cleaner
|
||
return nil, h.sdk.RegisterInputChannel(p.Name, def)
|
||
|
||
}
|
||
|
||
// 组内不应到达：Handle 的分派表与本函数的 case 标签同源，
|
||
// 出现即说明两处不同步。
|
||
return nil, fmt.Errorf("未知 method: %s", method)
|
||
}
|