Method CreateWorkflowTool

pkg/workflow/workflow_agent.go:57–84

CreateWorkflowTool 创建 workflow 执行工具

(
	session *WorkflowSession,
	executionInput *WorkflowInput,
	stream bool,
)

// CreateWorkflowTool 创建 workflow 执行工具
func (wa *WorkflowAgent) CreateWorkflowTool(
	session *WorkflowSession,
	executionInput *WorkflowInput,
	stream bool,
) WorkflowToolFunc {
	return func(ctx context.Context, query string) (any, error) {
		if wa.workflow == nil {
			return nil, errors.New("no workflow attached to agent")
		}

		workflowInput := &WorkflowInput{
			Input:          query,
			AdditionalData: executionInput.AdditionalData,
			Images:         executionInput.Images,
			Videos:         executionInput.Videos,
			Audio:          executionInput.Audio,
			Files:          executionInput.Files,
			SessionID:      session.ID,
			SessionState:   session.State,
		}

		if stream {
			return wa.executeStreamingWorkflow(ctx, workflowInput)
		}

		return wa.executeWorkflow(ctx, workflowInput)
	}
}

// executeWorkflow 执行 workflow（非流式）
func (wa *WorkflowAgent) executeWorkflow(ctx context.Context, input *WorkflowInput) (any, error) {