Function createTaskAwaitTool

src/node/services/tools/task_await.ts:243–994

(config: ToolConfiguration)

}

export const createTaskAwaitTool: ToolFactory = (config: ToolConfiguration) => {
  return tool({
    description: TOOL_DEFINITIONS.task_await.description,
    inputSchema: TOOL_DEFINITIONS.task_await.schema,
    execute: async (args, { abortSignal }): Promise<unknown> => {
      const workspaceId = requireWorkspaceId(config, "task_await");
      const taskService = requireTaskService(config, "task_await");

      const timeoutMs = coerceTimeoutMs(args.timeout_secs);
      // Preserve the documented 600s default when the model sends null
      // (Zod .default() only replaces undefined, not null).
      const timeoutSecsForBash = args.timeout_secs ?? 600;

      const requestedIds: string[] | null =
        args.task_ids && args.task_ids.length > 0 ? args.task_ids : null;

      const activeDescendantAgentTaskIds = taskService.listActiveDescendantAgentTaskIds(
        workspaceId,
        { excludeWorkflowTasks: true }
      );
      const isWorkflowOwnedDescendantAgentTask = async (taskId: string): Promise<boolean> =>
        (await taskService.isWorkflowOwnedDescendantAgentTask?.(workspaceId, taskId)) ?? false;

      const listInScopeBackgroundBashTaskIds = async (): Promise<string[]> => {
        if (!config.backgroundProcessManager) {
          return [];
        }

        const bashTaskIds: string[] = [];
        const processes = await config.backgroundProcessManager.list();
        for (const proc of processes) {
          if (proc.status !== "running") continue;
          const inScope =
            proc.workspaceId === workspaceId ||
            (await taskService.isDescendantAgentTask(workspaceId, proc.workspaceId));
          if (!inScope) continue;
          if (
            proc.workspaceId !== workspaceId &&
            (await isWorkflowOwnedDescendantAgentTask(proc.workspaceId))
          ) {
            continue;
          }

          bashTaskIds.push(toBashTaskId(proc.id));
        }

        return dedupeStrings(bashTaskIds);
      };
      const listInScopeWorkflowRunIds = async (): Promise<string[]> => {
        if (config.workflowService?.listRuns == null) {
          return [];
        }

        const workflowRunIds: string[] = [];
        const runs = await config.workflowService.listRuns({ workspaceId });
        for (const rawRun of runs) {
          const parsed = WorkflowRunRecordSchema.safeParse(rawRun);
          if (