Function TestIsContextLimitKey

coderd/x/chatd/chatloop/contextlimit_internal_test.go:79–130

(t *testing.T)

}

func TestIsContextLimitKey(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name string
		key  string
		want bool
	}{ // Exact matches after normalization.
		{name: "context_limit", key: "context_limit", want: true},
		{name: "context_window", key: "context_window", want: true},
		{name: "context_length", key: "context_length", want: true},
		{name: "max_context", key: "max_context", want: true},
		{name: "max_context_tokens", key: "max_context_tokens", want: true},
		{name: "max_input_tokens", key: "max_input_tokens", want: true},
		{name: "max_input_token", key: "max_input_token", want: true},
		{name: "input_token_limit", key: "input_token_limit", want: true},

		// Case and separator variations.
		{name: "Context-Window mixed case", key: "Context-Window", want: true},
		{name: "MAX_CONTEXT_TOKENS screaming", key: "MAX_CONTEXT_TOKENS", want: true},
		{name: "contextLimit camelCase", key: "contextLimit", want: true},
		{name: "modelContextLimit camelCase", key: "modelContextLimit", want: true},

		// Fallback heuristic: tokenized "context" + limit/window/length.
		{name: "model_context_limit", key: "model_context_limit", want: true},
		{name: "context_window_size", key: "context_window_size", want: true},
		{name: "context_length_max", key: "context_length_max", want: true},

		// Exact matches remain valid after separator stripping.
		{name: "max_context_", key: "max_context_", want: true},
		{name: "max_context_limit", key: "max_context_limit", want: true},

		// Non-matching keys should not be treated as context limits.
		{name: "max_context_version false positive", key: "max_context_version", want: false},
		{name: "context_tokens_used false positive", key: "context_tokens_used", want: false},
		{name: "context_length_used false positive", key: "context_length_used", want: false},
		{name: "context_window_used false positive", key: "context_window_used", want: false},
		{name: "context_id no limit keyword", key: "context_id", want: false},
		{name: "empty string", key: "", want: false},
		{name: "unrelated key", key: "model_name", want: false},
		{name: "limit without context", key: "rate_limit", want: false},
		{name: "max without context", key: "max_tokens", want: false},
		{name: "context alone", key: "context", want: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()
			got := isContextLimitKey(tt.key)
			require.Equal(t, tt.want, got)
		})
	}
}

func TestNumericContextLimitValue(t *testing.T) {
	t.Parallel()