1package provider
2
3import (
4 "errors"
5 "testing"
6)
7
8func TestIsContextOverflowError(t *testing.T) {
9 tests := []struct {
10 name string
11 err error
12 want bool
13 }{
14 {name: "nil error", err: nil, want: false},
15 {name: "http 413", err: errors.New("provider http 413: request rejected"), want: true},
16 {name: "response 413", err: errors.New("provider response 413: request rejected"), want: true},
17 {name: "HTTP status code 413", err: errors.New("provider request failed: HTTP status code: 413"), want: true},
18 {name: "payload too large", err: errors.New("proxy rejected request: Payload Too Large"), want: true},
19 {name: "request entity too large", err: errors.New("gateway: Request Entity Too Large"), want: true},
20 {name: "numeric value is not HTTP 413", err: errors.New("provider http 400: max_tokens cannot exceed 41344"), want: false},
21 {name: "unqualified numeric value is not HTTP 413", err: errors.New("max_tokens cannot exceed 413"), want: false},
22 {name: "context error code", err: errors.New("context_length_exceeded"), want: true},
23 {name: "maximum context length", err: errors.New("This model's maximum context length is 128000 tokens"), want: true},
24 {name: "context window exceeded", err: errors.New("context window exceeded"), want: true},
25 {name: "maximum input token count", err: errors.New("input token count exceeds the maximum number of tokens allowed"), want: true},
26 {name: "maximum output token count", err: errors.New("max_tokens exceeds the maximum number of tokens allowed"), want: false},
27 {name: "unrelated error", err: errors.New("rate limit exceeded"), want: false},
28 }
29
30 for _, tt := range tests {
31 t.Run(tt.name, func(t *testing.T) {
32 if got := IsContextOverflowError(tt.err); got != tt.want {
33 t.Fatalf("IsContextOverflowError(%q) = %t, want %t", tt.err, got, tt.want)
34 }
35 })
36 }
37}