修复 AI 网关流式断流:错误透传、自动降级,max_tokens 可缺省

2026-07-13 15:06:01 +08:00
parent 81d4650f3d
commit 9309ad1ffc
11 changed files with 420 additions and 28 deletions
@@ -85,6 +85,8 @@ func TestAiGatewayKeyModelRestrict(t *testing.T) {
 			`{"model":"meta.llama-3.3-70b-instruct","messages":[{"role":"user","content":"hi"}]}`, 404, deny},
 		{"chat completions 白名单内穿透", "limited-key-1234", "/ai/v1/chat/completions",
 			`{"model":"ghost-model","messages":[{"role":"user","content":"hi"}]}`, 404, pass},
+		{"messages 缺 max_tokens 按默认值放行", "open-key-12345", "/ai/v1/messages",
+			`{"model":"ghost-model","messages":[{"role":"user","content":"hi"}]}`, 404, pass},
 		{"chat completions 缺 messages 拒绝", "open-key-12345", "/ai/v1/chat/completions",
 			`{"model":"ghost-model"}`, 400, []string{"invalid_request_error"}},
 		{"chat completions 非 function 工具拒绝", "open-key-12345", "/ai/v1/chat/completions",