Function completeQueuedTaskSync

packages/db/src/task-queue.ts:385–473

(input: {
  taskId: string;
  resultJson?: Record<string, unknown>;
  sessionId?: string;
  workDir?: string;
})

}

export function completeQueuedTaskSync(input: {
  taskId: string;
  resultJson?: Record<string, unknown>;
  sessionId?: string;
  workDir?: string;
}): QueuedTaskRecord {
  const db = getDatabase();
  const now = new Date().toISOString();
  const previous = readQueuedTaskSync(input.taskId);
  db.prepare(
    `UPDATE agent_task_queue
     SET status = 'completed',
         result_json = ?,
         session_id = ?,
         work_dir = ?,
         finished_at = ?,
         updated_at = ?
     WHERE id = ?`,
  ).run(
    input.resultJson ? JSON.stringify(input.resultJson) : null,
    input.sessionId ?? null,
    input.workDir ?? null,
    now,
    now,
    input.taskId,
  );

  const task = readQueuedTaskSync(input.taskId);
  if (!task) {
    throw new Error(`Queued task "${input.taskId}" does not exist.`);
  }
  if (previous?.status !== "completed") {
    const attempt = readLatestAgentTaskAttemptForTaskSync(task.id);
    const runtime = readAgentRuntimeSync(task.runtimeId);
    if (attempt) {
      updateAgentTaskAttemptSync({
        attemptId: attempt.id,
        status: "completed",
        providerSessionId: input.sessionId ?? null,
        metadata: mergeJsonObject(attempt.metadataJson, {
          workDir: input.workDir,
          completedAt: task.finishedAt,
          resumeMode: attempt.providerSessionId ? "same_provider_resume" : "cold_rebuild",
        }),
      });
    }
    if (runtime && task.routerSessionId && input.sessionId) {
      upsertAgentRouterProviderSessionSync({
        workspaceId: task.workspaceId,
        routerSessionId: task.routerSessionId,
        runtimeId: task.runtimeId,
        provider: runtime.provider,
        providerSessionId: input.sessionId,
        metadata: {
          taskQueueId: task.id,
          attemptId: attempt?.id,
          workDir: input.workDir,
        },