From a3ba1529e0989e15215f30ce8625eb1d9b1471fa Mon Sep 17 00:00:00 2001
From: ajmallesh <ajmallesh@gmail.com>
Date: Wed, 11 Feb 2026 13:37:23 -0800
Subject: [PATCH] fix: restore max_output_tokens error handling

---
 src/ai/message-handlers.ts | 9 +++++++++
 src/ai/types.ts            | 1 +
 2 files changed, 10 insertions(+)

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 {