Method WrapModelCall

pkg/security/pii_middleware.go:56–93

WrapModelCall 包装模型调用，在发送前脱敏 PII。

(ctx context.Context, req *middleware.ModelRequest, handler middleware.ModelCallHandler)

// WrapModelCall 包装模型调用，在发送前脱敏 PII。
func (m *PIIRedactionMiddleware) WrapModelCall(ctx context.Context, req *middleware.ModelRequest, handler middleware.ModelCallHandler) (*middleware.ModelResponse, error) {
	// 提取 Agent ID（如果有）
	agentID := "unknown"
	if req.Metadata != nil {
		if id, ok := req.Metadata["agent_id"].(string); ok {
			agentID = id
		}
	}

	// 处理每个消息
	for i, msg := range req.Messages {
		redactedMsg, err := m.redactMessage(ctx, msg, agentID)
		if err != nil {
			return nil, fmt.Errorf("failed to redact message %d: %w", i, err)
		}
		req.Messages[i] = redactedMsg
	}

	// 处理 System Prompt
	if req.SystemPrompt != "" {
		redacted, _, err := m.redactWithTracking(ctx, req.SystemPrompt, agentID)
		if err != nil {
			return nil, fmt.Errorf("failed to redact system prompt: %w", err)
		}
		req.SystemPrompt = redacted
	}

	// 调用下一层处理器
	resp, err := handler(ctx, req)
	if err != nil {
		return nil, err
	}

	// 响应通常不需要脱敏（LLM 生成的内容）
	// 但如果需要，可以在这里添加

	return resp, nil
}

// redactMessage 脱敏单个消息。
func (m *PIIRedactionMiddleware) redactMessage(ctx context.Context, msg types.Message, agentID string) (types.Message, error) {