Function NewHumanInTheLoopMiddleware

pkg/middleware/hitl.go:84–114

NewHumanInTheLoopMiddleware 创建 HITL 中间件

(config *HumanInTheLoopMiddlewareConfig)

// NewHumanInTheLoopMiddleware 创建 HITL 中间件
func NewHumanInTheLoopMiddleware(config *HumanInTheLoopMiddlewareConfig) (*HumanInTheLoopMiddleware, error) {
	if config == nil {
		config = &HumanInTheLoopMiddlewareConfig{}
	}

	// 默认允许的决策
	defaultAllowedDecisions := config.DefaultAllowedDecisions
	if len(defaultAllowedDecisions) == 0 {
		defaultAllowedDecisions = []DecisionType{DecisionApprove, DecisionEdit, DecisionReject}
	}

	m := &HumanInTheLoopMiddleware{
		BaseMiddleware:          NewBaseMiddleware("hitl", 150),
		interruptConfigs:        make(map[string]*InterruptConfig),
		approvalHandler:         config.ApprovalHandler,
		defaultAllowedDecisions: defaultAllowedDecisions,
	}

	// 解析 InterruptOn 配置
	if config.InterruptOn != nil {
		for toolName, cfg := range config.InterruptOn {
			interruptCfg := m.parseInterruptConfig(toolName, cfg)
			if interruptCfg != nil && interruptCfg.Enabled {
				m.interruptConfigs[toolName] = interruptCfg
			}
		}
	}

	hitlLog.Info(context.Background(), "initialized", map[string]any{"tools_count": len(m.interruptConfigs)})
	return m, nil
}

var hitlLog = logging.ForComponent("HumanInTheLoopMiddleware")