@posthog/wizard 2.2.0 → 2.4.0

package/dist/src/lib/__tests__/agent-interface.test.js

@@ -263,5 +263,27 @@ describe('createStopHook', () => {
         const extra = hook(hookInput); // still allow
         expect(extra).toEqual({});
     });
+    it('allows stop immediately on API error (401)', () => {
+        const collectedText = [
+            'Failed to authenticate. API Error: 401 {"detail":"Authentication required"}',
+        ];
+        const hook = (0, agent_interface_1.createStopHook)([wizard_session_1.AdditionalFeature.LLM], collectedText);
+        const result = hook(hookInput);
+        expect(result).toEqual({});
+    });
+    it('allows stop immediately on generic API error', () => {
+        const collectedText = ['API Error: 500 Internal Server Error'];
+        const hook = (0, agent_interface_1.createStopHook)([wizard_session_1.AdditionalFeature.LLM], collectedText);
+        const result = hook(hookInput);
+        expect(result).toEqual({});
+    });
+    it('proceeds normally when collectedText has no API error', () => {
+        const collectedText = ['Some normal agent output'];
+        const hook = (0, agent_interface_1.createStopHook)([], collectedText);
+        // First call → remark prompt (normal behavior)
+        const first = hook(hookInput);
+        expect(first).toHaveProperty('decision', 'block');
+        expect(first.reason).toContain('WIZARD-REMARK');
+    });
 });
 //# sourceMappingURL=agent-interface.test.js.map

package/dist/src/lib/__tests__/agent-interface.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-interface.test.js","sourceRoot":"","sources":["../../../../src/lib/__tests__/agent-interface.test.ts"],"names":[],"mappings":";;AAAA,wDAA8D;AAG9D,sDAG2B;AAE3B,oBAAoB;AACpB,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AACnC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAE/B,sBAAsB;AACtB,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AAC5B,IAAI,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC;IACjD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;CAClD,CAAC,CAAC,CAAC;AAEJ,oBAAoB;AACpB,MAAM,cAAc,GAAG;IACrB,GAAG,EAAE;QACH,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;QAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;KAChB;IACD,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;IAClB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;IACjB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;IAClB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;IACf,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;IACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;IACf,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;IACrB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;IACtB,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC7B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;IACpB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;CACvB,CAAC;AACF,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3B,KAAK,EAAE,GAAG,EAAE,CAAC,cAAc;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,IAAI,WAIH,CAAC;IAEF,MAAM,cAAc,GAAkB;QACpC,KAAK,EAAE,KAAK;QACZ,UAAU,EAAE,WAAW;QACvB,YAAY,EAAE,KAAK;QACnB,OAAO,EAAE,KAAK;QACd,MAAM,EAAE,KAAK;QACb,QAAQ,EAAE,KAAK;QACf,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,KAAK;QACX,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,KAAK;KAClB,CAAC;IAEF,MAAM,kBAAkB,GAAG;QACzB,gBAAgB,EAAE,WAAW;QAC7B,UAAU,EAAE,EAAE;QACd,KAAK,EAAE,0BAA0B;KAClC,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,WAAW,GAAG;YACZ,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;SACnB,CAAC;QAEF,cAAc,CAAC,OAAO,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACpD,kBAAkB;QAClB,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;YAC9F,qCAAqC;YACrC,yCAAyC;YACzC,6DAA6D;YAC7D,iEAAiE;YACjE,yBAAyB;YACzB,8EAA8E;YAE9E,QAAQ,CAAC,CAAC,6BAA6B;gBACrC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,MAAM,EAAE,8BAA8B;iBACvC,CAAC;gBAEF,2DAA2D;gBAC3D,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;YAC1E,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,6BAA6B,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,qEAAqE;YAErE,QAAQ,CAAC,CAAC,0BAA0B;gBAClC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,mCAAmC;gBACnC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtC,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,0BAA0B,EAAE,CAAC,CAAC;YAExD,MAAM,MAAM,CACV,IAAA,0BAAQ,EACN,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YAEtC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,2DAA2D;YAC3D,mEAAmE;YAEnE,QAAQ,CAAC,CAAC,4BAA4B;gBACpC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS,EAAE,2CAA2C;oBAC/D,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,sCAAsC;iBAC/C,CAAC;gBAEF,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,4BAA4B,EAAE,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,uCAAuC;YACvC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,gFAAgF;YAChF,iEAAiE;YACjE,+DAA+D;YAC/D,0CAA0C;YAC1C,mEAAmE;YACnE,EAAE;YACF,0EAA0E;YAC1E,4DAA4D;YAE5D,QAAQ,CAAC,CAAC,yCAAyC;gBACjD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,sDAAsD;gBACtD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,SAAS,EAAE,GAAG;oBACd,MAAM,EAAE,oDAAoD;oBAC5D,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,IAAI;iBACrB,CAAC;gBAEF,iEAAiE;gBACjE,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,wBAAwB;oBACjC,QAAQ,EAAE,IAAI;oBACd,SAAS,EAAE,CAAC;oBACZ,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,CAAC;oBACjB,MAAM,EAAE;wBACN,sDAAsD;wBACtD,qDAAqD;wBACrD,qDAAqD;wBACrD,mCAAmC;qBACpC;iBACF,CAAC;YACJ,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,yCAAyC,EAAE,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAE9D,wEAAwE;YACxE,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,SAAS,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC;IAE9C,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,EAAE,CAAC,CAAC;QAEhC,6BAA6B;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAExE,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAErD,kCAAkC;QAClC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,IAAI,CAC/C,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,8BAA8B;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,CAAE,MAA6B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAEzE,0BAA0B;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,+DAA+D;QAC/D,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,EAAE,kCAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAE5E,0BAA0B;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,IAAI,CAC/C,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,iCAAiC;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,CAAE,MAA6B,CAAC,MAAM,CAAC,CAAC,IAAI,CAChD,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,6BAA6B;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAExE,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,EAAE,CAAC,CAAC;QAEhC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QAC1B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { runAgent, createStopHook } from '../agent-interface';\nimport type { WizardOptions } from '../../utils/types';\nimport type { SpinnerHandle } from '../../ui';\nimport {\n  AdditionalFeature,\n  ADDITIONAL_FEATURE_PROMPTS,\n} from '../wizard-session';\n\n// Mock dependencies\njest.mock('../../utils/analytics');\njest.mock('../../utils/debug');\n\n// Mock the SDK module\nconst mockQuery = jest.fn();\njest.mock('@anthropic-ai/claude-agent-sdk', () => ({\n  query: (...args: unknown[]) => mockQuery(...args),\n}));\n\n// Mock the UI layer\nconst mockUIInstance = {\n  log: {\n    step: jest.fn(),\n    success: jest.fn(),\n    error: jest.fn(),\n    warn: jest.fn(),\n    info: jest.fn(),\n  },\n  spinner: jest.fn(),\n  select: jest.fn(),\n  confirm: jest.fn(),\n  text: jest.fn(),\n  intro: jest.fn(),\n  outro: jest.fn(),\n  cancel: jest.fn(),\n  note: jest.fn(),\n  isCancel: jest.fn(),\n  setDetectedFramework: jest.fn(),\n  setCredentials: jest.fn(),\n  pushStatus: jest.fn(),\n  setLoginUrl: jest.fn(),\n  showBlockingOutage: jest.fn(),\n  setReadinessWarnings: jest.fn(),\n  showSettingsOverride: jest.fn(),\n  startRun: jest.fn(),\n  syncTodos: jest.fn(),\n  groupMultiselect: jest.fn(),\n  multiselect: jest.fn(),\n};\njest.mock('../../ui', () => ({\n  getUI: () => mockUIInstance,\n}));\n\ndescribe('runAgent', () => {\n  let mockSpinner: {\n    start: jest.Mock;\n    stop: jest.Mock;\n    message: jest.Mock;\n  };\n\n  const defaultOptions: WizardOptions = {\n    debug: false,\n    installDir: '/test/dir',\n    forceInstall: false,\n    default: false,\n    signup: false,\n    localMcp: false,\n    ci: false,\n    menu: false,\n    benchmark: false,\n    yaraReport: false,\n  };\n\n  const defaultAgentConfig = {\n    workingDirectory: '/test/dir',\n    mcpServers: {},\n    model: 'claude-opus-4-5-20251101',\n  };\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    mockSpinner = {\n      start: jest.fn(),\n      stop: jest.fn(),\n      message: jest.fn(),\n    };\n\n    mockUIInstance.spinner.mockReturnValue(mockSpinner);\n    // Reset log mocks\n    Object.values(mockUIInstance.log).forEach((fn) => fn.mockReset());\n  });\n\n  describe('race condition handling', () => {\n    it('should return success when agent completes successfully then SDK cleanup fails', async () => {\n      // This simulates the race condition:\n      // 1. Agent completes with success result\n      // 2. signalDone() is called, completing the prompt generator\n      // 3. SDK tries to send cleanup command while streaming is active\n      // 4. SDK throws an error\n      // The fix should recognize we already got a success and return success anyway\n\n      function* mockGeneratorWithCleanupError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          result: 'Agent completed successfully',\n        };\n\n        // Simulate the SDK cleanup error that occurs after success\n        throw new Error('only prompt commands are supported in streaming mode');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithCleanupError());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not throw\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n    });\n\n    it('should still throw when no success result was received before error', async () => {\n      // If we never got a success result, errors should propagate normally\n\n      function* mockGeneratorWithOnlyError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // No success result, just an error\n        throw new Error('Actual SDK error');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithOnlyError());\n\n      await expect(\n        runAgent(\n          defaultAgentConfig,\n          'test prompt',\n          defaultOptions,\n          mockSpinner as unknown as SpinnerHandle,\n          {\n            successMessage: 'Test success',\n            errorMessage: 'Test error',\n          },\n        ),\n      ).rejects.toThrow('Actual SDK error');\n\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test error');\n    });\n\n    it('should not treat error results as success', async () => {\n      // A result with is_error: true should not count as success\n      // Even if subtype is 'success', the is_error flag takes precedence\n\n      function* mockGeneratorWithErrorResult() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success', // subtype can be success but is_error true\n          is_error: true,\n          result: 'API Error: 500 Internal Server Error',\n        };\n\n        throw new Error('Process exited with code 1');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithErrorResult());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return API error, not success\n      expect(result.error).toBe('WIZARD_API_ERROR');\n      expect(result.message).toContain('API Error');\n    });\n\n    it('should suppress user-facing errors when SDK yields error result after success', async () => {\n      // This test models actual SDK behavior where the SDK emits TWO result messages:\n      // 1. SDK yields success result (num_turns: 105, is_error: false)\n      // 2. SDK yields a SECOND result with is_error: true containing\n      //    accumulated cleanup/telemetry errors\n      // 3. The errors should be logged to file but NOT shown to the user\n      //\n      // This differs from the thrown exception test above - here the SDK YIELDS\n      // an error result message instead of THROWING an exception.\n\n      function* mockGeneratorWithYieldedErrorAfterSuccess() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // First result: success (this is the real completion)\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          num_turns: 105,\n          result: '[WIZARD-REMARK] Integration completed successfully',\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 5.83,\n        };\n\n        // Second result: error (SDK cleanup noise - yielded, not thrown)\n        yield {\n          type: 'result',\n          subtype: 'error_during_execution',\n          is_error: true,\n          num_turns: 0,\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 0,\n          errors: [\n            'only prompt commands are supported in streaming mode',\n            'Error: 1P event logging: 14 events failed to export',\n            'Error: 1P event logging: 13 events failed to export',\n            'Error: Failed to export 14 events',\n          ],\n        };\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithYieldedErrorAfterSuccess());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not error\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n\n      // ui.log.error should NOT have been called (errors suppressed for user)\n      expect(mockUIInstance.log.error).not.toHaveBeenCalled();\n    });\n  });\n});\n\ndescribe('createStopHook', () => {\n  const hookInput = { stop_hook_active: false };\n\n  it('empty queue: first call blocks for remark, second allows stop', () => {\n    const hook = createStopHook([]);\n\n    // First call → remark prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Second call → allow stop\n    const second = hook(hookInput);\n    expect(second).toEqual({});\n  });\n\n  it('single feature: feature prompt, then remark, then allow stop', () => {\n    const hook = createStopHook([AdditionalFeature.LLM]);\n\n    // First call → LLM feature prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Second call → remark prompt\n    const second = hook(hookInput);\n    expect(second).toHaveProperty('decision', 'block');\n    expect((second as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Third call → allow stop\n    const third = hook(hookInput);\n    expect(third).toEqual({});\n  });\n\n  it('multiple queue entries: drains all, then remark, then allow stop', () => {\n    // Queue the same feature twice to exercise multi-item draining\n    const hook = createStopHook([AdditionalFeature.LLM, AdditionalFeature.LLM]);\n\n    // First call → LLM prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Second call → LLM prompt again\n    const second = hook(hookInput);\n    expect(second).toHaveProperty('decision', 'block');\n    expect((second as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Third call → remark prompt\n    const third = hook(hookInput);\n    expect(third).toHaveProperty('decision', 'block');\n    expect((third as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Fourth call → allow stop\n    const fourth = hook(hookInput);\n    expect(fourth).toEqual({});\n  });\n\n  it('allow stop is idempotent after all phases complete', () => {\n    const hook = createStopHook([]);\n\n    hook(hookInput); // remark\n    hook(hookInput); // allow\n    const extra = hook(hookInput); // still allow\n    expect(extra).toEqual({});\n  });\n});\n"]}
+{"version":3,"file":"agent-interface.test.js","sourceRoot":"","sources":["../../../../src/lib/__tests__/agent-interface.test.ts"],"names":[],"mappings":";;AAAA,wDAA8D;AAG9D,sDAG2B;AAE3B,oBAAoB;AACpB,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AACnC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAE/B,sBAAsB;AACtB,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AAC5B,IAAI,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC;IACjD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;CAClD,CAAC,CAAC,CAAC;AAEJ,oBAAoB;AACpB,MAAM,cAAc,GAAG;IACrB,GAAG,EAAE;QACH,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;QAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;KAChB;IACD,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;IAClB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;IACjB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;IAClB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;IACf,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;IACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;IACf,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;IACrB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;IACtB,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC7B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;IACpB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;CACvB,CAAC;AACF,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3B,KAAK,EAAE,GAAG,EAAE,CAAC,cAAc;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,IAAI,WAIH,CAAC;IAEF,MAAM,cAAc,GAAkB;QACpC,KAAK,EAAE,KAAK;QACZ,UAAU,EAAE,WAAW;QACvB,YAAY,EAAE,KAAK;QACnB,OAAO,EAAE,KAAK;QACd,MAAM,EAAE,KAAK;QACb,QAAQ,EAAE,KAAK;QACf,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,KAAK;QACX,SAAS,EAAE,KAAK;QAChB,UAAU,EAAE,KAAK;KAClB,CAAC;IAEF,MAAM,kBAAkB,GAAG;QACzB,gBAAgB,EAAE,WAAW;QAC7B,UAAU,EAAE,EAAE;QACd,KAAK,EAAE,0BAA0B;KAClC,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,WAAW,GAAG;YACZ,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;SACnB,CAAC;QAEF,cAAc,CAAC,OAAO,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACpD,kBAAkB;QAClB,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;YAC9F,qCAAqC;YACrC,yCAAyC;YACzC,6DAA6D;YAC7D,iEAAiE;YACjE,yBAAyB;YACzB,8EAA8E;YAE9E,QAAQ,CAAC,CAAC,6BAA6B;gBACrC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,MAAM,EAAE,8BAA8B;iBACvC,CAAC;gBAEF,2DAA2D;gBAC3D,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;YAC1E,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,6BAA6B,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,qEAAqE;YAErE,QAAQ,CAAC,CAAC,0BAA0B;gBAClC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,mCAAmC;gBACnC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtC,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,0BAA0B,EAAE,CAAC,CAAC;YAExD,MAAM,MAAM,CACV,IAAA,0BAAQ,EACN,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YAEtC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,2DAA2D;YAC3D,mEAAmE;YAEnE,QAAQ,CAAC,CAAC,4BAA4B;gBACpC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS,EAAE,2CAA2C;oBAC/D,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,sCAAsC;iBAC/C,CAAC;gBAEF,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,4BAA4B,EAAE,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,uCAAuC;YACvC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,gFAAgF;YAChF,iEAAiE;YACjE,+DAA+D;YAC/D,0CAA0C;YAC1C,mEAAmE;YACnE,EAAE;YACF,0EAA0E;YAC1E,4DAA4D;YAE5D,QAAQ,CAAC,CAAC,yCAAyC;gBACjD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,sDAAsD;gBACtD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,SAAS,EAAE,GAAG;oBACd,MAAM,EAAE,oDAAoD;oBAC5D,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,IAAI;iBACrB,CAAC;gBAEF,iEAAiE;gBACjE,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,wBAAwB;oBACjC,QAAQ,EAAE,IAAI;oBACd,SAAS,EAAE,CAAC;oBACZ,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,CAAC;oBACjB,MAAM,EAAE;wBACN,sDAAsD;wBACtD,qDAAqD;wBACrD,qDAAqD;wBACrD,mCAAmC;qBACpC;iBACF,CAAC;YACJ,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,yCAAyC,EAAE,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAE9D,wEAAwE;YACxE,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,SAAS,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC;IAE9C,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,EAAE,CAAC,CAAC;QAEhC,6BAA6B;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAExE,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAErD,kCAAkC;QAClC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,IAAI,CAC/C,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,8BAA8B;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,CAAE,MAA6B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAEzE,0BAA0B;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,+DAA+D;QAC/D,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,EAAE,kCAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAE5E,0BAA0B;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,IAAI,CAC/C,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,iCAAiC;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,CAAE,MAA6B,CAAC,MAAM,CAAC,CAAC,IAAI,CAChD,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,6BAA6B;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAExE,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,EAAE,CAAC,CAAC;QAEhC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QAC1B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,aAAa,GAAG;YACpB,6EAA6E;SAC9E,CAAC;QACF,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,CAAC,EAAE,aAAa,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,aAAa,GAAG,CAAC,sCAAsC,CAAC,CAAC;QAC/D,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,CAAC,EAAE,aAAa,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,aAAa,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,EAAE,EAAE,aAAa,CAAC,CAAC;QAE/C,+CAA+C;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { runAgent, createStopHook } from '../agent-interface';\nimport type { WizardOptions } from '../../utils/types';\nimport type { SpinnerHandle } from '../../ui';\nimport {\n  AdditionalFeature,\n  ADDITIONAL_FEATURE_PROMPTS,\n} from '../wizard-session';\n\n// Mock dependencies\njest.mock('../../utils/analytics');\njest.mock('../../utils/debug');\n\n// Mock the SDK module\nconst mockQuery = jest.fn();\njest.mock('@anthropic-ai/claude-agent-sdk', () => ({\n  query: (...args: unknown[]) => mockQuery(...args),\n}));\n\n// Mock the UI layer\nconst mockUIInstance = {\n  log: {\n    step: jest.fn(),\n    success: jest.fn(),\n    error: jest.fn(),\n    warn: jest.fn(),\n    info: jest.fn(),\n  },\n  spinner: jest.fn(),\n  select: jest.fn(),\n  confirm: jest.fn(),\n  text: jest.fn(),\n  intro: jest.fn(),\n  outro: jest.fn(),\n  cancel: jest.fn(),\n  note: jest.fn(),\n  isCancel: jest.fn(),\n  setDetectedFramework: jest.fn(),\n  setCredentials: jest.fn(),\n  pushStatus: jest.fn(),\n  setLoginUrl: jest.fn(),\n  showBlockingOutage: jest.fn(),\n  setReadinessWarnings: jest.fn(),\n  showSettingsOverride: jest.fn(),\n  startRun: jest.fn(),\n  syncTodos: jest.fn(),\n  groupMultiselect: jest.fn(),\n  multiselect: jest.fn(),\n};\njest.mock('../../ui', () => ({\n  getUI: () => mockUIInstance,\n}));\n\ndescribe('runAgent', () => {\n  let mockSpinner: {\n    start: jest.Mock;\n    stop: jest.Mock;\n    message: jest.Mock;\n  };\n\n  const defaultOptions: WizardOptions = {\n    debug: false,\n    installDir: '/test/dir',\n    forceInstall: false,\n    default: false,\n    signup: false,\n    localMcp: false,\n    ci: false,\n    menu: false,\n    benchmark: false,\n    yaraReport: false,\n  };\n\n  const defaultAgentConfig = {\n    workingDirectory: '/test/dir',\n    mcpServers: {},\n    model: 'claude-opus-4-5-20251101',\n  };\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    mockSpinner = {\n      start: jest.fn(),\n      stop: jest.fn(),\n      message: jest.fn(),\n    };\n\n    mockUIInstance.spinner.mockReturnValue(mockSpinner);\n    // Reset log mocks\n    Object.values(mockUIInstance.log).forEach((fn) => fn.mockReset());\n  });\n\n  describe('race condition handling', () => {\n    it('should return success when agent completes successfully then SDK cleanup fails', async () => {\n      // This simulates the race condition:\n      // 1. Agent completes with success result\n      // 2. signalDone() is called, completing the prompt generator\n      // 3. SDK tries to send cleanup command while streaming is active\n      // 4. SDK throws an error\n      // The fix should recognize we already got a success and return success anyway\n\n      function* mockGeneratorWithCleanupError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          result: 'Agent completed successfully',\n        };\n\n        // Simulate the SDK cleanup error that occurs after success\n        throw new Error('only prompt commands are supported in streaming mode');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithCleanupError());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not throw\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n    });\n\n    it('should still throw when no success result was received before error', async () => {\n      // If we never got a success result, errors should propagate normally\n\n      function* mockGeneratorWithOnlyError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // No success result, just an error\n        throw new Error('Actual SDK error');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithOnlyError());\n\n      await expect(\n        runAgent(\n          defaultAgentConfig,\n          'test prompt',\n          defaultOptions,\n          mockSpinner as unknown as SpinnerHandle,\n          {\n            successMessage: 'Test success',\n            errorMessage: 'Test error',\n          },\n        ),\n      ).rejects.toThrow('Actual SDK error');\n\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test error');\n    });\n\n    it('should not treat error results as success', async () => {\n      // A result with is_error: true should not count as success\n      // Even if subtype is 'success', the is_error flag takes precedence\n\n      function* mockGeneratorWithErrorResult() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success', // subtype can be success but is_error true\n          is_error: true,\n          result: 'API Error: 500 Internal Server Error',\n        };\n\n        throw new Error('Process exited with code 1');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithErrorResult());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return API error, not success\n      expect(result.error).toBe('WIZARD_API_ERROR');\n      expect(result.message).toContain('API Error');\n    });\n\n    it('should suppress user-facing errors when SDK yields error result after success', async () => {\n      // This test models actual SDK behavior where the SDK emits TWO result messages:\n      // 1. SDK yields success result (num_turns: 105, is_error: false)\n      // 2. SDK yields a SECOND result with is_error: true containing\n      //    accumulated cleanup/telemetry errors\n      // 3. The errors should be logged to file but NOT shown to the user\n      //\n      // This differs from the thrown exception test above - here the SDK YIELDS\n      // an error result message instead of THROWING an exception.\n\n      function* mockGeneratorWithYieldedErrorAfterSuccess() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // First result: success (this is the real completion)\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          num_turns: 105,\n          result: '[WIZARD-REMARK] Integration completed successfully',\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 5.83,\n        };\n\n        // Second result: error (SDK cleanup noise - yielded, not thrown)\n        yield {\n          type: 'result',\n          subtype: 'error_during_execution',\n          is_error: true,\n          num_turns: 0,\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 0,\n          errors: [\n            'only prompt commands are supported in streaming mode',\n            'Error: 1P event logging: 14 events failed to export',\n            'Error: 1P event logging: 13 events failed to export',\n            'Error: Failed to export 14 events',\n          ],\n        };\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithYieldedErrorAfterSuccess());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not error\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n\n      // ui.log.error should NOT have been called (errors suppressed for user)\n      expect(mockUIInstance.log.error).not.toHaveBeenCalled();\n    });\n  });\n});\n\ndescribe('createStopHook', () => {\n  const hookInput = { stop_hook_active: false };\n\n  it('empty queue: first call blocks for remark, second allows stop', () => {\n    const hook = createStopHook([]);\n\n    // First call → remark prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Second call → allow stop\n    const second = hook(hookInput);\n    expect(second).toEqual({});\n  });\n\n  it('single feature: feature prompt, then remark, then allow stop', () => {\n    const hook = createStopHook([AdditionalFeature.LLM]);\n\n    // First call → LLM feature prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Second call → remark prompt\n    const second = hook(hookInput);\n    expect(second).toHaveProperty('decision', 'block');\n    expect((second as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Third call → allow stop\n    const third = hook(hookInput);\n    expect(third).toEqual({});\n  });\n\n  it('multiple queue entries: drains all, then remark, then allow stop', () => {\n    // Queue the same feature twice to exercise multi-item draining\n    const hook = createStopHook([AdditionalFeature.LLM, AdditionalFeature.LLM]);\n\n    // First call → LLM prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Second call → LLM prompt again\n    const second = hook(hookInput);\n    expect(second).toHaveProperty('decision', 'block');\n    expect((second as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Third call → remark prompt\n    const third = hook(hookInput);\n    expect(third).toHaveProperty('decision', 'block');\n    expect((third as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Fourth call → allow stop\n    const fourth = hook(hookInput);\n    expect(fourth).toEqual({});\n  });\n\n  it('allow stop is idempotent after all phases complete', () => {\n    const hook = createStopHook([]);\n\n    hook(hookInput); // remark\n    hook(hookInput); // allow\n    const extra = hook(hookInput); // still allow\n    expect(extra).toEqual({});\n  });\n\n  it('allows stop immediately on API error (401)', () => {\n    const collectedText = [\n      'Failed to authenticate. API Error: 401 {\"detail\":\"Authentication required\"}',\n    ];\n    const hook = createStopHook([AdditionalFeature.LLM], collectedText);\n\n    const result = hook(hookInput);\n    expect(result).toEqual({});\n  });\n\n  it('allows stop immediately on generic API error', () => {\n    const collectedText = ['API Error: 500 Internal Server Error'];\n    const hook = createStopHook([AdditionalFeature.LLM], collectedText);\n\n    const result = hook(hookInput);\n    expect(result).toEqual({});\n  });\n\n  it('proceeds normally when collectedText has no API error', () => {\n    const collectedText = ['Some normal agent output'];\n    const hook = createStopHook([], collectedText);\n\n    // First call → remark prompt (normal behavior)\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toContain('WIZARD-REMARK');\n  });\n});\n"]}

package/dist/src/lib/agent-interface.d.ts

@@ -36,12 +36,30 @@ export declare enum AgentErrorType {
     /** YARA scanner detected a security violation */
     YARA_VIOLATION = "WIZARD_YARA_VIOLATION"
 }
+/** Where a settings conflict was found. */
+export type SettingsConflictSource = 'project' | 'managed';
+/** A single settings conflict detected during startup. */
+export interface SettingsConflict {
+    /** Where the conflict was found. */
+    source: SettingsConflictSource;
+    /** The blocking keys found (e.g. 'ANTHROPIC_BASE_URL', 'apiKeyHelper'). */
+    keys: string[];
+    /** Whether the wizard can back up / remove this file. Managed settings are read-only. */
+    writable: boolean;
+}
 /**
  * Check if .claude/settings.json in the project directory contains env
- * overrides for blocking keys that block the Wizard from accessing the PostHog LLM Gateway.
+ * overrides or apiKeyHelper that block the Wizard from accessing the PostHog LLM Gateway.
  * Returns the list of matched key names, or an empty array if none found.
+ *
+ * @deprecated Use {@link checkAllSettingsConflicts} for comprehensive detection.
  */
 export declare function checkClaudeSettingsOverrides(workingDirectory: string): string[];
+/**
+ * Check project and org-managed settings for blocking keys that conflict
+ * with the wizard's proxy auth.
+ */
+export declare function checkAllSettingsConflicts(workingDirectory: string): SettingsConflict[];
 /**
  * Copy .claude/settings.json to .wizard-backup (overwriting if it exists),
  * then remove the original so the SDK doesn't load the blocking overrides.
@@ -83,7 +101,7 @@ export type StopHookResult = Record<string, never> | {
  *   Phase 2 — collect remark (once): block with remark prompt
  *   Phase 3 — allow stop: return {}
  */
-export declare function createStopHook(featureQueue: readonly AdditionalFeature[]): (input: {
+export declare function createStopHook(featureQueue: readonly AdditionalFeature[], collectedText?: string[]): (input: {
     stop_hook_active: boolean;
 }) => StopHookResult;
 /**

package/dist/src/lib/agent-interface.js

@@ -42,6 +42,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.isSkillInstallCommand = exports.AgentErrorType = exports.AgentSignals = void 0;
 exports.checkClaudeSettingsOverrides = checkClaudeSettingsOverrides;
+exports.checkAllSettingsConflicts = checkAllSettingsConflicts;
 exports.backupAndFixClaudeSettings = backupAndFixClaudeSettings;
 exports.restoreClaudeSettings = restoreClaudeSettings;
 exports.createStopHook = createStopHook;
@@ -109,11 +110,41 @@ var AgentErrorType;
     /** YARA scanner detected a security violation */
     AgentErrorType["YARA_VIOLATION"] = "WIZARD_YARA_VIOLATION";
 })(AgentErrorType || (exports.AgentErrorType = AgentErrorType = {}));
-const BLOCKING_ENV_KEYS = ['ANTHROPIC_BASE_URL', 'ANTHROPIC_AUTH_TOKEN'];
+const BLOCKING_ENV_KEYS = [
+    'ANTHROPIC_API_KEY',
+    'ANTHROPIC_BASE_URL',
+    'ANTHROPIC_AUTH_TOKEN',
+];
+const BLOCKING_SETTINGS_KEYS = ['apiKeyHelper'];
+/**
+ * Check a single settings file for blocking env keys and top-level settings keys.
+ * Returns matched key names, or an empty array if none found.
+ */
+function checkSettingsFile(filePath) {
+    try {
+        const raw = fs.readFileSync(filePath, 'utf-8');
+        const parsed = JSON.parse(raw);
+        const matched = [];
+        // Check env block for blocking env keys
+        const envBlock = parsed?.env;
+        if (envBlock && typeof envBlock === 'object') {
+            matched.push(...BLOCKING_ENV_KEYS.filter((key) => key in envBlock));
+        }
+        // Check top-level settings keys
+        matched.push(...BLOCKING_SETTINGS_KEYS.filter((key) => key in parsed && parsed[key] !== '' && parsed[key] != null));
+        return matched;
+    }
+    catch {
+        // File doesn't exist or isn't valid JSON — skip
+        return [];
+    }
+}
 /**
  * Check if .claude/settings.json in the project directory contains env
- * overrides for blocking keys that block the Wizard from accessing the PostHog LLM Gateway.
+ * overrides or apiKeyHelper that block the Wizard from accessing the PostHog LLM Gateway.
  * Returns the list of matched key names, or an empty array if none found.
+ *
+ * @deprecated Use {@link checkAllSettingsConflicts} for comprehensive detection.
  */
 function checkClaudeSettingsOverrides(workingDirectory) {
     const candidates = [
@@ -121,19 +152,48 @@ function checkClaudeSettingsOverrides(workingDirectory) {
         path_1.default.join(workingDirectory, '.claude', 'settings'),
     ];
     for (const filePath of candidates) {
-        try {
-            const raw = fs.readFileSync(filePath, 'utf-8');
-            const parsed = JSON.parse(raw);
-            const envBlock = parsed?.env;
-            if (envBlock && typeof envBlock === 'object') {
-                return BLOCKING_ENV_KEYS.filter((key) => key in envBlock);
+        const matched = checkSettingsFile(filePath);
+        if (matched.length > 0)
+            return matched;
+    }
+    return [];
+}
+/**
+ * Managed settings path on macOS.
+ * IT/MDM-deployed settings — readable by all users, writable only by root.
+ */
+const MANAGED_SETTINGS_PATH = '/Library/Application Support/ClaudeCode/managed-settings.json';
+/**
+ * Check project and org-managed settings for blocking keys that conflict
+ * with the wizard's proxy auth.
+ */
+function checkAllSettingsConflicts(workingDirectory) {
+    const conflicts = [];
+    const sources = [
+        {
+            source: 'managed',
+            paths: [MANAGED_SETTINGS_PATH],
+            writable: false,
+        },
+        {
+            source: 'project',
+            paths: [
+                path_1.default.join(workingDirectory, '.claude', 'settings.json'),
+                path_1.default.join(workingDirectory, '.claude', 'settings'),
+            ],
+            writable: true,
+        },
+    ];
+    for (const { source, paths, writable } of sources) {
+        for (const filePath of paths) {
+            const keys = checkSettingsFile(filePath);
+            if (keys.length > 0) {
+                conflicts.push({ source, keys, writable });
+                break; // Only one conflict per source (settings.json vs settings fallback)
             }
         }
-        catch {
-            // File doesn't exist or isn't valid JSON — skip
-        }
     }
-    return [];
+    return conflicts;
 }
 /**
  * Copy .claude/settings.json to .wizard-backup (overwriting if it exists),
@@ -189,7 +249,7 @@ function restoreClaudeSettings(workingDirectory) {
  *   Phase 2 — collect remark (once): block with remark prompt
  *   Phase 3 — allow stop: return {}
  */
-function createStopHook(featureQueue) {
+function createStopHook(featureQueue, collectedText) {
     let featureIndex = 0;
     let remarkRequested = false;
     return (input) => {
@@ -199,6 +259,15 @@ function createStopHook(featureQueue) {
             remarkRequested,
             queueLength: featureQueue.length,
         });
+        // On API errors, allow stop immediately — blocking with remark/feature
+        // prompts would just fail again. The auth error screen is shown separately.
+        if (collectedText) {
+            const text = collectedText.join('\n');
+            if (text.includes('API Error:')) {
+                (0, debug_1.logToFile)('Stop hook: API error detected, allowing immediate stop');
+                return {};
+            }
+        }
         // Phase 1: drain feature queue
         if (featureIndex < featureQueue.length) {
             const feature = featureQueue[featureIndex++];
@@ -674,7 +743,9 @@ async function runAgent(agentConfig, prompt, options, spinner, config, middlewar
                     PostToolUse: (0, yara_hooks_1.createPostToolUseYaraHooks)(),
                     Stop: [
                         {
-                            hooks: [createStopHook(config?.additionalFeatureQueue ?? [])],
+                            hooks: [
+                                createStopHook(config?.additionalFeatureQueue ?? [], collectedText),
+                            ],
                             timeout: 30,
                         },
                     ],
@@ -722,6 +793,18 @@ async function runAgent(agentConfig, prompt, options, spinner, config, middlewar
             // Pass receivedSuccessResult so handleSDKMessage can suppress user-facing error
             // output for post-success cleanup errors while still logging them to file
             handleSDKMessage(message, options, spinner, collectedText, receivedSuccessResult);
+            // 401: show auth error screen and exit immediately
+            if (message.type === 'assistant' &&
+                collectedText.join('\n').includes('API Error: 401')) {
+                signalDone();
+                spinner.stop('Authentication failed');
+                (0, debug_1.logToFile)('Agent error: 401, showing auth error screen');
+                (0, ui_1.getUI)().showAuthError();
+                await (0, wizard_abort_1.wizardAbort)({
+                    message: 'Authentication failed (401)',
+                    error: new wizard_abort_1.WizardError('Authentication failed'),
+                });
+            }
             try {
                 middleware?.onMessage(message);
             }