Restore synced queued message UI

This commit is contained in:
Codex
2026-05-23 20:14:10 +08:00
committed by ekko
parent 79196304c4
commit 771d122f44
4 changed files with 137 additions and 12 deletions
@@ -19,6 +19,7 @@ import { handleBridgeRun } from './handle-bridge-run'
import { handleAbort } from './abort'
import { getOrCreateSession } from './compression'
import { handleSessionCommand, isSessionCommand, parseSessionCommand } from './session-command'
import { contentBlocksToString } from './content-blocks'
import type { ContentBlock, QueuedRun, SessionState } from './types'
import { authenticateUserToken, isAuthEnabled, type AuthenticatedUser } from '../../../middleware/user-auth'
import { userCanAccessProfile } from '../../../db/hermes/users-store'
@@ -162,6 +163,7 @@ export class ChatRunSocket {
event: 'run.queued',
session_id: data.session_id,
queue_length: state.queue.length,
queued_messages: this.serializeQueuedMessages(state.queue),
})
logger.info('[chat-run-socket] queued run for session %s (queue: %d)', data.session_id, state.queue.length)
return
@@ -199,6 +201,7 @@ export class ChatRunSocket {
event: 'run.queued',
session_id: data.session_id,
queue_length: state.queue.length,
queued_messages: this.serializeQueuedMessages(state.queue),
})
logger.info('[chat-run-socket] cancelled queued run %s for session %s (queue: %d)',
data.queue_id, data.session_id, state.queue.length)
@@ -308,6 +311,7 @@ export class ChatRunSocket {
outputTokens: state.outputTokens,
contextTokens: state.contextTokens,
queueLength: state.queue?.length || 0,
queueMessages: this.serializeQueuedMessages(state.queue || []),
})
logger.info('[chat-run-socket] socket %s resumed session %s (working: %s, messages: %d)',
@@ -326,6 +330,8 @@ export class ChatRunSocket {
event: 'run.queued',
session_id: sessionId,
queue_length: state.queue.length,
dequeued_queue_id: next.queue_id,
queued_messages: this.serializeQueuedMessages(state.queue),
})
this.runQueuedItem(socket, sessionId, next, fallbackProfile)
return true
@@ -355,6 +361,16 @@ export class ChatRunSocket {
}
}
private serializeQueuedMessages(queue: QueuedRun[]) {
return queue.map(item => ({
id: item.queue_id,
role: 'user',
content: contentBlocksToString(item.input),
timestamp: Math.floor(Date.now() / 1000),
queued: true,
}))
}
private canAccessProfile(user: AuthenticatedUser, profile: string): boolean {
return user.role === 'super_admin' || userCanAccessProfile(user.id, profile)
}
@@ -6,6 +6,7 @@ import { flushBridgePendingToDb } from './bridge-message'
import { buildDbHistory, estimateSnapshotAwareHistoryUsage, forceCompressBridgeHistory, getOrCreateSession, replaceState } from './compression'
import { handleAbort } from './abort'
import { calcAndUpdateUsage, contextTokensWithCachedOverhead, updateMessageContextTokenUsage } from './usage'
import { contentBlocksToString } from './content-blocks'
import type { ContentBlock, QueuedRun, SessionState } from './types'
type CommandName =
@@ -150,8 +151,9 @@ export async function handleSessionCommand(
emitCommand({ ok: false, action: 'queue', message: 'Session is idle. Send the message normally instead.' })
return
}
const queueId = `queue_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 8)}`
state.queue.push({
queue_id: `queue_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 8)}`,
queue_id: queueId,
input: command.args,
model: ctx.model,
instructions: ctx.instructions,
@@ -163,6 +165,13 @@ export async function handleSessionCommand(
event: 'run.queued',
session_id: sessionId,
queue_length: state.queue.length,
queued_messages: state.queue.map(item => ({
id: item.queue_id,
role: 'user',
content: contentBlocksToString(item.input),
timestamp: Math.floor(Date.now() / 1000),
queued: true,
})),
})
emitCommand({
action: 'queue',