mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 18:08:04 +00:00
1d4f2beeea
用户现场指出：**qq 插件的输出通道判据太严了**（那条判据在插件侧，已单独修：
`output_send__qq` 不再受"当前会话身份"限制）。同时内核提示词里还有一条**同类的凭空限制**：
「每轮对话**通常只需调用一次** output_send__{通道名} 即可完成回复。
仅在内容确实超过单条消息长度上限（如 >4000 字）时才拆分为多条」
可设计上输出是 agent 的**主动调用**：收到一次输入后，可以往**任意（已授权的）通道**
发**任意多次**（分段播报、先回执后结论、同时通知多个通道都合法）。这句话会让模型
自己收起合理的多次输出 —— 而且它不是任何机制的要求，只是当初为压 output-loop 写的
措辞（真正的防环机制是"回执只回 ok、不回传富结果"，那条保留）。
改法：
- 提示词改为明确授权：**输出次数与目标通道由你自己决定**，没有「一轮只能发一次」的限制；
只保留两条真话：单条长度上限（超长拆完整段落）、别反复重发**完全相同**的内容。
- `output_send__*` 的 `type` 参数改为**可选**（缺省 text）：判据该拦的是"不知道发什么"，
不是"没写众所周知的默认值"——此前缺 type 会直接失败并让模型重试一次。
判据 3 条（新增 `output_rules_test.go`）：提示词不得含输出次数限制且必须显式授权 /
省略 type 时按 text 发送成功且 schema 的 required 只有 payload / 空 payload 仍被拦。
(cherry picked from commit 17ea7fd5f0)
101 lines
3.4 KiB
Go
101 lines
3.4 KiB
Go
package core
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
|
||
agentAPI "gitcode.com/JianFeeeee/HomeAgent/internal/agent/api"
|
||
)
|
||
|
||
// 设计口径：输出是 agent 的**主动调用** —— 收到一次输入后，可以往任意（已授权的）
|
||
// 通道发**任意多次**（分段播报、先回执后结论、同时通知多个通道都合法）。
|
||
//
|
||
// 这条判据钉住的是"提示词里不得出现输出次数限制"。此前 `tooldefs.go` 里写着
|
||
// 「每轮对话通常只需调用一次 output_send__{通道名} 即可完成回复」—— 一条凭空的限制，
|
||
// 会让模型自己收起合理的多次输出（用户现场指出）。
|
||
func TestSystemPromptDoesNotRestrictOutputCount(t *testing.T) {
|
||
parent, _, _ := newRootForResidents(t)
|
||
defer parent.Stop()
|
||
|
||
prompt := parent.buildSystemPrompt("", "你好")
|
||
banned := []string{
|
||
"只需调用一次",
|
||
"只能调用一次",
|
||
"通常只需调用",
|
||
"不要重复发送",
|
||
}
|
||
for _, b := range banned {
|
||
if strings.Contains(prompt, b) {
|
||
t.Fatalf("系统提示词里仍有输出次数限制 %q —— 设计上次数不限", b)
|
||
}
|
||
}
|
||
if !strings.Contains(prompt, "输出次数与目标通道由你自己决定") {
|
||
t.Fatal("系统提示词应明确「输出次数与目标通道由你自己决定」")
|
||
}
|
||
if !strings.Contains(prompt, "没有任何「一轮只能发一次」的限制") {
|
||
t.Fatal("系统提示词应显式否认「一轮只能发一次」")
|
||
}
|
||
}
|
||
|
||
// 输出工具的 type 可省略，缺省按 text 处理（判据该拦的是"不知道发什么"，
|
||
// 不是"没写众所周知的默认值"）。
|
||
func TestOutputSendTypeDefaultsToText(t *testing.T) {
|
||
parent, _, _ := newRootForResidents(t)
|
||
defer parent.Stop()
|
||
|
||
dev := &outputTestDevice{name: "fakeout"}
|
||
if err := parent.io.RegisterDevice(dev); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
out := parent.executeOutputSendTool(agentAPI.ToolCall{
|
||
Name: "output_send__fakeout",
|
||
Arguments: map[string]interface{}{"payload": "只给 payload，不给 type"},
|
||
})
|
||
if out != "ok" {
|
||
t.Fatalf("省略 type 时应默认 text 并发送成功，得到 %q", out)
|
||
}
|
||
if len(dev.sent) != 1 {
|
||
t.Fatalf("通道应收到 1 次输出，得到 %d", len(dev.sent))
|
||
}
|
||
if args, _ := dev.sent[0]["args"].(map[string]interface{}); args["type"] != "text" {
|
||
t.Fatalf("缺省类型应为 text，实际 %v", args["type"])
|
||
}
|
||
|
||
// 工具 schema：required 只应含 payload
|
||
var found bool
|
||
for _, td := range parent.buildToolDefs() {
|
||
entry, _ := td.(map[string]interface{})
|
||
fn, _ := entry["function"].(map[string]interface{})
|
||
if n, _ := fn["name"].(string); n != "output_send__fakeout" {
|
||
continue
|
||
}
|
||
found = true
|
||
params, _ := fn["parameters"].(map[string]interface{})
|
||
req, _ := params["required"].([]string)
|
||
if len(req) != 1 || req[0] != "payload" {
|
||
t.Fatalf("output_send 的 required 应只有 payload，实际 %v", req)
|
||
}
|
||
}
|
||
if !found {
|
||
t.Fatal("未生成 output_send__fakeout 工具")
|
||
}
|
||
}
|
||
|
||
// 空 payload 仍应被拦（这条判据是对的：不知道发什么不能放过）。
|
||
func TestOutputSendStillRequiresPayload(t *testing.T) {
|
||
parent, _, _ := newRootForResidents(t)
|
||
defer parent.Stop()
|
||
|
||
if err := parent.io.RegisterDevice(&outputTestDevice{name: "fakeout"}); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
out := parent.executeOutputSendTool(agentAPI.ToolCall{
|
||
Name: "output_send__fakeout",
|
||
Arguments: map[string]interface{}{"type": "text"},
|
||
})
|
||
if !strings.Contains(out, "payload 不能为空") {
|
||
t.Fatalf("空 payload 应被拦，得到 %q", out)
|
||
}
|
||
}
|