From 839686c23cc9e60dba9336d34137ae784fda189b Mon Sep 17 00:00:00 2001
From: ajmallesh <ajmallesh@gmail.com>
Date: Fri, 20 Feb 2026 07:49:53 -0800
Subject: [PATCH] refactor: use SDK-exported SDKAssistantMessageError instead
 of local type definition

---
 src/ai/message-handlers.ts |  2 +-
 src/ai/types.ts            | 10 ++--------
 src/services/preflight.ts  |  3 ++-
 3 files changed, 5 insertions(+), 10 deletions(-)

diff --git a/src/ai/message-handlers.ts b/src/ai/message-handlers.ts
index e60ddb5..78068ad 100644
--- a/src/ai/message-handlers.ts
+++ b/src/ai/message-handlers.ts
@@ -19,9 +19,9 @@ import {
 } from './output-formatters.js';
 import type { AuditLogger } from './audit-logger.js';
 import type { ProgressManager } from './progress-manager.js';
+import type { SDKAssistantMessageError } from '@anthropic-ai/claude-agent-sdk';
 import type {
   AssistantMessage,
-  SDKAssistantMessageError,
   ResultMessage,
   ToolUseMessage,
   ToolResultMessage,
diff --git a/src/ai/types.ts b/src/ai/types.ts
index af1a6f6..ee98bdf 100644
--- a/src/ai/types.ts
+++ b/src/ai/types.ts
@@ -6,6 +6,8 @@
 
 // Type definitions for Claude executor message processing pipeline
 
+import type { SDKAssistantMessageError } from '@anthropic-ai/claude-agent-sdk';
+
 export interface ExecutionContext {
   isParallelExecution: boolean;
   useCleanOutput: boolean;
@@ -51,14 +53,6 @@ export interface ContentBlock {
   text?: string;
 }
 
-export type SDKAssistantMessageError =
-  | 'authentication_failed'
-  | 'billing_error'
-  | 'rate_limit'
-  | 'invalid_request'
-  | 'server_error'
-  | 'max_output_tokens'
-  | 'unknown';
 
 export interface AssistantMessage {
   type: 'assistant';
diff --git a/src/services/preflight.ts b/src/services/preflight.ts
index b5f45a6..95d6a08 100644
--- a/src/services/preflight.ts
+++ b/src/services/preflight.ts
@@ -18,7 +18,8 @@
  */
 
 import fs from 'fs/promises';
-import { query, type SDKAssistantMessageError } from '@anthropic-ai/claude-agent-sdk';
+import { query } from '@anthropic-ai/claude-agent-sdk';
+import type { SDKAssistantMessageError } from '@anthropic-ai/claude-agent-sdk';
 import { PentestError, isRetryableError } from './error-handling.js';
 import { ErrorCode } from '../types/errors.js';
 import { type Result, ok, err } from '../types/result.js';