Q46: parents wait for their subtasks and cannot close while any are open

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
ZacharyZhang-NYandClaude Opus 5.5 committed 2026-10-08 08:13:35 -04:00
1 parent 3bc341d757
commit b164677ea7
7 files changed
+138 -13

No files matched your search

+1
View File
@@ -46,6 +46,7 @@ export class HeartbeatScheduler {
and p.enabled = 1
and coalesce(u.request_count, 0) < p.monthly_request_limit
and not exists (select 1 from task_runs r where r.task_id = t.id and r.state in ('queued', 'running'))
and not exists (select 1 from tasks c where c.parent_task_id = t.id and c.status <> 'done')
`).all(window.start) as Array<{ task_id: string; org_id: string; provider_id: string }>;
this.observeQuotaBlocks(window.start);
const cooling = new Set(this.cooldowns.cooledProviderIds());
@@ -530,6 +530,7 @@ export class OrganizationService {
}
updateTaskStatus(orgId: string, taskId: string, status: string): void {
if (status === TASK_STATUS.DONE) this.assertNoOpenSubtasks(orgId, taskId);
const result = this.db.prepare("update tasks set status = ?, updated_at = ? where id = ? and org_id = ?")
.run(status, new Date().toISOString(), taskId, orgId);
if (result.changes === 0) throw new Error("Task was not found.");
@@ -543,6 +544,14 @@ export class OrganizationService {
payload: { status },
});
}
private assertNoOpenSubtasks(orgId: string, taskId: string): void {
const open = this.db.prepare("select title from tasks where org_id = ? and parent_task_id = ? and status <> 'done' order by created_at")
.all(orgId, taskId) as Array<{ title: string }>;
if (open.length === 0) return;
const parent = this.db.prepare("select title from tasks where id = ? and org_id = ?").get(taskId, orgId) as { title: string };
throw new Error(`"${parent.title}" has open subtasks: ${open.map((task) => task.title).join(", ")}. It can close once they are done.`);
}
}
/** Avatar colors persist as #rrggbb; anything else is a caller bug. */
@@ -121,6 +121,8 @@ export class TaskExecutionService {
if (!sameWorkspaceAccess(run.workspace_mount_json, workspaceAccess(task, workspace))) {
return this.supersedeRun(orgId, runId, run, "the task's workspace access changed after this run was queued.");
}
const openSubtask = this.db.prepare("select title from tasks where parent_task_id = ? and status <> 'done' limit 1").get(task.id) as { title: string } | undefined;
if (openSubtask) return this.supersedeRun(orgId, runId, run, `the task has open subtasks ("${openSubtask.title}"); it runs again once they are done.`);
const approvalProblem = executionApprovalProblem(this.db, orgId, task);
if (approvalProblem) return this.supersedeRun(orgId, runId, run, approvalProblem);
const isolation: TaskRunIsolation = {
@@ -213,6 +215,8 @@ export class TaskExecutionService {
handoffLines: renderTaskHandoff(this.db, orgId, task.handoff_json),
knowledgeLines: relevantKnowledgeLines(this.db, { orgId, agentId: agent.id, projectId: task.project_id, goalId: task.goal_id, taskId: task.id }, `${task.title}\n${task.description}`),
priorCompletedRuns: this.completedRunCount(task.id, runId),
subtasks: (this.db.prepare("select title, status = 'done' as done from tasks where parent_task_id = ? order by created_at").all(task.id) as Array<{ title: string; done: number }>)
.map((row) => ({ title: row.title, done: row.done === 1 })),
availableToolNames: runTools(externalTools).map((tool) => tool.name),
}),
toolContext,
+14 -2
View File
@@ -203,6 +203,8 @@ export function buildTaskPrompt(input: {
priorCompletedRuns: number;
/** Names of the tools the model can call this run. */
availableToolNames: string[];
/** Subtasks this task delegated; autonomously it only runs again once they are all done. */
subtasks: Array<{ title: string; done: boolean }>;
}): string {
const lines = [
"Execute this AgentCompany task.",
@@ -266,6 +268,15 @@ export function buildTaskPrompt(input: {
"This task carries filesystem access: read the existing code with list_files and read_file before changing anything, write changes with write_deliverable, and prove they work with run_command (install, build, test). Real output is code that builds and tests that pass — a document describing the work does not close an engineering task.",
);
}
const openSubtasks = input.subtasks.filter((subtask) => !subtask.done).map((subtask) => subtask.title);
if (openSubtasks.length > 0) {
lines.push("", `Subtasks you delegated from this task are still open: ${openSubtasks.join(", ")}. Do not redo them; this task can close only after they are done.`);
} else if (input.subtasks.length > 0) {
lines.push(
"",
`The subtasks you delegated from this task are all done: ${input.subtasks.map((subtask) => subtask.title).join(", ")}. Check their results (read_messages, list_files, list_documents) against what this task needs, then close it, or create the follow-up work that is still missing.`,
);
}
if (input.priorCompletedRuns > 0) {
lines.push(
"",
@@ -273,9 +284,10 @@ export function buildTaskPrompt(input: {
);
}
if (leadsPeople(input.agent.role_level, input.hasReports) && !input.hasReports) {
const park = `move it to "in_progress" with update_task_status(task_id="${input.task.id}", status="in_progress"): it waits while the tasks you created are open and comes back to you when they are done, so you can check the results and close it.`;
const announce = input.isManager
? `Post one company-wide kickoff to the #general channel. Then close this kickoff with update_task_status(task_id="${input.task.id}", status="done") and summarize.`
: `Tell your manager the plan in a direct message. Then close this task with update_task_status(task_id="${input.task.id}", status="done") if hiring and delegating was the work, or move it to "in_progress" and say what remains.`;
? `Post one company-wide kickoff to the #general channel. Then ${park}`
: `Tell your manager the plan in a direct message. Then ${park}`;
lines.push(
"",
`Your team is currently empty. First ${firstHiresFor(input.agent.role_level)}, create and assign their first tasks (create_task), and write the initial operating plan into the workspace (write_deliverable). Then use send_message with /handoff @Name and attachment_file_ids to pass the operating plan to the relevant report. Brief EACH report in a direct message (send_message with to="Their Name") about the work you assigned them — note the thread_id each returns so you can follow up in the same conversation. ${announce}`,
+8 -2
View File
@@ -49,6 +49,7 @@ export function WorkPage({ snapshot, actions }: PageProps) {
const [priorityFilter, setPriorityFilter] = useState<TaskPriority | "all">("all");
const [assigneeFilter, setAssigneeFilter] = useState<string>("all");
const [detailTaskId, setDetailTaskId] = useState<string | null>(null);
const [actionError, setActionError] = useState<string | null>(null);
const allTasks = useMemo(() => snapshot.tasks.filter((task) => task.orgId === orgId), [snapshot.tasks, orgId]);
const tasks = useMemo(
@@ -75,10 +76,14 @@ export function WorkPage({ snapshot, actions }: PageProps) {
const advance = (task: TaskRecord) => {
const next = nextStatus(task.status);
if (!next) return;
void actions.run(() => window.agentCompany.updateTaskStatus({ orgId, taskId: task.id, status: next }));
perform(() => actions.run(() => window.agentCompany.updateTaskStatus({ orgId, taskId: task.id, status: next })));
};
const runTask = (task: TaskRecord) => {
void actions.run(() => window.agentCompany.runTask({ orgId, taskId: task.id }));
perform(() => actions.run(() => window.agentCompany.runTask({ orgId, taskId: task.id })));
};
const perform = (operation: () => Promise<unknown>) => {
setActionError(null);
operation().catch((error: unknown) => setActionError(error instanceof Error ? error.message : String(error)));
};
const detailTask = detailTaskId ? allTasks.find((task) => task.id === detailTaskId) ?? null : null;
@@ -102,6 +107,7 @@ export function WorkPage({ snapshot, actions }: PageProps) {
{tasks.length === 1 ? t("{n} task", { n: tasks.length }) : t("{n} tasks", { n: tasks.length })} · {assigneeCount === 1 ? t("{n} assignee", { n: assigneeCount }) : t("{n} assignees", { n: assigneeCount })}
</p>
{actionError && <div className="form-error" role="alert">{actionError}</div>}
<ProjectOverview projects={projects} tasks={allTasks} agents={agents} workspaces={workspaces} chatSpaces={chatSpaces} />
<div className="work-page-body">
+8 -9
View File
@@ -22,7 +22,7 @@ function call(name: string, args: Record<string, unknown>): ToolCall {
return { id: `call_${randomUUID()}`, name, arguments: args };
}
/** Scripts the CEO: staff two reports, delegate work, hand off the file in chat, close the kickoff, then stop. */
/** Scripts the CEO: staff two reports, delegate work, hand off the file in chat, park the kickoff on its subtask, then stop. */
function scriptedCeo(): (input: ProviderToolTurnInput) => Promise<AgentTurnResult> {
const script: Array<{ text: string; toolCalls: ToolCall[] }> = [
{ text: "Staffing.", toolCalls: [
@@ -36,7 +36,7 @@ function scriptedCeo(): (input: ProviderToolTurnInput) => Promise<AgentTurnResul
call("write_deliverable", { path: "plan/operating-plan.md", content: "# OrbitOps Operating Plan\n\n- Validate demand\n- Launch booking site\n" }),
] },
{ text: "Handing off the plan.", toolCalls: [] },
{ text: "Closing the kickoff.", toolCalls: [] },
{ text: "Parking the kickoff.", toolCalls: [] },
{ text: "Team staffed, work delegated, plan written.", toolCalls: [] },
];
let index = 0;
@@ -57,13 +57,13 @@ function scriptedCeo(): (input: ProviderToolTurnInput) => Promise<AgentTurnResul
};
}
if (index === 4) {
// The prompt carries the task id so the agent can actually close its task.
// The prompt carries the task id so the agent can park its task until the delegated work is done.
const firstUser = input.messages.find((message) => message.role === "user");
const taskId = firstUser ? /Task id: (\S+)/.exec(firstUser.text)?.[1] : undefined;
index += 1;
return {
text: "Closing the kickoff.",
toolCalls: taskId ? [call("update_task_status", { task_id: taskId, status: "done" })] : [],
text: "Parking the kickoff.",
toolCalls: taskId ? [call("update_task_status", { task_id: taskId, status: "in_progress" })] : [],
rawResponseJson: JSON.stringify({ id: `resp_${index}` }),
};
}
@@ -162,13 +162,12 @@ describe("autonomous bootstrap run", () => {
const run = state.taskRuns.find((r) => r.id === claimed!.id);
expect(run?.state).toBe("completed");
// The CEO closed the kickoff task using the task id the prompt carries.
// The CEO parked the kickoff using the task id the prompt carries; it waits on its subtask.
const kickoffTask = connection.db.prepare("select status from tasks where id = (select task_id from task_runs where id = ?)")
.get(claimed!.id) as { status: string };
expect(kickoffTask).toEqual({ status: "done" });
expect(kickoffTask).toEqual({ status: "in_progress" });
// Stage 4 heartbeat: the CEO-delegated task (assigned to a hired, heartbeat-enabled
// agent, with no run yet) is enqueued exactly once for autonomous execution.
// Stage 4 heartbeat: only the delegated task is enqueued; the kickoff waits for it.
const enqueued = taskExecution.enqueueHeartbeatRuns();
expect(enqueued).toHaveLength(1);
const queued = connection.db.prepare("select task_id, state, trigger from task_runs where id = ?").get(enqueued[0]) as { task_id: string; state: string; trigger: string };
@@ -0,0 +1,94 @@
import { randomUUID } from "node:crypto";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { describe, expect, test } from "vitest";
import { DatabaseConnection } from "../../src/main/database/connection";
import { OnboardingService } from "../../src/main/services/onboarding-service";
import { OrganizationService } from "../../src/main/services/organization-service";
import { ProviderService } from "../../src/main/services/provider-service";
import { providerAdapters } from "../../src/main/services/provider-adapters";
import { TaskExecutionService } from "../../src/main/services/task-execution-service";
import { ChatService } from "../../src/main/services/chat-service";
import { dispatchAgentTool } from "../../src/main/services/agent-tools";
import { claimNextQueuedRun } from "../../src/main/app/task-queue-worker";
import { RISK_LEVEL, TASK_PRIORITY, TASK_STATUS } from "../../src/shared/domain";
import type { AgentTurnResult, ProviderToolTurnInput } from "../../src/main/services/provider-tool-turn";
const testVault = {
encryptSecret: (secret: string) => Buffer.from(secret, "utf8"),
decryptSecret: (ciphertext: Buffer) => ciphertext.toString("utf8"),
};
function company() {
const connection = new DatabaseConnection(join(tmpdir(), `ac-deps-${randomUUID()}.db`));
const prompts: string[] = [];
const turn = async (input: ProviderToolTurnInput): Promise<AgentTurnResult> => {
const first = input.messages[0];
if (first?.role !== "user") throw new Error("The run must open with the task prompt.");
prompts.push(first.text);
return { text: "done", toolCalls: [], rawResponseJson: "{}" };
};
const providers = new ProviderService(connection.db, providerAdapters, testVault, undefined, turn);
const onboarding = new OnboardingService({ db: connection.db, saveProviderCredential: (input) => providers.saveCredential(input) });
const execution = new TaskExecutionService(connection.db, providers);
const organization = new OrganizationService(connection.db);
let orgId = "";
let ceoId = "";
connection.transaction(() => {
const summary = onboarding.bootstrap({
companyName: "FlowPilot", companyDescription: "CSM analytics.", employeeName: "Yang", workspacePath: "",
executionMode: "ai_ceo", autoApproveHires: true, reviewDeliverables: false,
goalTitle: "Ship the MVP.", goalDescription: "A working product.",
providerKey: "openai", providerSecret: "sk-test-key", providerModel: "gpt-5.4",
});
orgId = summary.orgId;
ceoId = summary.ceoAgentId;
});
return { connection, execution, organization, orgId, ceoId, prompts };
}
describe("task dependencies", () => {
test("a parent cannot close while subtasks are open, and waits for them before running again", async () => {
const h = company();
const { prompts } = h;
const kickoff = claimNextQueuedRun(h.connection)!;
const parentId = (h.connection.db.prepare("select task_id from task_runs where id = ?").get(kickoff.id) as { task_id: string }).task_id;
await h.execution.executeRun(h.orgId, kickoff.id);
const childId = h.organization.createTask({
orgId: h.orgId, title: "Hire the CTO", description: "d", status: TASK_STATUS.BACKLOG, assigneeAgentId: h.ceoId,
risk: RISK_LEVEL.LOW, projectId: null, goalId: null, workspaceId: null, priority: TASK_PRIORITY.NORMAL, requiresFilesystem: false,
});
h.connection.db.prepare("update tasks set parent_task_id = ? where id = ?").run(parentId, childId);
const parentTitle = (h.connection.db.prepare("select title from tasks where id = ?").get(parentId) as { title: string }).title;
expect(() => h.organization.updateTaskStatus(h.orgId, parentId, TASK_STATUS.DONE))
.toThrow(`"${parentTitle}" has open subtasks: Hire the CTO. It can close once they are done.`);
const ctx = { orgId: h.orgId, agent: { id: h.ceoId, modelProviderId: null, modelId: null }, taskProjectId: null, taskGoalId: null, taskId: null, sourceThreadId: null, workspace: null };
const deps = { db: h.connection.db, organization: h.organization, chat: new ChatService(h.connection.db) };
expect(dispatchAgentTool(deps, ctx, { id: "c1", name: "update_task_status", arguments: { task_id: parentId, status: "done" } }))
.toBe(`ERROR: "${parentTitle}" has open subtasks: Hire the CTO. It can close once they are done.`);
// A run queued before the subtask existed is superseded instead of running ahead of it.
h.connection.db.prepare("insert into task_runs(id, org_id, task_id, agent_id, provider_id, execution_context_id, service_identity_id, workspace_mount_json, secret_scope_json, trigger, state, created_at) select 'run_early', org_id, task_id, agent_id, provider_id, 'ctx_early', service_identity_id, workspace_mount_json, secret_scope_json, 'heartbeat', 'running', created_at from task_runs where id = ?").run(kickoff.id);
await h.execution.executeRun(h.orgId, "run_early");
expect(h.connection.db.prepare("select state, error_message from task_runs where id = 'run_early'").get())
.toEqual({ state: "superseded", error_message: 'Superseded: the task has open subtasks ("Hire the CTO"); it runs again once they are done.' });
const waiting = h.connection.transaction(() => h.execution.enqueueHeartbeatRuns());
const queuedTasks = () => (h.connection.db.prepare("select task_id from task_runs where state = 'queued'").all() as Array<{ task_id: string }>).map((row) => row.task_id);
expect(waiting).toHaveLength(1);
expect(queuedTasks()).toEqual([childId]);
h.connection.db.prepare("update task_runs set state = 'completed' where state = 'queued'").run();
h.organization.updateTaskStatus(h.orgId, childId, TASK_STATUS.DONE);
h.connection.transaction(() => h.execution.enqueueHeartbeatRuns());
expect(queuedTasks()).toEqual([parentId]);
const resumed = claimNextQueuedRun(h.connection)!;
await h.execution.executeRun(h.orgId, resumed.id);
expect(prompts.at(-1)).toContain("The subtasks you delegated from this task are all done: Hire the CTO.");
expect(prompts[0]).toContain('move it to "in_progress"');
expect(prompts[0]).not.toContain(`update_task_status(task_id="${parentId}", status="done") and summarize`);
h.organization.updateTaskStatus(h.orgId, parentId, TASK_STATUS.DONE);
h.connection.close();
});
});