Method redactContentBlock

pkg/security/pii_middleware.go:121–148

redactContentBlock 脱敏内容块。

(ctx context.Context, block types.ContentBlock, agentID string)

// redactContentBlock 脱敏内容块。
func (m *PIIRedactionMiddleware) redactContentBlock(ctx context.Context, block types.ContentBlock, agentID string) (types.ContentBlock, error) {
	// 类型断言处理不同的内容块类型
	switch b := block.(type) {
	case *types.TextBlock:
		if b.Text != "" {
			redacted, _, err := m.redactWithTracking(ctx, b.Text, agentID)
			if err != nil {
				return block, err
			}
			return &types.TextBlock{Text: redacted}, nil
		}
	case *types.ToolResultBlock:
		// 工具结果也可能包含 PII
		if b.Content != "" {
			redacted, _, err := m.redactWithTracking(ctx, b.Content, agentID)
			if err != nil {
				return block, err
			}
			return &types.ToolResultBlock{
				ToolUseID: b.ToolUseID,
				Content:   redacted,
				IsError:   b.IsError,
			}, nil
		}
	}
	// 其他类型（如 ToolUseBlock）不需要脱敏
	return block, nil
}

// redactWithTracking 脱敏文本并追踪 PII 匹配。
func (m *PIIRedactionMiddleware) redactWithTracking(ctx context.Context, text string, agentID string) (string, []PIIMatch, error) {