Function testBasicWorkflow

examples/workflow_complete/main.go:40–77

测试 1: 基础 Workflow

(ctx context.Context)

// 测试 1: 基础 Workflow
func testBasicWorkflow(ctx context.Context) {
	wf := workflow.New("BasicTest").WithStream()

	wf.AddStep(workflow.NewFunctionStep("step1", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		return &workflow.StepOutput{
			Content:  "Step 1 完成",
			Metadata: make(map[string]any),
		}, nil
	}))

	wf.AddStep(workflow.NewFunctionStep("step2", func(ctx context.Context, input *workflow.StepInput) (*workflow.StepOutput, error) {
		return &workflow.StepOutput{
			Content:  fmt.Sprintf("Step 2 接收: %v", input.PreviousStepContent),
			Metadata: make(map[string]any),
		}, nil
	}))

	if err := wf.Validate(); err != nil {
		fmt.Printf("  ❌ 验证失败: %v\n", err)
		return
	}

	input := &workflow.WorkflowInput{Input: "测试输入"}
	reader := wf.Execute(ctx, input)
	for {
		event, err := reader.Recv()
		if err != nil {
			if errors.Is(err, io.EOF) {
				break
			}
			fmt.Printf("  ❌ 错误: %v\n", err)
			continue
		}
		if event.Type == workflow.EventWorkflowCompleted {
			fmt.Printf("  ✅ 成功: %v\n", event.Data.(map[string]any)["output"])
		}
	}
}

// 测试 2: 所有步骤类型
func testAllStepTypes(ctx context.Context) {