fix: recover fallback delivery retries across agent types

This commit is contained in:
Eyejoker
2026-04-04 05:21:25 +09:00
parent 0b4228b391
commit dfd9eb0f47
4 changed files with 192 additions and 12 deletions

View File

@@ -31,6 +31,7 @@ import {
getMessagesSinceSeq, getMessagesSinceSeq,
getNewMessagesBySeq, getNewMessagesBySeq,
getOpenWorkItem, getOpenWorkItem,
getOpenWorkItemForChat,
getPendingServiceHandoffs, getPendingServiceHandoffs,
recallMemories, recallMemories,
getRegisteredAgentTypesForJid, getRegisteredAgentTypesForJid,
@@ -2603,6 +2604,24 @@ describe('work items', () => {
expect(getOpenWorkItem('dc:123', 'claude-code')).toBeUndefined(); expect(getOpenWorkItem('dc:123', 'claude-code')).toBeUndefined();
}); });
it('finds pending delivery retries even when they were created by a fallback agent type', () => {
const fallbackItem = createProducedWorkItem({
group_folder: 'discord_test',
chat_jid: 'dc:fallback',
agent_type: 'codex',
delivery_role: 'reviewer',
start_seq: 20,
end_seq: 22,
result_payload: 'fallback reviewer output',
});
expect(getOpenWorkItem('dc:fallback', 'claude-code')).toBeUndefined();
expect(getOpenWorkItemForChat('dc:fallback')?.id).toBe(fallbackItem.id);
markWorkItemDelivered(fallbackItem.id, 'msg-fallback');
expect(getOpenWorkItemForChat('dc:fallback')).toBeUndefined();
});
it('stores service shadow as a derived compatibility field from role and agent type', () => { it('stores service shadow as a derived compatibility field from role and agent type', () => {
const reviewerItem = createProducedWorkItem({ const reviewerItem = createProducedWorkItem({
group_folder: 'discord_test', group_folder: 'discord_test',

View File

@@ -2039,6 +2039,20 @@ export function getOpenWorkItem(
return row ? hydrateWorkItem(row) : undefined; return row ? hydrateWorkItem(row) : undefined;
} }
export function getOpenWorkItemForChat(chatJid: string): WorkItem | undefined {
const row = db
.prepare(
`SELECT *
FROM work_items
WHERE chat_jid = ?
AND status IN ('produced', 'delivery_retry')
ORDER BY id ASC
LIMIT 1`,
)
.get(chatJid) as StoredWorkItemRow | undefined;
return row ? hydrateWorkItem(row) : undefined;
}
export function createProducedWorkItem(input: { export function createProducedWorkItem(input: {
group_folder: string; group_folder: string;
chat_jid: string; chat_jid: string;

View File

@@ -37,6 +37,13 @@ vi.mock('./paired-execution-context.js', () => ({
})); }));
vi.mock('./db.js', () => { vi.mock('./db.js', () => {
const getOpenWorkItem = vi.fn(
(
_chatJid?: string,
_agentType?: 'claude-code' | 'codex',
_serviceId?: string,
) => undefined,
);
const getMessagesSince = vi.fn( const getMessagesSince = vi.fn(
( (
_chatJid?: string, _chatJid?: string,
@@ -114,7 +121,8 @@ vi.mock('./db.js', () => {
}; };
}, },
), ),
getOpenWorkItem: vi.fn(() => undefined), getOpenWorkItem,
getOpenWorkItemForChat: vi.fn((chatJid: string) => getOpenWorkItem(chatJid)),
getLatestOpenPairedTaskForChat: vi.fn(() => undefined), getLatestOpenPairedTaskForChat: vi.fn(() => undefined),
getPairedTurnOutputs: vi.fn(() => []), getPairedTurnOutputs: vi.fn(() => []),
getRecentChatMessages: vi.fn(() => []), getRecentChatMessages: vi.fn(() => []),
@@ -782,6 +790,90 @@ describe('createMessageRuntime', () => {
expect(enqueueMessageCheck).toHaveBeenCalledWith(chatJid); expect(enqueueMessageCheck).toHaveBeenCalledWith(chatJid);
}); });
it('retries a stale fallback reviewer work item even when the room owner agent type is different', async () => {
const chatJid = 'group@test';
const group = makeGroup('claude-code');
const ownerChannel = makeChannel(chatJid);
const reviewerChannel = makeChannel(chatJid, 'discord-review', false);
const enqueueMessageCheck = vi.fn();
vi.mocked(serviceRouting.hasReviewerLease).mockReturnValue(true);
vi.mocked(db.getOpenWorkItem).mockReturnValue(undefined);
vi.mocked(db.getOpenWorkItemForChat).mockReturnValue({
id: 102,
group_folder: group.folder,
chat_jid: chatJid,
agent_type: 'codex',
service_id: 'codex-review',
delivery_role: 'reviewer',
status: 'delivery_retry',
start_seq: 12,
end_seq: 13,
result_payload: 'fallback reviewer final retry',
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.getLatestOpenPairedTaskForChat).mockReturnValue({
id: 'task-fallback-review-delivery-role',
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: 1,
status: 'merge_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',
});
const runtime = createMessageRuntime({
assistantName: 'Andy',
idleTimeout: 1_000,
pollInterval: 1_000,
timezone: 'UTC',
triggerPattern: /^@Andy\b/i,
channels: [ownerChannel, reviewerChannel],
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-fallback-review-work-item-persisted-role',
reason: 'messages',
});
expect(result).toBe(true);
expect(db.getOpenWorkItemForChat).toHaveBeenCalledWith(chatJid);
expect(reviewerChannel.sendMessage).toHaveBeenCalledWith(
chatJid,
'fallback reviewer final retry',
);
expect(ownerChannel.sendMessage).not.toHaveBeenCalled();
expect(enqueueMessageCheck).toHaveBeenCalledWith(chatJid);
});
it('does not inject filtered raw bot finals into workspace-based review prompts', async () => { it('does not inject filtered raw bot finals into workspace-based review prompts', async () => {
const chatJid = 'group@test'; const chatJid = 'group@test';
const group = makeGroup('codex'); const group = makeGroup('codex');
@@ -3388,4 +3480,65 @@ describe('createMessageRuntime', () => {
expect(enqueueTask).not.toHaveBeenCalled(); expect(enqueueTask).not.toHaveBeenCalled();
expect(db.getMessagesSinceSeq).not.toHaveBeenCalled(); expect(db.getMessagesSinceSeq).not.toHaveBeenCalled();
}); });
it('recovery also queues fallback delivery retries across agent types', () => {
const chatJid = 'group@test';
const group = makeGroup('claude-code');
const enqueueMessageCheck = vi.fn();
const enqueueTask = vi.fn();
vi.mocked(db.getOpenWorkItem).mockReturnValue(undefined);
vi.mocked(db.getOpenWorkItemForChat).mockReturnValue({
id: 199,
group_folder: group.folder,
chat_jid: chatJid,
agent_type: 'codex',
service_id: 'codex-review',
delivery_role: 'reviewer',
status: 'delivery_retry',
start_seq: 5,
end_seq: 6,
result_payload: '미전달 reviewer 결과',
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',
});
const runtime = createMessageRuntime({
assistantName: 'Andy',
idleTimeout: 1_000,
pollInterval: 1_000,
timezone: 'UTC',
triggerPattern: /^@Andy\b/i,
channels: [makeChannel(chatJid)],
queue: {
registerProcess: vi.fn(),
closeStdin: vi.fn(),
notifyIdle: vi.fn(),
enqueueMessageCheck,
enqueueTask,
} as any,
getRegisteredGroups: () => ({ [chatJid]: group }),
getSessions: () => ({}),
getLastTimestamp: () => '',
setLastTimestamp: vi.fn(),
getLastAgentTimestamps: () => ({}),
saveState: vi.fn(),
persistSession: vi.fn(),
clearSession: vi.fn(),
});
runtime.recoverPendingMessages();
expect(db.getOpenWorkItemForChat).toHaveBeenCalledWith(chatJid);
expect(enqueueMessageCheck).toHaveBeenCalledWith(
chatJid,
resolveGroupIpcPath(group.folder),
);
expect(enqueueTask).not.toHaveBeenCalled();
expect(db.getMessagesSinceSeq).not.toHaveBeenCalled();
});
}); });

View File

@@ -6,7 +6,7 @@ import {
createProducedWorkItem, createProducedWorkItem,
failServiceHandoff, failServiceHandoff,
getAllPendingServiceHandoffs, getAllPendingServiceHandoffs,
getOpenWorkItem, getOpenWorkItemForChat,
getMessagesSinceSeq, getMessagesSinceSeq,
getNewMessagesBySeq, getNewMessagesBySeq,
markWorkItemDelivered, markWorkItemDelivered,
@@ -830,12 +830,9 @@ export function createMessageRuntime(deps: MessageRuntimeDeps): {
); );
} }
// Deliver pending work items through the correct channel. // Delivery retries can come from forced fallback runs whose agent_type
// For paired rooms, check if the work item was from a reviewer turn. // differs from the room owner's registered agent type.
const openWorkItem = getOpenWorkItem( const openWorkItem = getOpenWorkItemForChat(chatJid);
chatJid,
(group.agentType || 'claude-code') as 'claude-code' | 'codex',
);
if (openWorkItem) { if (openWorkItem) {
const pendingTask = hasReviewerLease(chatJid) const pendingTask = hasReviewerLease(chatJid)
? getLatestOpenPairedTaskForChat(chatJid) ? getLatestOpenPairedTaskForChat(chatJid)
@@ -1313,10 +1310,7 @@ export function createMessageRuntime(deps: MessageRuntimeDeps): {
const recoverPendingMessages = (): void => { const recoverPendingMessages = (): void => {
const registeredGroups = deps.getRegisteredGroups(); const registeredGroups = deps.getRegisteredGroups();
for (const [chatJid, group] of Object.entries(registeredGroups)) { for (const [chatJid, group] of Object.entries(registeredGroups)) {
const openWorkItem = getOpenWorkItem( const openWorkItem = getOpenWorkItemForChat(chatJid);
chatJid,
(group.agentType || 'claude-code') as 'claude-code' | 'codex',
);
if (openWorkItem) { if (openWorkItem) {
logger.info( logger.info(
{ chatJid, group: group.name, workItemId: openWorkItem.id }, { chatJid, group: group.name, workItemId: openWorkItem.id },