Merge pull request #4 from phj1081/codex/fix-stale-final-leak
fix: isolate stale finals from next runs
This commit is contained in:
@@ -61,6 +61,7 @@ vi.mock('./db.js', () => {
|
|||||||
getAllChats: vi.fn(() => []),
|
getAllChats: vi.fn(() => []),
|
||||||
getAllTasks: vi.fn(() => []),
|
getAllTasks: vi.fn(() => []),
|
||||||
getLastHumanMessageTimestamp: vi.fn(() => null),
|
getLastHumanMessageTimestamp: vi.fn(() => null),
|
||||||
|
getLastHumanMessageContent: vi.fn(() => null),
|
||||||
getMessagesSince,
|
getMessagesSince,
|
||||||
getNewMessages,
|
getNewMessages,
|
||||||
getLatestMessageSeqAtOrBefore: vi.fn(() => 0),
|
getLatestMessageSeqAtOrBefore: vi.fn(() => 0),
|
||||||
@@ -458,6 +459,180 @@ describe('createMessageRuntime', () => {
|
|||||||
expect(saveState).toHaveBeenCalled();
|
expect(saveState).toHaveBeenCalled();
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('delivers a stale work item in a separate run before processing new messages', async () => {
|
||||||
|
const chatJid = 'group@test';
|
||||||
|
const group = makeGroup('codex');
|
||||||
|
const channel = makeChannel(chatJid);
|
||||||
|
const enqueueMessageCheck = vi.fn();
|
||||||
|
|
||||||
|
vi.mocked(db.getOpenWorkItem).mockReturnValue({
|
||||||
|
id: 99,
|
||||||
|
group_folder: group.folder,
|
||||||
|
chat_jid: chatJid,
|
||||||
|
agent_type: 'codex',
|
||||||
|
service_id: 'claude',
|
||||||
|
status: 'delivery_retry',
|
||||||
|
start_seq: 1,
|
||||||
|
end_seq: 1,
|
||||||
|
result_payload: '이전 final입니다.',
|
||||||
|
delivery_attempts: 1,
|
||||||
|
created_at: new Date().toISOString(),
|
||||||
|
updated_at: new Date().toISOString(),
|
||||||
|
delivered_at: null,
|
||||||
|
delivery_message_id: null,
|
||||||
|
last_error: 'discord send failed',
|
||||||
|
});
|
||||||
|
vi.mocked(db.getMessagesSince).mockReturnValue([
|
||||||
|
{
|
||||||
|
id: 'msg-1',
|
||||||
|
chat_jid: chatJid,
|
||||||
|
sender: 'user@test',
|
||||||
|
sender_name: 'User',
|
||||||
|
content: '새 작업입니다.',
|
||||||
|
timestamp: '2026-03-30T00:00:00.000Z',
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
|
||||||
|
const runtime = createMessageRuntime({
|
||||||
|
assistantName: 'Andy',
|
||||||
|
idleTimeout: 1_000,
|
||||||
|
pollInterval: 1_000,
|
||||||
|
timezone: 'UTC',
|
||||||
|
triggerPattern: /^@Andy\b/i,
|
||||||
|
channels: [channel],
|
||||||
|
queue: {
|
||||||
|
registerProcess: vi.fn(),
|
||||||
|
closeStdin: vi.fn(),
|
||||||
|
notifyIdle: vi.fn(),
|
||||||
|
enqueueMessageCheck,
|
||||||
|
} as any,
|
||||||
|
getRegisteredGroups: () => ({ [chatJid]: group }),
|
||||||
|
getSessions: () => ({}),
|
||||||
|
getLastTimestamp: () => '',
|
||||||
|
setLastTimestamp: vi.fn(),
|
||||||
|
getLastAgentTimestamps: () => ({}),
|
||||||
|
saveState: vi.fn(),
|
||||||
|
persistSession: vi.fn(),
|
||||||
|
clearSession: vi.fn(),
|
||||||
|
});
|
||||||
|
|
||||||
|
const result = await runtime.processGroupMessages(chatJid, {
|
||||||
|
runId: 'run-open-work-item-first',
|
||||||
|
reason: 'messages',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(result).toBe(true);
|
||||||
|
expect(channel.sendMessage).toHaveBeenCalledTimes(1);
|
||||||
|
expect(channel.sendMessage).toHaveBeenCalledWith(
|
||||||
|
chatJid,
|
||||||
|
'이전 final입니다.',
|
||||||
|
);
|
||||||
|
expect(agentRunner.runAgentProcess).not.toHaveBeenCalled();
|
||||||
|
expect(enqueueMessageCheck).toHaveBeenCalledWith(chatJid);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('does not inject filtered raw bot finals into workspace-based review prompts', async () => {
|
||||||
|
const chatJid = 'group@test';
|
||||||
|
const group = makeGroup('codex');
|
||||||
|
const saveState = vi.fn();
|
||||||
|
const lastAgentTimestamps: Record<string, string> = {};
|
||||||
|
const channel: Channel = {
|
||||||
|
...makeChannel(chatJid),
|
||||||
|
isOwnMessage: vi.fn((msg) => msg.sender === 'owner-bot@test'),
|
||||||
|
};
|
||||||
|
|
||||||
|
vi.mocked(config.isClaudeService).mockReturnValue(false);
|
||||||
|
vi.mocked(config.isReviewService).mockReturnValue(false);
|
||||||
|
vi.mocked(db.isPairedRoomJid).mockReturnValue(true);
|
||||||
|
vi.mocked(db.getLatestOpenPairedTaskForChat).mockReturnValue({
|
||||||
|
id: 'task-1',
|
||||||
|
chat_jid: chatJid,
|
||||||
|
group_folder: group.folder,
|
||||||
|
owner_service_id: 'claude',
|
||||||
|
reviewer_service_id: 'codex-main',
|
||||||
|
title: null,
|
||||||
|
source_ref: 'HEAD',
|
||||||
|
plan_notes: null,
|
||||||
|
review_requested_at: '2026-03-30T00:00:00.000Z',
|
||||||
|
round_trip_count: 0,
|
||||||
|
status: 'review_ready',
|
||||||
|
created_at: '2026-03-30T00:00:00.000Z',
|
||||||
|
updated_at: '2026-03-30T00:00:00.000Z',
|
||||||
|
});
|
||||||
|
vi.mocked(db.getLastHumanMessageContent).mockReturnValue(
|
||||||
|
'버전 올리고 dev 머지까지 해줘',
|
||||||
|
);
|
||||||
|
vi.mocked(db.getLatestMessageSeqAtOrBefore).mockReturnValue(41);
|
||||||
|
vi.mocked(db.getMessagesSinceSeq).mockReturnValue([
|
||||||
|
{
|
||||||
|
id: 'msg-1',
|
||||||
|
chat_jid: chatJid,
|
||||||
|
sender: 'owner-bot@test',
|
||||||
|
sender_name: 'Owner Bot',
|
||||||
|
content: 'DONE 이전 final이 여기 붙으면 안 됩니다.',
|
||||||
|
timestamp: '2026-03-30T00:00:05.000Z',
|
||||||
|
is_bot_message: true,
|
||||||
|
} as any,
|
||||||
|
]);
|
||||||
|
vi.mocked(agentRunner.runAgentProcess).mockImplementation(
|
||||||
|
async (_group, input, _onProcess, onOutput) => {
|
||||||
|
expect(input.prompt).toContain(
|
||||||
|
'User request:\n---\n버전 올리고 dev 머지까지 해줘\n---',
|
||||||
|
);
|
||||||
|
expect(input.prompt).toContain(
|
||||||
|
'Review the latest owner changes in the workspace.',
|
||||||
|
);
|
||||||
|
expect(input.prompt).not.toContain(
|
||||||
|
'DONE 이전 final이 여기 붙으면 안 됩니다.',
|
||||||
|
);
|
||||||
|
await onOutput?.({
|
||||||
|
status: 'success',
|
||||||
|
phase: 'final',
|
||||||
|
result: '리뷰 확인 완료',
|
||||||
|
newSessionId: 'session-review-sanitized',
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
status: 'success',
|
||||||
|
result: '리뷰 확인 완료',
|
||||||
|
newSessionId: 'session-review-sanitized',
|
||||||
|
};
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
const runtime = createMessageRuntime({
|
||||||
|
assistantName: 'Andy',
|
||||||
|
idleTimeout: 1_000,
|
||||||
|
pollInterval: 1_000,
|
||||||
|
timezone: 'UTC',
|
||||||
|
triggerPattern: /^@Andy\b/i,
|
||||||
|
channels: [channel],
|
||||||
|
queue: {
|
||||||
|
registerProcess: vi.fn(),
|
||||||
|
closeStdin: vi.fn(),
|
||||||
|
notifyIdle: vi.fn(),
|
||||||
|
} as any,
|
||||||
|
getRegisteredGroups: () => ({ [chatJid]: group }),
|
||||||
|
getSessions: () => ({}),
|
||||||
|
getLastTimestamp: () => '',
|
||||||
|
setLastTimestamp: vi.fn(),
|
||||||
|
getLastAgentTimestamps: () => lastAgentTimestamps,
|
||||||
|
saveState,
|
||||||
|
persistSession: vi.fn(),
|
||||||
|
clearSession: vi.fn(),
|
||||||
|
});
|
||||||
|
|
||||||
|
const result = await runtime.processGroupMessages(chatJid, {
|
||||||
|
runId: 'run-review-prompt-sanitized',
|
||||||
|
reason: 'messages',
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(result).toBe(true);
|
||||||
|
expect(agentRunner.runAgentProcess).toHaveBeenCalledTimes(1);
|
||||||
|
expect(lastAgentTimestamps[chatJid]).toBe('41');
|
||||||
|
expect(saveState).toHaveBeenCalled();
|
||||||
|
expect(channel.sendMessage).toHaveBeenCalledWith(chatJid, '리뷰 확인 완료');
|
||||||
|
});
|
||||||
|
|
||||||
it('allows follow-up messages without a trigger after a visible reply in non-main groups', async () => {
|
it('allows follow-up messages without a trigger after a visible reply in non-main groups', async () => {
|
||||||
const chatJid = 'group@test';
|
const chatJid = 'group@test';
|
||||||
const group: RegisteredGroup = {
|
const group: RegisteredGroup = {
|
||||||
|
|||||||
@@ -482,6 +482,11 @@ export function createMessageRuntime(deps: MessageRuntimeDeps): {
|
|||||||
openWorkItem,
|
openWorkItem,
|
||||||
);
|
);
|
||||||
if (!delivered) return false;
|
if (!delivered) return false;
|
||||||
|
// Keep stale final delivery isolated from the next conversational turn.
|
||||||
|
// A follow-up run will pick up any queued user messages or paired-room
|
||||||
|
// auto-review/finalize transitions after this retry succeeds.
|
||||||
|
deps.queue.enqueueMessageCheck(chatJid);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
const isMainGroup = group.isMain === true;
|
const isMainGroup = group.isMain === true;
|
||||||
@@ -508,32 +513,28 @@ export function createMessageRuntime(deps: MessageRuntimeDeps): {
|
|||||||
(pendingReviewTask.status === 'review_ready' ||
|
(pendingReviewTask.status === 'review_ready' ||
|
||||||
pendingReviewTask.status === 'in_review')
|
pendingReviewTask.status === 'in_review')
|
||||||
) {
|
) {
|
||||||
// Build review prompt with user's original request + owner's response
|
// No processable messages remain. Review workspace state using the
|
||||||
|
// user's request as context, but don't re-inject filtered raw bot
|
||||||
|
// output from older turns into the next review prompt.
|
||||||
const userMessage = getLastHumanMessageContent(chatJid);
|
const userMessage = getLastHumanMessageContent(chatJid);
|
||||||
const ownerContent = rawMissedMessages
|
|
||||||
.filter((m) => m.is_bot_message)
|
|
||||||
.map((m) => m.content)
|
|
||||||
.join('\n\n');
|
|
||||||
const parts: string[] = [];
|
const parts: string[] = [];
|
||||||
if (userMessage) {
|
if (userMessage) {
|
||||||
parts.push(`User request:\n---\n${userMessage}\n---`);
|
parts.push(`User request:\n---\n${userMessage}\n---`);
|
||||||
}
|
}
|
||||||
if (ownerContent) {
|
|
||||||
parts.push(`Owner response:\n---\n${ownerContent}\n---`);
|
|
||||||
}
|
|
||||||
const reviewPrompt =
|
const reviewPrompt =
|
||||||
parts.length > 0
|
parts.length > 0
|
||||||
? `${parts.join('\n\n')}\n\nReview the owner's response above.`
|
? `${parts.join('\n\n')}\n\nReview the latest owner changes in the workspace.`
|
||||||
: 'Review the latest owner changes in the workspace.';
|
: 'Review the latest owner changes in the workspace.';
|
||||||
|
|
||||||
// Advance cursor past the owner's messages so they aren't re-processed
|
// Advance cursor past filtered bot messages so they aren't re-processed
|
||||||
const lastRaw = rawMissedMessages[rawMissedMessages.length - 1];
|
const lastRaw = rawMissedMessages[rawMissedMessages.length - 1];
|
||||||
if (lastRaw?.seq != null) {
|
const cursor = lastRaw?.seq ?? lastRaw?.timestamp;
|
||||||
|
if (cursor != null) {
|
||||||
advanceLastAgentCursor(
|
advanceLastAgentCursor(
|
||||||
deps.getLastAgentTimestamps(),
|
deps.getLastAgentTimestamps(),
|
||||||
deps.saveState,
|
deps.saveState,
|
||||||
chatJid,
|
chatJid,
|
||||||
lastRaw.seq,
|
cursor,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -552,12 +553,13 @@ export function createMessageRuntime(deps: MessageRuntimeDeps): {
|
|||||||
// merge_ready: reviewer approved, owner gets final turn to finalize
|
// merge_ready: reviewer approved, owner gets final turn to finalize
|
||||||
if (pendingReviewTask && pendingReviewTask.status === 'merge_ready') {
|
if (pendingReviewTask && pendingReviewTask.status === 'merge_ready') {
|
||||||
const lastRaw = rawMissedMessages[rawMissedMessages.length - 1];
|
const lastRaw = rawMissedMessages[rawMissedMessages.length - 1];
|
||||||
if (lastRaw?.seq != null) {
|
const cursor = lastRaw?.seq ?? lastRaw?.timestamp;
|
||||||
|
if (cursor != null) {
|
||||||
advanceLastAgentCursor(
|
advanceLastAgentCursor(
|
||||||
deps.getLastAgentTimestamps(),
|
deps.getLastAgentTimestamps(),
|
||||||
deps.saveState,
|
deps.saveState,
|
||||||
chatJid,
|
chatJid,
|
||||||
lastRaw.seq,
|
cursor,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
const finalizePrompt =
|
const finalizePrompt =
|
||||||
|
|||||||
Reference in New Issue
Block a user