Files
AskaEth 3001fcc716 fix(model): providerUpstreamError 的 error.code 过令牌守卫
error.code 与 error.type 一样是**自由文本面**:主流上游里是枚举,但网关、代理与
兼容层可以往里塞任何东西(包括回显的用户原文)。M41 已给 type 加了守卫,
providerUpstreamError 的 code 一直原样带出,是这一族里唯一还漏的字段。

守卫本色从 runtime/model(safeErrorToken)与 runtime/multimodal/voice
(dashscopeErrorToken)两份逐字同形的实现上提到 trace.ErrorToken,model 侧改为
调用它(voice 侧仍是自己那份,不在本切片文件面内,见随附 finding)。

唯一的行为变化就是「不像令牌的 code 变空串」:status / type / body 体积与 sha256、
错误文案模板、双 %w 哨兵链(ErrUpstream、trace.ErrUpstream)都不变。

验收:go build ./... + go vet ./... + trace/model 包 test 全绿;新增用例覆盖
「带空格标点的 code 变空串且 status/type 仍在」与「令牌形状的 code 保留」。
2026-10-26 18:45:38 +08:00

64 lines
1.9 KiB
Go

package trace
// 本文件钉住 ErrorToken 的唯一职责:error.code / error.type 这类**自由文本面**只有
// 「像协议令牌」才允许出镜。它们在上游里是枚举,但网关、代理与兼容层可以往里面塞任何
// 东西 —— 包括回显的用户原文。
//
// 每条用例都是「把它放宽就变红」的形状。
import (
"strings"
"testing"
)
// TestErrorTokenKeepsProtocolTokens 真实枚举必须原样通过:丢掉它们的代价是少一条
// 分类提示(重试 / 降级 / 熔断的口径)。
func TestErrorTokenKeepsProtocolTokens(t *testing.T) {
t.Parallel()
for _, token := range []string{
"invalid_request_error",
"overloaded_error",
"Throttling.RateQuota",
"rate-limit",
"429",
" sparse_error ",
} {
if got := ErrorToken(token); got != strings.TrimSpace(token) {
t.Fatalf("ErrorToken(%q) = %q, want 令牌原值(去首尾空白)", token, got)
}
}
}
// TestErrorTokenDropsFreeText 只有像令牌才出镜:带空格、标点、引号的散文一律空串 ——
// 那正是「上游把用户原文回显进 code」的形态。
func TestErrorTokenDropsFreeText(t *testing.T) {
t.Parallel()
for _, text := range []string{
"Invalid value: '她刚才说的那句话'",
"prompt is too long",
"```json\n{\"a\":1}\n```",
"bad,code",
"code=1",
"",
" ",
"line\nbreak",
"emoji🙂",
} {
if got := ErrorToken(text); got != "" {
t.Fatalf("ErrorToken(%q) = %q, want 空串", text, got)
}
}
}
// TestErrorTokenRejectsOverlong 超长串不是令牌(边界 64 字节),否则「话」可以靠长度
// 混进来。
func TestErrorTokenRejectsOverlong(t *testing.T) {
t.Parallel()
if got := ErrorToken(strings.Repeat("a", 64)); got == "" {
t.Fatal("64 字节的令牌被误杀(边界应含)")
}
if got := ErrorToken(strings.Repeat("a", 65)); got != "" {
t.Fatalf("超长串被放行:%q", got)
}
}