mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-04 00:03:59 +00:00
fix(resident): 驻留子继承父的输出通道 —— 修「子侧 childIO 空壳、子不会发消息」
现场(用户在线上跑驻留子联调,日志实录):
父 agent 侧「通道装载完整」,子 `demo-resident` 侧 `childIO` 是**空壳**:
子的 `output_list_channels` 为空、`output_send__<通道>` 一律被判
「通道 [X] 不存在或不可用」,连 `output_send__*` 工具都不生成 ⇒ 子不会发消息。
根因:**输出通道在 io 层就是 Device**,而它们由插件登记在**父**的 `IOManager` 上。
`SpawnResident` 给子建的是全新 `IOManager`(它确实该有自己的输入入口与 outputCh),
却只共享了 inputch 登记表,**没有继承设备/输出通道视图**:
- `executeOutputSendTool` → `a.io.GetChannelCapabilities(ch)` 查的是 `devices[ch]` ⇒ 0
- 投递路径 `a.io.GetDevice(ch).Execute("output", …)` ⇒ nil
- 工具面 `tooldefs.go` 从 `a.io.ListChannels()` 生成 `output_send__*` ⇒ 空
改法:给 `IOManager` 增加**上级回退**(`SetParentIO`)——驻留子创建时把自己的 io 挂到
父的 io 上,`GetDevice` / `GetChannelCapabilities` / `ListChannels` / `ExecuteTool`
在自己没有时回退到上级。
为什么是**实时回退**而不是创建时复制快照:设备随资源生灭(远程设备上线/掉线以分钟计,
现场日志 60 秒一个来回),复制出来的表转瞬即过期;而回退永远与父一致。
**授权不受影响**:回退只解决"看得见",能不能用仍由各自的 `AllowedOutputs` 白名单把关
(`executeOutputSendTool` 的授权闸 + 工具生成时的过滤都在白名单之后);
自己的登记优先,子可以覆盖/屏蔽同名通道。
判据(新增 5 条):
- io 层:无上级时行为与以前完全一致 / 挂上级后看得见 / **实时**(父新登记立刻可见、
注销立刻不可见)/ 同名自己的优先且不重复列出 / `ExecuteTool` 同样回退
- 内核层:子看得见父通道 + 真能发出(父通道收到 1 次 output)/ 白名单外被拒且未送达 /
子工具面只生成授权通道(含 `_help`)/ 父后登记的通道立刻可见 / 默认即完整授权
This commit is contained in:
@ -104,6 +104,20 @@ type IOManager struct {
|
||||
nextReqID int64
|
||||
channelReg *ChannelRegistry
|
||||
|
||||
// parent 是"上级 IOManager"(驻留子的轻量内核指向父的内核)。
|
||||
//
|
||||
// 为什么需要:**输出通道在 io 层就是 Device**,而它们是由插件登记在**父**的
|
||||
// io 上的。驻留子有自己的 IOManager(自己的输入入口、自己的 outputCh),
|
||||
// 若只看自己那张空表,`output_send__<通道>` 会被判"通道不存在或不可用",
|
||||
// `output_list_channels` 是空的,`output_send__*` 工具也不会生成
|
||||
// —— 现场表现就是"驻留子不会说话/不会发消息"(联调实录:父侧通道装载完整、
|
||||
// 子侧 childIO 空壳)。
|
||||
//
|
||||
// 用**实时回退**而不是创建时复制快照:设备会随资源生灭(远程设备上线/掉线
|
||||
// 以分钟计),复制出来的表转瞬就过期。授权由各自的 AllowedOutputs 白名单把关,
|
||||
// 回退只解决"看得见",不解决"能不能用"。
|
||||
parent *IOManager
|
||||
|
||||
// toolBlocks:插件工具注入多模态内容块,process.go 在下一条 tool message 时消费。
|
||||
// 用 interface{}[] 避免 import api.ContentBlock 导致的循环依赖。
|
||||
toolBlocksMu sync.Mutex
|
||||
@ -120,6 +134,31 @@ func NewIOManager() *IOManager {
|
||||
}
|
||||
}
|
||||
|
||||
// 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) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
@ -158,9 +197,7 @@ func (m *IOManager) RegisterDevice(dev Device) error {
|
||||
}
|
||||
|
||||
func (m *IOManager) GetDevice(name string) Device {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
return m.devices[name]
|
||||
return m.lookupDevice(name)
|
||||
}
|
||||
|
||||
func (m *IOManager) StartAll() error {
|
||||
@ -542,6 +579,15 @@ func (m *IOManager) ExecuteTool(name string, args map[string]interface{}) (ret i
|
||||
m.mu.RUnlock()
|
||||
|
||||
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)
|
||||
}
|
||||
defer func() {
|
||||
@ -574,10 +620,22 @@ type ChannelInfo struct {
|
||||
|
||||
func (m *IOManager) ListChannels() []ChannelInfo {
|
||||
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
|
||||
for _, dev := range m.devices {
|
||||
appendDev := func(dev Device) {
|
||||
if seen[dev.Name()] {
|
||||
return
|
||||
}
|
||||
seen[dev.Name()] = true
|
||||
list = append(list, ChannelInfo{
|
||||
Name: dev.Name(),
|
||||
Type: dev.Type(),
|
||||
@ -586,13 +644,23 @@ func (m *IOManager) ListChannels() []ChannelInfo {
|
||||
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
|
||||
}
|
||||
|
||||
func (m *IOManager) GetChannelCapabilities(channel string) OutputCapability {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
if dev, ok := m.devices[channel]; ok {
|
||||
if dev := m.lookupDevice(channel); dev != nil {
|
||||
return dev.OutputCapabilities()
|
||||
}
|
||||
return 0
|
||||
|
||||
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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user