fix(model): providerUpstreamError 的 error.type 也过令牌守卫
同一份上游报文,兄弟出口 streamEventError 守的正是 type,而 providerUpstreamError 只守了 error.code —— 网关/兼容层把回显塞进 type 时会被原样渲染进错误与日志。 两个字段现在用同一道 trace.ErrorToken:不像令牌的变空串,status/code/body_bytes 等诊断面不受影响。
This commit is contained in:
1 parent
92b0f89492
commit
98e1ae6aa9
2 files changed
+42
-6
No files matched your search
@@ -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)
|
||||
}
|
||||
|
||||
|
||||
@@ -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)
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user