Method RunStream

pkg/workflow/workflow_agent.go:213–267

RunStream 流式运行 WorkflowAgent

(ctx context.Context, input string)

// RunStream 流式运行 WorkflowAgent
func (wa *WorkflowAgent) RunStream(ctx context.Context, input string) <-chan AgentStreamEvent {
	eventChan := make(chan AgentStreamEvent, 100)

	go func() {
		defer close(eventChan)

		eventChan <- AgentStreamEvent{
			Type:      AgentEventStart,
			Timestamp: time.Now(),
			Data:      map[string]any{"input": input},
		}

		workflowInput := &WorkflowInput{Input: input}
		reader := wa.workflow.Execute(ctx, workflowInput)
		for {
			event, err := reader.Recv()
			if err != nil {
				if errors.Is(err, io.EOF) {
					break
				}
				eventChan <- AgentStreamEvent{
					Type:      AgentEventError,
					Timestamp: time.Now(),
					Error:     err,
				}
				continue
			}

			eventChan <- AgentStreamEvent{
				Type:      AgentEventWorkflowEvent,
				Timestamp: time.Now(),
				Data:      map[string]any{"workflow_event": event},
			}

			if event.Type == EventWorkflowCompleted {
				if data, ok := event.Data.(map[string]any); ok {
					if output, ok := data["output"]; ok {
						eventChan <- AgentStreamEvent{
							Type:      AgentEventResponse,
							Timestamp: time.Now(),
							Data:      map[string]any{"response": output},
						}
					}
				}
			}
		}

		eventChan <- AgentStreamEvent{
			Type:      AgentEventComplete,
			Timestamp: time.Now(),
		}
	}()

	return eventChan
}

// ===== Types =====