Method Execute

pkg/workflow/router.go:49–181

Execute 执行 Router - 选择步骤并顺序链接执行

(ctx context.Context, input *StepInput)

// Execute 执行 Router - 选择步骤并顺序链接执行
func (r *Router) Execute(ctx context.Context, input *StepInput) *stream.Reader[*StepOutput] {
	reader, writer := stream.Pipe[*StepOutput](1)

	go func() {
		defer writer.Close()
		startTime := time.Now()

		// 调用选择器函数
		stepsToExecute := r.selector(input)

		if len(stepsToExecute) == 0 {
			// 没有选中任何步骤
			output := &StepOutput{
				StepID:      r.id,
				StepName:    r.name,
				StepType:    StepTypeRouter,
				Content:     fmt.Sprintf("Router %s: no steps selected", r.name),
				StartTime:   startTime,
				EndTime:     time.Now(),
				NestedSteps: []*StepOutput{},
				Metadata: map[string]any{
					"selected_steps": 0,
				},
				Metrics: &StepMetrics{ExecutionTime: time.Since(startTime).Seconds()},
			}
			output.Duration = output.EndTime.Sub(output.StartTime).Seconds()
			writer.Send(output, nil)
			return
		}

		// 收集所有执行结果
		allResults := make([]*StepOutput, 0, len(stepsToExecute))
		currentInput := input
		routerStepOutputs := make(map[string]*StepOutput)

		// 顺序执行选中的步骤（链接模式）
		for i, step := range stepsToExecute {
			// 更新输入：使用前一步的输出
			if i > 0 && len(allResults) > 0 {
				lastOutput := allResults[len(allResults)-1]
				currentInput = &StepInput{
					Input:               input.Input,
					PreviousStepContent: lastOutput.Content,
					PreviousStepOutputs: routerStepOutputs,
					AdditionalData:      input.AdditionalData,
					SessionState:        input.SessionState,
					Images:              input.Images,
					Videos:              input.Videos,
					Audio:               input.Audio,
					Files:               input.Files,
					WorkflowSession:     input.WorkflowSession,
				}
			}

			// 执行步骤
			var stepOutput *StepOutput
			var stepError error