CheapestInference
db9f4f81af
The rate limit detection was fundamentally broken because Claude Code
renders rate limit messages across multiple lines in its TUI (Ink/React),
but all regex patterns required keywords on a single line.
Key fixes:
- Split detection into LIMIT_PATTERNS + RESET_PATTERNS with a 6-line
sliding window, so multi-line TUI renders are detected correctly
- Capture only last 20 lines instead of 200 to avoid stale scrollback
- Add 30s post-retry cooldown to prevent re-detecting old messages
- Fix max-retries infinite loop (stay in waiting, check clear first)
- Add OSC/DCS sequence stripping for complete ANSI handling
- Parse relative times ("try again in 5 minutes") instead of 5h fallback
- Handle invalid timezones gracefully with fallback
- Fix zombie monitor on pane destruction (exit after 10 consecutive errors)
- Replace setInterval with recursive setTimeout to prevent concurrent ticks
- Restore user shell traps instead of clobbering them
- Remove overly broad /\blimit\b/ pattern that caused false positives
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
84 lines
3.8 KiB
JavaScript
84 lines
3.8 KiB
JavaScript
import { describe, it } from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import { createMonitorState, processOneTick } from '../src/monitor.js';
|
|
import { DEFAULT_CONFIG } from '../src/config.js';
|
|
|
|
function mockTmux(paneContent = '', paneCommand = 'node') {
|
|
const t = {
|
|
_sent: [],
|
|
capturePane: async () => paneContent,
|
|
getPaneCommand: async () => paneCommand,
|
|
sendKeys: async (_p, text) => { t._sent.push(text); },
|
|
};
|
|
return t;
|
|
}
|
|
|
|
describe('processOneTick', () => {
|
|
it('returns monitoring when no rate limit', async () => {
|
|
const t = mockTmux('Normal output');
|
|
const s = createMonitorState();
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'monitoring');
|
|
assert.equal(t._sent.length, 0);
|
|
});
|
|
it('enters waiting on rate limit', async () => {
|
|
const t = mockTmux('5-hour limit reached - resets 3pm (UTC)');
|
|
const s = createMonitorState();
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'waiting');
|
|
assert.ok(s.waitUntil > Date.now());
|
|
});
|
|
it('exits when PID dead', async () => {
|
|
const t = mockTmux('5-hour limit reached - resets 3pm (UTC)');
|
|
const s = createMonitorState();
|
|
s.waitUntil = Date.now() - 1000; s.status = 'waiting';
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => false), 'exit');
|
|
});
|
|
it('sends retry when wait expired and rate limit visible', async () => {
|
|
const t = mockTmux('5-hour limit reached - resets 3pm (UTC)');
|
|
const s = createMonitorState();
|
|
s.waitUntil = Date.now() - 1000; s.status = 'waiting';
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'retried');
|
|
assert.equal(t._sent.length, 1);
|
|
assert.equal(s.attempts, 1);
|
|
// Should stay in 'waiting' with a cooldown to let Claude process
|
|
assert.equal(s.status, 'waiting');
|
|
assert.ok(s.waitUntil > Date.now());
|
|
});
|
|
it('detects multi-line TUI rate limit', async () => {
|
|
const t = mockTmux('⚠ You\'ve hit your limit\n· resets 3pm (UTC)');
|
|
const s = createMonitorState();
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'waiting');
|
|
assert.ok(s.waitUntil > Date.now());
|
|
});
|
|
it('skips when foreground is not claude', async () => {
|
|
const t = mockTmux('5-hour limit reached - resets 3pm (UTC)', 'vim');
|
|
const s = createMonitorState();
|
|
s.waitUntil = Date.now() - 1000; s.status = 'waiting';
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'skipped-not-claude');
|
|
assert.equal(t._sent.length, 0);
|
|
});
|
|
it('resets counter when rate limit disappears', async () => {
|
|
const t = mockTmux('Claude is working normally');
|
|
const s = createMonitorState();
|
|
s.waitUntil = Date.now() - 1000; s.status = 'waiting'; s.attempts = 2;
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'user-continued');
|
|
assert.equal(s.attempts, 0);
|
|
});
|
|
it('stops retrying after max attempts and stays in waiting', async () => {
|
|
const t = mockTmux('5-hour limit reached - resets 3pm (UTC)');
|
|
const s = createMonitorState();
|
|
s.waitUntil = Date.now() - 1000; s.status = 'waiting'; s.attempts = 5;
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'max-retries');
|
|
// Should stay in 'waiting' to avoid re-detection loop
|
|
assert.equal(s.status, 'waiting');
|
|
assert.ok(s.waitUntil > Date.now());
|
|
});
|
|
it('resets from max-retries when rate limit clears', async () => {
|
|
const t = mockTmux('Claude is working normally');
|
|
const s = createMonitorState();
|
|
s.waitUntil = Date.now() - 1000; s.status = 'waiting'; s.attempts = 10;
|
|
// Rate limit cleared → should detect user-continued before max-retries check
|
|
assert.equal(await processOneTick(s, t, '%0', DEFAULT_CONFIG, () => true), 'user-continued');
|
|
assert.equal(s.attempts, 0);
|
|
});
|
|
});
|