diff --git a/src/ai/message-handlers.ts b/src/ai/message-handlers.ts index 1fc507e..57cdd4e 100644 --- a/src/ai/message-handlers.ts +++ b/src/ai/message-handlers.ts @@ -166,6 +166,15 @@ function handleStructuredError( false // Not retryable - needs code fix ), }; + case 'max_output_tokens': + return { + detected: true, + shouldThrow: new PentestError( + `Max output tokens reached: ${content.slice(0, 100)}`, + 'billing', + true // Retryable - may succeed with different content + ), + }; case 'unknown': default: return { detected: true }; diff --git a/src/ai/types.ts b/src/ai/types.ts index 006756d..f742dd3 100644 --- a/src/ai/types.ts +++ b/src/ai/types.ts @@ -73,6 +73,7 @@ export type SDKAssistantMessageError = | 'rate_limit' | 'invalid_request' | 'server_error' + | 'max_output_tokens' | 'unknown'; export interface AssistantMessage {