Method parseInterruptConfig

pkg/middleware/hitl.go:119–167

parseInterruptConfig 解析审核配置

(toolName string, cfg any)

// parseInterruptConfig 解析审核配置
func (m *HumanInTheLoopMiddleware) parseInterruptConfig(toolName string, cfg any) *InterruptConfig {
	switch v := cfg.(type) {
	case bool:
		if !v {
			return nil // 禁用审核
		}
		// 启用默认审核配置
		return &InterruptConfig{
			Enabled:          true,
			AllowedDecisions: m.defaultAllowedDecisions,
			Message:          fmt.Sprintf("Tool '%s' requires approval before execution", toolName),
		}

	case map[string]any:
		// 自定义配置
		interruptCfg := &InterruptConfig{
			Enabled: true,
		}

		// 解析 allowed_decisions
		if decisions, ok := v["allowed_decisions"].([]any); ok {
			interruptCfg.AllowedDecisions = make([]DecisionType, 0, len(decisions))
			for _, d := range decisions {
				if ds, ok := d.(string); ok {
					interruptCfg.AllowedDecisions = append(interruptCfg.AllowedDecisions, DecisionType(ds))
				}
			}
		}
		if len(interruptCfg.AllowedDecisions) == 0 {
			interruptCfg.AllowedDecisions = m.defaultAllowedDecisions
		}

		// 解析 message
		if msg, ok := v["message"].(string); ok {
			interruptCfg.Message = msg
		} else {
			interruptCfg.Message = fmt.Sprintf("Tool '%s' requires approval before execution", toolName)
		}

		return interruptCfg

	case *InterruptConfig:
		return v

	default:
		hitlLog.Warn(context.Background(), "invalid interrupt config", map[string]any{"tool": toolName, "type": fmt.Sprintf("%T", cfg)})
		return nil
	}
}

// WrapToolCall 拦截工具调用,请求人工审核
func (m *HumanInTheLoopMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error) {