fix(model): providerUpstreamError 的 error.type 也过令牌守卫

同一份上游报文,兄弟出口 streamEventError 守的正是 type,而 providerUpstreamError
只守了 error.code —— 网关/兼容层把回显塞进 type 时会被原样渲染进错误与日志。
两个字段现在用同一道 trace.ErrorToken:不像令牌的变空串,status/code/body_bytes
等诊断面不受影响。
This commit is contained in:
AskaEth committed 2026-10-26 19:01:16 +08:00
1 parent 92b0f89492
commit 98e1ae6aa9
2 files changed
+42 -6

No files matched your search

+5 -3
View File
@@ -40,9 +40,10 @@ func upstreamLabel(name, baseURL string) string {
// 而这类 error 会被 `chat run` / `loop step` 打日志。截断到 512 字节只是把
// 「整段泄漏」变成「前 512 字节泄漏」,不是修复。
//
// error.code 也要过令牌守卫(trace.ErrorToken):它在主流上游是枚举,但网关、代理与
// 兼容层可以往里面塞任何东西 —— 包括回显的输入。不像令牌的 code 一律变空串,代价只是
// 少一条分类提示;status / body 摘要等诊断面不受影响。
// error.code 与 error.type 也要过令牌守卫(trace.ErrorToken):它们在主流上游是枚举,
// 但网关、代理与兼容层可以往里面塞任何东西 —— 包括回显的输入。不像令牌的字段一律变空
// 串,代价只是少一条分类提示;status / body 摘要等诊断面不受影响。两个字段用同一道
// 守卫(不按字段分别处理),是因为「同一份上游报文,两个出口口径不一致」正是下发漏点。
//
// 双 %w:既保持 errors.Is(err, ErrUpstream)(降级链按它分类,主 → 备 → 本地),
// 也让 errors.Is(err, trace.ErrUpstream) 成立(调用方按上游失败分类)。
@@ -55,6 +56,7 @@ func providerUpstreamError(name, baseURL string, resp *http.Response) error {
return fmt.Errorf("model: %w: status %d", ErrUpstream, resp.StatusCode)
}
info.Code = trace.ErrorToken(info.Code)
info.Type = trace.ErrorToken(info.Type)
return fmt.Errorf("model: %w: %w", ErrUpstream, info)
}
+37 -3
View File
@@ -2,9 +2,10 @@ package model
// 本文件钉住 providerUpstreamError 的两条出镜边界:
//
// - **error.code 是自由文本面**:它在主流上游是枚举,但网关、代理与兼容层可以往
// 里面塞任何东西(包括回显的输入)。只有像令牌才出镜,其余变空串 —— 而 status /
// type / 体积摘要这些诊断面不受影响。
// - **error.code / error.type 是自由文本面**:它们在主流上游是枚举,但网关、代理与
// 兼容层可以往里面塞任何东西(包括回显的输入)。只有像令牌才出镜,其余变空串 ——
// 而 status / 体积摘要这些诊断面不受影响。两个字段必须同一道守卫:同一份上游报文
// 的两个出口(本文件与 streamEventError)口径不一致,就是一个漏点。
// - **响应体正文一个字节都不进错误**:这是 providerUpstreamError 存在的理由
// (旧实现把体截到 512B 拼进 error,而 `chat run` / `loop step` 会把它落日志)。
//
@@ -92,3 +93,36 @@ func TestProviderUpstreamErrorDropsBodyText(t *testing.T) {
t.Fatalf("error.message 的回显进了错误:%v", err)
}
}
// TestProviderUpstreamErrorDropsFreeTextType type 里是带空格标点的散文(网关 / 兼容层
// 把回显塞进 type 是真实写法)⇒ 该字段整体消失(连 ` type=` 这个键都不出现),而同一
// 份报文里的 status / code / body_bytes 仍在。去掉 type 的令牌守卫就变红 —— 兄弟出口
// streamEventError 守的正是 type,两个出口对同一份上游报文必须口径一致。
func TestProviderUpstreamErrorDropsFreeTextType(t *testing.T) {
t.Parallel()
const prose = "Invalid value: 她刚才说的那句话"
body := `{"error":{"code":"invalid_request_error","type":"` + prose + `"}}`
err := providerUpstreamError("openai", "https://api.example.com/v1/chat/completions",
upstreamErrResp(http.StatusBadRequest, body))
if err == nil {
t.Fatal("非 2xx 却报成功")
}
msg := err.Error()
if strings.Contains(msg, prose) {
t.Fatalf("type 里的回显进了错误(会随 chat run 落日志):%v", msg)
}
// ` type=` 带前导空格:content_type= 的尾巴里也有 "type=",不能用裸子串判。
if strings.Contains(msg, " type=") {
t.Fatalf("非令牌形状的 type 没有被丢弃:%v", msg)
}
// 诊断面不能一起丢:状态、令牌形状的 code、体积摘要都还在。
for _, want := range []string{"status=400", "code=invalid_request_error", "body_bytes=", "api.example.com"} {
if !strings.Contains(msg, want) {
t.Fatalf("错误里缺少 %q(诊断面不能一起丢):%v", want, msg)
}
}
if !errors.Is(err, ErrUpstream) || !errors.Is(err, trace.ErrUpstream) {
t.Fatalf("降级链的哨兵断了:%v", err)
}
}