Merge pull request #59 from Vasanthdev2004/gpt4o-max-tokens-test

test: cover OpenAI max token caps for gpt-4o and GPT-5.4

src/utils/context.test.ts

@@ -35,3 +35,54 @@ test('deepseek-chat clamps oversized max output overrides to the provider limit'
 
   expect(getMaxOutputTokensForModel('deepseek-chat')).toBe(8_192)
 })
+
+test('gpt-4o uses provider-specific context and output caps', () => {
+  process.env.CLAUDE_CODE_USE_OPENAI = '1'
+  delete process.env.CLAUDE_CODE_MAX_OUTPUT_TOKENS
+
+  expect(getContextWindowForModel('gpt-4o')).toBe(128_000)
+  expect(getModelMaxOutputTokens('gpt-4o')).toEqual({
+    default: 16_384,
+    upperLimit: 16_384,
+  })
+  expect(getMaxOutputTokensForModel('gpt-4o')).toBe(16_384)
+})
+
+test('gpt-4o clamps oversized max output overrides to the provider limit', () => {
+  process.env.CLAUDE_CODE_USE_OPENAI = '1'
+  process.env.CLAUDE_CODE_MAX_OUTPUT_TOKENS = '32000'
+
+  expect(getMaxOutputTokensForModel('gpt-4o')).toBe(16_384)
+})
+
+test('gpt-5.4 family uses provider-specific context and output caps', () => {
+  process.env.CLAUDE_CODE_USE_OPENAI = '1'
+  delete process.env.CLAUDE_CODE_MAX_OUTPUT_TOKENS
+
+  expect(getContextWindowForModel('gpt-5.4')).toBe(1_050_000)
+  expect(getModelMaxOutputTokens('gpt-5.4')).toEqual({
+    default: 128_000,
+    upperLimit: 128_000,
+  })
+
+  expect(getContextWindowForModel('gpt-5.4-mini')).toBe(400_000)
+  expect(getModelMaxOutputTokens('gpt-5.4-mini')).toEqual({
+    default: 128_000,
+    upperLimit: 128_000,
+  })
+
+  expect(getContextWindowForModel('gpt-5.4-nano')).toBe(400_000)
+  expect(getModelMaxOutputTokens('gpt-5.4-nano')).toEqual({
+    default: 128_000,
+    upperLimit: 128_000,
+  })
+})
+
+test('gpt-5.4 family keeps large max output overrides within provider limits', () => {
+  process.env.CLAUDE_CODE_USE_OPENAI = '1'
+  process.env.CLAUDE_CODE_MAX_OUTPUT_TOKENS = '200000'
+
+  expect(getMaxOutputTokensForModel('gpt-5.4')).toBe(128_000)
+  expect(getMaxOutputTokensForModel('gpt-5.4-mini')).toBe(128_000)
+  expect(getMaxOutputTokensForModel('gpt-5.4-nano')).toBe(128_000)
+})