Function runWorkflowAgentTask

src/tools/WorkflowTool/WorkflowTool.ts:215–372

({
  taskId,
  executionMode,
  agent,
  workflowAbortController,
  toolUseContext,
  canUseTool,
  parentMessage,
  rootSetAppState,
}: {
  taskId: string
  executionMode: WorkflowExecutionMode
  agent: ResolvedWorkflowAgent
  workflowAbortController: AbortController
  toolUseContext: ToolUseContext
  canUseTool: Parameters<ToolDef<InputSchema, Output>['call']>[2]
  parentMessage: Parameters<ToolDef<InputSchema, Output>['call']>[3]
  rootSetAppState: (f: (prev: AppState) => AppState) => void
})

}

async function runWorkflowAgentTask({
  taskId,
  executionMode,
  agent,
  workflowAbortController,
  toolUseContext,
  canUseTool,
  parentMessage,
  rootSetAppState,
}: {
  taskId: string
  executionMode: WorkflowExecutionMode
  agent: ResolvedWorkflowAgent
  workflowAbortController: AbortController
  toolUseContext: ToolUseContext
  canUseTool: Parameters<ToolDef<InputSchema, Output>['call']>[2]
  parentMessage: Parameters<ToolDef<InputSchema, Output>['call']>[3]
  rootSetAppState: (f: (prev: AppState) => AppState) => void
}): Promise<'completed' | 'failed' | 'killed'> {
  if (workflowAbortController.signal.aborted) {
    stopWorkflowAgent(taskId, agent.id, rootSetAppState, 'Workflow stopped')
    return 'killed'
  }

  const agentAbortController = createChildAbortController(
    workflowAbortController,
  )
  setWorkflowAgentController(taskId, agent.id, agentAbortController, rootSetAppState)
  startWorkflowAgent(taskId, agent.id, rootSetAppState)
  appendWorkflowLog(
    taskId,
    `Started ${agent.name} (${agent.agentType}) in ${executionMode} mode.`,
  )

  const workflowGetAppState = () =>
    getWorkflowAppState(toolUseContext.getAppState())
  const workflowToolUseContext = {
    ...toolUseContext,
    getAppState: workflowGetAppState,
  }
  const workflowAppState = workflowGetAppState()
  const workerPermissionContext = {
    ...workflowAppState.toolPermissionContext,
    mode:
      agent.selectedAgent.permissionMode ??
      workflowAppState.toolPermissionContext.mode,
  }
  const workerTools = assembleToolPool(
    workerPermissionContext,
    workflowAppState.mcp.tools,
  )
  const tracker = createProgressTracker()
  const resolveActivity = createActivityDescriptionResolver(workerTools)
  const agentMessages: Parameters<typeof finalizeAgentTool>[0] = []
  const resolvedModel = getAgentModel(
    agent.selectedAgent.model,
    toolUseContext.options.mainLoopModel,
    agent.model,