fix(background-task): retry transient missing output tasks

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
YeonGyu-Kim
2026-05-14 18:51:22 +09:00
parent b9beea1039
commit 1ea192b3f4
2 changed files with 62 additions and 3 deletions
@@ -13,6 +13,7 @@ import { getAgentDisplayName } from "../../shared/agent-display-names"
import { recordBackgroundOutputConsumption } from "../../shared/background-output-consumption"
const SISYPHUS_JUNIOR_AGENT = getAgentDisplayName("sisyphus-junior")
const MISSING_BACKGROUND_TASK_RETRY_DELAY_MS = 100
type ToolContextWithMetadata = {
sessionID: string
@@ -40,6 +41,23 @@ function isSessionId(value: string): boolean {
return /^ses[_-]/.test(value)
}
function isBackgroundTaskId(value: string): boolean {
return /^bg[_-]/.test(value)
}
async function getTaskWithMissingRetry(
manager: BackgroundOutputManager,
taskId: string,
): Promise<BackgroundTask | undefined> {
const task = manager.getTask(taskId)
if (task || !isBackgroundTaskId(taskId)) {
return task
}
await delay(MISSING_BACKGROUND_TASK_RETRY_DELAY_MS)
return manager.getTask(taskId)
}
function formatTaskNotFoundMessage(taskId: string): string {
if (!isSessionId(taskId)) {
return `Task not found: ${taskId}`
@@ -74,7 +92,7 @@ export function createBackgroundOutput(manager: BackgroundOutputManager, client:
async execute(args: BackgroundOutputArgs, toolContext) {
try {
const ctx = toolContext as ToolContextWithMetadata
const task = manager.getTask(args.task_id)
const task = await getTaskWithMissingRetry(manager, args.task_id)
if (!task) {
return formatTaskNotFoundMessage(args.task_id)
}
@@ -103,7 +121,7 @@ export function createBackgroundOutput(manager: BackgroundOutputManager, client:
while (Date.now() - startTime < timeoutMs) {
await delay(1000)
const currentTask = manager.getTask(args.task_id)
const currentTask = await getTaskWithMissingRetry(manager, args.task_id)
if (!currentTask) {
return `Task was deleted: ${args.task_id}`
}
@@ -116,7 +134,7 @@ export function createBackgroundOutput(manager: BackgroundOutputManager, client:
}
if (isTaskActiveStatus(resolvedTask.status)) {
const finalCheck = manager.getTask(args.task_id)
const finalCheck = await getTaskWithMissingRetry(manager, args.task_id)
if (finalCheck) {
resolvedTask = finalCheck
}