Method getTaskToolPartialState

src/node/services/taskService.ts:9907–9972

(workspaceId: string)

  }

  private async getTaskToolPartialState(workspaceId: string): Promise<{
    pendingBestOfTaskToolCount: number;
    pendingTaskToolCount: number;
    referencedTaskIds: Set<string>;
  }> {
    const partial = await this.historyService.readPartial(workspaceId);
    const referencedTaskIds = new Set<string>();
    if (!partial) {
      return {
        pendingBestOfTaskToolCount: 0,
        pendingTaskToolCount: 0,
        referencedTaskIds,
      };
    }

    let pendingBestOfTaskToolCount = 0;
    let pendingTaskToolCount = 0;
    for (const part of partial.parts) {
      if (!isDynamicToolPart(part) || part.toolName !== "task") {
        continue;
      }

      if (part.state === "input-available") {
        pendingTaskToolCount += 1;
        const parsedInput = TaskToolArgsSchema.safeParse(part.input);
        if (parsedInput.success && getTaskGroupCount(parsedInput.data) > 1) {
          pendingBestOfTaskToolCount += 1;
        }
        continue;
      }
      if (part.state !== "output-available") {
        continue;
      }

      const parsedOutput = TaskToolResultSchema.safeParse(part.output);
      if (!parsedOutput.success) {
        continue;
      }

      const output = parsedOutput.data;
      if (typeof output.taskId === "string") {
        referencedTaskIds.add(output.taskId);
      }
      if (Array.isArray(output.taskIds)) {
        for (const taskId of output.taskIds) {
          referencedTaskIds.add(taskId);
        }
      }
      if ("tasks" in output && Array.isArray(output.tasks)) {
        for (const task of output.tasks) {
          referencedTaskIds.add(task.taskId);
        }
      }
      if ("reports" in output && Array.isArray(output.reports)) {
        for (const report of output.reports) {
          referencedTaskIds.add(report.taskId);
        }
      }