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 保留」。
64 lines
1.9 KiB
Go
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)
|
|
}
|
|
}
|