mirror of
https://github.com/stablyai/orca.git
synced 2026-11-13 17:37:40 +00:00
Brennan Benson
1d945af231
Hosts take the shared domain database, reads go through named async journal reads and catch-up history pages, and released-build fixtures type the release's own synchronous journal API.
61 lines
2.4 KiB
TypeScript
61 lines
2.4 KiB
TypeScript
import type { AgentJournalMessageItem } from '../../../src/shared/agent-session/journal/agent-session-journal-types'
|
|
import { QUEUED_MESSAGE_PAUSED_SEND_FAILED } from '../../../src/shared/agent-session/requests/agent-session-queued-message-wire'
|
|
import type { AgentSessionJournal } from '../../../src/main/native-chat/agent-session-journal/journal-store'
|
|
|
|
// Queue edits as each release's queue writes them: two cards, an edit, a move, a hold and a
|
|
// withdrawal. Releases before cards moved into the chat's rows kept them in a table of their own,
|
|
// with no edit or move.
|
|
|
|
type TableQueue = {
|
|
insert: (input: {
|
|
messageId: string
|
|
body: AgentJournalMessageItem
|
|
fingerprint: string
|
|
hostInstance: string
|
|
}) => Promise<unknown>
|
|
hold: (input: { messageIds: readonly string[]; reason: 'send_failed' }) => Promise<void>
|
|
withdraw: (input: { messageIds: readonly string[]; settledByOp: string }) => Promise<unknown>
|
|
}
|
|
|
|
/** A table-backed queue answers an operation's replay from its stored rows. */
|
|
function isTableQueue(queue: unknown): queue is TableQueue {
|
|
return typeof queue === 'object' && queue !== null && 'receipts' in queue
|
|
}
|
|
|
|
const message = (text: string): AgentJournalMessageItem => ({
|
|
kind: 'message',
|
|
role: 'user',
|
|
blocks: [{ type: 'text', text }]
|
|
})
|
|
|
|
export async function queueReleasedCards(
|
|
queue: Pick<
|
|
AgentSessionJournal['queuedMessages'],
|
|
'insert' | 'update' | 'move' | 'hold' | 'withdraw'
|
|
>
|
|
): Promise<void> {
|
|
const table: unknown = queue
|
|
if (isTableQueue(table)) {
|
|
const card = { fingerprint: 'fp-card', hostInstance: 'host-instance-1' }
|
|
await table.insert({ ...card, messageId: 'card-1', body: message('later') })
|
|
await table.insert({ ...card, messageId: 'card-2', body: message('second') })
|
|
await table.hold({ messageIds: ['card-2'], reason: QUEUED_MESSAGE_PAUSED_SEND_FAILED })
|
|
await table.withdraw({ messageIds: ['card-1'], settledByOp: 'withdraw-card-1' })
|
|
return
|
|
}
|
|
const first = await queue.insert({
|
|
messageId: 'card-1',
|
|
body: message('later'),
|
|
options: { model: 'model-1' }
|
|
})
|
|
await queue.insert({ messageId: 'card-2', body: message('second') })
|
|
await queue.update({
|
|
messageId: 'card-1',
|
|
expectedBodyFingerprint: first.fingerprint,
|
|
text: 'later, edited'
|
|
})
|
|
await queue.move({ messageId: 'card-2', beforeMessageId: 'card-1' })
|
|
await queue.hold({ messageIds: ['card-2'], reason: QUEUED_MESSAGE_PAUSED_SEND_FAILED })
|
|
await queue.withdraw(['card-1'])
|
|
}
|