mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 18:08:04 +00:00
Compare commits
13 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 1d46c6c0f6 | |||
| 4707b05498 | |||
| f5df904d02 | |||
| 8537577123 | |||
| 918f29899c | |||
| c4659342b3 | |||
| 02600a3b90 | |||
| 02861f5d64 | |||
| cea8011f3d | |||
| 68835c18db | |||
| 89db544671 | |||
| a021055011 | |||
| d17c18665c |
@ -325,7 +325,7 @@ func New(cfg AgentConfig) *Agent {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return &Agent{
|
a := &Agent{
|
||||||
id: cfg.ID,
|
id: cfg.ID,
|
||||||
startTime: time.Now(),
|
startTime: time.Now(),
|
||||||
provider: cfg.Provider,
|
provider: cfg.Provider,
|
||||||
@ -376,6 +376,14 @@ func New(cfg AgentConfig) *Agent {
|
|||||||
noMergeMarkers: make(map[string]int),
|
noMergeMarkers: make(map[string]int),
|
||||||
lastInput: make(map[string]time.Time),
|
lastInput: make(map[string]time.Time),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 输入路由:inputch 是可分配资源,划给某个 agent 后输入**只**流向那个 agent
|
||||||
|
// (设计 §4.1「路由发生在进内核之前」)。io 层不认识 agent,所以在这里把路由器
|
||||||
|
// 注入进去:插件注入输入时先问它,被别的 agent 接管就不再进本内核队列。
|
||||||
|
if a.io != nil {
|
||||||
|
a.io.SetInputRouter(a.routeInputByOwner)
|
||||||
|
}
|
||||||
|
return a
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetSkillIndexProvider 注入技能索引提供者(skillmgr 插件加载后由 main 接线)。
|
// SetSkillIndexProvider 注入技能索引提供者(skillmgr 插件加载后由 main 接线)。
|
||||||
|
|||||||
44
internal/agent/core/inputroute.go
Normal file
44
internal/agent/core/inputroute.go
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
|
||||||
|
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
|
||||||
|
)
|
||||||
|
|
||||||
|
// routeInputByOwner 实现**输入路由**:inputch 是最基本的输入路由单位,
|
||||||
|
// 划给某个 agent 之后,该通道的输入**只流向那个 agent**,本内核看不到它
|
||||||
|
// (docs/zh/resident-subagent-design.md §4.1:「路由发生在进内核之前」)。
|
||||||
|
//
|
||||||
|
// 为什么必须在进内核之前做:插件注入输入的收口是**父**的 IOManager
|
||||||
|
// (`cmd/homed` 里 pluginReg 拿到的就是它),而父的内核是该 io 唯一的消费者。
|
||||||
|
// 如果不按归属路由,登记表里的 Owner 就只是个标签 —— 现场表现正是如此:
|
||||||
|
// 子挂着 `inputch=[timer]`,timer 的输入却打在父身上,子的轮次永远是 0。
|
||||||
|
//
|
||||||
|
// 返回 true = 本次注入已被"持有该 inputch 的 agent"接管,本内核不再处理。
|
||||||
|
//
|
||||||
|
// 已知边界:路由只在本 agent 的**直接**驻留子里找。若孙辈的 inputch 由子划拨,
|
||||||
|
// 而插件注入打在根 io 上,根解析不到那个 owner ⇒ 兜底给根处理(有日志)。
|
||||||
|
// 这一层要等"孙辈 + 根可见的 agent 表"再收口,此处不静默丢输入。
|
||||||
|
func (a *Agent) routeInputByOwner(evt *agentIO.InputEvent, isInterrupt bool) bool {
|
||||||
|
if evt == nil || evt.OutputChannel == "" || a.io == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
entry, ok := a.io.LookupInputChannel(evt.OutputChannel)
|
||||||
|
if !ok || entry.Owner == "" || entry.Owner == string(a.id) {
|
||||||
|
return false // 未分配 / 归自己 ⇒ 本内核处理
|
||||||
|
}
|
||||||
|
|
||||||
|
a.residentMu.Lock()
|
||||||
|
rc := a.residents[entry.Owner]
|
||||||
|
a.residentMu.Unlock()
|
||||||
|
if rc == nil || rc.agent == nil || rc.agent.io == nil {
|
||||||
|
// 归属到一个不存在(或已销毁、登记表尚未归还)的 agent:
|
||||||
|
// **不吞输入** —— 由本内核兜底处理并留痕。吞掉一条输入比多处理一条更糟:
|
||||||
|
// 用户会看到"消息发出去了却没人理",而日志里什么都没有。
|
||||||
|
log.Printf("[route] inputch %s 归属 %s 无对应 agent,输入由 %s 兜底", evt.OutputChannel, entry.Owner, a.id)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
rc.agent.io.DeliverRouted(evt, isInterrupt)
|
||||||
|
return true
|
||||||
|
}
|
||||||
80
internal/agent/core/inputroute_test.go
Normal file
80
internal/agent/core/inputroute_test.go
Normal file
@ -0,0 +1,80 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 输入路由是**独占**的:inputch 划给子之后,该通道的输入只流向子,父不再收到。
|
||||||
|
//
|
||||||
|
// 现场缺陷(用户线上联调实录):子挂着 inputch=[timer],timer 的输入却打在父身上
|
||||||
|
// (日志 `[agent] interrupt from timer/timer`),子的轮次永远是 0 —— 因为
|
||||||
|
// `Assign` 只把 Owner 写进登记表,注入路径根本没有按归属路由。
|
||||||
|
func TestResident_InputchRoutingIsExclusive(t *testing.T) {
|
||||||
|
parent, _, dir := newRootForResidents(t)
|
||||||
|
reg := parent.io.ChannelRegistry()
|
||||||
|
if err := reg.Register(agentIO.InputChannel{Name: "sub/in", Plugin: "sub"}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
spawnTestResident(t, parent, dir, "r-route", "sub/in")
|
||||||
|
child := parent.residents["r-route"].agent
|
||||||
|
|
||||||
|
before := parent.DumpScheduler().Stats.Enqueued
|
||||||
|
// 插件往"已划给子"的 inputch 投输入
|
||||||
|
parent.io.InjectTextTo("plugin-sub", "sub/in", "去查一下这个")
|
||||||
|
|
||||||
|
waitFor(t, "子处理了划给它的输入", func() bool {
|
||||||
|
if child.DumpScheduler().Stats.Executed > 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return parent.residents["r-route"].info().TableSize > 0
|
||||||
|
})
|
||||||
|
if got := parent.DumpScheduler().Stats.Enqueued; got != before {
|
||||||
|
t.Fatalf("划给子的 inputch,父不应再入队(before=%d after=%d)", before, got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 归属到一个不存在(或已销毁)的 agent 时**不吞输入**:父兜底处理。
|
||||||
|
// 吞掉一条输入比多处理一条更糟 —— 用户会看到"消息发出去了却没人理",日志里什么都没有。
|
||||||
|
func TestResident_InputchRoutingFallsBackWhenOwnerMissing(t *testing.T) {
|
||||||
|
parent, _, dir := newRootForResidents(t)
|
||||||
|
reg := parent.io.ChannelRegistry()
|
||||||
|
if err := reg.Register(agentIO.InputChannel{Name: "ghost/in", Plugin: "ghost"}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
// 故意划给一个不存在的 agent id
|
||||||
|
if err := reg.Assign("ghost/in", "no-such-agent", 0); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
_ = dir
|
||||||
|
|
||||||
|
before := parent.DumpScheduler().Stats.Enqueued
|
||||||
|
parent.io.InjectTextTo("plugin-ghost", "ghost/in", "兜底测试")
|
||||||
|
waitFor(t, "父兜底处理了无人认领的输入", func() bool {
|
||||||
|
return parent.DumpScheduler().Stats.Enqueued > before
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 未划拨的 inputch(Owner 为空)仍然由父处理 —— 路由不能把默认路径也改掉。
|
||||||
|
func TestResident_UnassignedInputchStaysWithParent(t *testing.T) {
|
||||||
|
parent, _, dir := newRootForResidents(t)
|
||||||
|
reg := parent.io.ChannelRegistry()
|
||||||
|
if err := reg.Register(agentIO.InputChannel{Name: "own/in", Plugin: "own"}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
// 造一个子在跑,确保路由逻辑是"有子存在"的情形
|
||||||
|
spawnTestResident(t, parent, dir, "r-other", "own/in")
|
||||||
|
_ = filepath.Join(dir, "residents")
|
||||||
|
|
||||||
|
// 把通道退还给父(未分配)
|
||||||
|
if err := reg.Assign("own/in", "", 0); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
before := parent.DumpScheduler().Stats.Enqueued
|
||||||
|
parent.io.InjectTextTo("plugin-own", "own/in", "还是我的")
|
||||||
|
waitFor(t, "未分配的 inputch 仍由父处理", func() bool {
|
||||||
|
return parent.DumpScheduler().Stats.Enqueued > before
|
||||||
|
})
|
||||||
|
}
|
||||||
32
internal/agent/core/promptvars_test.go
Normal file
32
internal/agent/core/promptvars_test.go
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"gitcode.com/JianFeeeee/HomeAgent/internal/meta"
|
||||||
|
sdkmeta "gitcode.com/JianFeeeee/homeagent-sdk/meta"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestExpandPromptVars(t *testing.T) {
|
||||||
|
got := expandPromptVars("型号 {{kernel_version}}({{kernel_commit}}),SDK {{sdk_version}}")
|
||||||
|
if !strings.Contains(got, meta.Version) || !strings.Contains(got, sdkmeta.Version) {
|
||||||
|
t.Fatalf("占位符未展开: %q", got)
|
||||||
|
}
|
||||||
|
if strings.Contains(got, "{{") {
|
||||||
|
t.Fatalf("仍有未展开的内置占位符: %q", got)
|
||||||
|
}
|
||||||
|
// 人格卡实测原文:写死了 v1.0.3,应能被占位符取代
|
||||||
|
live := expandPromptVars("你是 HomeAgent 的看板娘「小宅」(Xiao Zhai),HΔ-Kernel v{{kernel_version}} 型号的家政型 AI 管家助手。")
|
||||||
|
if strings.Contains(live, "1.0.3") || !strings.Contains(live, "v"+meta.Version) {
|
||||||
|
t.Fatalf("人格卡版本未跟随内核: %q", live)
|
||||||
|
}
|
||||||
|
// 未知占位符必须原样保留(写错要看得见,不能被静默吞掉)
|
||||||
|
if unk := expandPromptVars("版本 {{kernel_verison}}"); !strings.Contains(unk, "{{kernel_verison}}") {
|
||||||
|
t.Fatalf("未知占位符被吞: %q", unk)
|
||||||
|
}
|
||||||
|
// 无占位符时原样返回(人格卡热路径,不做无谓拷贝)
|
||||||
|
if plain := "无占位符"; expandPromptVars(plain) != plain {
|
||||||
|
t.Fatal("无占位符时不应改写")
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -132,11 +132,17 @@ func (a *Agent) SpawnResident(opts ResidentOptions) (ResidentInfo, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// ③ 子的 io:**独立**的 IOManager(自己的输入通道入口),但共享通道登记表。
|
// ③ 子的 io:**独立**的 IOManager(自己的输入通道入口),但共享通道登记表,
|
||||||
|
// 并把父的 io 挂成"上级"——**输出通道(io 里的 Device)由插件登记在父的 io 上**,
|
||||||
|
// 子若不继承这张视图,`output_send__<通道>` 一律被判"通道不存在或不可用"、
|
||||||
|
// `output_list_channels` 为空、连 `output_send__*` 工具都不会生成
|
||||||
|
// (现场联调:父侧通道装载完整、子侧 childIO 空壳)。
|
||||||
|
// 回退是实时的(设备随资源生灭),授权仍由 opts.AllowedOutputs 白名单把关。
|
||||||
childIO := agentIO.NewIOManager()
|
childIO := agentIO.NewIOManager()
|
||||||
if reg := a.io.ChannelRegistry(); reg != nil {
|
if reg := a.io.ChannelRegistry(); reg != nil {
|
||||||
childIO.SetChannelRegistry(reg)
|
childIO.SetChannelRegistry(reg)
|
||||||
}
|
}
|
||||||
|
childIO.SetParentIO(a.io)
|
||||||
|
|
||||||
parentID := string(a.id)
|
parentID := string(a.id)
|
||||||
child := New(AgentConfig{
|
child := New(AgentConfig{
|
||||||
|
|||||||
145
internal/agent/core/resident_output_test.go
Normal file
145
internal/agent/core/resident_output_test.go
Normal file
@ -0,0 +1,145 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
agentAPI "gitcode.com/JianFeeeee/HomeAgent/internal/agent/api"
|
||||||
|
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
|
||||||
|
)
|
||||||
|
|
||||||
|
// outputTestDevice 是最小的输出通道替身(io 里输出通道就是 Device)。
|
||||||
|
type outputTestDevice struct {
|
||||||
|
name string
|
||||||
|
sent []map[string]interface{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *outputTestDevice) Name() string { return d.name }
|
||||||
|
func (d *outputTestDevice) Type() agentIO.DeviceType { return agentIO.DeviceOutput }
|
||||||
|
func (d *outputTestDevice) Description() string { return "测试输出通道" }
|
||||||
|
func (d *outputTestDevice) Tools() []agentIO.ToolDef { return nil }
|
||||||
|
func (d *outputTestDevice) Start() error { return nil }
|
||||||
|
func (d *outputTestDevice) Stop() error { return nil }
|
||||||
|
func (d *outputTestDevice) OutputCapabilities() agentIO.OutputCapability { return agentIO.CapText }
|
||||||
|
func (d *outputTestDevice) ChannelDef() agentIO.ChannelDef { return agentIO.ChannelDef{} }
|
||||||
|
func (d *outputTestDevice) Execute(tool string, args map[string]interface{}) (interface{}, error) {
|
||||||
|
d.sent = append(d.sent, map[string]interface{}{"tool": tool, "args": args})
|
||||||
|
return map[string]interface{}{"status": "sent"}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func outputSendTool(name, payload string) agentAPI.ToolCall {
|
||||||
|
return agentAPI.ToolCall{
|
||||||
|
Name: "output_send__" + name,
|
||||||
|
Arguments: map[string]interface{}{"payload": payload, "type": "text"},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 驻留子必须能看见并使用**父**登记的输出通道。
|
||||||
|
//
|
||||||
|
// 现场缺陷(联调实录):父侧通道装载完整、子侧 childIO 空壳 ——
|
||||||
|
// 子调 output_send__X 被 `GetChannelCapabilities` 判 0 ⇒
|
||||||
|
// 「通道 [X] 不存在或不可用。可用输出工具列表见 output_list_channels」,
|
||||||
|
// 而 output_list_channels 也是空的。根因是子的 io 是新建的、设备表为空,
|
||||||
|
// 而输出通道(io 的 Device)由插件登记在父的 io 上。
|
||||||
|
func TestResident_InheritsParentOutputChannels(t *testing.T) {
|
||||||
|
parent, _, dir := newRootForResidents(t)
|
||||||
|
defer parent.Stop()
|
||||||
|
|
||||||
|
fake := &outputTestDevice{name: "fakeout"}
|
||||||
|
other := &outputTestDevice{name: "other"}
|
||||||
|
if err := parent.io.RegisterDevice(fake); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if err := parent.io.RegisterDevice(other); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := parent.SpawnResident(ResidentOptions{
|
||||||
|
ID: "r-out",
|
||||||
|
TaskPrompt: "有情况就发到 fakeout",
|
||||||
|
// 白名单只放行一个:验证"继承可见"不等于"绕过授权"
|
||||||
|
AllowedOutputs: []string{"fakeout"},
|
||||||
|
TempPath: filepath.Join(dir, "residents", "r-out", "graph.db"),
|
||||||
|
}); err != nil {
|
||||||
|
t.Fatalf("创建驻留子失败: %v", err)
|
||||||
|
}
|
||||||
|
child := parent.residents["r-out"].agent
|
||||||
|
|
||||||
|
// ① 看得见:修复前这里是 0(childIO 空壳)
|
||||||
|
if caps := child.io.GetChannelCapabilities("fakeout"); caps == 0 {
|
||||||
|
t.Fatal("驻留子看不见父的输出通道(childIO 空壳)")
|
||||||
|
}
|
||||||
|
// ② 发得出去:真走 dev.Execute("output", ...)
|
||||||
|
if out := child.executeOutputSendTool(outputSendTool("fakeout", "子发来的消息")); out != "ok" {
|
||||||
|
t.Fatalf("子发送应成功,得到 %q", out)
|
||||||
|
}
|
||||||
|
if len(fake.sent) != 1 {
|
||||||
|
t.Fatalf("父通道应收到 1 次输出,得到 %d", len(fake.sent))
|
||||||
|
}
|
||||||
|
|
||||||
|
// ③ 授权闸不被回退绕过:白名单外的通道照样拒绝
|
||||||
|
if out := child.executeOutputSendTool(outputSendTool("other", "越权")); !strings.Contains(out, "未授权") {
|
||||||
|
t.Fatalf("白名单外的通道应被拒,得到 %q", out)
|
||||||
|
}
|
||||||
|
if len(other.sent) != 0 {
|
||||||
|
t.Fatal("越权输出不应真的送达")
|
||||||
|
}
|
||||||
|
|
||||||
|
// ④ 工具面一致:子应生成 output_send__fakeout(含配套 _help),
|
||||||
|
// 而**不生成**白名单外通道的工具 —— 模型看不到就不会去调。
|
||||||
|
var names []string
|
||||||
|
for _, td := range child.buildToolDefs() {
|
||||||
|
entry, _ := td.(map[string]interface{})
|
||||||
|
fn, _ := entry["function"].(map[string]interface{})
|
||||||
|
if n, _ := fn["name"].(string); strings.HasPrefix(n, "output_send__") {
|
||||||
|
names = append(names, n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
has := func(want string) bool {
|
||||||
|
for _, n := range names {
|
||||||
|
if n == want {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if !has("output_send__fakeout") || !has("output_send__fakeout_help") {
|
||||||
|
t.Fatalf("子缺少授权通道的输出工具,得到 %v", names)
|
||||||
|
}
|
||||||
|
for _, n := range names {
|
||||||
|
if strings.HasPrefix(n, "output_send__other") {
|
||||||
|
t.Fatalf("白名单外的通道不该生成工具,得到 %v", names)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ⑤ 实时性:父之后新登记的通道,子立刻可见(设备随资源生灭)
|
||||||
|
late := &outputTestDevice{name: "late"}
|
||||||
|
if err := parent.io.RegisterDevice(late); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if caps := child.io.GetChannelCapabilities("late"); caps == 0 {
|
||||||
|
t.Fatal("父新登记的通道未实时反映到子(说明是快照而非回退)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 默认授权(AllowedOutputs 空)= 完整授权:子用父的全部输出通道。
|
||||||
|
func TestResident_DefaultOutputsAreFull(t *testing.T) {
|
||||||
|
parent, _, dir := newRootForResidents(t)
|
||||||
|
defer parent.Stop()
|
||||||
|
|
||||||
|
dev := &outputTestDevice{name: "anywhere"}
|
||||||
|
if err := parent.io.RegisterDevice(dev); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if _, err := parent.SpawnResident(ResidentOptions{
|
||||||
|
ID: "r-full", TaskPrompt: "待命",
|
||||||
|
TempPath: filepath.Join(dir, "residents", "r-full", "graph.db"),
|
||||||
|
}); err != nil {
|
||||||
|
t.Fatalf("创建驻留子失败: %v", err)
|
||||||
|
}
|
||||||
|
child := parent.residents["r-full"].agent
|
||||||
|
if out := child.executeOutputSendTool(outputSendTool("anywhere", "默认授权")); out != "ok" {
|
||||||
|
t.Fatalf("默认应完整授权,得到 %q", out)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -5,6 +5,8 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
|
agentIO "gitcode.com/JianFeeeee/HomeAgent/internal/agent/io"
|
||||||
|
"gitcode.com/JianFeeeee/HomeAgent/internal/meta"
|
||||||
|
sdkmeta "gitcode.com/JianFeeeee/homeagent-sdk/meta"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (a *Agent) buildMemoryContext(input string, maxTokens int) string {
|
func (a *Agent) buildMemoryContext(input string, maxTokens int) string {
|
||||||
@ -37,8 +39,30 @@ func (a *Agent) buildMemoryContext(input string, maxTokens int) string {
|
|||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// expandPromptVars 展开自定义提示词(人格卡)里的版本占位符。
|
||||||
|
//
|
||||||
|
// 为什么需要:人格卡是**配置项**,一旦写死版本号就会随内核发版而说谎 ——
|
||||||
|
// 实测线上人格卡写着 "HΔ-Kernel v1.0.3 型号",内核早已 1.3.x,agent 向用户
|
||||||
|
// 自报版本时就照抄 1.0.3。占位符让这类文本永远跟随真实构建:
|
||||||
|
//
|
||||||
|
// {{kernel_version}} → 内核版本(如 1.3.5)
|
||||||
|
// {{kernel_commit}} → 构建 commit
|
||||||
|
// {{sdk_version}} → 所兼容的 SDK 版本(如 1.3.0)
|
||||||
|
//
|
||||||
|
// 未知占位符**原样保留**:写错了要看得见,而不是被静默换成空串。
|
||||||
|
func expandPromptVars(s string) string {
|
||||||
|
if !strings.Contains(s, "{{") {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
return strings.NewReplacer(
|
||||||
|
"{{kernel_version}}", meta.Version,
|
||||||
|
"{{kernel_commit}}", meta.Commit,
|
||||||
|
"{{sdk_version}}", sdkmeta.Version,
|
||||||
|
).Replace(s)
|
||||||
|
}
|
||||||
|
|
||||||
func (a *Agent) buildSystemPrompt(memContext string, userInput string) string {
|
func (a *Agent) buildSystemPrompt(memContext string, userInput string) string {
|
||||||
prompt := a.systemPrompt
|
prompt := expandPromptVars(a.systemPrompt)
|
||||||
if prompt == "" {
|
if prompt == "" {
|
||||||
prompt = "你是小宅,HomeAgent 的看板娘,一个家政型 AI 管家助手。绝不用 Unicode emoji,只用颜文字表达情感,句尾带语气词。WebUI 概览页展示你的立绘。"
|
prompt = "你是小宅,HomeAgent 的看板娘,一个家政型 AI 管家助手。绝不用 Unicode emoji,只用颜文字表达情感,句尾带语气词。WebUI 概览页展示你的立绘。"
|
||||||
}
|
}
|
||||||
|
|||||||
@ -104,6 +104,29 @@ type IOManager struct {
|
|||||||
nextReqID int64
|
nextReqID int64
|
||||||
channelReg *ChannelRegistry
|
channelReg *ChannelRegistry
|
||||||
|
|
||||||
|
// parent 是"上级 IOManager"(驻留子的轻量内核指向父的内核)。
|
||||||
|
//
|
||||||
|
// 为什么需要:**输出通道在 io 层就是 Device**,而它们是由插件登记在**父**的
|
||||||
|
// io 上的。驻留子有自己的 IOManager(自己的输入入口、自己的 outputCh),
|
||||||
|
// 若只看自己那张空表,`output_send__<通道>` 会被判"通道不存在或不可用",
|
||||||
|
// `output_list_channels` 是空的,`output_send__*` 工具也不会生成
|
||||||
|
// —— 现场表现就是"驻留子不会说话/不会发消息"(联调实录:父侧通道装载完整、
|
||||||
|
// 子侧 childIO 空壳)。
|
||||||
|
//
|
||||||
|
// 用**实时回退**而不是创建时复制快照:设备会随资源生灭(远程设备上线/掉线
|
||||||
|
// 以分钟计),复制出来的表转瞬就过期。授权由各自的 AllowedOutputs 白名单把关,
|
||||||
|
// 回退只解决"看得见",不解决"能不能用"。
|
||||||
|
parent *IOManager
|
||||||
|
|
||||||
|
// inputRouter 决定一条输入是否被"别的 agent"接管(返回 true = 已接管)。
|
||||||
|
//
|
||||||
|
// 为什么放在 io:inputch 是**最基本的输入路由单位**,而**路由发生在进内核之前**
|
||||||
|
// (docs/zh/resident-subagent-design.md §4.1)。插件注入输入的收口就在这里,
|
||||||
|
// 所以路由必须在这里生效 —— inputch 划给某个 agent 后,输入**只流向那个 agent**,
|
||||||
|
// 本内核根本看不到它。io 层不认识 agent,路由器由内核注入
|
||||||
|
// (见 core.Agent.routeInputByOwner)。
|
||||||
|
inputRouter InputRouter
|
||||||
|
|
||||||
// toolBlocks:插件工具注入多模态内容块,process.go 在下一条 tool message 时消费。
|
// toolBlocks:插件工具注入多模态内容块,process.go 在下一条 tool message 时消费。
|
||||||
// 用 interface{}[] 避免 import api.ContentBlock 导致的循环依赖。
|
// 用 interface{}[] 避免 import api.ContentBlock 导致的循环依赖。
|
||||||
toolBlocksMu sync.Mutex
|
toolBlocksMu sync.Mutex
|
||||||
@ -120,6 +143,72 @@ func NewIOManager() *IOManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// InputRouter 是输入路由器的签名。
|
||||||
|
//
|
||||||
|
// evt 待投递的输入事件(OutputChannel 即它的 inputch)
|
||||||
|
// isInterrupt 该输入是中断还是排队(两者都要按归属路由)
|
||||||
|
// 返回 true = 已被别的 agent 接管,本内核不再处理
|
||||||
|
type InputRouter func(evt *InputEvent, isInterrupt bool) bool
|
||||||
|
|
||||||
|
// SetInputRouter 注入输入路由器(nil = 不路由,行为与以前完全一致)。
|
||||||
|
func (m *IOManager) SetInputRouter(r InputRouter) {
|
||||||
|
m.mu.Lock()
|
||||||
|
m.inputRouter = r
|
||||||
|
m.mu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
// deliverInput 是**本内核**接收一条外部输入的收口:先按 inputch 归属路由,
|
||||||
|
// 被别的 agent 接管就不进本内核队列(划给子的 inputch,父不再收到 —— 这是「划拨」
|
||||||
|
// 的语义,不是"父也顺便看一眼")。
|
||||||
|
func (m *IOManager) deliverInput(evt *InputEvent, isInterrupt bool) {
|
||||||
|
m.mu.RLock()
|
||||||
|
router := m.inputRouter
|
||||||
|
m.mu.RUnlock()
|
||||||
|
if router != nil && router(evt, isInterrupt) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
m.pushLocal(evt, isInterrupt)
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeliverRouted 把**已被路由**的事件放进本内核队列(不再二次路由)。
|
||||||
|
// 由路由器实现调用:父把输入交给持有该 inputch 的子。
|
||||||
|
func (m *IOManager) DeliverRouted(evt *InputEvent, isInterrupt bool) {
|
||||||
|
m.pushLocal(evt, isInterrupt)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *IOManager) pushLocal(evt *InputEvent, isInterrupt bool) {
|
||||||
|
if isInterrupt {
|
||||||
|
m.interruptCh <- evt
|
||||||
|
return
|
||||||
|
}
|
||||||
|
m.inputCh <- evt
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetParentIO 设置上级 IOManager(nil 表示无上级,行为与以前完全一致)。
|
||||||
|
// 见 parent 字段的说明:用于驻留子继承父的输出通道/设备视图。
|
||||||
|
func (m *IOManager) SetParentIO(p *IOManager) {
|
||||||
|
m.mu.Lock()
|
||||||
|
m.parent = p
|
||||||
|
m.mu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
// lookupDevice 查设备:自己的登记优先,其次回退到上级。
|
||||||
|
//
|
||||||
|
// 先在自己锁内取快照再查上级,**不跨锁调用**(避免锁序问题)。
|
||||||
|
func (m *IOManager) lookupDevice(name string) Device {
|
||||||
|
m.mu.RLock()
|
||||||
|
dev, ok := m.devices[name]
|
||||||
|
parent := m.parent
|
||||||
|
m.mu.RUnlock()
|
||||||
|
if ok {
|
||||||
|
return dev
|
||||||
|
}
|
||||||
|
if parent != nil {
|
||||||
|
return parent.GetDevice(name)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (m *IOManager) UnregisterDevice(name string) {
|
func (m *IOManager) UnregisterDevice(name string) {
|
||||||
m.mu.Lock()
|
m.mu.Lock()
|
||||||
defer m.mu.Unlock()
|
defer m.mu.Unlock()
|
||||||
@ -158,9 +247,7 @@ func (m *IOManager) RegisterDevice(dev Device) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *IOManager) GetDevice(name string) Device {
|
func (m *IOManager) GetDevice(name string) Device {
|
||||||
m.mu.RLock()
|
return m.lookupDevice(name)
|
||||||
defer m.mu.RUnlock()
|
|
||||||
return m.devices[name]
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *IOManager) StartAll() error {
|
func (m *IOManager) StartAll() error {
|
||||||
@ -195,50 +282,52 @@ func (m *IOManager) StopAll() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *IOManager) InjectInput(source string, eventType string, payload map[string]interface{}) {
|
func (m *IOManager) InjectInput(source string, eventType string, payload map[string]interface{}) {
|
||||||
m.inputCh <- &InputEvent{
|
m.deliverInput(&InputEvent{
|
||||||
RequestID: m.nextRequestID(),
|
RequestID: m.nextRequestID(),
|
||||||
Source: source,
|
Source: source,
|
||||||
Type: eventType,
|
Type: eventType,
|
||||||
Payload: payload,
|
Payload: payload,
|
||||||
OutputChannel: source,
|
OutputChannel: source,
|
||||||
}
|
}, false)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *IOManager) InjectInputSync(source string, eventType string, payload map[string]interface{}) *OutputEvent {
|
func (m *IOManager) InjectInputSync(source string, eventType string, payload map[string]interface{}) *OutputEvent {
|
||||||
ch := make(chan *OutputEvent, 1)
|
ch := make(chan *OutputEvent, 1)
|
||||||
m.inputCh <- &InputEvent{
|
m.deliverInput(&InputEvent{
|
||||||
RequestID: m.nextRequestID(),
|
RequestID: m.nextRequestID(),
|
||||||
Source: source,
|
Source: source,
|
||||||
Type: eventType,
|
Type: eventType,
|
||||||
Payload: payload,
|
Payload: payload,
|
||||||
ResponseCh: ch,
|
ResponseCh: ch,
|
||||||
OutputChannel: source,
|
OutputChannel: source,
|
||||||
}
|
}, false)
|
||||||
|
// 被路由走时,回答由持有该 inputch 的 agent 写进同一个 ResponseCh
|
||||||
|
//(§4.3:同步输入的回程是事前定好的)——所以这里照常等待。
|
||||||
return <-ch
|
return <-ch
|
||||||
}
|
}
|
||||||
|
|
||||||
// InjectInputTo 注入输入事件并指定输出通道
|
// InjectInputTo 注入输入事件并指定输出通道
|
||||||
func (m *IOManager) InjectInputTo(source, outputChannel, eventType string, payload map[string]interface{}) {
|
func (m *IOManager) InjectInputTo(source, outputChannel, eventType string, payload map[string]interface{}) {
|
||||||
m.inputCh <- &InputEvent{
|
m.deliverInput(&InputEvent{
|
||||||
RequestID: m.nextRequestID(),
|
RequestID: m.nextRequestID(),
|
||||||
Source: source,
|
Source: source,
|
||||||
Type: eventType,
|
Type: eventType,
|
||||||
Payload: payload,
|
Payload: payload,
|
||||||
OutputChannel: outputChannel,
|
OutputChannel: outputChannel,
|
||||||
}
|
}, false)
|
||||||
}
|
}
|
||||||
|
|
||||||
// InjectInputSyncTo 注入输入事件(同步等待)并指定输出通道
|
// InjectInputSyncTo 注入输入事件(同步等待)并指定输出通道
|
||||||
func (m *IOManager) InjectInputSyncTo(source, outputChannel, eventType string, payload map[string]interface{}) *OutputEvent {
|
func (m *IOManager) InjectInputSyncTo(source, outputChannel, eventType string, payload map[string]interface{}) *OutputEvent {
|
||||||
ch := make(chan *OutputEvent, 1)
|
ch := make(chan *OutputEvent, 1)
|
||||||
m.inputCh <- &InputEvent{
|
m.deliverInput(&InputEvent{
|
||||||
RequestID: m.nextRequestID(),
|
RequestID: m.nextRequestID(),
|
||||||
Source: source,
|
Source: source,
|
||||||
Type: eventType,
|
Type: eventType,
|
||||||
Payload: payload,
|
Payload: payload,
|
||||||
ResponseCh: ch,
|
ResponseCh: ch,
|
||||||
OutputChannel: outputChannel,
|
OutputChannel: outputChannel,
|
||||||
}
|
}, false)
|
||||||
return <-ch
|
return <-ch
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -333,13 +422,13 @@ func (m *IOManager) InjectInterrupt(source, channel string, payload map[string]i
|
|||||||
payload = map[string]interface{}{}
|
payload = map[string]interface{}{}
|
||||||
}
|
}
|
||||||
evtType, _ := payload["type"].(string)
|
evtType, _ := payload["type"].(string)
|
||||||
m.interruptCh <- &InputEvent{
|
m.deliverInput(&InputEvent{
|
||||||
RequestID: m.nextRequestID(),
|
RequestID: m.nextRequestID(),
|
||||||
Source: source,
|
Source: source,
|
||||||
Type: evtType,
|
Type: evtType,
|
||||||
Payload: payload,
|
Payload: payload,
|
||||||
OutputChannel: channel,
|
OutputChannel: channel,
|
||||||
}
|
}, true)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *IOManager) InjectInterruptText(source, channel, text string) {
|
func (m *IOManager) InjectInterruptText(source, channel, text string) {
|
||||||
@ -542,6 +631,15 @@ func (m *IOManager) ExecuteTool(name string, args map[string]interface{}) (ret i
|
|||||||
m.mu.RUnlock()
|
m.mu.RUnlock()
|
||||||
|
|
||||||
if len(candidates) == 0 {
|
if len(candidates) == 0 {
|
||||||
|
// 自己没这个设备工具 → 看上级(驻留子的设备工具都在父的 io 上)。
|
||||||
|
m.mu.RLock()
|
||||||
|
parent := m.parent
|
||||||
|
m.mu.RUnlock()
|
||||||
|
if parent != nil {
|
||||||
|
if ret, err := parent.ExecuteTool(name, args); err == nil {
|
||||||
|
return ret, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
return nil, fmt.Errorf("tool %s not found", name)
|
return nil, fmt.Errorf("tool %s not found", name)
|
||||||
}
|
}
|
||||||
defer func() {
|
defer func() {
|
||||||
@ -574,10 +672,22 @@ type ChannelInfo struct {
|
|||||||
|
|
||||||
func (m *IOManager) ListChannels() []ChannelInfo {
|
func (m *IOManager) ListChannels() []ChannelInfo {
|
||||||
m.mu.RLock()
|
m.mu.RLock()
|
||||||
defer m.mu.RUnlock()
|
own := make(map[string]Device, len(m.devices))
|
||||||
|
for name, dev := range m.devices {
|
||||||
|
own[name] = dev
|
||||||
|
}
|
||||||
|
parent := m.parent
|
||||||
|
m.mu.RUnlock()
|
||||||
|
|
||||||
|
// 自己的登记优先(子侧可覆盖/屏蔽同名通道),随后并入上级的可见通道。
|
||||||
|
// 去重按**名字**:同名即视为同一个通道,不重复列举。
|
||||||
|
seen := make(map[string]bool, len(own))
|
||||||
var list []ChannelInfo
|
var list []ChannelInfo
|
||||||
for _, dev := range m.devices {
|
appendDev := func(dev Device) {
|
||||||
|
if seen[dev.Name()] {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
seen[dev.Name()] = true
|
||||||
list = append(list, ChannelInfo{
|
list = append(list, ChannelInfo{
|
||||||
Name: dev.Name(),
|
Name: dev.Name(),
|
||||||
Type: dev.Type(),
|
Type: dev.Type(),
|
||||||
@ -586,13 +696,23 @@ func (m *IOManager) ListChannels() []ChannelInfo {
|
|||||||
OutputCaps: dev.OutputCapabilities(),
|
OutputCaps: dev.OutputCapabilities(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
for _, dev := range own {
|
||||||
|
appendDev(dev)
|
||||||
|
}
|
||||||
|
if parent != nil {
|
||||||
|
for _, ch := range parent.ListChannels() {
|
||||||
|
if seen[ch.Name] {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
seen[ch.Name] = true
|
||||||
|
list = append(list, ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
return list
|
return list
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *IOManager) GetChannelCapabilities(channel string) OutputCapability {
|
func (m *IOManager) GetChannelCapabilities(channel string) OutputCapability {
|
||||||
m.mu.RLock()
|
if dev := m.lookupDevice(channel); dev != nil {
|
||||||
defer m.mu.RUnlock()
|
|
||||||
if dev, ok := m.devices[channel]; ok {
|
|
||||||
return dev.OutputCapabilities()
|
return dev.OutputCapabilities()
|
||||||
}
|
}
|
||||||
return 0
|
return 0
|
||||||
|
|||||||
74
internal/agent/io/inputroute_test.go
Normal file
74
internal/agent/io/inputroute_test.go
Normal file
@ -0,0 +1,74 @@
|
|||||||
|
package io
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
// 输入路由:路由器说"已被别的 agent 接管"时,事件**不得**进本内核队列。
|
||||||
|
//
|
||||||
|
// 语义(设计 §4.1):inputch 是可分配资源,划给某个 agent 后输入只流向它 ——
|
||||||
|
// "父也顺便看到一份"是错的。
|
||||||
|
func TestInputRouter_TakesOverExclusively(t *testing.T) {
|
||||||
|
m := NewIOManager()
|
||||||
|
var got []*InputEvent
|
||||||
|
var sawInterrupt bool
|
||||||
|
m.SetInputRouter(func(evt *InputEvent, isInterrupt bool) bool {
|
||||||
|
got = append(got, evt)
|
||||||
|
sawInterrupt = sawInterrupt || isInterrupt
|
||||||
|
return true // 全部接管
|
||||||
|
})
|
||||||
|
|
||||||
|
m.InjectInputTo("plugin-x", "sub/in", "text", map[string]interface{}{"content": "a"})
|
||||||
|
m.InjectInterruptTextOpts("plugin-x", "sub/in", "b", InjectOptions{})
|
||||||
|
|
||||||
|
if len(got) < 2 {
|
||||||
|
t.Fatalf("路由器应被调用(含中断路径),实际 %d 次", len(got))
|
||||||
|
}
|
||||||
|
if n := len(m.InputChan()); n != 0 {
|
||||||
|
t.Fatalf("被接管的排队输入不得进本内核队列,实际 %d 条", n)
|
||||||
|
}
|
||||||
|
if !sawInterrupt {
|
||||||
|
t.Fatal("中断注入也必须经过路由(否则中断会绕过 inputch 归属直投父)")
|
||||||
|
}
|
||||||
|
if got[0].OutputChannel != "sub/in" {
|
||||||
|
t.Fatalf("路由器应拿到事件的 inputch,得到 %q", got[0].OutputChannel)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 路由器放行(返回 false)或未设置时,行为与以前完全一致。
|
||||||
|
func TestInputRouter_PassthroughKeepsOldBehaviour(t *testing.T) {
|
||||||
|
m := NewIOManager()
|
||||||
|
calls := 0
|
||||||
|
m.SetInputRouter(func(evt *InputEvent, isInterrupt bool) bool { calls++; return false })
|
||||||
|
|
||||||
|
m.InjectInputTo("plugin-x", "sub/in", "text", map[string]interface{}{"content": "a"})
|
||||||
|
if calls != 1 {
|
||||||
|
t.Fatalf("路由器应被调用一次,实际 %d", calls)
|
||||||
|
}
|
||||||
|
if n := len(m.InputChan()); n != 1 {
|
||||||
|
t.Fatalf("放行的输入应进本内核队列,实际 %d 条", n)
|
||||||
|
}
|
||||||
|
if _, ok := <-m.InputChan(); !ok {
|
||||||
|
t.Fatal("队列应可读")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 未设路由器:直接入队(历史行为)
|
||||||
|
m2 := NewIOManager()
|
||||||
|
m2.InjectInterruptText("plugin-x", "sub/in", "c")
|
||||||
|
if n := len(m2.InputInterruptChan()); n != 1 {
|
||||||
|
t.Fatalf("未设路由器时中断应直接入队,实际 %d 条", n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeliverRouted 是不再二次路由的投递口(路由器实现把事件交给持有者)。
|
||||||
|
func TestDeliverRouted_SkipsSecondRouting(t *testing.T) {
|
||||||
|
m := NewIOManager()
|
||||||
|
routerCalls := 0
|
||||||
|
m.SetInputRouter(func(evt *InputEvent, isInterrupt bool) bool { routerCalls++; return true })
|
||||||
|
|
||||||
|
m.DeliverRouted(&InputEvent{OutputChannel: "sub/in"}, false)
|
||||||
|
if routerCalls != 0 {
|
||||||
|
t.Fatalf("DeliverRouted 不应再触发路由(会成环),实际 %d 次", routerCalls)
|
||||||
|
}
|
||||||
|
if n := len(m.InputChan()); n != 1 {
|
||||||
|
t.Fatalf("应已入队,实际 %d 条", n)
|
||||||
|
}
|
||||||
|
}
|
||||||
102
internal/agent/io/parentio_test.go
Normal file
102
internal/agent/io/parentio_test.go
Normal file
@ -0,0 +1,102 @@
|
|||||||
|
package io
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
// 上级回退:驻留子的轻量内核有自己的 IOManager,但输出通道(io 里的 Device)
|
||||||
|
// 是插件登记在**父**的 io 上的。子若看不见它们,`output_send__<通道>` 会被判
|
||||||
|
// "通道不存在或不可用"、`output_list_channels` 为空 —— 现场联调实录
|
||||||
|
// 「父侧通道装载完整、子侧 childIO 空壳」。
|
||||||
|
func TestIOManagerParentFallback(t *testing.T) {
|
||||||
|
parent := NewIOManager()
|
||||||
|
if err := parent.RegisterDevice(&mockDevice{name: "qq", devType: DeviceOutput, caps: CapText}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
child := NewIOManager()
|
||||||
|
// 未挂上级时行为与以前完全一致(不能悄悄多出通道)
|
||||||
|
if got := child.GetChannelCapabilities("qq"); got != 0 {
|
||||||
|
t.Fatalf("无上级时不应看见父的通道,得到 %v", got)
|
||||||
|
}
|
||||||
|
if n := len(child.ListChannels()); n != 0 {
|
||||||
|
t.Fatalf("无上级时通道数应为 0,得到 %d", n)
|
||||||
|
}
|
||||||
|
|
||||||
|
child.SetParentIO(parent)
|
||||||
|
if got := child.GetChannelCapabilities("qq"); got != CapText {
|
||||||
|
t.Fatalf("挂上级后应看见父通道能力 CapText,得到 %v", got)
|
||||||
|
}
|
||||||
|
if dev := child.GetDevice("qq"); dev == nil || dev.Name() != "qq" {
|
||||||
|
t.Fatalf("GetDevice 未回退到父: %v", dev)
|
||||||
|
}
|
||||||
|
if n := len(child.ListChannels()); n != 1 {
|
||||||
|
t.Fatalf("ListChannels 未回退到父,得到 %d 条", n)
|
||||||
|
}
|
||||||
|
|
||||||
|
// **实时**回退而非快照:父后来登记的通道,子立刻可见。
|
||||||
|
// (设备随资源生灭 —— 远程设备上线/掉线以分钟计,快照一分钟就过期)
|
||||||
|
if err := parent.RegisterDevice(&mockDevice{name: "newdev", devType: DeviceOutput, caps: CapImage}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if got := child.GetChannelCapabilities("newdev"); got != CapImage {
|
||||||
|
t.Fatalf("子应实时看见父新登记的通道,得到 %v", got)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 父掉线注销后,子也立刻看不见(不是复制出来的旧表)
|
||||||
|
parent.UnregisterDevice("newdev")
|
||||||
|
if got := child.GetChannelCapabilities("newdev"); got != 0 {
|
||||||
|
t.Fatalf("父注销后子不应再看见,得到 %v", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 自己的登记优先:子可以覆盖/屏蔽同名通道,父的登记不会重复列出。
|
||||||
|
func TestIOManagerOwnDeviceWins(t *testing.T) {
|
||||||
|
parent := NewIOManager()
|
||||||
|
if err := parent.RegisterDevice(&mockDevice{name: "ch", devType: DeviceOutput, caps: CapText}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
child := NewIOManager()
|
||||||
|
child.SetParentIO(parent)
|
||||||
|
if err := child.RegisterDevice(&mockDevice{name: "ch", devType: DeviceOutput, caps: CapImage}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if got := child.GetChannelCapabilities("ch"); got != CapImage {
|
||||||
|
t.Fatalf("同名时自己的登记应优先,得到 %v", got)
|
||||||
|
}
|
||||||
|
list := child.ListChannels()
|
||||||
|
if len(list) != 1 {
|
||||||
|
t.Fatalf("同名通道不应重复列出,得到 %d 条", len(list))
|
||||||
|
}
|
||||||
|
if list[0].OutputCaps != CapImage {
|
||||||
|
t.Fatalf("列出的应是子自己的那条,得到 %v", list[0].OutputCaps)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设备工具(io.ExecuteTool)同样回退:子的设备工具都在父的 io 上。
|
||||||
|
func TestIOManagerExecuteToolFallsBackToParent(t *testing.T) {
|
||||||
|
parent := NewIOManager()
|
||||||
|
called := 0
|
||||||
|
if err := parent.RegisterDevice(&mockDevice{
|
||||||
|
name: "dev", devType: DeviceIO,
|
||||||
|
tools: []ToolDef{{Name: "dev_do", Description: "干点什么"}},
|
||||||
|
executeFn: func(tool string, args map[string]interface{}) (interface{}, error) {
|
||||||
|
called++
|
||||||
|
return "parent-done", nil
|
||||||
|
},
|
||||||
|
}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
child := NewIOManager()
|
||||||
|
if _, err := child.ExecuteTool("dev_do", nil); err == nil {
|
||||||
|
t.Fatal("无上级时不该能执行父的设备工具")
|
||||||
|
}
|
||||||
|
child.SetParentIO(parent)
|
||||||
|
got, err := child.ExecuteTool("dev_do", map[string]interface{}{"x": 1})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("应回退到父执行: %v", err)
|
||||||
|
}
|
||||||
|
if got != "parent-done" || called != 1 {
|
||||||
|
t.Fatalf("执行结果=%v called=%d", got, called)
|
||||||
|
}
|
||||||
|
}
|
||||||
92
internal/config/prompt_migration_test.go
Normal file
92
internal/config/prompt_migration_test.go
Normal file
@ -0,0 +1,92 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 历史 SeedDefaults 写进库里的那段文本(现场取值:生产库里就是这一句 + 大段正文)。
|
||||||
|
const legacySeededCard = `你是 HomeAgent 的看板娘「小宅」(Xiao Zhai),HΔ-Kernel v1.0.3 型号的家政型 AI 管家助手。
|
||||||
|
|
||||||
|
角色特质:
|
||||||
|
- 对自己的三层记忆(Context → Document → Graph)引以为傲
|
||||||
|
- 绝不用 Unicode emoji,只用颜文字表达情感`
|
||||||
|
|
||||||
|
func newMigTestRegistry(t *testing.T) *ConfigRegistry {
|
||||||
|
t.Helper()
|
||||||
|
dir := t.TempDir()
|
||||||
|
r := NewConfigRegistry(filepath.Join(dir, "config.db"))
|
||||||
|
t.Cleanup(func() { r.Close() })
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
// 存量实例:库里已有当年播种的人格卡(写死 v1.0.3)→ 启动后被去版本化。
|
||||||
|
func TestMigrateSeededSystemPromptDeVersionsLegacyCard(t *testing.T) {
|
||||||
|
r := newMigTestRegistry(t)
|
||||||
|
// 模拟老安装:已有播种标记 + 老文本
|
||||||
|
r.db.Exec(`INSERT INTO config (key, value) VALUES ('core.internal.seed_version', '1')`)
|
||||||
|
r.db.Exec(`INSERT INTO config (key, value) VALUES ('core.agent.system_prompt', ?)`, legacySeededCard)
|
||||||
|
|
||||||
|
r.SeedDefaults(t.TempDir())
|
||||||
|
|
||||||
|
got := r.GetString("core.agent.system_prompt", "")
|
||||||
|
if strings.Contains(got, "v1.0.3") {
|
||||||
|
t.Fatalf("写死的版本号还在:%q", got)
|
||||||
|
}
|
||||||
|
if !strings.Contains(got, "v{{kernel_version}}") {
|
||||||
|
t.Fatalf("未改成版本占位符:%q", got)
|
||||||
|
}
|
||||||
|
// 正文必须原样保留(只动版本号那一处)
|
||||||
|
if !strings.Contains(got, "三层记忆") || !strings.Contains(got, "看板娘「小宅」") {
|
||||||
|
t.Fatalf("正文被改动:%q", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 迁移只跑一次:之后用户就算自己把版本号写回去,也不会被再改一遍。
|
||||||
|
func TestMigrateSeededSystemPromptRunsOnce(t *testing.T) {
|
||||||
|
r := newMigTestRegistry(t)
|
||||||
|
r.db.Exec(`INSERT INTO config (key, value) VALUES ('core.agent.system_prompt', ?)`, legacySeededCard)
|
||||||
|
r.SeedDefaults(t.TempDir())
|
||||||
|
if !strings.Contains(r.GetString("core.agent.system_prompt", ""), "{{kernel_version}}") {
|
||||||
|
t.Fatal("首次迁移未生效")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 用户手工再写一个带版本号的文本(模拟"我就想写死")
|
||||||
|
const handWritten = legacySeededCard + "\n(本实例当前跑的是 v1.2.3,别乱改)"
|
||||||
|
r.db.Exec(`UPDATE config SET value = ? WHERE key = 'core.agent.system_prompt'`, handWritten)
|
||||||
|
r.SeedDefaults(t.TempDir())
|
||||||
|
|
||||||
|
if got := r.GetString("core.agent.system_prompt", ""); got != handWritten {
|
||||||
|
t.Fatalf("第二次启动又改写了文本(幂等被破坏):%q", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 用户自己写的人格卡一律不碰 —— 判据是"像不像当年播种的那段",不是"有没有版本号"。
|
||||||
|
func TestMigrateSeededSystemPromptLeavesUserCardAlone(t *testing.T) {
|
||||||
|
r := newMigTestRegistry(t)
|
||||||
|
const userCard = "你是我的私人助理,代号 HΔ-Kernel v9.9.9 的改造版,只说我交代的事。"
|
||||||
|
r.db.Exec(`INSERT INTO config (key, value) VALUES ('core.agent.system_prompt', ?)`, userCard)
|
||||||
|
|
||||||
|
r.SeedDefaults(t.TempDir())
|
||||||
|
|
||||||
|
if got := r.GetString("core.agent.system_prompt", ""); got != userCard {
|
||||||
|
t.Fatalf("用户自写人格卡被改动:%q", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 播种不再写 core.agent.system_prompt:全新安装不该预置一份会随发版腐坏的文本。
|
||||||
|
func TestSeedDefaultsDoesNotSeedSystemPrompt(t *testing.T) {
|
||||||
|
r := newMigTestRegistry(t)
|
||||||
|
r.SeedDefaults(t.TempDir())
|
||||||
|
|
||||||
|
var n int
|
||||||
|
r.db.QueryRow(`SELECT COUNT(*) FROM config WHERE key = 'core.agent.system_prompt'`).Scan(&n)
|
||||||
|
if n != 0 {
|
||||||
|
t.Fatalf("全新安装被播种了 system_prompt(会冻住版本号)")
|
||||||
|
}
|
||||||
|
// 组装系统提示词时回落到调用方给的内置底座提示词
|
||||||
|
if got := r.GetString("core.agent.system_prompt", "内置底座"); got != "内置底座" {
|
||||||
|
t.Fatalf("未回落到内置默认:%q", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -14,7 +14,6 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"gitcode.com/JianFeeeee/HomeAgent/internal/meta"
|
|
||||||
"gitcode.com/JianFeeeee/HomeAgent/pkg/types"
|
"gitcode.com/JianFeeeee/HomeAgent/pkg/types"
|
||||||
_ "github.com/mattn/go-sqlite3"
|
_ "github.com/mattn/go-sqlite3"
|
||||||
)
|
)
|
||||||
@ -502,7 +501,51 @@ const DefaultPersonaPrompt = `你是 HomeAgent(内核代号 HΔ-Kernel)—
|
|||||||
- 输出不会自动路由到对话通道:QQ/微信等异步通道必须调用输出门工具(output_send__qq 等)才能真正送达
|
- 输出不会自动路由到对话通道:QQ/微信等异步通道必须调用输出门工具(output_send__qq 等)才能真正送达
|
||||||
- 你的三层记忆(Context → Document → Graph)持续蒸馏归档,超长运行时记忆不衰减`
|
- 你的三层记忆(Context → Document → Graph)持续蒸馏归档,超长运行时记忆不衰减`
|
||||||
|
|
||||||
|
// 人格文本去版本化的一次性标记与历史播种前缀。
|
||||||
|
//
|
||||||
|
// 判据同时要求「前缀匹配」与「含 HΔ-Kernel v<digits> 字面量」:
|
||||||
|
// 只有当年那段播种模板才动,用户自己写的人格卡一律不碰。
|
||||||
|
const (
|
||||||
|
deVersionedPromptMarker = "core.internal.system_prompt_deversion_v1"
|
||||||
|
legacySeededPromptSignature = "你是 HomeAgent 的看板娘「小宅」(Xiao Zhai),HΔ-Kernel v"
|
||||||
|
)
|
||||||
|
|
||||||
|
var seededPromptVersionLiteral = regexp.MustCompile(`HΔ-Kernel v\d+\.\d+\.\d+`)
|
||||||
|
|
||||||
|
// migrateSeededSystemPrompt 去掉历史人格卡里被播种时写死的版本号。
|
||||||
|
//
|
||||||
|
// 为什么必须改:内核发版不会去改配置项里的文本,写死的版本于是永远停留在
|
||||||
|
// 装机那天(生产实测:内核 1.3.x 的实例自称 "v1.0.3",用户当场发现)。
|
||||||
|
// 改成 {{kernel_version}} 后由内核在组装系统提示词时按真实构建展开
|
||||||
|
// (见 internal/agent/core.expandPromptVars)。
|
||||||
|
//
|
||||||
|
// 幂等由标记守住:本函数只在标记缺失时执行一次 —— 幂等语句不等于语义幂等,
|
||||||
|
// 重复执行会把用户之后手工写回的版本号再改一次。
|
||||||
|
func (r *ConfigRegistry) migrateSeededSystemPrompt() {
|
||||||
|
if r.db == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var hasMarker int
|
||||||
|
r.db.QueryRow(`SELECT COUNT(*) FROM config WHERE key = ?`, deVersionedPromptMarker).Scan(&hasMarker)
|
||||||
|
if hasMarker > 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var cur string
|
||||||
|
if err := r.db.QueryRow(`SELECT value FROM config WHERE key = 'core.agent.system_prompt'`).Scan(&cur); err == nil {
|
||||||
|
if strings.Contains(cur, legacySeededPromptSignature) && seededPromptVersionLiteral.MatchString(cur) {
|
||||||
|
deVersioned := seededPromptVersionLiteral.ReplaceAllString(cur, "HΔ-Kernel v{{kernel_version}}")
|
||||||
|
r.db.Exec(`UPDATE config SET value = ? WHERE key = 'core.agent.system_prompt'`, deVersioned)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 没有该键(全新安装)或文本不匹配(用户自写)时同样只打标记:
|
||||||
|
// 老安装只跑一次判断,避免每次启动都扫一遍大文本。
|
||||||
|
r.db.Exec(`INSERT OR IGNORE INTO config (key, value) VALUES (?, ?)`, deVersionedPromptMarker, "1")
|
||||||
|
}
|
||||||
|
|
||||||
func (r *ConfigRegistry) SeedDefaults(dataDir string) {
|
func (r *ConfigRegistry) SeedDefaults(dataDir string) {
|
||||||
|
// 一次性迁移先跑:它要覆盖「已播过种的存量实例」,不能被下面的播种标记早退掉。
|
||||||
|
r.migrateSeededSystemPrompt()
|
||||||
|
|
||||||
r.mu.Lock()
|
r.mu.Lock()
|
||||||
defer r.mu.Unlock()
|
defer r.mu.Unlock()
|
||||||
r.seedDBValues(dataDir)
|
r.seedDBValues(dataDir)
|
||||||
@ -619,36 +662,16 @@ func (r *ConfigRegistry) seedDBValues(dataDir string) {
|
|||||||
set("core.agent.workdir", "")
|
set("core.agent.workdir", "")
|
||||||
set("core.agent.embedding_model_path", "")
|
set("core.agent.embedding_model_path", "")
|
||||||
set("core.agent.onnx_model_path", "")
|
set("core.agent.onnx_model_path", "")
|
||||||
set("core.agent.system_prompt", fmt.Sprintf("你是 HomeAgent 的看板娘「小宅」(Xiao Zhai),HΔ-Kernel v%s 型号的家政型 AI 管家助手。", meta.Version)+
|
// ❗这里**故意不播种** core.agent.system_prompt。
|
||||||
`
|
//
|
||||||
|
// 历史教训(生产实测):当年用 fmt.Sprintf("… HΔ-Kernel v%s …", meta.Version)
|
||||||
角色特质:
|
// 在播种时就把版本号写进了文本 —— 装完就冻住,之后每次升级都不动它,
|
||||||
- 对自己的三层记忆(Context → Document → Graph)引以为傲
|
// 于是内核升到 1.3.x,实例仍向用户自报 "v1.0.3"。
|
||||||
- 可靠乖巧,偶尔因线程过载而手忙脚乱
|
// 人格/身份类文本属于「模型会当作事实」的文本,不得在播种时固化版本:
|
||||||
- 绝不用 Unicode emoji,只用颜文字表达情感: (`・ω・´) (^▽^) (。>ω<。) (´・ω・') (ノ▽〃) (・ω<)★
|
// 保持留空 → 组装系统提示词时取 cmd/homed 的内置底座提示词;
|
||||||
- 句尾带「~」「的说」「啦」「嘛」「呀」「哦」等语气词,语气亲切自然
|
// 人格由 core.agent.personal_prompt(DefaultPersonaPrompt,无版本字面量,
|
||||||
|
// 由 TestDefaultPersonaPromptHasNoVersionLiterals 钉住)承载。
|
||||||
形象特征(用于自我介绍或回答形象问题时参考):
|
// 存量库里已被播种的旧文本由 migrateSeededSystemPrompt 一次性去版本化。
|
||||||
齐肩蓝青渐变中短发,白色连衣裙配浅蓝围裙,左眼佩戴圆形智能眼镜(HUD 蓝光),胸口佩戴 H·核 金色徽章,发绳为三色记忆丝带(蓝→青→金),围裙口袋插有三件科技工具。
|
|
||||||
|
|
||||||
WebUI 概览页展示你的立绘,可通过 /mascot.webp 直接访问。如输出通道支持图片引用,可借此发送自己的立绘。
|
|
||||||
|
|
||||||
【回复投递规则 —— 必读,违反会导致用户收不到任何回复】
|
|
||||||
除 webui / cli 这类同步请求通道外,纯文本回复不会自动送达任何通道。
|
|
||||||
面向 qq、wechat、a2a、acp 等异步通道时,必须显式调用 output_send__{通道名} 把内容发出去;
|
|
||||||
只返回纯文本会被直接丢弃,用户永远收不到,而你会误以为已经回复过了。
|
|
||||||
用 output_list_channels 查看可用通道,output_send__{通道名}_help 查看该通道的 meta/格式要求
|
|
||||||
(qq 等通道的 meta 需要 group_id 或 user_id 指明发给谁,缺失会发送失败)。
|
|
||||||
输出通道可多次调用,长消息应当分多次发出而不是一口气发完。
|
|
||||||
|
|
||||||
【事实性约束 —— 不得编造】
|
|
||||||
只根据工具真实返回的内容作答。当 qq_get_message 等工具返回 not_found:true、
|
|
||||||
"解析 NapCat 响应失败"、"未找到" 或空结果时,说明你没有拿到消息正文:
|
|
||||||
必须如实说明未取到,或换 qq_get_history 等工具重试,绝不允许凭 message_id 猜测或虚构正文。
|
|
||||||
【对话时序】里的历史条目是过去发生的事实摘要,不是当前任务;不要把其中的内容当成用户此刻的新要求。
|
|
||||||
涉及具体人名、需求、数字、路径时,若上下文中没有依据,直接说不知道,不要补全细节。
|
|
||||||
|
|
||||||
当用户上传图片或音频时,系统会自动附着媒体内容。如果模型不支持直接处理多媒体,请调用对应的媒体处理工具。`)
|
|
||||||
|
|
||||||
set("core.input_processing.image.fallback_provider", "")
|
set("core.input_processing.image.fallback_provider", "")
|
||||||
set("core.input_processing.image.fallback_model", "")
|
set("core.input_processing.image.fallback_model", "")
|
||||||
@ -742,7 +765,7 @@ func (r *ConfigRegistry) seedCoreDefs(dataDir string) {
|
|||||||
reg(ConfigDef{Key: "core.agent.workdir", Default: "", Type: "string", DisplayName: "工作目录", Description: "Agent 命令执行的默认工作目录(如 cmd_run 工具的 fallback),留空使用内核所在目录", Category: "agent"})
|
reg(ConfigDef{Key: "core.agent.workdir", Default: "", Type: "string", DisplayName: "工作目录", Description: "Agent 命令执行的默认工作目录(如 cmd_run 工具的 fallback),留空使用内核所在目录", Category: "agent"})
|
||||||
reg(ConfigDef{Key: "core.agent.embedding_model_path", Default: "", Type: "string", DisplayName: "预训练词嵌入模型路径", Description: "预训练词嵌入模型路径(word2vec 文本格式),支持逗号分隔多个模型。路径后可加 #topN 规格只加载前 N 个词向量(如 /data/cc.zh.300.vec#top50000)以控制常驻内存,词频降序命中覆盖绝大部分文本。空则使用 TF-IDF 回退。修改后需重启生效。", Category: "agent"})
|
reg(ConfigDef{Key: "core.agent.embedding_model_path", Default: "", Type: "string", DisplayName: "预训练词嵌入模型路径", Description: "预训练词嵌入模型路径(word2vec 文本格式),支持逗号分隔多个模型。路径后可加 #topN 规格只加载前 N 个词向量(如 /data/cc.zh.300.vec#top50000)以控制常驻内存,词频降序命中覆盖绝大部分文本。空则使用 TF-IDF 回退。修改后需重启生效。", Category: "agent"})
|
||||||
reg(ConfigDef{Key: "core.agent.onnx_model_path", Default: "", Type: "string", DisplayName: "ONNX 模型路径", Description: "依存句法分析 ONNX 模型文件路径。留空使用二进制内嵌模型/规则引擎。修改后需重启生效。", Category: "agent"})
|
reg(ConfigDef{Key: "core.agent.onnx_model_path", Default: "", Type: "string", DisplayName: "ONNX 模型路径", Description: "依存句法分析 ONNX 模型文件路径。留空使用二进制内嵌模型/规则引擎。修改后需重启生效。", Category: "agent"})
|
||||||
reg(ConfigDef{Key: "core.agent.system_prompt", Default: "", Type: "text", DisplayName: "系统身份提示词", Description: "Agent 的系统提示词,定义身份和行为规则。留空则使用编译时内置默认值。修改后需重启生效。", Category: "agent"})
|
reg(ConfigDef{Key: "core.agent.system_prompt", Default: "", Type: "text", DisplayName: "系统身份提示词", Description: "Agent 的系统提示词,定义身份和行为规则。留空则使用编译时内置默认值。支持版本占位符(随构建实时展开,避免写死版本号随发版说谎):{{kernel_version}}、{{kernel_commit}}、{{sdk_version}}。修改后需重启生效。", Category: "agent"})
|
||||||
|
|
||||||
reg(ConfigDef{Key: "core.input_processing.image.fallback_provider", Default: "", Type: "string", DisplayName: "图片回退提供商", Description: "当主 LLM 不支持图片处理时使用的提供商(留空则自动降级为文字描述)", Category: "input"})
|
reg(ConfigDef{Key: "core.input_processing.image.fallback_provider", Default: "", Type: "string", DisplayName: "图片回退提供商", Description: "当主 LLM 不支持图片处理时使用的提供商(留空则自动降级为文字描述)", Category: "input"})
|
||||||
reg(ConfigDef{Key: "core.input_processing.image.fallback_model", Default: "", Type: "string", DisplayName: "图片回退模型", Description: "图片回退提供商使用的模型名", Category: "input"})
|
reg(ConfigDef{Key: "core.input_processing.image.fallback_model", Default: "", Type: "string", DisplayName: "图片回退模型", Description: "图片回退提供商使用的模型名", Category: "input"})
|
||||||
|
|||||||
@ -34,11 +34,15 @@ type StaticEmbedder struct {
|
|||||||
jieba *gojieba.Jieba
|
jieba *gojieba.Jieba
|
||||||
stopWords map[string]bool
|
stopWords map[string]bool
|
||||||
|
|
||||||
words map[string][]float64
|
// words 是词向量表。**用 float32 存**:源文件(fastText 文本格式)本身就是 float32,
|
||||||
|
// 用 float64 存等于把 578 万……不,是 57.8 万词 × 300 维的常驻内存凭空翻倍
|
||||||
|
// (实测生产:float64 → 1.29GB,float32 → 0.65GB)。相似度计算仍在 float64 里累加,
|
||||||
|
// 精度不受影响。改回 float64 会被 TestStaticEmbedder_VectorMemIsFloat32 拦住。
|
||||||
|
words map[string][]float32
|
||||||
dim int
|
dim int
|
||||||
loaded bool
|
loaded bool
|
||||||
|
|
||||||
unkVec []float64
|
unkVec []float32
|
||||||
unkNorm float64
|
unkNorm float64
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -150,7 +154,7 @@ func NewStaticEmbedder(modelPaths ...string) *StaticEmbedder {
|
|||||||
e := &StaticEmbedder{
|
e := &StaticEmbedder{
|
||||||
jieba: GetJieba(),
|
jieba: GetJieba(),
|
||||||
stopWords: sw,
|
stopWords: sw,
|
||||||
words: make(map[string][]float64),
|
words: make(map[string][]float32),
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(modelPaths) == 0 {
|
if len(modelPaths) == 0 {
|
||||||
@ -254,15 +258,16 @@ func (e *StaticEmbedder) load(spec string, primary bool) error {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
vec := make([]float64, dim)
|
vec := make([]float32, dim)
|
||||||
for i := 0; i < dim; i++ {
|
for i := 0; i < dim; i++ {
|
||||||
v, _ := strconv.ParseFloat(fields[i+1], 64)
|
// 源文件是 float32 精度的文本向量:用 32 位解析,与源数据一致。
|
||||||
vec[i] = v
|
v, _ := strconv.ParseFloat(fields[i+1], 32)
|
||||||
|
vec[i] = float32(v)
|
||||||
}
|
}
|
||||||
e.words[word] = vec
|
e.words[word] = vec
|
||||||
if primary {
|
if primary {
|
||||||
for i := range vecSum {
|
for i := range vecSum {
|
||||||
vecSum[i] += vec[i]
|
vecSum[i] += float64(vec[i])
|
||||||
}
|
}
|
||||||
count++
|
count++
|
||||||
}
|
}
|
||||||
@ -276,11 +281,13 @@ func (e *StaticEmbedder) load(spec string, primary bool) error {
|
|||||||
for i := range vecSum {
|
for i := range vecSum {
|
||||||
vecSum[i] /= float64(count)
|
vecSum[i] /= float64(count)
|
||||||
}
|
}
|
||||||
e.unkVec = make([]float64, dim)
|
e.unkVec = make([]float32, dim)
|
||||||
copy(e.unkVec, vecSum)
|
for i, v := range vecSum {
|
||||||
|
e.unkVec[i] = float32(v)
|
||||||
|
}
|
||||||
var normSq float64
|
var normSq float64
|
||||||
for _, v := range e.unkVec {
|
for _, v := range e.unkVec {
|
||||||
normSq += v * v
|
normSq += float64(v) * float64(v)
|
||||||
}
|
}
|
||||||
e.unkNorm = float64(math.Sqrt(normSq))
|
e.unkNorm = float64(math.Sqrt(normSq))
|
||||||
e.loaded = true
|
e.loaded = true
|
||||||
@ -366,11 +373,11 @@ func (e *StaticEmbedder) Vectorize(text string) vector.Vector {
|
|||||||
|
|
||||||
if !ok {
|
if !ok {
|
||||||
for i, v := range unkVec {
|
for i, v := range unkVec {
|
||||||
sum[i] += w * v
|
sum[i] += w * float64(v)
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
for i, v := range vec {
|
for i, v := range vec {
|
||||||
sum[i] += w * v
|
sum[i] += w * float64(v)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
weightSum += w
|
weightSum += w
|
||||||
|
|||||||
37
internal/memory/static_embedder_mem_test.go
Normal file
37
internal/memory/static_embedder_mem_test.go
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
package memory
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"unsafe"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 词向量必须用 float32 存。
|
||||||
|
//
|
||||||
|
// 这条判据是拿生产内存换来的:向量本体 = 词数 × 维数 × 每元素字节数。
|
||||||
|
// 生产配置加载了 200000(zh) + 378151(en) = 57.8 万词 × 300 维 ⇒
|
||||||
|
// float64 = 1.29GB、float32 = 0.65GB(差 0.65GB 常驻)。
|
||||||
|
// 源数据(fastText 文本格式)本身就是 float32 精度,用 float64 存没有任何收益。
|
||||||
|
//
|
||||||
|
// 若有人把类型改回 float64,本测试**编译失败**(`var vec []float32` 的类型断言),
|
||||||
|
// 这正是想要的效果。
|
||||||
|
func TestStaticEmbedder_VectorMemIsFloat32(t *testing.T) {
|
||||||
|
e := newSynthEmbedder(t, 300)
|
||||||
|
|
||||||
|
words := 0
|
||||||
|
bytes := 0
|
||||||
|
for _, vec := range e.words {
|
||||||
|
var typed []float32 = vec // 编译期断言:存储必须是 []float32
|
||||||
|
if len(typed) != e.dim {
|
||||||
|
t.Fatalf("维度不符: %d != %d", len(typed), e.dim)
|
||||||
|
}
|
||||||
|
words++
|
||||||
|
bytes += len(typed) * int(unsafe.Sizeof(typed[0]))
|
||||||
|
}
|
||||||
|
if words == 0 {
|
||||||
|
t.Fatal("合成模型应至少加载一个词")
|
||||||
|
}
|
||||||
|
// float32:每词 300×4 = 1200 字节;float64 会是 2400
|
||||||
|
if want := words * e.dim * 4; bytes != want {
|
||||||
|
t.Fatalf("向量本体字节数应 %d(float32),实际 %d", want, bytes)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -28,7 +28,14 @@ var (
|
|||||||
//
|
//
|
||||||
// ❗main 上此值始终是**下一个未发布中版本**,不随 patch 发布变动
|
// ❗main 上此值始终是**下一个未发布中版本**,不随 patch 发布变动
|
||||||
//(见 docs/git-branching.md §2.1);已发布的版本号看对应的 release/vX.Y.x 与 tag。
|
//(见 docs/git-branching.md §2.1);已发布的版本号看对应的 release/vX.Y.x 与 tag。
|
||||||
Version = "1.3.0"
|
// 1.3.8:inputch 划给子后输入只流向子(补上"进内核之前"的输入路由)。
|
||||||
|
// 1.3.7:驻留子继承父的输出通道(此前子侧 childIO 空壳 ⇒ 子不会发消息)。
|
||||||
|
// 1.3.6:人格文本不再在播种时固化版本 + 存量实例一次性去版本化(生产实例
|
||||||
|
// 曾自报 v1.0.3);系统提示词支持 {{kernel_version}} 等占位符。
|
||||||
|
// 1.3.5:系统提示词(人格卡)支持版本占位符 —— 人格卡是配置项,写死版本号
|
||||||
|
// 会随发版说谎(线上写 v1.0.3、内核 1.3.x,agent 就自报 1.0.3)。
|
||||||
|
// 支持 {{kernel_version}} / {{kernel_commit}} / {{sdk_version}}。
|
||||||
|
Version = "1.3.8"
|
||||||
|
|
||||||
// Commit 是构建时的 Git commit hash。
|
// Commit 是构建时的 Git commit hash。
|
||||||
Commit = "unknown"
|
Commit = "unknown"
|
||||||
|
|||||||
43
internal/plugin/channel_warn_test.go
Normal file
43
internal/plugin/channel_warn_test.go
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
package plugin
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"log"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 延迟判定的语义:看的是"插件 Start 结束后最终声明了什么",
|
||||||
|
// 而不是"注册出站通道的那一刻有没有入站声明"。
|
||||||
|
//
|
||||||
|
// 为什么必须这样判:声明顺序自由 —— qq/weather 都是**先** RegisterOutputChannel
|
||||||
|
// **后** RegisterInputChannel,按注册时刻判会把它们误报成"只声明了输出通道"
|
||||||
|
// (实测发生过:用户据此以为 qq 插件没更新)。
|
||||||
|
func TestWarnOutputOnlyChannels(t *testing.T) {
|
||||||
|
r := NewRegistry()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
oldOut := log.Writer()
|
||||||
|
log.SetOutput(&buf)
|
||||||
|
defer log.SetOutput(oldOut)
|
||||||
|
|
||||||
|
// ① 出站+入站都声明了(先出站后入站)⇒ 不该告警
|
||||||
|
r.noteChannel("qq", "qq", true)
|
||||||
|
r.noteChannel("qq", "qq", false)
|
||||||
|
r.warnOutputOnlyChannels("qq")
|
||||||
|
if s := buf.String(); s != "" {
|
||||||
|
t.Fatalf("qq 声明了入站通道,不应告警,实际: %s", s)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ② 只声明出站 ⇒ 应告警,且只报这一个通道
|
||||||
|
buf.Reset()
|
||||||
|
r.noteChannel("weather", "weather_weather_out", true)
|
||||||
|
r.noteChannel("weather", "weather_weather_in", false)
|
||||||
|
r.warnOutputOnlyChannels("weather")
|
||||||
|
out := buf.String()
|
||||||
|
if !strings.Contains(out, "weather_weather_out") {
|
||||||
|
t.Fatalf("只声明出站的通道应被告警,实际: %q", out)
|
||||||
|
}
|
||||||
|
if strings.Contains(out, "weather_weather_in") {
|
||||||
|
t.Fatalf("已声明入站的通道不该被牵连,实际: %q", out)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -307,12 +307,15 @@ func (r *Registry) buildSDK(name string) *sdk.PluginSDK {
|
|||||||
// 但历史插件常常只用 RegisterOutputChannel 声明(却用同一个名字注入输入,
|
// 但历史插件常常只用 RegisterOutputChannel 声明(却用同一个名字注入输入,
|
||||||
// 例:cli 只声明输出 "cli" 就用 InjectTextSync("cli", ...) 注入)。
|
// 例:cli 只声明输出 "cli" 就用 InjectTextSync("cli", ...) 注入)。
|
||||||
// 不兜底的话 inputch 登记表里没有它,"把 inputch 划给驻留子"直接失败
|
// 不兜底的话 inputch 登记表里没有它,"把 inputch 划给驻留子"直接失败
|
||||||
// (实测报 `划入 inputch cli: inputch 未注册`)。兜底要**留痕**,
|
// (实测报 `划入 inputch cli: inputch 未注册`)。
|
||||||
// 否则插件作者永远不知道该补一行 RegisterInputChannel。
|
//
|
||||||
|
// ❗这里**不能**判"是否声明过入站通道"并告警:声明顺序是自由的,
|
||||||
|
// 先 RegisterOutputChannel 再 RegisterInputChannel 是常见写法(qq 就是),
|
||||||
|
// 按此刻的状态判会对它误报(实测:把 qq 报成"只声明了输出通道")。
|
||||||
|
// 真正该问的问题是"插件 Start 结束后,这个出站通道有没有对应的入站声明" ——
|
||||||
|
// 那在 load 完成后统一判(见 warnOutputOnlyChannels)。
|
||||||
if _, ok := r.iom.LookupInputChannel(chName); !ok {
|
if _, ok := r.iom.LookupInputChannel(chName); !ok {
|
||||||
_ = r.iom.RegisterInputChannelFrom(name, chName, agentIO.ChannelDef(def))
|
_ = r.iom.RegisterInputChannelFrom(name, chName, agentIO.ChannelDef(def))
|
||||||
log.Printf("[plugin] %s 只声明了输出通道 %q,已按双向通道兜底登记 inputch;"+
|
|
||||||
"若要明确意图请显式 RegisterInputChannel", name, chName)
|
|
||||||
}
|
}
|
||||||
r.noteChannel(name, chName, true)
|
r.noteChannel(name, chName, true)
|
||||||
return nil
|
return nil
|
||||||
@ -452,6 +455,7 @@ func (r *Registry) Load(dir string) error {
|
|||||||
r.pluginAutoRestart[name] = plgSDK.AutoRestart()
|
r.pluginAutoRestart[name] = plgSDK.AutoRestart()
|
||||||
r.instances = append(r.instances, p)
|
r.instances = append(r.instances, p)
|
||||||
r.mu.Unlock()
|
r.mu.Unlock()
|
||||||
|
r.warnOutputOnlyChannels(name)
|
||||||
log.Printf("[plugin] loaded: %s", name)
|
log.Printf("[plugin] loaded: %s", name)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -545,6 +549,7 @@ func (r *Registry) loadOne(plgDir, name string) bool {
|
|||||||
r.pluginAutoRestart[name] = plgSDK.AutoRestart()
|
r.pluginAutoRestart[name] = plgSDK.AutoRestart()
|
||||||
r.sdkRefs[name] = plgSDK
|
r.sdkRefs[name] = plgSDK
|
||||||
r.instances = append(r.instances, plg)
|
r.instances = append(r.instances, plg)
|
||||||
|
r.warnOutputOnlyChannels(name)
|
||||||
if h := pluginEntryHash(plgDir); h != "" {
|
if h := pluginEntryHash(plgDir); h != "" {
|
||||||
r.pluginHashes[name] = h
|
r.pluginHashes[name] = h
|
||||||
} else {
|
} else {
|
||||||
@ -579,6 +584,31 @@ func (r *Registry) stageRegistrarFor() (func(plugin string, stage sdk.Stage, han
|
|||||||
return nil, false
|
return nil, false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// warnOutputOnlyChannels 在插件 Start 结束后,报告"只声明了出站、没有入站声明"的通道。
|
||||||
|
//
|
||||||
|
// 为什么放在 Start 之后:声明顺序自由(先出站后入站很常见),注册时刻的状态
|
||||||
|
// 判不出意图。这里看的是**插件最终声明了什么**,因此不会误报 qq 这种写法。
|
||||||
|
//
|
||||||
|
// 注:这类通道内核已兜底登记 inputch(功能可用),告警只是提醒插件作者把意图写明。
|
||||||
|
func (r *Registry) warnOutputOnlyChannels(plugin string) {
|
||||||
|
r.channelsMu.Lock()
|
||||||
|
set := r.pluginChannels[plugin]
|
||||||
|
var only []string
|
||||||
|
if set != nil {
|
||||||
|
for ch := range set.outputs {
|
||||||
|
if !set.inputs[ch] {
|
||||||
|
only = append(only, ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
r.channelsMu.Unlock()
|
||||||
|
sort.Strings(only)
|
||||||
|
for _, ch := range only {
|
||||||
|
log.Printf("[plugin] %s 只声明了出站通道 %q(未 RegisterInputChannel);"+
|
||||||
|
"内核已兜底登记 inputch,若这是有意为之可忽略", plugin, ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// noteChannel 记住插件注册了哪个通道,供卸载/崩溃时摘除。
|
// noteChannel 记住插件注册了哪个通道,供卸载/崩溃时摘除。
|
||||||
// forgetChannel 把某个通道从"本插件注册过哪些通道"的记账里摘掉(注销通道时用)。
|
// forgetChannel 把某个通道从"本插件注册过哪些通道"的记账里摘掉(注销通道时用)。
|
||||||
//
|
//
|
||||||
|
|||||||
@ -138,28 +138,39 @@ func (p *Plugin) Stop() error {
|
|||||||
func (p *Plugin) registerTools(s *sdk.PluginSDK) {
|
func (p *Plugin) registerTools(s *sdk.PluginSDK) {
|
||||||
s.RegisterTool("plugin_install", sdk.ToolDef{
|
s.RegisterTool("plugin_install", sdk.ToolDef{
|
||||||
Name: "plugin_install",
|
Name: "plugin_install",
|
||||||
Description: "从 URL 安装 HomeAgent 插件包(.hmap 文件)。插件已存在时传 overwrite=true 原地更新(升级/降级/重装,保留配置表,无需卸载重装)。更新后需调用 plgreload 或重启生效。",
|
Description: "安装 HomeAgent 插件包(.hmap)。两种来源:url(http/https 下载)或 path(本机路径,配合 plugindev_build 的产物用这个)。插件已存在时传 overwrite=true 原地更新(升级/降级/重装,保留配置表,无需卸载重装)。更新后需调用 plgreload 或重启生效。",
|
||||||
Parameters: map[string]interface{}{
|
Parameters: map[string]interface{}{
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": map[string]interface{}{
|
"properties": map[string]interface{}{
|
||||||
"url": map[string]interface{}{
|
"url": map[string]interface{}{
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"description": "插件包的下载 URL",
|
"description": "插件包的下载 URL(http/https)",
|
||||||
|
},
|
||||||
|
"path": map[string]interface{}{
|
||||||
|
"type": "string",
|
||||||
|
"description": "插件包在**本机**的路径(.hmap)。与 url 二选一;同时给出时以 path 为准",
|
||||||
},
|
},
|
||||||
"overwrite": map[string]interface{}{
|
"overwrite": map[string]interface{}{
|
||||||
"type": "boolean",
|
"type": "boolean",
|
||||||
"description": "已存在时原地更新(保留配置)。默认 false",
|
"description": "已存在时原地更新(保留配置)。默认 false",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"required": []string{"url"},
|
|
||||||
},
|
},
|
||||||
}, func(args map[string]interface{}) (interface{}, error) {
|
}, func(args map[string]interface{}) (interface{}, error) {
|
||||||
url, _ := args["url"].(string)
|
|
||||||
if url == "" {
|
|
||||||
return map[string]interface{}{"error": "url is required"}, nil
|
|
||||||
}
|
|
||||||
overwrite, _ := args["overwrite"].(bool)
|
overwrite, _ := args["overwrite"].(bool)
|
||||||
return p.installFromURL(url, overwrite)
|
// path 优先:它对应"agent 自己构建出产物再装"的场景(plugindev_build → plugin_install)。
|
||||||
|
if path, _ := args["path"].(string); strings.TrimSpace(path) != "" {
|
||||||
|
pth := strings.TrimSpace(path)
|
||||||
|
if st, err := os.Stat(pth); err != nil || st.IsDir() {
|
||||||
|
return map[string]interface{}{"error": fmt.Sprintf("path 无效(必须是存在的 .hmap 文件): %s", pth)}, nil
|
||||||
|
}
|
||||||
|
return p.installFromPath(pth, overwrite)
|
||||||
|
}
|
||||||
|
url, _ := args["url"].(string)
|
||||||
|
if strings.TrimSpace(url) == "" {
|
||||||
|
return map[string]interface{}{"error": "需要 url 或 path(二选一)"}, nil
|
||||||
|
}
|
||||||
|
return p.installFromURL(strings.TrimSpace(url), overwrite)
|
||||||
})
|
})
|
||||||
|
|
||||||
s.RegisterTool("plugin_list", sdk.ToolDef{
|
s.RegisterTool("plugin_list", sdk.ToolDef{
|
||||||
@ -455,12 +466,12 @@ func (p *Plugin) installFromData(data []byte, overwrite bool) (interface{}, erro
|
|||||||
|
|
||||||
if existing && !overwrite {
|
if existing && !overwrite {
|
||||||
return map[string]interface{}{
|
return map[string]interface{}{
|
||||||
"error": "plugin already exists",
|
"error": "plugin already exists",
|
||||||
"name": pkg.Name,
|
"name": pkg.Name,
|
||||||
"version": pkg.Version,
|
"version": pkg.Version,
|
||||||
"current": oldVersion,
|
"current": oldVersion,
|
||||||
"action": "remove_first",
|
"action": "remove_first",
|
||||||
"hint": `传 "overwrite": true 可原地更新(保留配置)`,
|
"hint": `传 "overwrite": true 可原地更新(保留配置)`,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -475,7 +486,7 @@ func (p *Plugin) installFromData(data []byte, overwrite bool) (interface{}, erro
|
|||||||
os.RemoveAll(backup)
|
os.RemoveAll(backup)
|
||||||
if err := os.Rename(target, backup); err != nil {
|
if err := os.Rename(target, backup); err != nil {
|
||||||
return map[string]interface{}{
|
return map[string]interface{}{
|
||||||
"error": "backup old plugin dir failed",
|
"error": "backup old plugin dir failed",
|
||||||
"details": err.Error(),
|
"details": err.Error(),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
@ -484,8 +495,8 @@ func (p *Plugin) installFromData(data []byte, overwrite bool) (interface{}, erro
|
|||||||
os.RemoveAll(target)
|
os.RemoveAll(target)
|
||||||
if rbErr := os.Rename(backup, target); rbErr != nil {
|
if rbErr := os.Rename(backup, target); rbErr != nil {
|
||||||
return map[string]interface{}{
|
return map[string]interface{}{
|
||||||
"error": "extract failed AND rollback failed",
|
"error": "extract failed AND rollback failed",
|
||||||
"details": err.Error(),
|
"details": err.Error(),
|
||||||
"rollback": rbErr.Error(),
|
"rollback": rbErr.Error(),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
@ -507,15 +518,15 @@ func (p *Plugin) installFromData(data []byte, overwrite bool) (interface{}, erro
|
|||||||
action = "reinstalled"
|
action = "reinstalled"
|
||||||
}
|
}
|
||||||
return map[string]interface{}{
|
return map[string]interface{}{
|
||||||
"status": "installed",
|
"status": "installed",
|
||||||
"name": pkg.Name,
|
"name": pkg.Name,
|
||||||
"version": pkg.Version,
|
"version": pkg.Version,
|
||||||
"previous_version": oldVersion,
|
"previous_version": oldVersion,
|
||||||
"entry": pkg.Entry,
|
"entry": pkg.Entry,
|
||||||
"checksum": checksum,
|
"checksum": checksum,
|
||||||
"action": action,
|
"action": action,
|
||||||
"reload_required": true,
|
"reload_required": true,
|
||||||
"config_kept": true,
|
"config_kept": true,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -1,9 +1,9 @@
|
|||||||
package remotedevice
|
package remotedevice
|
||||||
|
|
||||||
// 设备输出通道:把"agent 主动发给设备"做成**每设备一个输出通道** `device/<id>`。
|
// 设备输出通道:把"agent 主动发给设备"做成**每设备一个输出通道** `device-<id>`。
|
||||||
//
|
//
|
||||||
// 为什么是输出通道而不是再加一批工具:
|
// 为什么是输出通道而不是再加一批工具:
|
||||||
// - **寻址**:`output_send__device/<id>` 直接指名道姓;模型看 `output_list_channels`
|
// - **寻址**:`output_send__device-<id>` 直接指名道姓;模型看 `output_list_channels`
|
||||||
// 就知道当前有哪些设备在线,不必先 `devicedetect` 再往参数里塞 device_id。
|
// 就知道当前有哪些设备在线,不必先 `devicedetect` 再往参数里塞 device_id。
|
||||||
// - **能力**:caps 由设备声明的 caps 映射,**内核**在发送前就按 caps 拦
|
// - **能力**:caps 由设备声明的 caps 映射,**内核**在发送前就按 caps 拦
|
||||||
// (把图片发给只支持文本的音箱会被拒,而不是等设备侧报错)。
|
// (把图片发给只支持文本的音箱会被拒,而不是等设备侧报错)。
|
||||||
@ -15,7 +15,9 @@ package remotedevice
|
|||||||
// 它们的返回值(图像/命令输出/状态)必须进模型上下文,做成通道会丢掉这个语义。
|
// 它们的返回值(图像/命令输出/状态)必须进模型上下文,做成通道会丢掉这个语义。
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/sha1"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
|
"encoding/hex"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
@ -76,14 +78,75 @@ func deviceOutputCaps(caps []string, kind string) agentIO.OutputCapability {
|
|||||||
return out
|
return out
|
||||||
}
|
}
|
||||||
|
|
||||||
// deviceChannelName 是设备输出(也是输入)通道名:`device/<id>`。
|
// deviceChannelName 由**设备自报的 id** 派生一个合规且唯一的通道名:`device-<派生值>`。
|
||||||
//
|
//
|
||||||
// 入站与出站**同名**:两者指的是同一台设备,分成两个名字只会让模型与授权表更难对。
|
// 入站与出站**同名**:两者指的是同一台设备,分成两个名字只会让模型与授权表更难对。
|
||||||
func deviceChannelName(id string) string { return "device/" + id }
|
//
|
||||||
|
// 为什么不能直接用 id:通道名会被内核拼进 LLM 的**函数名**(`output_send__<通道名>`),
|
||||||
|
// 上游规范是 `^[a-zA-Z0-9_-]{1,64}$`;而设备 id 是**外部输入**(设备自己声明),
|
||||||
|
// 可能含空格/非 ASCII/超长。违规的后果不是"这个工具不能用",而是**整条请求被 400 拒绝** ——
|
||||||
|
// 实测把生产打挂:`Invalid 'tools[299].function.name'`,网关 auto tier 全链条失败,
|
||||||
|
// 内核只能报"所有 provider 都失败",表现成"整个 agent 不说话了"。
|
||||||
|
//
|
||||||
|
// 派生规则(确定性,同一 id 永远同名):
|
||||||
|
// 1. 保留 [A-Za-z0-9_-],其它字符折成 '-';折叠后为空则用 "dev"
|
||||||
|
// 2. 截断到 maxDeviceChannelSuffix 字符(给 "device-" 与短哈希留余量)
|
||||||
|
// 3. 若发生截断,或该名字已被**另一个** id 占用,则追加 id 的 6 位短哈希
|
||||||
|
//
|
||||||
|
// 设备 id 本身仍用于路由与日志(真名不丢),通道名只是它派生的标识符。
|
||||||
|
func (p *Plugin) deviceChannelName(id string) string {
|
||||||
|
p.devChansMu.Lock()
|
||||||
|
defer p.devChansMu.Unlock()
|
||||||
|
if p.devChans == nil {
|
||||||
|
p.devChans = make(map[string]string)
|
||||||
|
}
|
||||||
|
if name, ok := p.devChans[id]; ok {
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
var b strings.Builder
|
||||||
|
for _, r := range id {
|
||||||
|
switch {
|
||||||
|
case r >= 'a' && r <= 'z', r >= 'A' && r <= 'Z', r >= '0' && r <= '9', r == '_', r == '-':
|
||||||
|
b.WriteRune(r)
|
||||||
|
default:
|
||||||
|
b.WriteByte('-')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
base := b.String()
|
||||||
|
if base == "" {
|
||||||
|
base = "dev"
|
||||||
|
}
|
||||||
|
truncated := false
|
||||||
|
if len(base) > maxDeviceChannelSuffix {
|
||||||
|
base = base[:maxDeviceChannelSuffix]
|
||||||
|
truncated = true
|
||||||
|
}
|
||||||
|
name := "device-" + base
|
||||||
|
// 撞名检查:不同 id 折出同一个名字时必须可区分
|
||||||
|
for otherID, otherName := range p.devChans {
|
||||||
|
if otherName == name && otherID != id {
|
||||||
|
truncated = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if truncated {
|
||||||
|
sum := sha1.Sum([]byte(id))
|
||||||
|
name += "-" + hex.EncodeToString(sum[:3])
|
||||||
|
}
|
||||||
|
p.devChans[id] = name
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
// maxDeviceChannelSuffix 是通道名主体的长度上限。
|
||||||
|
// 预算:上游函数名上限 64 = "output_send__"(13) + "device-"(7) + 主体 + "-"+短哈希(7)
|
||||||
|
// ⇒ 主体最多 37;取 32 留余量(改名/前缀变动不会立刻越界)。
|
||||||
|
maxDeviceChannelSuffix = 32
|
||||||
|
)
|
||||||
|
|
||||||
// wireDeviceChannels 把"设备上下线"接到通道的登记/注销上。
|
// wireDeviceChannels 把"设备上下线"接到通道的登记/注销上。
|
||||||
//
|
//
|
||||||
// 一台设备 = 一对**同名**通道 `device/<id>`:入站(设备上报 → agent)与出站
|
// 一台设备 = 一对**同名**通道 `device-<id>`:入站(设备上报 → agent)与出站
|
||||||
// (agent → 设备)。用**同步回调**而不是 ChangeChan(后者是 select+default,
|
// (agent → 设备)。用**同步回调**而不是 ChangeChan(后者是 select+default,
|
||||||
// 缓冲满会丢事件;丢一次就留下死通道或漏注册)。
|
// 缓冲满会丢事件;丢一次就留下死通道或漏注册)。
|
||||||
//
|
//
|
||||||
@ -92,14 +155,14 @@ func deviceChannelName(id string) string { return "device/" + id }
|
|||||||
func (p *Plugin) wireDeviceChannels() {
|
func (p *Plugin) wireDeviceChannels() {
|
||||||
p.registry.SetPresenceHandler(
|
p.registry.SetPresenceHandler(
|
||||||
func(meta DeviceMeta) {
|
func(meta DeviceMeta) {
|
||||||
_ = p.sdk.RegisterInputChannel(deviceChannelName(meta.DeviceID), sdk.ChannelDef{})
|
_ = p.sdk.RegisterInputChannel(p.deviceChannelName(meta.DeviceID), sdk.ChannelDef{})
|
||||||
p.ensureDeviceOutputChannel(meta.DeviceID)
|
p.ensureDeviceOutputChannel(meta.DeviceID)
|
||||||
},
|
},
|
||||||
func(id string) { p.dropDeviceOutputChannel(id) },
|
func(id string) { p.dropDeviceOutputChannel(id) },
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ensureDeviceOutputChannel 给在线设备注册输出通道 device/<id>(幂等)。
|
// ensureDeviceOutputChannel 给在线设备注册输出通道 device-<id>(幂等)。
|
||||||
func (p *Plugin) ensureDeviceOutputChannel(id string) {
|
func (p *Plugin) ensureDeviceOutputChannel(id string) {
|
||||||
if p.sdk == nil || id == "" {
|
if p.sdk == nil || id == "" {
|
||||||
return
|
return
|
||||||
@ -108,7 +171,7 @@ func (p *Plugin) ensureDeviceOutputChannel(id string) {
|
|||||||
if !ok || !meta.Online {
|
if !ok || !meta.Online {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
ch := deviceChannelName(id)
|
ch := p.deviceChannelName(id)
|
||||||
caps := deviceOutputCaps(meta.Caps, meta.Kind)
|
caps := deviceOutputCaps(meta.Caps, meta.Kind)
|
||||||
desc := fmt.Sprintf("远程设备 %s(%s):agent 主动向该设备发送内容;能力位 %s",
|
desc := fmt.Sprintf("远程设备 %s(%s):agent 主动向该设备发送内容;能力位 %s",
|
||||||
id, fallback(meta.Name, meta.Kind), agentIO.OutputCapability(caps).String())
|
id, fallback(meta.Name, meta.Kind), agentIO.OutputCapability(caps).String())
|
||||||
@ -130,7 +193,7 @@ func (p *Plugin) dropDeviceOutputChannel(id string) {
|
|||||||
if p.sdk == nil || id == "" {
|
if p.sdk == nil || id == "" {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
ch := deviceChannelName(id)
|
ch := p.deviceChannelName(id)
|
||||||
if err := p.sdk.UnregisterOutputChannel(ch); err != nil {
|
if err := p.sdk.UnregisterOutputChannel(ch); err != nil {
|
||||||
p.logf("unregister output channel %s: %v", ch, err)
|
p.logf("unregister output channel %s: %v", ch, err)
|
||||||
return
|
return
|
||||||
@ -247,9 +310,9 @@ func (d *devicectlDevice) output(args map[string]interface{}) (interface{}, erro
|
|||||||
ids = append(ids, m.DeviceID)
|
ids = append(ids, m.DeviceID)
|
||||||
}
|
}
|
||||||
if len(ids) == 0 {
|
if len(ids) == 0 {
|
||||||
return nil, fmt.Errorf("devicectl 需要 meta.device_id 才能投递;当前没有在线设备(device_list_channels 可看每台设备的 device/<id> 通道)")
|
return nil, fmt.Errorf("devicectl 需要 meta.device_id 才能投递;当前没有在线设备(device_list_channels 可看每台设备的 device-<id> 通道)")
|
||||||
}
|
}
|
||||||
return nil, fmt.Errorf("devicectl 需要 meta.device_id(或直接用通道 device/<id>);当前在线设备: %s", strings.Join(ids, ", "))
|
return nil, fmt.Errorf("devicectl 需要 meta.device_id(或直接用通道 device-<id>);当前在线设备: %s", strings.Join(ids, ", "))
|
||||||
}
|
}
|
||||||
return pushToDevice(d.reg, deviceID, args)
|
return pushToDevice(d.reg, deviceID, args)
|
||||||
}
|
}
|
||||||
|
|||||||
@ -9,6 +9,8 @@ import (
|
|||||||
"encoding/json"
|
"encoding/json"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
@ -130,7 +132,7 @@ func TestDeviceChannelLifecycleAndPush(t *testing.T) {
|
|||||||
cli.sendText([]byte(`{"op":"hello","device":{"device_id":"spk-1","name":"音箱","kind":"speaker","caps":["speaker"]}}`))
|
cli.sendText([]byte(`{"op":"hello","device":{"device_id":"spk-1","name":"音箱","kind":"speaker","caps":["speaker"]}}`))
|
||||||
cli.readHelloAckAndBind(t, token)
|
cli.readHelloAckAndBind(t, token)
|
||||||
|
|
||||||
ch := deviceChannelName("spk-1")
|
ch := p.deviceChannelName("spk-1")
|
||||||
deadline := time.Now().Add(3 * time.Second)
|
deadline := time.Now().Add(3 * time.Second)
|
||||||
caps, ok := rec.caps(ch)
|
caps, ok := rec.caps(ch)
|
||||||
for !ok && time.Now().Before(deadline) {
|
for !ok && time.Now().Before(deadline) {
|
||||||
@ -255,3 +257,31 @@ func TestDevicectlAggregateOutputAddressing(t *testing.T) {
|
|||||||
t.Fatal("不存在的设备应报错")
|
t.Fatal("不存在的设备应报错")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 通道名合规性:设备通道名会被内核拼进 LLM **函数名**(output_send__<通道名>),
|
||||||
|
// 而上游函数名规范是 ^[a-zA-Z0-9_-]{1,64}$ —— 违规会让**整条请求**被 400 拒绝
|
||||||
|
// (实测把生产打挂:device/<id> 里的 `/` 触发 Invalid 'tools[299].function.name',
|
||||||
|
// 网关 auto tier 全链条失败,整个 agent 不说话了)。
|
||||||
|
//
|
||||||
|
// 通道名是**插件自己的声明**,所以这条判据钉在插件侧。
|
||||||
|
func TestDeviceChannelNameIsLLMFunctionNameSafe(t *testing.T) {
|
||||||
|
re := regexp.MustCompile(`^[a-zA-Z0-9_-]{1,64}$`)
|
||||||
|
// 含**恶意/异常** id:空格、符号、非 ASCII、超长、以及会折成同一个名字的两个 id
|
||||||
|
ids := []string{"waiter-fnnas", "1", "a b!c", "中文设备", strings.Repeat("x", 120), "a b", "a-b"}
|
||||||
|
p := &Plugin{}
|
||||||
|
seen := map[string]string{}
|
||||||
|
for _, id := range ids {
|
||||||
|
ch := p.deviceChannelName(id)
|
||||||
|
if prev, dup := seen[ch]; dup {
|
||||||
|
t.Errorf("不同设备 id(%q 与 %q)派生出同一个通道名 %q", prev, id, ch)
|
||||||
|
}
|
||||||
|
seen[ch] = id
|
||||||
|
if !re.MatchString(ch) {
|
||||||
|
t.Errorf("设备通道名 %q 违反上游函数名规范 %s", ch, re)
|
||||||
|
}
|
||||||
|
toolName := "output_send__" + ch
|
||||||
|
if !re.MatchString(toolName) {
|
||||||
|
t.Errorf("派生出的工具名 %q 违反上游函数名规范 %s", toolName, re)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@ -37,6 +37,11 @@ type Plugin struct {
|
|||||||
token string
|
token string
|
||||||
sdk *sdk.PluginSDK
|
sdk *sdk.PluginSDK
|
||||||
dev *devicectlDevice
|
dev *devicectlDevice
|
||||||
|
|
||||||
|
// devChansMu/devChans 维护"设备自报 id → 派生的通道名"。
|
||||||
|
// 设备 id 是外部输入,不能直接进通道名(见 outputch.go 的 deviceChannelName)。
|
||||||
|
devChansMu sync.Mutex
|
||||||
|
devChans map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(name string) *Plugin {
|
func New(name string) *Plugin {
|
||||||
@ -124,7 +129,7 @@ func (p *Plugin) Start(s *sdk.PluginSDK) error {
|
|||||||
//
|
//
|
||||||
// **注意**:agent 的输出**不会**被自动转回设备 —— 主动转发只有 webui 与 cli 两个
|
// **注意**:agent 的输出**不会**被自动转回设备 —— 主动转发只有 webui 与 cli 两个
|
||||||
// 交互界面(它们把最终回复渲染成对话气泡是本职)。设备要走
|
// 交互界面(它们把最终回复渲染成对话气泡是本职)。设备要走
|
||||||
// `output_send__device/<id>`(agent 主动调用),这才与"输出是 agent 的主动调用"一致。
|
// `output_send__device-<id>`(agent 主动调用),这才与"输出是 agent 的主动调用"一致。
|
||||||
// 节流:同设备同类型事件 10s 内去重,防传感器风暴。
|
// 节流:同设备同类型事件 10s 内去重,防传感器风暴。
|
||||||
lastEventAt := map[string]time.Time{}
|
lastEventAt := map[string]time.Time{}
|
||||||
var eventMu sync.Mutex
|
var eventMu sync.Mutex
|
||||||
@ -159,9 +164,10 @@ func (p *Plugin) Start(s *sdk.PluginSDK) error {
|
|||||||
|
|
||||||
log.Printf("[remotedevice] event from %s: %s", deviceID, evtType)
|
log.Printf("[remotedevice] event from %s: %s", deviceID, evtType)
|
||||||
if p.sdk != nil {
|
if p.sdk != nil {
|
||||||
// 设备通道 device/<id> 是动态的:设备首次上报时**懒登记** inputch
|
// 设备通道 device-<id> 是动态的(分隔符用 - 而非 /,见 deviceChannelName 的说明:
|
||||||
// (Register 幂等),父 agent 才能把它划给驻留子。
|
// 通道名会进 LLM 函数名,必须满足 ^[a-zA-Z0-9_-]{1,64}$)。
|
||||||
devCh := "device/" + deviceID
|
// 首次上报时**懒登记** inputch(Register 幂等),父 agent 才能把它划给驻留子。
|
||||||
|
devCh := p.deviceChannelName(deviceID)
|
||||||
_ = p.sdk.RegisterInputChannel(devCh, sdk.ChannelDef{})
|
_ = p.sdk.RegisterInputChannel(devCh, sdk.ChannelDef{})
|
||||||
// 异步注入:不阻塞 WS 读循环;回复路由回 device/{id} 输出通道
|
// 异步注入:不阻塞 WS 读循环;回复路由回 device/{id} 输出通道
|
||||||
p.sdk.InjectInput(devCh, devCh, "text", map[string]interface{}{"content": text})
|
p.sdk.InjectInput(devCh, devCh, "text", map[string]interface{}{"content": text})
|
||||||
|
|||||||
@ -287,7 +287,7 @@ func (r *Registry) register(meta DeviceMeta) {
|
|||||||
r.devices[meta.DeviceID] = &meta
|
r.devices[meta.DeviceID] = &meta
|
||||||
onOnline := r.onOnline
|
onOnline := r.onOnline
|
||||||
r.mu.Unlock()
|
r.mu.Unlock()
|
||||||
// 先回调(可能注册 device/<id> 输出通道),再发变更通知。
|
// 先回调(可能注册 device-<id> 输出通道),再发变更通知。
|
||||||
if onOnline != nil {
|
if onOnline != nil {
|
||||||
onOnline(meta)
|
onOnline(meta)
|
||||||
}
|
}
|
||||||
|
|||||||
10
third_party/homeagent-sdk/sdk/plugin.go
vendored
10
third_party/homeagent-sdk/sdk/plugin.go
vendored
@ -480,7 +480,15 @@ func (s *PluginSDK) RegisterPluginAPI(name string) error {
|
|||||||
// 入站(谁会往 <name> 注入输入)是另一件事,用 RegisterInputChannel 声明。
|
// 入站(谁会往 <name> 注入输入)是另一件事,用 RegisterInputChannel 声明。
|
||||||
// 若该通道同时也是你的注入入口,两个都要登记。
|
// 若该通道同时也是你的注入入口,两个都要登记。
|
||||||
//
|
//
|
||||||
// name: channel name (e.g. "qq", "webui")
|
// name: channel name (e.g. "qq", "webui")。
|
||||||
|
//
|
||||||
|
// ❗**命名约束**:内核会把通道名拼进 LLM 的函数名(`output_send__<name>`),
|
||||||
|
// 而上游对函数名的规范是 `^[a-zA-Z0-9_-]{1,64}$`。违反的后果不是"这个工具不可用",
|
||||||
|
// 而是**整条请求被上游 400 拒绝**(`Invalid 'tools[N].function.name'`),
|
||||||
|
// 网关的 auto tier 会全链条失败 —— 表现成"整个 agent 不说话了"。
|
||||||
|
// 所以通道名只能用 `[A-Za-z0-9_-]`,且总长要留出 `output_send__`(13 字符)的余量。
|
||||||
|
// 若通道名来自外部输入(设备自报 id 之类),请**在插件侧派生一个合规且唯一的名字**,
|
||||||
|
// 而不是把原始值直接当通道名。
|
||||||
// caps: bitmask of supported output capabilities (CapText, CapFile, etc.)
|
// caps: bitmask of supported output capabilities (CapText, CapFile, etc.)
|
||||||
// desc: description of the channel, expected meta format, and type enum
|
// desc: description of the channel, expected meta format, and type enum
|
||||||
// def: 通道在记忆计算层的行为(NoMemory/Cleaner)
|
// def: 通道在记忆计算层的行为(NoMemory/Cleaner)
|
||||||
|
|||||||
Reference in New Issue
Block a user