fix: treat CI watcher completions as system input (#191)
This commit is contained in:
@@ -144,4 +144,33 @@ describe('processLoopGroupMessages', () => {
|
|||||||
expect(args.enqueueMessageCheck).not.toHaveBeenCalled();
|
expect(args.enqueueMessageCheck).not.toHaveBeenCalled();
|
||||||
expect(args.lastAgentTimestamps).toEqual({ [chatJid]: '12' });
|
expect(args.lastAgentTimestamps).toEqual({ [chatJid]: '12' });
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('pipes trusted external bot events instead of treating them as human interrupts', async () => {
|
||||||
|
hasReviewerLeaseMock.mockReturnValue(true);
|
||||||
|
const args = defaultArgs({
|
||||||
|
groupMessages: [
|
||||||
|
message({
|
||||||
|
id: 'watch-ci-completed:task-1',
|
||||||
|
sender: 'ci-watcher',
|
||||||
|
sender_name: 'CI watcher',
|
||||||
|
content: '[CI watcher completed]\nCI succeeded',
|
||||||
|
is_from_me: false,
|
||||||
|
is_bot_message: false,
|
||||||
|
message_source_kind: 'trusted_external_bot',
|
||||||
|
seq: 13,
|
||||||
|
}),
|
||||||
|
],
|
||||||
|
isRunningMessageTurn: vi.fn(() => true),
|
||||||
|
});
|
||||||
|
|
||||||
|
await processLoopGroupMessages(args);
|
||||||
|
|
||||||
|
expect(args.sendQueuedMessage).toHaveBeenCalledWith(
|
||||||
|
chatJid,
|
||||||
|
'[CI watcher completed]\nCI succeeded',
|
||||||
|
);
|
||||||
|
expect(args.closeStdin).not.toHaveBeenCalled();
|
||||||
|
expect(args.enqueueMessageCheck).not.toHaveBeenCalled();
|
||||||
|
expect(args.lastAgentTimestamps).toEqual({ [chatJid]: '13' });
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ import {
|
|||||||
advanceLastAgentCursor,
|
advanceLastAgentCursor,
|
||||||
filterLoopingPairedBotMessages,
|
filterLoopingPairedBotMessages,
|
||||||
getProcessableMessages,
|
getProcessableMessages,
|
||||||
|
isExternalHumanMessage,
|
||||||
resolveCursorKey,
|
resolveCursorKey,
|
||||||
shouldSkipBotOnlyCollaboration,
|
shouldSkipBotOnlyCollaboration,
|
||||||
} from './message-runtime-rules.js';
|
} from './message-runtime-rules.js';
|
||||||
@@ -32,9 +33,7 @@ import type {
|
|||||||
type RuntimeLog = Pick<typeof logger, 'info' | 'debug'>;
|
type RuntimeLog = Pick<typeof logger, 'info' | 'debug'>;
|
||||||
|
|
||||||
function hasExternalHumanMessage(messages: NewMessage[]): boolean {
|
function hasExternalHumanMessage(messages: NewMessage[]): boolean {
|
||||||
return messages.some(
|
return messages.some(isExternalHumanMessage);
|
||||||
(message) => message.is_from_me !== true && !message.is_bot_message,
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function enqueueGenericFollowUpAfterDeliveryRetry(args: {
|
export function enqueueGenericFollowUpAfterDeliveryRetry(args: {
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ import {
|
|||||||
} from './message-runtime-follow-up.js';
|
} from './message-runtime-follow-up.js';
|
||||||
import {
|
import {
|
||||||
advanceLastAgentCursor,
|
advanceLastAgentCursor,
|
||||||
|
isBotOrTrustedSystemMessage,
|
||||||
resolveCursorKey,
|
resolveCursorKey,
|
||||||
resolveNextTurnAction,
|
resolveNextTurnAction,
|
||||||
} from './message-runtime-rules.js';
|
} from './message-runtime-rules.js';
|
||||||
@@ -83,10 +84,7 @@ export function isBotOnlyPairedRoomTurn(
|
|||||||
messages: NewMessage[],
|
messages: NewMessage[],
|
||||||
): boolean {
|
): boolean {
|
||||||
return (
|
return (
|
||||||
hasReviewerLease(chatJid) &&
|
hasReviewerLease(chatJid) && messages.every(isBotOrTrustedSystemMessage)
|
||||||
messages.every(
|
|
||||||
(message) => message.is_from_me === true || !!message.is_bot_message,
|
|
||||||
)
|
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -83,6 +83,14 @@ describe('message-runtime-preflight-messages', () => {
|
|||||||
seq: 13,
|
seq: 13,
|
||||||
is_bot_message: true,
|
is_bot_message: true,
|
||||||
}),
|
}),
|
||||||
|
makeMessage({
|
||||||
|
id: 'trusted-system',
|
||||||
|
content: '[CI watcher completed]',
|
||||||
|
seq: 14,
|
||||||
|
is_from_me: false,
|
||||||
|
is_bot_message: false,
|
||||||
|
message_source_kind: 'trusted_external_bot',
|
||||||
|
}),
|
||||||
]);
|
]);
|
||||||
|
|
||||||
const messages = getFreshHumanPreflightMessages({
|
const messages = getFreshHumanPreflightMessages({
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ import { getMessagesSinceSeq } from './db.js';
|
|||||||
import {
|
import {
|
||||||
filterLoopingPairedBotMessages,
|
filterLoopingPairedBotMessages,
|
||||||
getProcessableMessages,
|
getProcessableMessages,
|
||||||
|
isExternalHumanMessage,
|
||||||
} from './message-runtime-rules.js';
|
} from './message-runtime-rules.js';
|
||||||
import type { WorkItem } from './db/work-items.js';
|
import type { WorkItem } from './db/work-items.js';
|
||||||
import type { Channel, NewMessage } from './types.js';
|
import type { Channel, NewMessage } from './types.js';
|
||||||
@@ -24,9 +25,7 @@ export function getFreshHumanPreflightMessages(args: {
|
|||||||
getProcessableMessages(args.chatJid, preflightRawMessages, args.channel),
|
getProcessableMessages(args.chatJid, preflightRawMessages, args.channel),
|
||||||
args.failureFinalText,
|
args.failureFinalText,
|
||||||
);
|
);
|
||||||
return preflightMessages.filter(
|
return preflightMessages.filter(isExternalHumanMessage);
|
||||||
(message) => message.is_from_me !== true && !message.is_bot_message,
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function hasHumanMessageAfterWorkItem(
|
export function hasHumanMessageAfterWorkItem(
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ import { resolveStoredVisibleVerdict } from './paired-verdict.js';
|
|||||||
import {
|
import {
|
||||||
advanceLastAgentCursor,
|
advanceLastAgentCursor,
|
||||||
finalizeQueuedRunCursor,
|
finalizeQueuedRunCursor,
|
||||||
|
isExternalHumanMessage,
|
||||||
resolveActiveRole,
|
resolveActiveRole,
|
||||||
resolveCursorKeyForRole,
|
resolveCursorKeyForRole,
|
||||||
resolveQueuedPairedTurnRole,
|
resolveQueuedPairedTurnRole,
|
||||||
@@ -243,9 +244,7 @@ export async function runQueuedGroupTurn(args: {
|
|||||||
args;
|
args;
|
||||||
let currentTask = task;
|
let currentTask = task;
|
||||||
const hasHumanMsg = task
|
const hasHumanMsg = task
|
||||||
? !missedMessages.every(
|
? missedMessages.some(isExternalHumanMessage)
|
||||||
(message) => message.is_from_me === true || !!message.is_bot_message,
|
|
||||||
)
|
|
||||||
: !isBotOnlyPairedRoomTurn(chatJid, missedMessages);
|
: !isBotOnlyPairedRoomTurn(chatJid, missedMessages);
|
||||||
let fallbackMessages = missedMessages;
|
let fallbackMessages = missedMessages;
|
||||||
if (currentTask !== undefined && hasHumanMsg) {
|
if (currentTask !== undefined && hasHumanMsg) {
|
||||||
|
|||||||
@@ -26,6 +26,35 @@ import {
|
|||||||
|
|
||||||
const BOT_COLLABORATION_WINDOW_MS = 12 * 60 * 60 * 1000;
|
const BOT_COLLABORATION_WINDOW_MS = 12 * 60 * 60 * 1000;
|
||||||
|
|
||||||
|
export type MessageOriginFlags = Pick<
|
||||||
|
NewMessage,
|
||||||
|
'is_from_me' | 'is_bot_message' | 'message_source_kind'
|
||||||
|
>;
|
||||||
|
|
||||||
|
export function isTrustedExternalBotMessage(
|
||||||
|
message: Pick<NewMessage, 'message_source_kind'>,
|
||||||
|
): boolean {
|
||||||
|
return message.message_source_kind === 'trusted_external_bot';
|
||||||
|
}
|
||||||
|
|
||||||
|
export function isExternalHumanMessage(message: MessageOriginFlags): boolean {
|
||||||
|
return (
|
||||||
|
message.is_from_me !== true &&
|
||||||
|
!message.is_bot_message &&
|
||||||
|
!isTrustedExternalBotMessage(message)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function isBotOrTrustedSystemMessage(
|
||||||
|
message: MessageOriginFlags,
|
||||||
|
): boolean {
|
||||||
|
return (
|
||||||
|
message.is_from_me === true ||
|
||||||
|
!!message.is_bot_message ||
|
||||||
|
isTrustedExternalBotMessage(message)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
export function advanceLastAgentCursor(
|
export function advanceLastAgentCursor(
|
||||||
lastAgentTimestamps: Record<string, string>,
|
lastAgentTimestamps: Record<string, string>,
|
||||||
saveState: () => void,
|
saveState: () => void,
|
||||||
@@ -439,9 +468,7 @@ export function createImplicitContinuationTracker(idleTimeout: number) {
|
|||||||
implicitContinuationUntil.delete(chatJid);
|
implicitContinuationUntil.delete(chatJid);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
return messages.some(
|
return messages.some(isExternalHumanMessage);
|
||||||
(message) => message.is_from_me !== true && !message.is_bot_message,
|
|
||||||
);
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
@@ -451,9 +478,7 @@ export function shouldSkipBotOnlyCollaboration(
|
|||||||
messages: NewMessage[],
|
messages: NewMessage[],
|
||||||
): boolean {
|
): boolean {
|
||||||
if (hasReviewerLease(chatJid)) return false;
|
if (hasReviewerLease(chatJid)) return false;
|
||||||
const allFromBots = messages.every(
|
const allFromBots = messages.every(isBotOrTrustedSystemMessage);
|
||||||
(message) => message.is_from_me || !!message.is_bot_message,
|
|
||||||
);
|
|
||||||
if (!allFromBots) return false;
|
if (!allFromBots) return false;
|
||||||
const lastHuman = getLastHumanMessageTimestamp(chatJid);
|
const lastHuman = getLastHumanMessageTimestamp(chatJid);
|
||||||
if (!lastHuman) return true;
|
if (!lastHuman) return true;
|
||||||
|
|||||||
142
src/message-runtime-trusted-system.test.ts
Normal file
142
src/message-runtime-trusted-system.test.ts
Normal file
@@ -0,0 +1,142 @@
|
|||||||
|
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
||||||
|
|
||||||
|
import { _initTestDatabase, createPairedTask } from './db.js';
|
||||||
|
import { runQueuedGroupTurn } from './message-runtime-queue.js';
|
||||||
|
import {
|
||||||
|
isBotOrTrustedSystemMessage,
|
||||||
|
isExternalHumanMessage,
|
||||||
|
} from './message-runtime-rules.js';
|
||||||
|
import { resetPairedFollowUpScheduleState } from './paired-follow-up-scheduler.js';
|
||||||
|
import type { Channel, PairedTask, RegisteredGroup } from './types.js';
|
||||||
|
|
||||||
|
function makeGroup(): RegisteredGroup {
|
||||||
|
return {
|
||||||
|
name: 'Test Group',
|
||||||
|
folder: 'test-group',
|
||||||
|
trigger: '@Andy',
|
||||||
|
added_at: '2026-03-30T00:00:00.000Z',
|
||||||
|
requiresTrigger: false,
|
||||||
|
agentType: 'codex',
|
||||||
|
workDir: '/repo',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function makeTask(): PairedTask {
|
||||||
|
return {
|
||||||
|
id: 'task-trusted-system',
|
||||||
|
chat_jid: 'group@test',
|
||||||
|
group_folder: 'test-group',
|
||||||
|
owner_service_id: 'claude',
|
||||||
|
reviewer_service_id: 'codex-main',
|
||||||
|
owner_agent_type: 'claude-code',
|
||||||
|
reviewer_agent_type: 'codex',
|
||||||
|
arbiter_agent_type: null,
|
||||||
|
title: null,
|
||||||
|
source_ref: 'HEAD',
|
||||||
|
plan_notes: null,
|
||||||
|
review_requested_at: '2026-03-30T00:00:00.000Z',
|
||||||
|
round_trip_count: 1,
|
||||||
|
status: 'review_ready',
|
||||||
|
arbiter_verdict: null,
|
||||||
|
arbiter_requested_at: null,
|
||||||
|
completion_reason: null,
|
||||||
|
created_at: '2026-03-30T00:00:00.000Z',
|
||||||
|
updated_at: '2026-03-30T00:00:00.000Z',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function makeChannel(): Channel {
|
||||||
|
return {
|
||||||
|
name: 'discord-review',
|
||||||
|
connect: vi.fn(),
|
||||||
|
sendMessage: vi.fn(),
|
||||||
|
isConnected: vi.fn(() => true),
|
||||||
|
ownsJid: vi.fn(() => true),
|
||||||
|
disconnect: vi.fn(),
|
||||||
|
} as unknown as Channel;
|
||||||
|
}
|
||||||
|
|
||||||
|
describe('trusted system message routing', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
_initTestDatabase();
|
||||||
|
resetPairedFollowUpScheduleState();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('does not classify trusted external bot events as human interruptions', () => {
|
||||||
|
const trustedEvent = {
|
||||||
|
is_from_me: false,
|
||||||
|
is_bot_message: false,
|
||||||
|
message_source_kind: 'trusted_external_bot' as const,
|
||||||
|
};
|
||||||
|
|
||||||
|
expect(isExternalHumanMessage(trustedEvent)).toBe(false);
|
||||||
|
expect(isBotOrTrustedSystemMessage(trustedEvent)).toBe(true);
|
||||||
|
expect(
|
||||||
|
isExternalHumanMessage({
|
||||||
|
is_from_me: false,
|
||||||
|
is_bot_message: false,
|
||||||
|
message_source_kind: 'human',
|
||||||
|
}),
|
||||||
|
).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('keeps trusted CI completions on the pending reviewer path', async () => {
|
||||||
|
const task = makeTask();
|
||||||
|
createPairedTask(task);
|
||||||
|
const executeTurn = vi.fn(async () => ({
|
||||||
|
outputStatus: 'success' as const,
|
||||||
|
deliverySucceeded: true,
|
||||||
|
visiblePhase: 'final',
|
||||||
|
}));
|
||||||
|
|
||||||
|
const outcome = await runQueuedGroupTurn({
|
||||||
|
chatJid: task.chat_jid,
|
||||||
|
group: makeGroup(),
|
||||||
|
runId: 'run-ci-completion-reviewer',
|
||||||
|
log: { info: vi.fn(), warn: vi.fn(), error: vi.fn() } as any,
|
||||||
|
timezone: 'UTC',
|
||||||
|
missedMessages: [
|
||||||
|
{
|
||||||
|
id: 'watch-ci-completed:task-1',
|
||||||
|
chat_jid: task.chat_jid,
|
||||||
|
sender: 'ci-watcher',
|
||||||
|
sender_name: 'CI watcher',
|
||||||
|
content: '[CI watcher completed]\nCI succeeded',
|
||||||
|
timestamp: '2026-03-30T00:00:02.000Z',
|
||||||
|
seq: 46,
|
||||||
|
is_from_me: false,
|
||||||
|
is_bot_message: false,
|
||||||
|
message_source_kind: 'trusted_external_bot',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
task,
|
||||||
|
roleToChannel: {
|
||||||
|
owner: null,
|
||||||
|
reviewer: makeChannel(),
|
||||||
|
arbiter: null,
|
||||||
|
},
|
||||||
|
ownerChannel: makeChannel(),
|
||||||
|
lastAgentTimestamps: {},
|
||||||
|
saveState: vi.fn(),
|
||||||
|
executeTurn,
|
||||||
|
getFixedRoleChannelName: () => 'discord-review',
|
||||||
|
labelPairedSenders: (_chatJid, messages) => messages,
|
||||||
|
formatMessages: () => 'formatted prompt',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(outcome).toBe(true);
|
||||||
|
expect(executeTurn).toHaveBeenCalledWith(
|
||||||
|
expect.objectContaining({
|
||||||
|
deliveryRole: 'reviewer',
|
||||||
|
forcedRole: 'reviewer',
|
||||||
|
pairedTurnIdentity: {
|
||||||
|
turnId: 'task-trusted-system:2026-03-30T00:00:00.000Z:reviewer-turn',
|
||||||
|
taskId: 'task-trusted-system',
|
||||||
|
taskUpdatedAt: '2026-03-30T00:00:00.000Z',
|
||||||
|
intentKind: 'reviewer-turn',
|
||||||
|
role: 'reviewer',
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user