Function NewConditionalPIIMiddleware

pkg/security/pii_middleware.go:263–273

NewConditionalPIIMiddleware 创建条件 PII 中间件。

(cfg ConditionalPIIConfig)

// NewConditionalPIIMiddleware 创建条件 PII 中间件。
func NewConditionalPIIMiddleware(cfg ConditionalPIIConfig) *ConditionalPIIMiddleware {
	if cfg.Priority == 0 {
		cfg.Priority = 200
	}

	return &ConditionalPIIMiddleware{
		BaseMiddleware: middleware.NewBaseMiddleware("conditional-pii-redaction", cfg.Priority),
		redactor:       NewRedactor(cfg.Detector, cfg.Strategy),
		condition:      cfg.Condition,
	}
}

// WrapModelCall 包装模型调用。
func (m *ConditionalPIIMiddleware) WrapModelCall(ctx context.Context, req *middleware.ModelRequest, handler middleware.ModelCallHandler) (*middleware.ModelResponse, error) {