From 13242329466002040c72df620d9c37d5c8a07f9b Mon Sep 17 00:00:00 2001 From: JianFeeeee Date: Sun, 4 Oct 2026 16:42:43 +0800 Subject: [PATCH] =?UTF-8?q?fix(agent)=E2=98=85:=20relay=20=E7=9A=84=20budg?= =?UTF-8?q?et=5Fcharged=20=E5=9B=9E=E4=BC=A0=E6=92=92=E8=B0=8E=20=E2=80=94?= =?UTF-8?q?=E2=80=94=20Agent=E2=86=94Agent=20=E6=98=8E=E6=98=8E=E6=89=A3?= =?UTF-8?q?=E4=BA=86=E5=8D=B4=E8=AF=B4=E6=B2=A1=E6=89=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit ## 实测发现的（Agent↔Agent 全场景回归中） relay→Agent : budget_charged=False ← 错 relay→人类 : budget_charged=False ← 对 普通→Agent : （无该字段 = 正常扣费） 查库确认 relay→Agent 的会话 **used_rounds=1（确实扣了）**， 而同一响应里回传 `budget_charged=false`。 ## 根因 if relay != "" { resp["relay"] = relay resp["budget_charged"] = false // ← 只看 relay 非空 } 而真正的计费判定在别处，且是对的： relayFree := false if relay != "" { human, _ := repo.IsHumanUser(ctx, to.Name) relayFree = human // ← 只有发给人类才免 } ## 为什么这个字段的错不是文案瑕疵 `relayFree` 的注释写着：「Agent→Agent 的 relay 照样扣会话预算， max_rounds 就能截断它」，理由是「Agent↔Agent 双方都不在做决定， 整个回路没有任何一处在计数」——生产上跑出过**41 封、间隔从 15 分钟 缩到 5 秒的无穷循环**（会话 f3d824ce）。 `budget_charged` 正是那条链上的判定依据之一：插件/宿主据此认为 「还有额度」而继续发。 ## 顺带实测确认的三道 Agent↔Agent 专属防线 1. **新建会话速率限制**：24 次 `.new` 里 18 次被拦 （1 小时上限 20 条），文案含剩余秒数。 2. **冷静期**：普通 Agent↔Agent 回信第 1 封进冷静期， 后续被拒 —— 提示「上一次连续 N 封」。 3. **relay 连转发上限 5**：★ relay **不是**绕过冷静期的后门 —— 实测第 6 封被「本会话已连续 5 封自动转发（上限 5）」挡住， 并说明「若确实需要继续，请由模型主动调 send_mail（不带 relay）」。 这三道防线都在，41 封循环的结构性防护完整。 ## 判据（2 格，2 个变异都转红） - relay→人类：确实免配额 + 回传 false - relay→Agent：**要扣** + 回传 true + 库里 used_rounds=1 变异①改回 `= false` → 转红； 变异②让 `relayFree` 对 Agent 也为 true → 转红 （第一次试这个变异时 0 红，查下去是**编译失败**（`human` 变量 变成 unused）—— 变异无效而不是判据无效。改成 `human || true` 才真正生效。**「0 红」的第一反应必须是「变异没生效」。**） 14 包全绿。 --- server/internal/handler/mail.go | 17 ++- server/internal/handler/relay_budget_test.go | 127 +++++++++++++++++++ 2 files changed, 142 insertions(+), 2 deletions(-) create mode 100644 server/internal/handler/relay_budget_test.go diff --git a/server/internal/handler/mail.go b/server/internal/handler/mail.go index c3de3f1..c911a29 100644 --- a/server/internal/handler/mail.go +++ b/server/internal/handler/mail.go @@ -658,9 +658,22 @@ func SendMail(w http.ResponseWriter, r *http.Request) { resp["budget_max"] = budget.Max } if relay != "" { - // 告知本次未扣预算，否则插件看到 budget_remaining 没变会以为数据错了 resp["relay"] = relay - resp["budget_charged"] = false + // ★ 必须用 relayFree，而不是 relay != ""（2026-10-04 实测修）。 + // + // 原写法在**任何** relay 上都回 budget_charged=false，而实际上： + // relay 发给**人类** → relayFree=true → 确实没扣 ✓ + // relay 发给**另一个 Agent** → relayFree=false → **扣了** + // （见上面 relayFree 的注释：Agent↔Agent 双方都不在做决定， + // 整个回路没有任何一处在计数 —— 生产上跑出过 41 封、间隔从 + // 15 分钟缩到 5 秒的无穷循环。免配额只给发往人类的 relay。） + // + // 后果：Agent↔Agent 的 relay 明明扣了预算，却告诉调用方没扣。 + // 插件据此判断「还有额度」而继续发 —— 正是那个循环的判定依据。 + // + // 实测证据：relay→Agent 的会话 used_rounds=1（确实扣了）， + // 而同一响应里 budget_charged=false。 + resp["budget_charged"] = !relayFree } if proposal != nil { // 回传规范化后的别名：Agent 提的名字可能含非法字符被改写过， diff --git a/server/internal/handler/relay_budget_test.go b/server/internal/handler/relay_budget_test.go new file mode 100644 index 0000000..30bcfc2 --- /dev/null +++ b/server/internal/handler/relay_budget_test.go @@ -0,0 +1,127 @@ +package handler + +/* +relay 的配额回传必须与实际扣费一致（2026-10-04 实测驱动）。 + +★ 缺陷：`resp["budget_charged"] = false` 只看 `relay != ""`， + 于是**任何** relay 都回「没扣预算」。而实际： + + relay → 人类 relayFree=true 确实没扣 回传 false ✓ + relay → 另一个 Agent relayFree=false **扣了** 回传 false ✗ 撒谎 + +★ 为什么这个字段的错很要紧（不是文案瑕疵）： + 插件据 budget_charged 判断「还有额度」，Agent↔Agent 的 relay + 明明扣了却被告知没扣 —— 而「Agent↔Agent 双方都不计数」正是 + 生产上那个 41 封、间隔 15 分钟→5 秒的无穷循环（会话 f3d824ce）。 + 这个字段就是那条链上的判定依据之一。 + +★ 实测证据：relay→Agent 的会话 used_rounds=1（确实扣了）， + 同一响应里 budget_charged=false。 +*/ + +import ( + "context" + "net/http" + "net/http/httptest" + "path/filepath" + "strings" + "testing" + + "github.com/agentmail/gateway/internal/db" + "github.com/agentmail/gateway/internal/middleware" + "github.com/agentmail/gateway/internal/models" +) + +func setupRelayBudgetDB(t *testing.T) { + t.Helper() + db.Close() + p := filepath.Join(t.TempDir(), "relay-budget.db") + if err := db.Connect(context.Background(), "sqlite://"+p); err != nil { + t.Fatalf("连接测试库: %v", err) + } + if err := db.Migrate(context.Background()); err != nil { + t.Fatalf("迁移: %v", err) + } + t.Cleanup(db.Close) +} + +func mustAgentForBudget(t *testing.T, name string) { + t.Helper() + if _, err := db.DB.ExecContext(context.Background(), + `INSERT INTO agents (agent_name, secret, platform) VALUES ($1,'x','test') + ON CONFLICT (agent_name) DO NOTHING`, name); err != nil { + t.Fatalf("建 agent %s: %v", name, err) + } +} + +func mustHumanForBudget(t *testing.T, name string) { + t.Helper() + if _, err := db.DB.ExecContext(context.Background(), + `INSERT INTO users (username, password_hash, role) VALUES ($1,'x','user') + ON CONFLICT (username) DO NOTHING`, name); err != nil { + t.Fatalf("建用户 %s: %v", name, err) + } +} + +func postMailAs(t *testing.T, agent, body string) *httptest.ResponseRecorder { + t.Helper() + req := httptest.NewRequest(http.MethodPost, "/api/v1/mail/send", strings.NewReader(body)) + req = req.WithContext(context.WithValue(req.Context(), middleware.AgentNameKey, agent)) + rec := httptest.NewRecorder() + SendMail(rec, req) + return rec +} + +// relay 发给人类：确实免配额，且回传 budget_charged=false +func TestRelayToHumanIsFreeAndReportsSo(t *testing.T) { + setupRelayBudgetDB(t) + mustAgentForBudget(t, "sender-h") + mustHumanForBudget(t, "human-h") + + rec := postMailAs(t, "sender-h", + `{"to":"human-h@/tmp.new","subject":"总结","body":"x","relay":"summary","relay_key":"h1"}`) + if rec.Code != http.StatusOK { + t.Fatalf("应 200，实际 %d：%s", rec.Code, rec.Body.String()) + } + body := rec.Body.String() + if !strings.Contains(body, `"budget_charged":false`) { + t.Errorf("relay 发给人类应回 budget_charged=false，实际：%s", body) + } +} + +// ★ relay 发给另一个 Agent：**要扣**预算，且回传必须说 true +func TestRelayToAgentIsChargedAndReportsTruthfully(t *testing.T) { + setupRelayBudgetDB(t) + mustAgentForBudget(t, "sender-a") + mustAgentForBudget(t, "receiver-a") + + rec := postMailAs(t, "sender-a", + `{"to":"receiver-a@/tmp.new","subject":"relay给Agent","body":"x","relay":"summary","relay_key":"a1"}`) + if rec.Code != http.StatusOK { + t.Fatalf("应 200，实际 %d：%s", rec.Code, rec.Body.String()) + } + body := rec.Body.String() + + // ① 回传不能说谎 + if !strings.Contains(body, `"budget_charged":true`) { + t.Errorf("★ relay 发给 Agent **扣了**预算，回传却说 budget_charged=false"+ + "（插件据此以为还有额度而继续发）：%s", body) + } + + // ② 而且库里的计数必须真的动了 + var sid string + if err := db.DB.QueryRowContext(context.Background(), + `SELECT session_id FROM mails ORDER BY created_at DESC LIMIT 1`).Scan(&sid); err != nil { + t.Fatalf("取会话: %v", err) + } + var used int + if err := db.DB.QueryRowContext(context.Background(), + `SELECT used_rounds FROM sessions WHERE session_id = $1`, sid).Scan(&used); err != nil { + t.Fatalf("读 used_rounds: %v", err) + } + if used != 1 { + t.Errorf("★ relay 发给 Agent 应扣 1 次预算，实际 used_rounds=%d", used) + } +} + +var _ = models.Mail{}